MCPMT.hh

 
#ifndef __RAT_DS_MCPMT__
#define __RAT_DS_MCPMT__
 
#include <RAT/DS/MCPhoton.hh>
#include <RAT/Log.hh>
#include <vector>
 
namespace RAT {
namespace DS {
 
class MCPMT : public TObject {
 public:
  MCPMT() : TObject() {}
  virtual ~MCPMT() {}
 
  virtual Int_t GetID() const { return id; };
  virtual void SetID(Int_t _id) { id = _id; };
 
  virtual Double_t GetCharge() const;
 
  virtual std::string GetCreatorProcess() const;
 
  virtual Int_t GetType() const { return type; };
  virtual void SetType(Int_t _type) { type = _type; };
 
  MCPhoton *GetMCPhoton(Int_t i) { return &photon[i]; }
  Int_t GetMCPhotonCount() const { return photon.size(); }
  MCPhoton *AddNewMCPhoton() {
    photon.resize(photon.size() + 1);
    return &photon.back();
  }
  void RemoveMCPhoton(Int_t i) { photon.erase(photon.begin() + i); }
  void PruneMCPhoton() { photon.resize(0); }
  void SortMCPhotons() { std::sort(photon.begin(), photon.end()); }
 
  void PruneNoiseMCPhotons() {
    photon.erase(std::remove_if(photon.begin(), photon.end(), [](const MCPhoton ph) { return ph.IsDarkHit(); }),
                 photon.end());
  }
  void PruneAfterPulseMCPhotons() {
    photon.erase(std::remove_if(photon.begin(), photon.end(), [](const MCPhoton ph) { return ph.IsAfterPulse(); }),
                 photon.end());
  }
 
  ClassDef(MCPMT, 3);
 
 protected:
  Int_t id;
  Int_t type;
  std::vector<MCPhoton> photon;
};
 
}  // namespace DS
}  // namespace RAT
 
#endif