MCTrack.hh

 
#ifndef __RAT_DS_MCTrack__
#define __RAT_DS_MCTrack__
 
#include <TObject.h>
 
#include <RAT/DS/MCTrackStep.hh>
#include <RAT/ListHelp.hh>
 
namespace RAT {
namespace DS {
 
class MCTrack : public TObject {
 public:
  MCTrack() : TObject() {}
  virtual ~MCTrack() {}
 
  virtual Int_t GetID() const { return id; }
  virtual void SetID(Int_t _id) { id = _id; }
 
  virtual Int_t GetParentID() const { return parentID; }
  virtual void SetParentID(Int_t _parentID) { parentID = _parentID; }
 
  virtual Int_t GetPDGCode() const { return pdgcode; }
  virtual void SetPDGCode(Int_t _pdgcode) { pdgcode = _pdgcode; }
 
  virtual const std::string &GetParticleName() const { return particleName; }
  virtual void SetParticleName(const std::string &_particleName) { particleName = _particleName; }
 
  virtual MCTrackStep *GetMCTrackStep(Int_t i) { return &step[i]; }
  virtual Int_t GetMCTrackStepCount() const { return step.size(); }
  virtual MCTrackStep *AddNewMCTrackStep() {
    step.resize(step.size() + 1);
    return &step.back();
  }
  virtual void PruneMCTrackStep() { step.resize(0); }
  virtual MCTrackStep *GetLastMCTrackStep() {
    if (step.size() == 0) {
      return NULL;
    }
    return &step.back();
  }
 
  virtual void PruneIntermediateMCTrackSteps() {
    if (step.size() > 2) {
      step[1] = step.back();
      step.resize(2);
    }
  }
 
  virtual Double_t GetLength() const { return length; }
  virtual void SetLength(Double_t _length) { length = _length; }
 
  virtual Double_t GetDepositedEnergy() const { return depositedEnergy; }
  virtual void SetDepositedEnergy(Double_t _depositedEnergy) { depositedEnergy = _depositedEnergy; }
 
  ClassDef(MCTrack, 2);
 
 protected:
  Int_t id;
  Int_t parentID;
  Int_t pdgcode;
  Double_t length;
  Double_t depositedEnergy;
  std::string particleName;
  std::vector<MCTrackStep> step;
};
 
}  // namespace DS
}  // namespace RAT
 
#endif