BSEmcDigitizerTask.h

Go to the documentation of this file.

#ifndef BSEMCDIGITIZERTASK_HH
#define BSEMCDIGITIZERTASK_HH

#include <string>

#include "FairLogger.h"
#include "FairRootManager.h"

#include "PndContainerI.h"
#include "PndInputContainer.h"
#include "PndOutputContainer.h"
#include "PndProcess.h"
#include "PndProcessTask.h"

#include "BSEmcDigi.h"
#include "BSEmcDigitizationData.h"
#include "BSEmcMCHit.h"
#include "BSEmcMultiWaveform.h"

class BSEmcDigitizerTask : public PndProcessTask<BSEmcDigitizationData> {
 public:
  BSEmcDigitizerTask(const std::string &t_detectorname = "Barrel", Bool_t t_storedigis = kTRUE);
  virtual ~BSEmcDigitizerTask();

  void SetHitBranchName(const std::string &t_hitbranchname) { fHitBranchname = t_hitbranchname; }
  void SetDigiBranchName(const std::string &t_digibranchname) { fDigiBranchname = t_digibranchname; }
  void SetWaveformBranchName(const std::string &t_waveformbranchname) { fWaveformBranchname = t_waveformbranchname; }

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

 private:
  PndInputContainer<BSEmcMCHit> *fMCHitInput{nullptr};
  PndOutputContainer<BSEmcMultiWaveform> *fWaveformOutput{nullptr};
  PndOutputContainer<BSEmcDigi> *fDigiOutput{nullptr};

  std::string fHitBranchname{""};
  std::string fDigiBranchname{""};
  std::string fWaveformBranchname{""};
  ClassDef(BSEmcDigitizerTask, 1);
};

#endif /*BSEMCDIGITIZERTASK_HH*/