qa/CbmRichRecoQa.h

Go to the documentation of this file.

/* Copyright (C) 2016-2024 GSI/JINR-LIT, Darmstadt/Dubna
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Andrey Lebedev [committer], Semen Lebedev, Martin Beyer */
 
#ifndef CBM_RICH_RECO_QA
#define CBM_RICH_RECO_QA
 
#include "FairTask.h"
class TClonesArray;
class CbmMCDataArray;
class CbmRichRing;
class CbmHistManager;
class TH1D;
class TH2D;
class TH1;
class TH2;
class TH3;
class CbmMCTrack;
class CbmDigiManager;
class CbmMCEventList;
class CbmLink;
 
#include <map>
#include <vector>
 
class CbmRichRecoQa : public FairTask {
 
 public:
  CbmRichRecoQa();
 
  virtual ~CbmRichRecoQa() {}
 
  virtual InitStatus Init();
 
  virtual void Exec(Option_t* option);
 
  virtual void Finish();
 
  static bool IsMcPrimaryElectron(const CbmMCTrack* mctrack);
 
  static bool IsMcPion(const CbmMCTrack* mctrack);
 
  void SetOutputDir(const std::string& dir) { fOutputDir = dir; }
 
  void DrawFromFile(const std::string& fileName, const std::string& outputDir);
 
 private:
  void InitHistograms();
 
  void FillRichRingNofHits();
 
  void FillRingTrackDistance();
 
  void RingTrackMismatchSource();
 
  void DrawMismatchSrc();
 
  void DrawHist();
 
  std::string GetMeanRmsOverflowString(TH1* h, bool withOverflow = true);
 
  void DrawRingTrackDist(const std::string& opt);
 
  /*
     * \brief Check that the ring with an input MCTrackId was found
     */
  bool WasRingFound(const CbmLink& mcTrackLink);
 
  /*
     * \brief Check that the ring was matched with some global track
     */
  bool WasRingMatched(const CbmLink& mcTrackLink);
 
  /*
     * \brief Check that the Sts track projection was matched with RICH ring
     */
  bool WasRichProjectionMatched(int stsTrackId);
 
  /*
     * Check that STS track has projection in the RICH
     */
  bool HasRichProjection(int stsTrackId);
 
 
  CbmRichRecoQa(const CbmRichRecoQa&);
 
  CbmRichRecoQa& operator=(const CbmRichRecoQa&);
 
 
  CbmHistManager* fHM = nullptr;
 
  int fEventNum = 0;
 
  std::string fOutputDir = "";  // output dir for results
 
  CbmMCDataArray* fMcTracks      = nullptr;
  CbmMCDataArray* fRichPoints    = nullptr;
  TClonesArray* fRichHits        = nullptr;
  TClonesArray* fRichRings       = nullptr;
  TClonesArray* fRichRingMatches = nullptr;
  TClonesArray* fGlobalTracks    = nullptr;
  TClonesArray* fStsTracks       = nullptr;
  TClonesArray* fStsTrackMatches = nullptr;
  TClonesArray* fRichProjections = nullptr;
  CbmDigiManager* fDigiMan       = nullptr;
  CbmMCEventList* fEventList     = nullptr;
 
  // Number of hits in the MC RICH ring
  std::map<CbmLink, int> fNofHitsInRingMap;
 
  ClassDef(CbmRichRecoQa, 1)
};
 
#endif