CbmRichEventDisplay.h

Go to the documentation of this file.

/* Copyright (C) 2012-2023 UGiessen/JINR-LIT, Giessen/Dubna
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Semen Lebedev [committer], Martin Beyer */
 
#ifndef CBM_RICH_EVENT_DISPLAY
#define CBM_RICH_EVENT_DISPLAY
 
#include "CbmDigiManager.h"
#include "CbmHistManager.h"
 
#include <FairTask.h>
 
#include <TClonesArray.h>
 
#include <string>
 
class CbmRichRing;
class CbmEvent;
 
class CbmRichEventDisplay : public FairTask {
public:
  CbmRichEventDisplay() : FairTask("CbmRichEventDisplay") {};
 
  ~CbmRichEventDisplay() = default;
 
  CbmRichEventDisplay(const CbmRichEventDisplay&) = delete;
 
  CbmRichEventDisplay& operator=(const CbmRichEventDisplay&) = delete;
 
  InitStatus Init();
 
  void Exec(Option_t* opt);
 
  void Finish();
 
  void SetDrawRings(bool b) { fDrawRings = b; }
 
  void SetDrawHits(bool b) { fDrawHits = b; }
 
  void SetDrawPoints(bool b) { fDrawPoints = b; }
 
  void SetDrawProjections(bool b) { fDrawProjections = b; }
 
  void SetOutputDir(const std::string& dir) { fOutputDir = dir; }
 
  void SetOutputFormat(const std::string& format) { fOutputFormat = format; }
 
private:
  int fEventNum {};
 
  CbmDigiManager* fDigiManager {nullptr};
  CbmHistManager* fHM {nullptr};
 
  TClonesArray* fCbmEvents {nullptr};
  TClonesArray* fRichPoints {nullptr};
  TClonesArray* fRichHits {nullptr};
  TClonesArray* fRichRings {nullptr};
  TClonesArray* fRichMatches {nullptr};
  TClonesArray* fRichProjections {nullptr};
 
  std::string fOutputDir {"RichEventDisplays"};
  std::string fOutputFormat {"png,pdf"};
 
  bool fDrawRings {true};
  bool fDrawHits {true};
  bool fDrawPoints {false};
  bool fDrawProjections {true};
 
  void DrawOneEvent(CbmEvent* event);
 
  void DrawOnePmtPlane(const std::string& plane, CbmEvent* event);
 
  void DrawEllipse(CbmRichRing* ring);
 
  ClassDef(CbmRichEventDisplay, 2);
};
 
#endif