CbmRichGeoTest.h

Go to the documentation of this file.

 
#ifndef CBMRICHGEOTEST
#define CBMRICHGEOTEST
 
#include "FairTask.h"
 
class TH1;
class TH2;
class TH3;
class TH1D;
class TH2D;
class TH3D;
class TClonesArray;
class CbmRichRingFitterCOP;
class CbmRichRingFitterEllipseTau;
class CbmRichRing;
class CbmRichRingLight;
class TCanvas;
class CbmHistManager;
class TVector3;
class CbmMCDataArray;
class CbmMCEventList;
class CbmDigiManager;
 
#include <string>
#include <vector>
 
using namespace std;
 
class CbmRichGeoTest : public FairTask {
 
public:
  CbmRichGeoTest();
 
  virtual ~CbmRichGeoTest();
 
  virtual InitStatus Init();
 
  virtual void Exec(Option_t* option);
 
  virtual void Finish();
 
  void SetOutputDir(const string& dir) { fOutputDir = dir; }
 
  void DrawFromFile(const string& fileName, const string& outputDir);
 
  void SetDrawPmts(Bool_t draw) { fDrawPmts = draw; }
  void SetDrawEventDisplay(Bool_t draw) { fDrawEventDisplay = draw; }
 
private:
  void InitHistograms();
 
 
  void ProcessMc();
 
  void RingParameters();
 
  void FitAndFillHistEllipse(Int_t histIndex,
                             CbmRichRingLight* ring,
                             Double_t momentum);
 
  void FitAndFillHistCircle(Int_t histIndex,
                            CbmRichRingLight* ring,
                            Double_t momentum);
 
  void FillMcVsHitFitEllipse(CbmRichRingLight* ring, CbmRichRingLight* ringMc);
 
  void FillMcVsHitFitCircle(CbmRichRingLight* ring, CbmRichRingLight* ringMc);
 
  void ProcessHits();
 
  TH1D* CreateAccVsMinNofHitsHist();
 
  void DrawHist();
 
  void DrawRing(CbmRichRingLight* ringHit, CbmRichRingLight* ringPoint);
 
 
  void DrawH2MeanRms(TH2* hist, const string& canvasName);
 
  void DrawPmts();
 
  void DrawPmtPoint(const string& coordinates,
                    const vector<Int_t>& ids,
                    Bool_t isDrawPixel);
 
  string CalcEfficiency(TH1* histRec, TH1* histAcc);
 
  CbmRichGeoTest(const CbmRichGeoTest&);
 
  CbmRichGeoTest& operator=(const CbmRichGeoTest&);
 
  string fOutputDir;  // output directory for results
 
  TClonesArray* fRichHits;
  TClonesArray* fRichRings;
  CbmMCDataArray* fRichRefPlanePoints;
  CbmDigiManager* fDigiMan = nullptr;
  CbmMCDataArray* fRichPoints;
  CbmMCDataArray* fMcTracks;
  TClonesArray* fRichRingMatches;
  CbmMCEventList* fEventList;
 
  // rings will be fitted on a fly
  CbmRichRingFitterCOP* fCopFit;
  CbmRichRingFitterEllipseTau* fTauFit;
 
  CbmHistManager* fHM;  // Histogram manager
 
  Int_t fEventNum;
  Int_t
    fMinNofHits;  // Min number of hits in ring for detector acceptance calculation.
 
  // fitting efficiency
  Double_t fMinAaxis;
  Double_t fMaxAaxis;
  Double_t fMinBaxis;
  Double_t fMaxBaxis;
  Double_t fMinRadius;
  Double_t fMaxRadius;
 
  Int_t fNofDrawnRings;  // store number of drawn rings
  Bool_t fDrawPmts;      // draw pixels and PMTs to test rotation procedure
  Bool_t fDrawEventDisplay;
 
  ClassDef(CbmRichGeoTest, 1)
};
 
#endif