PndMvdSttGemRiemannTrackFinderTask.h

Go to the documentation of this file.

#ifndef PNDMVDSTTGEMRIEMANNTRACKFINDERTASK_H_
#define PNDMVDSTTGEMRIEMANNTRACKFINDERTASK_H_

#include "FairTask.h"
//#include "PndSdsHit.h"
#include "TH2F.h"
#include "PndGeoHandling.h"
#include "PndGeoSttPar.h"

#include "FairTSBufferFunctional.h"

#include <vector>

class PndMvdSttGemRiemannTrackFinderTask : public FairTask {
 public:
  PndMvdSttGemRiemannTrackFinderTask();
  virtual ~PndMvdSttGemRiemannTrackFinderTask();
  PndMvdSttGemRiemannTrackFinderTask(const PndMvdSttGemRiemannTrackFinderTask &) = delete;
  PndMvdSttGemRiemannTrackFinderTask &operator=(const PndMvdSttGemRiemannTrackFinderTask &) = delete;

  virtual void SetParContainers();
  virtual InitStatus Init();
  virtual InitStatus ReInit();

  virtual void Exec(Option_t *opt);
  virtual void FinishEvent();

  void AddHitBranch(TString branchName); // Has to be called before Init() is used otherwise the default names are taken!

  void SetMaxSZChi2(double val) { fMaxSZChi2 = val; }
  void SetMaxSZDist(double val) { fMaxSZDist = val; }
  void SetMinPointDist(double val) { fMinPointDist = val; }
  void SetMaxDist(double val) { fMaxDist = val; }

  void SetCutDistH(TH2F *hist) { fCutDistH = hist; }
  void SetCutChi2H(TH2F *hist) { fCutChi2H = hist; }

  void SetB(Double_t B) { fB = B; }

  // void PrintResult();  // not implemented
  void SetVerbose(Int_t verbose) { fVerbose = verbose; };
  void SetPersistence(Bool_t val) { fPersistence = val; };
  //  void SetGeoH(PndGeoHandling geoH){ fGeoH=geoH;};

  void FillHitArray();
  void InitHitArray(TString branchName);

 private:
  std::vector<TString> fHitBranch;
  TString fTrackBranch;

  int fEventNr;

  double fMaxSZChi2;
  double fMaxSZDist;
  double fMinPointDist;
  double fMaxDist;

  double fB;

  std::vector<TClonesArray *> fHitArray;
  TClonesArray *fTrackCandArray;
  TClonesArray *fRiemannTrackArray;
  TClonesArray *fTrackArray;
  TClonesArray *fCorrectedSttHitArray;

  TH2F *fCutDistH;
  TH2F *fCutChi2H;

  Bool_t fInitDone;
  Bool_t fPersistence;

  BinaryFunctor *fStopFunctor;
  BinaryFunctor *fTimeGapFunctor;

  PndGeoSttPar *fSttParameters; //  CHECK added
  TClonesArray *fTubeArray;

  // PndGeoHandling* fGeoH;

  void Register();
  void Reset();
  void ProduceHits();

  ClassDef(PndMvdSttGemRiemannTrackFinderTask, 1);
};

#endif /*PNDMVDRIEMANNTRACKFINDERTASK_H_*/