BSEmcMCHitProducerTask.h

Go to the documentation of this file.

#ifndef BSEMCMCHITPRODUCERTASK_HH
#define BSEMCMCHITPRODUCERTASK_HH

#include "FairLogger.h"

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

#include "BSEmcMCHitProductionData.h"

class BSEmcMCHitProducerTask : public PndProcessTask<BSEmcMCHitProductionData> {
 public:
  BSEmcMCHitProducerTask(const std::string &t_detectorName = "Barrel", Bool_t t_storeHits = kTRUE);
  virtual ~BSEmcMCHitProducerTask();

  void SetPointBranchName(const std::string &t_hitbranchname) { fPointBranchname = t_hitbranchname; }
  void SetHitBranchName(const std::string &t_hitbranchname) { fHitBranchname = t_hitbranchname; }

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

  virtual void SetupDataArrays() /*override*/;

 private:
  PndOutputContainer<BSEmcMCPoint> *fMCPointInput{nullptr};
  PndInputContainer<PndMCTrack> *fMCTrackInput{nullptr};
  PndOutputContainer<BSEmcMCHit> *fMCHitOutput{nullptr};

  std::string fPointBranchname{""};
  std::string fHitBranchname{""};
  ClassDef(BSEmcMCHitProducerTask, 1);
};

#endif /*BSEMCMCHITPRODUCERTASK_HH*/