CbmRichMCbmQaRichOnly.h

Go to the documentation of this file.

/* Copyright (C) 2020 GSI Helmholtzzentrum fuer Schwerionenforschung, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Florian Uhlig [committer] */
 
#ifndef MCBM_RICH_QA_RICH_ONLY
#define MCBM_RICH_QA_RICH_ONLY
 
#include "CbmEvent.h"
#include "CbmRichRingFinderHoughImpl.h"
#include "FairTask.h"
class TClonesArray;
class CbmRichRing;
class CbmRichHit;
class CbmTofTracklet;
class CbmHistManager;
class TVector3;
class CbmDigiManager;
class CbmRichMCbmSEDisplay;
 
#include <array>
#include <map>
#include <tuple>
#include <vector>
 
using namespace std;
 
 
class CbmRichMCbmQaRichOnly : public FairTask {
 
 public:
  CbmRichMCbmQaRichOnly();
 
  virtual ~CbmRichMCbmQaRichOnly(){};
 
  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 DoRestrictToAcc() { fRestrictToAcc = true; }
 
 
  void DoRestrictToFullAcc(bool val = true) { fRestrictToFullAcc = val; }
 
 
  void DoDrawCanvas(bool val = true) { fDoDrawCanvas = val; }
 
  void DoWriteHistToFile(bool val = true) { fDoWriteHistToFile = val; }
 
  void XOffsetHistos(Double_t val = 0.) { fXOffsetHisto = val; }
 
  void SetMaxNofDrawnEvents(Int_t val = 100) { fMaxNofDrawnEvents = val; }
 
  void SetTriggerRichHits(Int_t val = 0) { fTriggerRichHits = val; }
 
  void SetIcdGeneration(bool val = true) { fGenerateICD = val; }
 
  void SetTotRich(Double_t min, Double_t max)
  {
    fTotMin = min;
    fTotMax = max;
  }
 
 
 private:
  CbmDigiManager* fDigiMan = nullptr;
 
  TClonesArray* fRichHits;
 
  TClonesArray* fRichRings;
 
  TClonesArray* fCbmEvent;
 
  CbmHistManager* fHM;
 
  Double_t fXOffsetHisto;
 
  Double_t fTotMin = 0.0;
 
  Double_t fTotMax = 100.0;
 
  Int_t fEventNum;
 
  Int_t fNofDrawnRings;
 
  Int_t fNofDrawnRichTofEv;
 
  Int_t fNofDrawnEvents;
 
  Int_t fMaxNofDrawnEvents;
 
  Int_t fTriggerRichHits;
 
 
  string fOutputDir;  // output dir for results
 
  bool fRestrictToAcc     = false;
  bool fRestrictToFullAcc = false;
 
  bool fDoWriteHistToFile = true;
  bool fDoDrawCanvas      = true;
 
  bool fGenerateICD = false;
 
  std::array<Double_t, 2304> ICD_offset_read;
  std::array<Double_t, 2304> ICD_offset;
  std::array<uint32_t, 2304> ICD_offset_cnt;
 
  CbmRichMCbmSEDisplay* fSeDisplay = nullptr;
 
 
  void InitHistograms();
 
  void DrawHist();
 
  void RichRings();
 
  void DrawEvent(CbmEvent* ev, std::vector<int>& ringIndx, bool full);
 
  void DrawRing(CbmRichRing* ring) { DrawRing(ring, false); };
 
  void DrawRing(CbmRichRing* ring, bool full);
 
 
  bool doToT(CbmRichHit* hit);
 
 
  void analyseRing(CbmRichRing* ring, CbmEvent* ev);
 
  Bool_t cutRadius(CbmRichRing* ring);
 
  void save_ICD(std::array<Double_t, 2304>& offsets, unsigned int iteration);
 
  void read_ICD(std::array<Double_t, 2304>& offsets, unsigned int iteration);
 
  CbmRichMCbmQaRichOnly(const CbmRichMCbmQaRichOnly&);
 
  CbmRichMCbmQaRichOnly& operator=(const CbmRichMCbmQaRichOnly&);
 
 
  ClassDef(CbmRichMCbmQaRichOnly, 1)
};
 
#endif