Adding option to input to cluster width

Rosie-Hasan · Oct 15, 2024 · 2c1e7e7 · 2c1e7e7
1 parent 6571520
commit 2c1e7e7
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 7 deletions.
diff --git a/Core/include/Acts/Seeding/GbtsDataStorage.hpp b/Core/include/Acts/Seeding/GbtsDataStorage.hpp
@@ -30,15 +30,17 @@ struct GbtsSP {
   bool m_isPixel;
   float m_phi;
   float m_r; 
-  GbtsSP(const space_point_t *sp, int id, int combined_id, bool isPixel)
-      : SP(sp), gbtsID(id), combined_ID{combined_id}, m_isPixel(isPixel) {
+  float m_ClusterWidth ; 
+  GbtsSP(const space_point_t *sp, int id, int combined_id, bool isPixel, float ClusterWidth)
+      : SP(sp), gbtsID(id), combined_ID{combined_id}, m_isPixel(isPixel), m_ClusterWidth(ClusterWidth) {
     m_phi = std::atan(SP->x() / SP->y());
     m_r = std::sqrt((SP->x()*SP->x())+ (SP->y()*SP->y())) ;   
   };
   bool isPixel() const { return m_isPixel; }
   bool isSCT() const { return !m_isPixel; }
   float phi() const { return m_phi; }
   float r() const {return m_r;}
+  bool ClusterWidth() const { return m_ClusterWidth; }
 };
 
 template <typename space_point_t>
@@ -164,7 +166,7 @@ class GbtsDataStorage {
       return -1;
     }
 
-    int binIndex = pL->getEtaBin(sp.SP->z(), sp.SP->r());
+    int binIndex = pL->getEtaBin(sp.SP->z(), sp.r());
 
     if (binIndex == -1) {
       return -2;
@@ -177,7 +179,8 @@ class GbtsDataStorage {
       float max_tau = 100.0;
       // can't do this bit yet as dont have cluster width
       if (useClusterWidth) {
-        float cluster_width = 1;  // temporary while cluster width not available
+        float cluster_width = sp.ClusterWidth(); 
+        // if (cluster_width == 0) {continue;} //catch for casses that dont have cluster width available 
         min_tau = 6.7 * (cluster_width - 0.2);
         max_tau =
             1.6 + 0.15 / (cluster_width + 0.2) + 6.1 * (cluster_width - 0.2);
@@ -187,7 +190,8 @@ class GbtsDataStorage {
           sp, min_tau, max_tau));  // adding ftf member to nodes
     } else {
       if (useClusterWidth) {
-        float cluster_width = 1;  // temporary while cluster width not available
+        float cluster_width = sp.ClusterWidth();  
+        // if (cluster_width == 0) {continue;} //catch for casses that dont have cluster width available 
         if (cluster_width > 0.2) {
           return -3;
         }

diff --git a/Examples/Algorithms/TrackFinding/src/GbtsSeedingAlgorithm.cpp b/Examples/Algorithms/TrackFinding/src/GbtsSeedingAlgorithm.cpp
@@ -225,14 +225,15 @@ ActsExamples::GbtsSeedingAlgorithm::MakeGbtsSpacePoints(
 
       //check if SP is pixel, dependant of type of SP so must be done in examples 
       bool isPixel = false ;
-      if (source_link.size() == 1) {  // pixels have 1 SL
+      if (sourceLink.size() == 1) {  // pixels have 1 SL
         isPixel = true;
       } else {
         isPixel = false;
       }
 
+      float ClusterWidth = 0 ; //false input as this is not available in examples 
       // fill Gbts vector with current sapce point and ID
-      gbtsSpacePoints.emplace_back(&spacePoint, Gbts_id, combined_id, isPixel); //make new GbtsSP here ! 
+      gbtsSpacePoints.emplace_back(&spacePoint, Gbts_id, combined_id, isPixel, ClusterWidth); //make new GbtsSP here ! 
     }
   }
   ACTS_VERBOSE("Space points successfully assigned Gbts ID");