BSEmcPlotTask.h

Go to the documentation of this file.

#ifndef BSEMCPLOTTASK_HH
#define BSEMCPLOTTASK_HH

#include <stdexcept>
#include <string>
#include <vector>

#include "FairLogger.h"

#include "PndConstContainer.h"
#include "PndContainerI.h"
#include "PndProcess.h"
#include "PndProcessTask.h"

#include "BSEmcCluster.h"
#include "BSEmcDigi.h"
#include "BSEmcMCDeposit.h"
#include "BSEmcMCPoint.h"
#include "BSEmcMultiWaveform.h"
#include "BSEmcPrecluster.h"
#include "BSEmcRecoHit.h"
#include "BSEmcSubCluster.h"
#include "PndMCTrack.h"

#include "BSEmcPlotProcess.h"

class BSEmcPlotTask : public PndProcessTask {
 public:
  // Constructors
  BSEmcPlotTask(const std::string &t_detectorname, BSEmcHistogrammer *t_histogrammer);
  // Destructor
  virtual ~BSEmcPlotTask();

  void SetMCPointBranchName(const std::string &t_branchname) { fMCPointBranchname = t_branchname; }
  void SetMCDepositBranchName(const std::string &t_branchname) { fMCDepositBranchname = t_branchname; }
  void SetWaveformBranchName(const std::string &t_branchname) { fWaveformBranchname = t_branchname; }
  void SetDigiBranchName(const std::string &t_branchname) { fDigiBranchname = t_branchname; }
  void SetPreclusterBranchName(const std::string &t_branchname) { fPreclusterBranchname = t_branchname; }
  void SetClusterBranchName(const std::string &t_branchname) { fClusterBranchname = t_branchname; }
  void SetSubClusterBranchName(const std::string &t_branchname) { fSubClusterBranchname = t_branchname; }
  void SetRecoHitBranchName(const std::string &t_branchname) { fRecoHitBranchname = t_branchname; }

 protected:
  virtual void SetupDataArrays() /*override*/;
  virtual void PreProcessing(Option_t * /*unused*/) /*override*/;
  virtual void PostProcessing(Option_t * /*unused*/) /*override*/;
  virtual void ExecuteTasks(Option_t *t_option) /*override*/;

  virtual void SetupProcess(PndProcess *t_process) /*override*/;

 private:
  BSEmcHistogrammer *fHistogrammer{nullptr};

  std::string fMCPointBranchname{""};
  std::string fMCDepositBranchname{""};
  std::string fWaveformBranchname{""};
  std::string fDigiBranchname{""};
  std::string fPreclusterBranchname{""};
  std::string fClusterBranchname{""};
  std::string fSubClusterBranchname{""};
  std::string fRecoHitBranchname{""};

  ClassDef(BSEmcPlotTask, 1)
};

#endif /*BSEMCPLOTTASK_HH*/