CbmRichGeoTest.h

Go to the documentation of this file.

/* Copyright (C) 2009-2020 UGiessen/JINR-LIT, Giessen/Dubna
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Andrey Lebedev, Semen Lebedev [committer] */
 
#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 CbmHistManager;
class CbmMCDataArray;
class CbmMCEventList;
class CbmDigiManager;
 
#include <string>
#include <vector>
 
class CbmRichGeoTest : public FairTask {
 
 public:
  CbmRichGeoTest();
 
  virtual ~CbmRichGeoTest();
 
  virtual InitStatus Init();
 
  virtual void Exec(Option_t* option);
 
  virtual void Finish();
 
  void SetOutputDir(const std::string& dir) { fOutputDir = dir; }
 
  void DrawFromFile(const std::string& fileName, const std::string& outputDir);
 
  void SetDrawPmts(bool draw) { fDrawPmts = draw; }
  void SetDrawEventDisplay(bool draw) { fDrawEventDisplay = draw; }
 
 private:
  void InitHistograms();
 
 
  void ProcessMc();
 
  CbmRichRingLight CreateRingLightWithPoints(int fileId, int mcEventId, int mcTrackId);
 
  void RingParameters();
 
  void FitAndFillHistEllipse(int histIndex, CbmRichRingLight* ring, double momentum);
 
  void FitAndFillHistCircle(int histIndex, CbmRichRingLight* ring, double 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 std::string& canvasName);
 
  void DrawPmts();
 
  void DrawPmtPoint(const std::string& coordOpt, const std::vector<int>& ids, bool isDrawPixel);
 
  std::string CalcEfficiency(TH1* histRec, TH1* histAcc);
 
  CbmRichGeoTest(const CbmRichGeoTest&);
 
  CbmRichGeoTest& operator=(const CbmRichGeoTest&);
 
  std::string fOutputDir = "";  // output directory for results
 
  TClonesArray* fRichHits             = nullptr;
  TClonesArray* fRichRings            = nullptr;
  CbmMCDataArray* fRichRefPlanePoints = nullptr;
  CbmDigiManager* fDigiMan            = nullptr;
  CbmMCDataArray* fRichPoints         = nullptr;
  CbmMCDataArray* fMcTracks           = nullptr;
  TClonesArray* fRichRingMatches      = nullptr;
  CbmMCEventList* fEventList          = nullptr;
 
  // rings will be fitted on a fly
  CbmRichRingFitterCOP* fCopFit        = nullptr;
  CbmRichRingFitterEllipseTau* fTauFit = nullptr;
 
  CbmHistManager* fHM = nullptr;  // Histogram manager
 
  int fEventNum   = 0;
  int fMinNofHits = 7;  // Min number of hits in ring for detector acceptance calculation.
 
  // fitting efficiency
  double fMinAaxis  = 3.;
  double fMaxAaxis  = 7.;
  double fMinBaxis  = 3.;
  double fMaxBaxis  = 7.;
  double fMinRadius = 3.;
  double fMaxRadius = 7.;
 
  int fNofDrawnRings     = 0;     // store number of drawn rings
  bool fDrawPmts         = true;  // draw pixels and PMTs to test rotation procedure
  bool fDrawEventDisplay = true;
 
  ClassDef(CbmRichGeoTest, 1)
};
 
#endif