BmnInnerTrackingRun7.h

Go to the documentation of this file.

/*
 * File:   BmnInnerTrackingRun7.h
 * Author: Sergei Merts
 *
 * Created on April 6, 2020
 *
 *
 */
 
#ifndef BMNINNERTRACKINGRUN7_H
#define BMNINNERTRACKINGRUN7_H
 
// Base Class Headers ----------------
 
#include "BmnCellDuet.h"
#include "BmnGemStripStationSet.h"
#include "BmnGlobalTrack.h"
#include "BmnKalmanFilter.h"
#include "BmnMath.h"
#include "BmnSiliconStationSet.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 BmnInnerTrackingRun7 : public FairTask
{
  public:
    // Constructors/Destructors ---------
 
    BmnInnerTrackingRun7() {};
    BmnInnerTrackingRun7(Int_t run, Bool_t target);
    void SetFiltration(Bool_t f) { fDoHitAsymFiltration = f; }
    virtual ~BmnInnerTrackingRun7();
 
    virtual InitStatus Init();
    virtual void Exec(Option_t* opt);
    virtual void Finish();
 
  private:
    BmnStatus SortHits(vector<BmnHit*>* sortedHits);
 
    BmnStatus FindTracks_4of4_OnLastGEMStations();
    BmnStatus FindTracks_3of4_OnLastGEMStations();
    BmnStatus FindTracks_2of2_OnFirstGEMStationsDownstream();
    BmnStatus FindTracks_2of2_OnFirstGEMStationsUpstream();
 
    BmnStatus MatchHit(BmnTrack* cand, vector<BmnHit*> sortedHits, Bool_t downstream);
    BmnStatus FindCandidateByThreeStations(Short_t st0,
                                           Short_t st1,
                                           Short_t st2,
                                           vector<BmnTrack>& candidates,
                                           vector<BmnHit*>* sortedHits);
    BmnStatus FindCandidateByTwoStations(Short_t st0,
                                         Short_t st1,
                                         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;   //!
    BmnSiliconStationSet* fSilDetector;    //!
    TString fGemHitsBranchName;
    TString fSilHitsBranchName;
    TString fGlobTracksBranchName;
    TString fGemTracksBranchName;
    TString fSilTracksBranchName;
 
    TClonesArray* fGlobTracksArray;   //!
    TClonesArray* fSilTracksArray;    //!
    TClonesArray* fGemTracksArray;    //!
    TClonesArray* fSilHitsArray;      //!
    TClonesArray* fGemHitsArray;      //!
    TClonesArray* fHitsArray;         //!
 
    TClonesArray* fMCTracksArray;    //!
    TClonesArray* fSilPointsArray;   //!
    TClonesArray* fGemPointsArray;   //!
 
    BmnKalmanFilter* fKalman;   //!
 
    Bool_t fIsTarget;   // run with target or not
    Bool_t fDoHitAsymFiltration;
 
    UInt_t fEventNo;
    Short_t fNStations;
    Short_t fNSiliconStations;
 
    TVector3 fRoughVertex;   // for correct transformation
 
    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(BmnInnerTrackingRun7, 1);
};
 
#endif /* BMNINNERTRACKINGRUN7_H */