PndHoughTrackFinderTask.h

Go to the documentation of this file.

//****************************************************************************
//*                   This file is part of PandaRoot.                        *
//*                                                                          *
//*            PandaRoot is distributed under the terms of the               *
//*              GNU General Public License (GPL) version 3,                 *
//*                 copied verbatim in the file "LICENSE".                   *
//*                                                                          *
//*  Copyright (C) 2006 - 2024 FAIR GmbH and copyright holders of PandaRoot  *
//*     The copyright holders are listed in the file "COPYRIGHTHOLDERS".     *
//*               The authors are listed in the file "AUTHORS".              *
//****************************************************************************

//  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 SetPreselectedTrackCandName(TString name) { fPreselectedTrackCandName = name; }

  virtual void SetWithGhostReduction(bool ghostred) { fWithGhostReduction = ghostred; }

  virtual void Exec(Option_t *opt);
  bool CheckZInfo(PndTrackCand &cand);

  virtual void Finish();

 private:
  FairRootManager *ioman = nullptr;

  float fCutMergeByHoughSpace;
  Int_t fNbins1;
  Int_t fNbins2;
  Int_t fNBinsSeg;
  bool fWithCuda;
  bool fWithGhostReduction;
  TString fPreselectedTrackCandName;
  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_*/