extend cluster info

LDMX-Software · Apr 17, 2024 · 325f160 · 325f160
1 parent 1e0f9dc
commit 325f160
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/include/Ecal/Event/EcalCluster.h b/include/Ecal/Event/EcalCluster.h
@@ -72,6 +72,11 @@ class EcalCluster {
    */
   void setIDs(std::vector<unsigned int>& hitIDs) { hitIDs_ = hitIDs; }
 
+  void setHitValsX(std::vector<float>& x) { hitX_ = x; }
+  void setHitValsY(std::vector<float>& x) { hitY_ = x; }
+  void setHitValsZ(std::vector<float>& x) { hitZ_ = x; }
+  void setHitValsE(std::vector<float>& x) { hitE_ = x; }
+
   /**
    * Sets the three coordinates of the cluster centroid
    * @param x The x coordinate.
@@ -123,6 +128,12 @@ class EcalCluster {
 
   const std::vector<unsigned int>& getHitIDs() const { return hitIDs_; }
 
+  // ability to store limited hit info
+  const std::vector<float>& getHitX() const { return hitX_; }
+  const std::vector<float>& getHitY() const { return hitY_; }
+  const std::vector<float>& getHitZ() const { return hitZ_; }
+  const std::vector<float>& getHitE() const { return hitE_; }
+
   bool operator<(const EcalCluster& rhs) const {
     return this->getEnergy() < rhs.getEnergy();
   }
@@ -141,6 +152,10 @@ class EcalCluster {
   double DYDZ_{0};
   double errDXDZ_{0};
   double errDYDZ_{0};
+  std::vector<float> hitX_;
+  std::vector<float> hitY_;
+  std::vector<float> hitZ_;
+  std::vector<float> hitE_;
 
   ClassDef(EcalCluster, 1);
 };