alignment/CbmRichRecoQa.h

Go to the documentation of this file.

/* Copyright (C) 2017 Justus-Liebig-Universitaet Giessen, Giessen
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Jordan Bendarouach [committer] */
 
#ifndef CBM_RICH_RECO_QA
#define CBM_RICH_RECO_QA
 
#include "FairTask.h"
class TClonesArray;
class CbmRichRing;
class TCanvas;
class CbmHistManager;
class TH1D;
class TH2D;
class TH2;
class TH3;
class CbmMCTrack;
 
#include <map>
#include <vector>
 
using namespace std;
 
class CbmRichRecoQa : public FairTask {
 
 public:
  CbmRichRecoQa();
 
  virtual ~CbmRichRecoQa() {}
 
  virtual InitStatus Init();
 
  virtual void Exec(Option_t* option);
 
  virtual void Finish();
 
  static Bool_t IsMcPrimaryElectron(const CbmMCTrack* mctrack);
 
  static Bool_t IsMcPion(const CbmMCTrack* mctrack);
 
  void SetOutputDir(const string& dir) { fOutputDir = dir; }
 
  void DrawFromFile(const string& fileName, const string& outputDir);
 
  void SetCorrection(string s) { fCorrection = s; }
 
 private:
  void InitHistograms();
 
  void FillRichRingNofHits();
 
  void FillRingTrackDistance();
 
  void RingTrackMismatchSource();
 
  void DrawHist();
 
  string GetMeanRmsOverflowString(TH1* h, Bool_t withOverflow = true);
 
  void DrawRingTrackDistHistWithSuffix(const string& suffix);
 
  void DrawRingTrackDistHistWithSuffixPmtHalf(const string& suffix);
 
  /*
     * \brief Check that the ring with an input MCTrackId was found
     */
  bool WasRingFound(Int_t mcTrackId);
 
  /*
     * \brief Check that the ring was matched with some global track
     */
  bool WasRingMatched(Int_t mcTrackId);
 
  /*
     * \brief Check that the Sts track projection was matched with RICH ring
     */
  bool WasRichProjectionMatched(Int_t stsTrackId);
 
  /*
     * Check that STS track has projection in the RICH
     */
  bool HasRichProjection(Int_t stsTrackId);
 
 
  CbmRichRecoQa(const CbmRichRecoQa&);
 
  CbmRichRecoQa& operator=(const CbmRichRecoQa&);
 
 
  CbmHistManager* fHM;
 
  Int_t fEventNum;
 
  string fOutputDir;  // output dir for results
 
  TClonesArray* fMCTracks;
  TClonesArray* fRichPoints;
  TClonesArray* fRichDigis;
  TClonesArray* fRichHits;
  TClonesArray* fRichRings;
  TClonesArray* fRichRingMatches;
  TClonesArray* fGlobalTracks;
  TClonesArray* fStsTracks;
  TClonesArray* fStsTrackMatches;
  TClonesArray* fRichProjections;
 
  // Number of hits in the MC RICH ring
  std::map<Int_t, Int_t> fNofHitsInRingMap;
 
  string fCorrection;  // Is correction on or not
 
  vector<TCanvas*> fCanvas;
 
  ClassDef(CbmRichRecoQa, 1)
};
 
#endif