alignment/CbmRichReconstruction.h

Go to the documentation of this file.

/* Copyright (C) 2012-2017 Justus-Liebig-Universitaet Giessen, Giessen
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Semen Lebedev, Jordan Bendarouach [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;
 
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 SetMirrorMisalignmentCorrectionParameterFile(string s)
  {
    fPathToMirrorMisalignmentCorrectionParameterFile = s + "/correction_table/correction_param_array.txt";
  }
 
  void SetZTrackExtrapolation(Double_t z) { fZTrackExtrapolation = z; }
 
 private:
  TClonesArray* fRichHits;
  TClonesArray* fRichRings;
  TClonesArray* fRichProjections;
  TClonesArray* fRichTrackParamZ;
  TClonesArray* fGlobalTracks;
 
  CbmRichRingFinder* fRingFinder;                      // pointer to ring finder algorithm
  CbmRichRingFitterBase* fRingFitter;                  // pointer to ring fitting algorithm
  CbmRichTrackExtrapolationBase* fTrackExtrapolation;  // pointer to track extrapolation algorithm
  CbmRichProjectionProducerBase* fProjectionProducer;  // pointer to projection producer
  CbmRichRingTrackAssignBase* fRingTrackAssign;        // pointer to track assignment algorithm
 
  // What do you wan to run.
  bool fRunExtrapolation;
  bool fRunProjection;
  bool fRunFinder;
  bool fRunFitter;
  bool fRunTrackAssign;
 
  // Algorithm names for each step of reconstruction.
  string fExtrapolationName;                                // name of extrapolation algorithm
  string fProjectionName;                                   // name of track projection algorithm
  string fFinderName;                                       // name of ring finder algorithm
  string fFitterName;                                       // name of ring fitter algorithm
  string fTrackAssignName;                                  // name of track-ring matching algorithm
  string fPathToMirrorMisalignmentCorrectionParameterFile;  // path to the mirror misalignment parameter file
 
  Double_t fZTrackExtrapolation;  // Z coordinate to which one wants to extrapolate STS tracks
 
  void InitExtrapolation();
 
  void InitProjection();
 
  void InitFinder();
 
  void InitFitter();
 
  void InitTrackAssign();
 
  void RunExtrapolation();
 
  void RunProjection();
 
  void RunFinder();
 
  void RunFitter();
 
  void RunTrackAssign();
 
  CbmRichReconstruction(const CbmRichReconstruction&);
 
  CbmRichReconstruction& operator=(const CbmRichReconstruction&);
 
  ClassDef(CbmRichReconstruction, 1);
};
 
#endif