CbmRichReconstruction.h

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file CbmRichReconstruction.h

#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();

   void SetParContainers();

   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 SetZTrackExtrapolation(Double_t z){fZTrackExtrapolation = z;}

   void SetMinNofStsHits(Int_t minNofStsHits){fMinNofStsHits = minNofStsHits;}

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

   Double_t fZTrackExtrapolation; // Z coordinate to which one wants to extrapolate STS tracks
   Int_t fMinNofStsHits; // minimum number of Sts hits for extrapolation to RICH detector

   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