PndCaloDraw.h

Go to the documentation of this file.

#ifndef PndCaloDraw_H
#define PndCaloDraw_H

#include "FairTask.h" // for FairTask, InitStatus

//#include "FairTSBufferFunctional.h" // IWYU pragma: keep needed for cint
#include "FairDataSourceI.h"

#include "Rtypes.h" // for Double_t, Int_t, Bool_t, etc
#include "TH2.h"
#include "TEveCaloData.h"
#include "TEveCalo.h"

class FairBoxSet;
class TObject;
class TVector3;
class TClonesArray;
class FairEventManager;
class FairRootManager;

class PndCaloDraw : public FairTask {

 public:
  PndCaloDraw();

  PndCaloDraw(const char *name, FairDataSourceI *dataSource = nullptr);

  virtual ~PndCaloDraw();

  virtual Double_t GetTimeWindowPlus() { return fTimeWindowPlus; }
  virtual Double_t GetTimeWindowMinus() { return fTimeWindowMinus; }

  virtual void SetTimeWindowMinus(Double_t val);
  virtual void SetTimeWindowPlus(Double_t val);
  virtual void SetStartTime(Double_t val) { fStartTime = val; }
  virtual void UseEventTimeAsStartTime(Bool_t val = kTRUE) { fUseEventTime = val; }

  virtual void Exec(Option_t *option);

  FairBoxSet *CreateBoxSet();

  void Reset();

 protected:
  virtual void SetParContainers();
  virtual InitStatus Init();
  virtual void Finish();

  virtual void FillHisto(TObject *obj);
  virtual TVector3 GetVector(TObject *obj) = 0;
  virtual Double_t GetEnergy(TObject *obj) = 0;

  TClonesArray *fList = nullptr;             
  FairEventManager *fEventManager = nullptr; 
  FairRootManager *fManager = nullptr;       
  TH2F *fCaloHisto = nullptr;                
  TEveCaloDataHist *fEveHist = nullptr;      
  TEveCalo3D *fCalo3D = nullptr;             

  Double_t fTimeWindowPlus = -1.0;
  Double_t fTimeWindowMinus = -1.0;
  Double_t fStartTime = 0.;
  Bool_t fUseEventTime = kFALSE;

 private:
  PndCaloDraw(const PndCaloDraw &);
  PndCaloDraw &operator=(const PndCaloDraw &);
  FairDataSourceI *fDataSource = nullptr;

  ClassDef(PndCaloDraw, 2);
};

#endif