PndIdealTrackFinder.h

Go to the documentation of this file.

#ifndef PndIdealTrackFinder_H_
#define PndIdealTrackFinder_H_

// framework includes
#include "PndPersistencyTask.h"
#include "FairMCPoint.h"
#include "PndTrackCand.h"
#include "PndTrackFunctor.h"
#include "FairTSBufferFunctional.h"

#include "TClonesArray.h"
#include "TDatabasePDG.h"
#include <math.h>

class PndIdealTrackFinder : public PndPersistencyTask {
 public:
  PndIdealTrackFinder(TString name = "PndIdealTrackFinder");
  virtual ~PndIdealTrackFinder();

  virtual InitStatus Init();

  virtual void AddBranchName(TString name)
  {
    fBranchNames.push_back(name);
  }

  virtual void AddBranchName(TString name, TString mcName)
  {
    fBranchNames.push_back(name);
    fPointBranchMap[name] = mcName;
  }

  virtual void SetOutputBranchName(TString name) { fOutBranchName = name; };

  virtual void Exec(Option_t *opt);

  //  virtual void Finish();

  // taken from sttmvdtracking/PndSttMvdGemTrackingIdeal.h

  void SetMomentumSmearing(Double_t sigmax, Double_t sigmay, Double_t sigmaz)
  {
    fMomSigma.SetXYZ(fabs(sigmax), fabs(sigmay), fabs(sigmaz));
    fRelative = kFALSE;
  };

  void SetRelativeMomentumSmearing(Double_t dpop)
  {
    fDPoP = fabs(dpop);
    fRelative = kTRUE;
  };

  void SetRunTimeBased(bool valRunTimeBased = true) { fRunTimeBased = valRunTimeBased; };

  void SetVertexSmearing(Double_t sigmax = -1., Double_t sigmay = -1., Double_t sigmaz = -1.) { fVtxSigma.SetXYZ(fabs(sigmax), fabs(sigmay), fabs(sigmaz)); }; // in cm

  void SetTrackingEfficiency(Double_t eff = 1.) { fEfficiency = eff; };

  void SetTrackSelector(TString selector)
  {
    fTrackSelector = PndTrackFunctor::make_PndTrackFunctor(selector.Data());
  }

  void SetPersistence(Bool_t persistence) { SetPersistency(persistence); }

 protected:
  virtual void CreateTrackCands();
  virtual void CreateTracks();
  virtual void FilterTrackCands();
  virtual FairMCPoint *GetFairMCPoint(TString hitBranch, FairMultiLinkedData_Interface *links, FairMultiLinkedData &array);
  // taken from sttmvdtracking/PndSttMvdGemTrackingIdeal.h
  virtual void SmearVector(TVector3 &vec, const TVector3 &sigma);

 protected:
  // For time-based data
  // Functor for start time

  Bool_t fRunTimeBased;

  BinaryFunctor *fFunctor;
  double fStopTimeValue;

  TString fOutBranchName;
  TClonesArray *fTrackCand;
  TClonesArray *fTrack;
  TClonesArray *fMCTrack;
  PndTrackFunctor *fTrackSelector;
  std::map<TString, TClonesArray *> fBranchMap;
  std::vector<TString> fBranchNames;
  std::map<FairLink, PndTrackCand> fTrackCandMap;
  std::map<FairLink, FairMCPoint> fFirstPointMap;
  std::map<FairLink, FairMCPoint> fLastPointMap;
  std::map<TString, TString> fPointBranchMap;

  TDatabasePDG *fPdg; 

  Int_t fHitCount;

  // Parameters for fake tracking taken from sttmvdtracking/PndSttMvdGemTrackingIdeal.h
  TVector3 fMomSigma;   
  Double_t fDPoP;       
  Bool_t fRelative;     
  TVector3 fVtxSigma;   
  Double_t fEfficiency; 

  ClassDef(PndIdealTrackFinder, 2);
};

#endif /* PndIdealTrackFinder_H_ */