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 "PndContainerI.h"
#include "PndInputContainer.h"
#include "PndProcess.h"
#include "PndProcessTask.h"

#include "BSEmcCluster.h"
#include "BSEmcDigi.h"
#include "BSEmcMCHit.h"
#include "BSEmcMCPoint.h"
#include "BSEmcMultiWaveform.h"
#include "BSEmcPlotData.h"
#include "BSEmcPlotProcess.h"
#include "BSEmcPrecluster.h"
#include "BSEmcRecoHit.h"
#include "BSEmcSubCluster.h"

class BSEmcPlotTask : public PndProcessTask<BSEmcPlotData> {
 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 SetMCHitBranchName(const std::string &t_branchname) { fMCHitBranchname = 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<BSEmcPlotData> *t_process) /*override*/;

 private:
  BSEmcHistogrammer *fHistogrammer{nullptr};

  PndInputContainer<PndMCTrack> *fMCTrackInput{new PndInputContainer<PndMCTrack>()};
  PndInputContainer<BSEmcMCPoint> *fPointInput{new PndInputContainer<BSEmcMCPoint>()};
  PndInputContainer<BSEmcMCHit> *fHitInput{new PndInputContainer<BSEmcMCHit>()};
  PndInputContainer<BSEmcMultiWaveform> *fWaveformInput{new PndInputContainer<BSEmcMultiWaveform>()};
  PndInputContainer<BSEmcDigi> *fDigiInput{new PndInputContainer<BSEmcDigi>()};
  PndInputContainer<BSEmcPrecluster> *fPreclusterInput{new PndInputContainer<BSEmcPrecluster>()};
  PndInputContainer<BSEmcCluster> *fClusterInput{new PndInputContainer<BSEmcCluster>()};
  PndInputContainer<BSEmcSubCluster> *fSubClusterInput{new PndInputContainer<BSEmcSubCluster>()};
  PndInputContainer<BSEmcRecoHit> *fRecoHitInput{new PndInputContainer<BSEmcRecoHit>()};

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

  ClassDef(BSEmcPlotTask, 1)
};

#endif /*BSEMCPLOTTASK_HH*/