BmnBeamTracking.h

Go to the documentation of this file.

#ifndef BMNBEAMTRACKING_H
#define BMNBEAMTRACKING_H
 
#include "BmnKalmanFilter.h"
#include "BmnMath.h"
#include "BmnSiBTHit.h"
#include "BmnSiBTStationSet.h"
#include "BmnTrack.h"
#include "FairTask.h"
#include "FairTrackParam.h"
#include "TClonesArray.h"
#include "TString.h"
 
#include <TStopwatch.h>
#include <algorithm>
#include <map>
#include <vector>
 
class BmnBeamTracking : public FairTask
{
  public:
    BmnBeamTracking();
    BmnBeamTracking(Int_t run);
 
    virtual ~BmnBeamTracking();
 
    virtual InitStatus Init();
    virtual void Exec(Option_t* opt);
    virtual void Finish();
 
    void SetNTrMax(Int_t n) { fNTrMax = n; }
 
  private:
    BmnStatus SortHits(std::vector<std::vector<BmnSiBTHit*>>& sortedHits);
    BmnStatus FindTracks();
    BmnStatus TrackUpdateByKalman(BmnTrack* cand);
    BmnStatus CalculateTrackParams(BmnTrack* tr);
    Double_t CalcQp(BmnTrack* track);
    BmnStatus CalcCovMatrix(BmnTrack* tr);
 
    BmnSiBTStationSet* fSiBTDetector;   //!
    TString fSiBTHitsBranchName;
    TString fBeamTracksBranchName;
 
    // output array of Beam tracks
    TClonesArray* fBeamTracksArray;   //!
    // input array of hits
    TClonesArray* fSiBTHitsArray;   //!
 
    BmnKalmanFilter* fKalman;   //!
    UInt_t fEventNo;
    Int_t fNTrMax;       // max number of output tracks
    FairField* fField;   //!
    Double_t fTime;
 
    ClassDef(BmnBeamTracking, 2);
};
 
#endif /* BMNBEAMTRACKING_H */