BSEmcAbsWaveformSimulator.h

Go to the documentation of this file.

#ifndef BSEMCABSWAVEFORMSIMULATOR_HH
#define BSEMCABSWAVEFORMSIMULATOR_HH

#include "TClonesArray.h"
#include "TList.h"
#include "TMath.h"
#include "TObject.h"
#include "TROOT.h"

#include "BSEmcAbsWaveformModifier.h"

class TClonesArray;
class BSEmcWaveform;
class BSEmcWaveformData;

class BSEmcAbsWaveformSimulator : public TObject {
 public:
  virtual BSEmcWaveform *Simulate(BSEmcWaveformData *t_wfData, TClonesArray *t_arrayToStore = nullptr) { return CallModifiers(MakeWaveform(t_wfData, t_arrayToStore)); }

  virtual BSEmcWaveform *
  SimulateSinglePulse(Double_t t_depositEnergy, Double_t t_depositTime, TClonesArray *t_arrayToStore = nullptr, Int_t t_detId = -1, Int_t t_trackId = -1, Int_t t_depositIndex = -1)
  {
    return CallModifiers(MakeSingleWaveform(t_depositEnergy, t_depositTime, t_arrayToStore, t_detId, t_trackId, t_depositIndex));
  }

  virtual BSEmcWaveform *Get1GevWaveform() = 0;

  virtual void GetAbsoluteTimeInterval(BSEmcWaveformData *t_wfData, Double_t &t_startTime, Double_t &t_activeTime) = 0; // in ns

  virtual Double_t GetTimeBeforeFirstDeposit(BSEmcWaveformData * /*unused*/) { return 0; } // in ns  // wfData //[R.K.03/2017] unused variable(s)

  virtual Double_t GetScale() = 0;

  virtual Double_t GetTotalScale()
  {
    Double_t totalScale = GetScale();

    TIter iterModifiers(&fListOfWaveformModifiers);
    while (BSEmcAbsWaveformModifier *wfmod = dynamic_cast<BSEmcAbsWaveformModifier *>(iterModifiers())) {
      totalScale *= wfmod->GetScale();
    }

    return totalScale;
  }

  virtual Double_t GetSampleRate() { return fSampleRate; }

  virtual void AddModifier(BSEmcAbsWaveformModifier *t_wfModifier) { fListOfWaveformModifiers.Add(t_wfModifier); }

  void SetMCDepositBranchName(const TString &t_branchname) { fMCDepositBranchName = t_branchname; }

 protected:
  BSEmcAbsWaveformSimulator() : fSampleRate(0){};
  BSEmcAbsWaveformSimulator(Double_t sampleRate) : fSampleRate(sampleRate){};

  virtual BSEmcWaveform *MakeWaveform(BSEmcWaveformData *t_wfData, TClonesArray *t_arrayToStore = nullptr) = 0;
  virtual BSEmcWaveform *MakeSingleWaveform(Double_t t_depositEnergy, Double_t t_depositTime, TClonesArray *t_arrayToStore = nullptr, Int_t t_detId = -1, Int_t t_trackId = -1,
                                            Int_t t_depositIndex = -1) = 0;

  /* sync adc clock with absolute time */
  inline void SyncWithADCClock(Double_t &t_time) { t_time = TMath::Floor(t_time * fSampleRate) / fSampleRate; };

  virtual BSEmcWaveform *CallModifiers(BSEmcWaveform *t_wf)
  {
    TIter iterModifiers(&fListOfWaveformModifiers);
    while (BSEmcAbsWaveformModifier *wfmod = dynamic_cast<BSEmcAbsWaveformModifier *>(iterModifiers())) {
      wfmod->Modify(t_wf);
    }
    return t_wf;
  }

  TList fListOfWaveformModifiers{};
  Double_t fSampleRate{-1}; 

  TString fMCDepositBranchName{""};
  ClassDef(BSEmcAbsWaveformSimulator, 3);
};

#endif /*BSEMCABSWAVEFORMSIMULATOR_HH*/