SrcInnerTrackingRun7.h

Go to the documentation of this file.

/*
 * File:   SrcInnerTrackingRun7.h
 * Author: Sergei Merts
 *
 * Created on April 6, 2020
 *
 *
 */
 
#ifndef SRCINNERTRACKINGRUN7_H
#define SRCINNERTRACKINGRUN7_H
 
// Base Class Headers ----------------
 
#include "BmnCellDuet.h"
#include "BmnGemStripStationSet.h"
#include "BmnGlobalTrack.h"
#include "BmnKalmanFilter.h"
#include "BmnMath.h"
#include "BmnSteering.h"
#include "BmnTrack.h"
#include "CbmMCTrack.h"
#include "FairMCPoint.h"
#include "FairTask.h"
#include "FairTrackParam.h"
#include "TClonesArray.h"
#include "TString.h"
 
#include <map>
#include <vector>
 
using namespace std;
 
class SrcInnerTrackingRun7 : public FairTask
{
  public:
    // Constructors/Destructors ---------
 
    SrcInnerTrackingRun7() {};
    SrcInnerTrackingRun7(Int_t run, Bool_t target);
    virtual ~SrcInnerTrackingRun7();
 
    virtual InitStatus Init();
    virtual void Exec(Option_t* opt);
    virtual void Finish();
 
  private:
    BmnStatus SortHits(vector<BmnHit*>* sortedHits);
 
    BmnStatus FindTracks_3of6();
    BmnStatus FindTracks_6of6();
    BmnStatus FindTracks_5of6();
    BmnStatus FindTracks_4of6();
 
    BmnStatus MatchHit(BmnTrack* cand, vector<BmnHit*> sortedHits, Bool_t downstream);
    // BmnStatus FindCandidateByFourStations(Short_t st0, Short_t st1, Short_t st2, Short_t st3, Short_t skip0, Short_t
    // skip1);
    BmnStatus FindCandidateByThreeStations(Short_t st0,
                                           Short_t st1,
                                           Short_t st2,
                                           vector<BmnTrack>& candidates,
                                           vector<BmnHit*>* sortedHits);
    BmnStatus FindCandidateByFourStations(Short_t st0,
                                          Short_t st1,
                                          Short_t st2,
                                          Short_t st3,
                                          vector<BmnTrack>& candidates,
                                          vector<BmnHit*>* sortedHits);
    BmnStatus FindCandidateByFiveStations(Short_t st0,
                                          Short_t st1,
                                          Short_t st2,
                                          Short_t st3,
                                          Short_t st4,
                                          vector<BmnTrack>& candidates,
                                          vector<BmnHit*>* sortedHits);
 
    BmnStatus TrackUpdateByKalman(vector<BmnTrack>& cands);
    BmnStatus TrackUpdateByLine(vector<BmnTrack>& cands);
    BmnStatus SortTracks(vector<BmnTrack>& inTracks, vector<BmnTrack>& sortedTracks);
    BmnStatus TrackSelection(vector<BmnTrack>& cands);
 
    Double_t CalcQp(BmnTrack* track, Bool_t doSimple);
    BmnStatus CalculateTrackParams(BmnTrack* tr);
    BmnStatus CalcCovMatrix(BmnTrack* tr);
    TVector2 CalcMeanSigma(vector<Double_t>);
    Double_t CalculateLength(BmnTrack* tr);
    BmnStatus CheckSharedHits(vector<BmnTrack>& sortedTracks);
    void SetHitsUsing(BmnTrack* tr, Bool_t use);
    BmnStatus DrawHits();
 
    BmnGemStripStationSet* fGemDetector;   //!
    TString fGemHitsBranchName;
    TString fGlobTracksBranchName;
    TString fGemTracksBranchName;
 
    TClonesArray* fGlobTracksArray;   //!
    TClonesArray* fGemTracksArray;    //!
    TClonesArray* fGemHitsArray;      //!
    TClonesArray* fHitsArray;         //!
 
    TClonesArray* fMCTracksArray;    //!
    TClonesArray* fGemPointsArray;   //!
 
    BmnKalmanFilter* fKalman;   //!
 
    Bool_t fIsTarget;   // run with target or not
 
    UInt_t fEventNo;
    Short_t fNStations;
 
    FairField* fField;   //!
 
    Double_t fChiSquareCut;
    Double_t* fHitXCutMin;   //!
    Double_t* fHitXCutMax;   //!
    Double_t* fHitYCutMin;   //!
    Double_t* fHitYCutMax;   //!
 
    Int_t fNHitsCut;
    Double_t fDistCut;
 
    TString fSteerFile;
    BmnSteering* fSteering;   //!
 
    ClassDef(SrcInnerTrackingRun7, 1);
};
 
#endif /* SRCINNERTRACKINGRUN7_H */