TopDownSelector.h

Go to the documentation of this file.

#ifndef _TopDownSelector_h_
#define _TopDownSelector_h_

#include <fwk/VModule.h>
#include <utl/Vector.h>
#include <utl/TimeInterval.h>

#include <evt/Event.h>
#include <vector>


namespace sevt {
  class Station;
  class SEvent;
};

namespace sdet {
  class Station;
};


namespace TopDownSelectorNS {

  // The following is a simple struct so we can add and remove stations easily by just changing the 'candidate' flag.
  // It also makes it easier to access detector and event data from the same place.
  struct StationPairStatus {
    const sdet::Station* detector;
    sevt::Station* event;
    bool candidate;
  };

  class TopDownSelector : public fwk::VModule {

  public:
    TopDownSelector();
    virtual ~TopDownSelector() { }

    fwk::VModule::ResultFlag Init();
    fwk::VModule::ResultFlag Run(evt::Event& event);
    fwk::VModule::ResultFlag Finish();

  private:
    double fBaryTime;
    utl::Vector fBarycenter;
    utl::Vector fAxis;
    sevt::SEvent* fSEvent;
    
    bool Select();
    
    std::vector<double> fResiduals;
    std::vector<StationPairStatus> fStations;
    
    // Module config parameters
    int fVerbose;
    bool fRejectNonT4Events;
    double kBaryPower;
    double kTolRes;
    double kTolResM;
    double kIsoDist1;
    double kIsoDist2;
    double kCurv0;
    double kAreaMin;
    double kAreaMax;
    double kAna3RatMin;
    double kDistMax;
    double kIsoTime1;
    double kIsoTime2;

    enum TimeCorrectionType {
      eNone = 0,
      eAltitude = 1,
      eCurved = 2,
      eAltitudeCurved = 3
    };

  private:
    // method to initialize.
    void InitializeEvent();
    void RemoveEA();
    void RemoveDoublets();
    void RemoveIsolatedStations();
    bool TryRemovingStations(int);
    bool EstimateCore();
    bool IsGoodTimeConfig(TimeCorrectionType correctionType);
    bool IsGoodSpaceConfig();
    double CorrectTimeAltitude(int);
    double CorrectTimeAltitudeCurv(int);
    bool IsAligned();
    double fU, fV, fT0;
    double fXCore, fYCore, fZCore, fTCore;
    std::vector<unsigned int> fIndexes;
    unsigned int fNMinStations, fNMaxRemovedStations;
    bool fIsLastTimingIteration;
 
    bool PatternRecovering();
    void RemoveRandomStations();
    bool Recover3Stations();

    double EstimatedZenith(const std::string& s);
    double EstimatedAzimuth(const std::string& s);

    REGISTER_MODULE("TopDownSelector", TopDownSelector);

  };

}


#endif