PndHoughTrackFinderTask.h

Go to the documentation of this file.

//  PndHoughTrackFinderTask
//  Finds Track for one event

#ifndef PndHoughTrackFinderTask_H_
#define PndHoughTrackFinderTask_H_

#include "FairTask.h"
#include "PndGeoSttPar.h"
#include <vector>
#include "PndHoughTrackFinder.h"

class PndHoughTrackFinderTask : public FairTask {
 public:

  PndHoughTrackFinderTask();

  virtual ~PndHoughTrackFinderTask();

  virtual void SetParContainers();

  virtual InitStatus Init();

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

  virtual void SetBinningX(double bin) { fNbins1 = bin; };
  virtual void SetBinningY(double bin) { fNbins2 = bin; };
  virtual void SetNBinsSeg(Int_t n) { fNBinsSeg = n; };
  virtual void SetCutMergeByHoughSpace(float cut) { fCutMergeByHoughSpace = cut; }
  virtual void SetWithCuda(bool cuda) { fWithCuda = cuda; }
  virtual void SetWithGhostReduction(bool ghostred) { fWithGhostReduction = ghostred; }

  virtual void Exec(Option_t *opt);

  virtual void Finish();

 private:
  FairRootManager *ioman = nullptr;

  float fCutMergeByHoughSpace;
  Int_t fNbins1;
  Int_t fNbins2;
  Int_t fNBinsSeg;
  bool fWithCuda;
  bool fWithGhostReduction;

  std::vector<TString> fBranchNames;
  std::map<TString, TClonesArray *> fBranchMap;

  PndGeoSttPar *fSttParameters = nullptr; // needed for SttStrawMap
  TClonesArray *fSttTubeArray = nullptr;

  PndHoughTrackFinder *fPndHoughTrackFinder = nullptr;

  TClonesArray *fPreselectedTrackArrayCand = nullptr;
  TClonesArray *fApolloniusTrackArrayMerged = nullptr;
  TClonesArray *fApolloniusTrackArrayMergedCand = nullptr;
  TClonesArray *fCombinedSkewed = nullptr;

  ClassDef(PndHoughTrackFinderTask, 1);
};

#endif /*PndHoughTrackFinderTask_H_*/