CbmRichReconstruction.h

Go to the documentation of this file.

/* Copyright (C) 2012-2021 UGiessen/JINR-LIT, Giessen/Dubna
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Semen Lebedev [committer] */
 
#ifndef CBM_RICH_RECONSTRUCTION
#define CBM_RICH_RECONSTRUCTION
 
#include <FairTask.h>
 
#include <string>
 
class TClonesArray;
class CbmRichRingFinder;
class CbmRichRingFitterBase;
class CbmRichTrackExtrapolationBase;
class CbmRichProjectionProducerBase;
class CbmRichRingTrackAssignBase;
class CbmEvent;
 
using std::string;
 
class CbmRichReconstruction : public FairTask {
 public:
  CbmRichReconstruction();
 
  virtual ~CbmRichReconstruction();
 
 
  virtual InitStatus Init();
 
  virtual void Exec(Option_t* opt);
 
  virtual void Finish();
 
  void SetRunExtrapolation(bool b) { fRunExtrapolation = b; }
  void SetRunProjection(bool b) { fRunProjection = b; }
  void SetRunFinder(bool b) { fRunFinder = b; }
  void SetRunFitter(bool b) { fRunFitter = b; }
  void SetRunTrackAssign(bool b) { fRunTrackAssign = b; }
 
  void SetExtrapolationName(const string& n) { fExtrapolationName = n; }
  void SetProjectionName(const string& n) { fProjectionName = n; }
  void SetFinderName(const string& n) { fFinderName = n; }
  void SetFitterName(const string& n) { fFitterName = n; }
  void SetTrackAssignName(const string& n) { fTrackAssignName = n; }
 
  void SetUseHTAnnSelect(bool select) { fUseHTAnnSelect = select; }
  void SetUseHTSubdivide(bool select) { fUseHTSubdivide = select; }
 
  void SetZTrackExtrapolation(Double_t z) { fZTrackExtrapolation = z; }
 
  void UseMCbmSetup()
  {
    this->SetRunExtrapolation(false);
    this->SetRunProjection(false);
    this->SetRunTrackAssign(false);
    this->SetUseHTAnnSelect(false);
    this->SetUseHTSubdivide(false);
  }
 
 private:
  TClonesArray* fRichHits        = nullptr;
  TClonesArray* fRichRings       = nullptr;
  TClonesArray* fRichProjections = nullptr;
  TClonesArray* fRichTrackParamZ = nullptr;
  TClonesArray* fGlobalTracks    = nullptr;
  TClonesArray* fCbmEvents       = nullptr;
 
  std::array<Double_t, 6> fCalcTime{0., 0., 0., 0., 0., 0.};
 
  Int_t fNofTs                = 0;
  Int_t fNofEvents            = 0;
  Int_t fTotalNofHits         = 0;
  Int_t fTotalNofRings        = 0;
  Int_t fTotalNofTrackProj    = 0;
  Int_t fTotalNofGlobalTracks = 0;
 
 
  // pointers to the algorithms
  CbmRichRingFinder* fRingFinder                     = nullptr;
  CbmRichRingFitterBase* fRingFitter                 = nullptr;
  CbmRichTrackExtrapolationBase* fTrackExtrapolation = nullptr;
  CbmRichProjectionProducerBase* fProjectionProducer = nullptr;
  CbmRichRingTrackAssignBase* fRingTrackAssign       = nullptr;
 
  // What do you want to run.
  bool fRunExtrapolation = true;
  bool fRunProjection    = true;
  bool fRunFinder        = true;
  bool fRunFitter        = true;
  bool fRunTrackAssign   = true;
 
  // Run ring-candidate selection algorithm based on ANN
  bool fUseHTAnnSelect = true;
 
  // Subdivide the RICH plain at y=0 to run both parts in parallel
  bool fUseHTSubdivide = true;
 
  // Algorithm names for each step of reconstruction.
  string fExtrapolationName = "littrack";
  string fProjectionName    = "analytical";
  string fFinderName        = "hough";
  string fFitterName        = "ellipse_tau";
  string fTrackAssignName   = "closest_distance";
 
  // Z coordinate where STS tracks will be extrapolated.
  // Initialized later using geometry.
  Double_t fZTrackExtrapolation = -1;  //260.
 
  void InitExtrapolation();
 
  void InitProjection();
 
  void InitFinder();
 
  void InitFitter();
 
  void InitTrackAssign();
 
  void ProcessData(CbmEvent* event);
 
  void RunExtrapolation(CbmEvent* event);
 
  void RunProjection(CbmEvent* event);
 
  void RunFinder(CbmEvent* event);
 
  void RunFitter(CbmEvent* event);
 
  void RunTrackAssign(CbmEvent* event);
 
  CbmRichReconstruction(const CbmRichReconstruction&);
 
  CbmRichReconstruction& operator=(const CbmRichReconstruction&);
 
  ClassDef(CbmRichReconstruction, 1);
};
 
#endif