SimCalorimeterHit.h

Go to the documentation of this file.

 
#ifndef SIMCORE_EVENT_SIMCALORIMETERHIT_H_
#define SIMCORE_EVENT_SIMCALORIMETERHIT_H_
 
// ROOT
#include "TObject.h"  //For ClassDef
 
// LDMX
#include "SimCore/Event/SimParticle.h"
 
namespace ldmx {
 
class SimCalorimeterHit {
 public:
  static const std::string ECAL_COLLECTION;
 
  static const std::string HCAL_COLLECTION;
 
  struct Contrib {
    int incidentID{-1};
 
    int trackID{-1};
 
    int pdgCode{0};
 
    float edep{0};
 
    float time{0};
  };
 
  SimCalorimeterHit() = default;
 
  virtual ~SimCalorimeterHit() = default;
 
  void Clear();
 
  void Print() const;
 
  int getID() const { return id_; }
 
  void setID(const int id) { this->id_ = id; }
 
  float getEdep() const { return edep_; }
 
  void setEdep(const float edep) { this->edep_ = edep; }
 
  std::vector<float> getPosition() const { return {x_, y_, z_}; }
 
  std::vector<float> getPreStepPosition() const {
    return {preStepX_, preStepY_, preStepZ_};
  }
  std::vector<float> getPostStepPosition() const {
    return {postStepX_, postStepY_, postStepZ_};
  }
  void setPosition(const float x, const float y, const float z) {
    this->x_ = x;
    this->y_ = y;
    this->z_ = z;
  }
  void setPreStepPosition(const float x, const float y, const float z) {
    preStepX_ = x;
    preStepY_ = y;
    preStepZ_ = z;
  }
  void setPostStepPosition(const float x, const float y, const float z) {
    postStepX_ = x;
    postStepY_ = y;
    postStepZ_ = z;
  }
 
  void setPathLength(const float length) { pathLength_ = length; }
  float getPathLength() const { return pathLength_; }
  void setPreStepTime(const float time) { preStepTime_ = time; }
  void setPostStepTime(const float time) { postStepTime_ = time; }
 
  void setVelocity(float velocity) { velocity_ = velocity; }
 
  float getTime() const { return time_; }
  float getPreStepTime() const { return preStepTime_; }
  float getPostStepTime() const { return postStepTime_; }
 
  float getVelocity() const { return velocity_; }
 
  void setTime(const float time) { this->time_ = time; }
 
  unsigned getNumberOfContribs() const { return nContribs_; }
 
  void addContrib(int incidentID, int trackID, int pdgCode, float edep,
                  float time);
 
  Contrib getContrib(int i) const;
 
  int findContribIndex(int trackID, int pdgCode) const;
 
  void updateContrib(int i, float edep, float time);
 
  bool operator<(const SimCalorimeterHit &rhs) const {
    return this->getTime() < rhs.getTime();
  }
 
 private:
  int id_{0};
 
  float edep_{0};
 
  float x_{0};
 
  float y_{0};
 
  float z_{0};
 
  float time_{0};
 
  std::vector<int> trackIDContribs_;
 
  std::vector<int> incidentIDContribs_;
 
  std::vector<int> pdgCodeContribs_;
 
  std::vector<float> edepContribs_;
 
  std::vector<float> timeContribs_;
 
  unsigned nContribs_{0};
 
  /*
   * Parameters used only for hits corresponding to a single interactions
   * (currently Hcal and TS).
   */
 
  float pathLength_{-1};
 
  float preStepX_{0};
  float preStepY_{0};
  float preStepZ_{0};
  float preStepTime_{0};
  float postStepX_{0};
  float postStepY_{0};
  float postStepZ_{0};
  float postStepTime_{0};
 
  float velocity_{-1};
 
  ClassDef(SimCalorimeterHit, 4)
};
}  // namespace ldmx
 
#endif