feat: Vertex classification for VertexPerformanceWriter in Examples (…

…acts-project#3044) This modifies the truth matching in the `VertexPerformanceWriter` and adds classification for clean/merged/split reco vertices. blocked by - acts-project#2998
Ragansu · Apr 19, 2024 · c938eb4 · c938eb4
1 parent 862e996
commit c938eb4
Show file tree

Hide file tree

Showing 13 changed files with 441 additions and 278 deletions.
diff --git a/CI/physmon/reference/performance_amvf_gridseeder_seeded_hist.root b/CI/physmon/reference/performance_amvf_gridseeder_seeded_hist.root
diff --git a/CI/physmon/reference/performance_amvf_gridseeder_ttbar_hist.root b/CI/physmon/reference/performance_amvf_gridseeder_ttbar_hist.root
diff --git a/CI/physmon/reference/performance_amvf_orthogonal_hist.root b/CI/physmon/reference/performance_amvf_orthogonal_hist.root
diff --git a/CI/physmon/reference/performance_amvf_seeded_hist.root b/CI/physmon/reference/performance_amvf_seeded_hist.root
diff --git a/CI/physmon/reference/performance_amvf_truth_estimated_hist.root b/CI/physmon/reference/performance_amvf_truth_estimated_hist.root
diff --git a/CI/physmon/reference/performance_amvf_truth_smeared_hist.root b/CI/physmon/reference/performance_amvf_truth_smeared_hist.root
diff --git a/CI/physmon/reference/performance_amvf_ttbar_hist.root b/CI/physmon/reference/performance_amvf_ttbar_hist.root
diff --git a/CI/physmon/reference/performance_ivf_orthogonal_hist.root b/CI/physmon/reference/performance_ivf_orthogonal_hist.root
diff --git a/CI/physmon/reference/performance_ivf_seeded_hist.root b/CI/physmon/reference/performance_ivf_seeded_hist.root
diff --git a/CI/physmon/reference/performance_ivf_truth_estimated_hist.root b/CI/physmon/reference/performance_ivf_truth_estimated_hist.root
diff --git a/CI/physmon/reference/performance_ivf_truth_smeared_hist.root b/CI/physmon/reference/performance_ivf_truth_smeared_hist.root
diff --git a/Examples/Io/Performance/ActsExamples/Io/Performance/VertexPerformanceWriter.cpp b/Examples/Io/Performance/ActsExamples/Io/Performance/VertexPerformanceWriter.cpp
diff --git a/Examples/Io/Performance/ActsExamples/Io/Performance/VertexPerformanceWriter.hpp b/Examples/Io/Performance/ActsExamples/Io/Performance/VertexPerformanceWriter.hpp
@@ -30,6 +30,13 @@ class TTree;
 
 namespace ActsExamples {
 
+enum class RecoVertexClassification {
+  Unknown = 0,
+  Clean,
+  Merged,
+  Split,
+};
+
 /// @class VertexPerformanceWriter
 ///
 /// Writes out the number of reconstructed primary vertices along with
@@ -106,6 +113,10 @@ class VertexPerformanceWriter final
   int m_nRecoVtx = -1;
   /// Number of true vertices
   int m_nTrueVtx = -1;
+  /// Number of merged vertices
+  int m_nMergedVtx = -1;
+  /// Number of split vertices
+  int m_nSplitVtx = -1;
   /// Number of vertices in detector acceptance
   int m_nVtxDetAcceptance = -1;
   /// Max. number of reconstructable vertices (detector acceptance + tracking
@@ -155,6 +166,9 @@ class VertexPerformanceWriter final
   /// Number of tracks associated with the truth vertex
   std::vector<int> m_nTracksOnTruthVertex;
 
+  /// Classification of the reconstructed vertex see RecoVertexClassification
+  std::vector<int> m_recoVertexClassification;
+
   // True 4D vertex position
   std::vector<double> m_truthX;
   std::vector<double> m_truthY;