CbmRichReconstruction.h

Go to the documentation of this file.

 
#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 SetUseHTAnnSelect(bool select) { fUseHTAnnSelect = select; }
 
  void SetZTrackExtrapolation(Double_t z) { fZTrackExtrapolation = z; }
 
  void UseMCbmSetup() {
    this->SetRunExtrapolation(false);
    this->SetRunProjection(false);
    this->SetRunTrackAssign(false);
    this->SetUseHTAnnSelect(false);
  }
 
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;
 
  bool fUseHTAnnSelect;
 
  // 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
 
  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