BSEmcFwEndcapAPDAndVPTTWaveformGenerator.h

Go to the documentation of this file.

#ifndef BSEMCFWENDCAPAPDANDVPTTWAVEFORMGENERATOR_HH
#define BSEMCFWENDCAPAPDANDVPTTWAVEFORMGENERATOR_HH
#include <algorithm>
#include <memory>
#include <string>
#include <vector>

#include "Rtypes.h"
#include "RtypesCore.h"

#include "PndContainerI.h"
#include "PndParameterRegister.h"
#include "PndProcess.h"

#include "BSEmcFwEndcapCosmicWaveformGeneration.h"

class TClonesArray;
class BSEmcDigiPar;
class BSEmcMCHit;
class BSEmcDigi;
class BSEmcDigitizationData;
class PndParameterRegister;
class TBuffer;
class TClass;
class TMemberInspector;
template <class T>
class PndInputContainerI;
template <class T>
class PndOutputContainerI;

class BSEmcFwEndcapAPDAndVPTTWaveformGenerator : public PndProcess<BSEmcDigitizationData> {
 public:
  BSEmcFwEndcapAPDAndVPTTWaveformGenerator();
  virtual ~BSEmcFwEndcapAPDAndVPTTWaveformGenerator();

  virtual void SetDetectorName(const std::string &t_detectorName) /*override*/;
  virtual void SetupParameters(const PndParameterRegister *t_parameterRegister) /*override*/;
  virtual void SetData(BSEmcDigitizationData *t_data) /*override*/;
  virtual void Process() /*override*/;

  void SetVpttIds(const std::vector<Int_t> &t_vpttIds)
  {
    fVpttIds = t_vpttIds;
    std::sort(fVpttIds.begin(), fVpttIds.end());
  }
  void SetWaveformBranchName(const TString &t_branchName) { fWaveformBranchName = t_branchName; }
  void SetWaveformDataBranchName(const TString &t_branchName) { fWaveformDataBranchName = t_branchName; }
  void RunTimebased(Bool_t t_timebased = kTRUE)
  {
    fAPDWaveformGenerator->RunTimebased(t_timebased);
    fVPTTWaveformGenerator->RunTimebased(t_timebased);
  };
  void SetStorageOfData(Bool_t t_storeWaves = kTRUE)
  {
    fAPDWaveformGenerator->SetStorageOfData(t_storeWaves);
    fVPTTWaveformGenerator->SetStorageOfData(t_storeWaves);
  };
  void StoreDataClass(Bool_t t_storeData = kTRUE)
  {
    fAPDWaveformGenerator->StoreDataClass(t_storeData);
    fVPTTWaveformGenerator->StoreDataClass(t_storeData);
  };

 protected:
  virtual Bool_t IsVpttHit(const BSEmcMCHit *t_hit) const;

 private:
  PndInputContainerI<BSEmcMCHit> *fHitArray{nullptr};
  TString fWaveformBranchName{""};
  TString fWaveformDataBranchName{""};
  std::unique_ptr<BSEmcFwEndcapCosmicWaveformGeneration> fAPDWaveformGenerator{new BSEmcFwEndcapCosmicWaveformGeneration()};
  std::unique_ptr<BSEmcFwEndcapCosmicWaveformGeneration> fVPTTWaveformGenerator{new BSEmcFwEndcapCosmicWaveformGeneration()};
  std::vector<Int_t> fVpttIds{}; // searching a vector for each hit is bad. Need a different approach.

  ClassDef(BSEmcFwEndcapAPDAndVPTTWaveformGenerator, 1);
};

#endif /*BSEMCFWENDCAPAPDANDVPTTWAVEFORMGENERATOR_HH*/