MCSummary.hh

 
#ifndef __RAT_DS_MCSummary__
#define __RAT_DS_MCSummary__
 
#include <TObject.h>
#include <TVector3.h>
 
#include <map>
#include <string>
 
namespace RAT {
namespace DS {
 
typedef std::map<std::string, double> StringDoubleMap;
 
class MCSummary : public TObject {
 public:
  MCSummary() : TObject() {}
  virtual ~MCSummary() {}
 
  virtual const TVector3 &GetEnergyCentroid() const { return energyCentroid; }
  virtual void SetEnergyCentroid(const TVector3 &_energyCentroid) { energyCentroid = _energyCentroid; }
 
  virtual const TVector3 &GetEnergyRMS() const { return energyRMS; }
  virtual void SetEnergyRMS(const TVector3 &_energyRMS) { energyRMS = _energyRMS; }
 
  virtual Double_t GetEnergyLossByVolume(const char *volume) { return energyLoss[volume]; }
  virtual void SetEnergyLossByVolume(StringDoubleMap _energyLoss) { energyLoss = _energyLoss; }
 
  virtual Double_t GetTotalScintEdep() const { return totalScintEdep; };
  virtual void SetTotalScintEdep(Double_t _totalScintEdep) { totalScintEdep = _totalScintEdep; }
 
  virtual Double_t GetTotalScintEdepQuenched() const { return totalScintEdepQuenched; }
  virtual void SetTotalScintEdepQuenched(Double_t _totalScintEdepQuenched) {
    totalScintEdepQuenched = _totalScintEdepQuenched;
  }
 
  virtual const TVector3 &GetTotalScintCentroid() const { return totalScintCentroid; }
  virtual void SetTotalScintCentroid(const TVector3 &_totalScintCentroid) { totalScintCentroid = _totalScintCentroid; }
 
  virtual Double_t GetInitialScintTime() const { return initialScintTime; }
  virtual void SetInitialScintTime(Double_t _initialScintTime) { initialScintTime = _initialScintTime; }
 
  virtual Int_t GetNumScintPhoton() const { return numScintPhoton; }
  virtual void SetNumScintPhoton(Int_t _numScintPhoton) { numScintPhoton = _numScintPhoton; }
 
  virtual Int_t GetNumReemitPhoton() const { return numReemitPhoton; }
  virtual void SetNumReemitPhoton(Int_t _numReemitPhoton) { numReemitPhoton = _numReemitPhoton; }
 
  virtual Int_t GetNumCerenkovPhoton() const { return numCerenkovPhoton; }
  virtual void SetNumCerenkovPhoton(Int_t _numCerenkovPhoton) { numCerenkovPhoton = _numCerenkovPhoton; }
 
  virtual void SetPhotonInfo(std::vector<std::vector<double>> _photonComposition) {
    photonComposition = _photonComposition;
  }
  virtual std::vector<std::vector<double>> GetPhotonInfo() const { return photonComposition; }
  virtual double GetNPhotonsTriggers() const { return photonComposition.size(); }
  virtual double GetCherenkovPhotons(int _trigger) const { return photonComposition[_trigger][2]; }
  virtual double GetScintPhotons(int _trigger) const { return photonComposition[_trigger][0]; }
  virtual double GetRemmitPhotons(int _trigger) const { return photonComposition[_trigger][1]; }
  virtual double GetTimePhotons(int _trigger) const { return photonComposition[_trigger][3]; }
 
  virtual void SetPMTPhotonInfo(std::vector<std::vector<double>> _pmtHitVector) { pmtHitVector = _pmtHitVector; }
 
  virtual std::vector<std::vector<double>> GetPMTPhotonInfo() { return pmtHitVector; }
 
  ClassDef(MCSummary, 4);
 
 protected:
  Int_t numScintPhoton;
  Int_t numReemitPhoton;
  Int_t numCerenkovPhoton;
  Double_t totalScintEdep;
  Double_t totalScintEdepQuenched;
  Double_t initialScintTime;
  TVector3 energyCentroid;
  TVector3 energyRMS;
  TVector3 opticalCentroid;
  TVector3 opticalRMS;
  TVector3 totalScintCentroid;
  StringDoubleMap energyLoss;
  std::vector<std::vector<double>> photonComposition;
  std::vector<std::vector<double>> pmtHitVector;
};
 
}  // namespace DS
}  // namespace RAT
 
#endif