TrigScintCluster.h

 
#ifndef TRIGSCINT_EVENT_TRIGSCINTCLUSTER_H_
#define TRIGSCINT_EVENT_TRIGSCINTCLUSTER_H_
 
// STL
#include <iostream>
#include <set>
 
// ldmx-sw
#include "Recon/Event/EventConstants.h"
#include "TrigScint/Event/TestBeamHit.h"
#include "TrigScint/Event/TrigScintHit.h"
 
namespace ldmx {
 
class TrigScintCluster {
 public:
  TrigScintCluster() = default;
 
  virtual ~TrigScintCluster();
 
  friend std::ostream &operator<<(std::ostream &o, const TrigScintCluster &d);
 
  void clear(Option_t *option = "");  // override;
 
  void addHit(uint idx, const ldmx::TrigScintHit *hit);
 
  void setSeed(int idx) { seed_ = idx; }
 
  void setEnergy(double energy) { energy_ = energy; }
 
  void setPE(float PE) { pe_ = PE; }
 
  void setNHits(int nHits) { n_hits_ = nHits; }
 
  void setIDs(std::vector<unsigned int> &hitIDs) { hit_ids_ = hitIDs; }
 
  void setCentroidXYZ(double x, double y, double z) {
    centroid_x_ = x;
    centroid_y_ = y;
    centroid_z_ = z;
  }
 
  void setCentroid(double centroid) { centroid_ = centroid; }
 
  void setTime(float t) { time_ = t; }
 
  float getTime() const { return time_; }
 
  void setBeamEfrac(float e) { beam_efrac_ = e; }
 
  float getBeamEfrac() const { return beam_efrac_; }
 
  int getSeed() const { return seed_; }
 
  double getEnergy() const { return energy_; }
 
  double getPE() const { return pe_; }
 
  int getNHits() const { return n_hits_; }
 
  double getCentroidX() const { return centroid_x_; }
 
  double getCentroidY() const { return centroid_y_; }
 
  double getCentroidZ() const { return centroid_z_; }
 
  const std::vector<unsigned int> &getHitIDs() const { return hit_ids_; }
 
  double getCentroid() const { return centroid_; }
 
  bool operator<(const TrigScintCluster &rhs) const {
    return this->getEnergy() < rhs.getEnergy();
  }
 
 private:
  // hits forming the cluster
  std::vector<unsigned int> hit_ids_;
 
  // total cluster energy depostion
  double energy_{0};
 
  // number of hits forming the cluster
  int n_hits_{0};
 
  // total cluster photoelectron count
  float pe_{0};
 
  // index of cluster seeding hit
  int seed_{-1};
 
  // hit centroid in units of channel nb: energy weighted average of the IDs of
  // the hits forming the cluster
  double centroid_{-1};
 
  // hit centroid in x [mm] (not implemented)
  double centroid_x_{0};
 
  // hit centroid in y [mm] (not implemented)
  double centroid_y_{0};
 
  // hit centroid in z [mm] (not implemented)
  double centroid_z_{0};
 
  // fraction of cluster energy deposited in a sim hit associated with beam
  // electrons
  float beam_efrac_{0.};
 
  // cluster time: energy weighted average of the times of the hits forming the
  // cluster
  float time_{0.};
 
  ClassDef(TrigScintCluster, 3);
};
}  // namespace ldmx
 
#endif  // TRIGSCINT_EVENT_TRIGSCINTCLUSTER_H_