draft

andiwand · Jan 30, 2024 · ff32af7 · ff32af7
1 parent 223ae74
commit ff32af7
Show file tree

Hide file tree

Showing 14 changed files with 362 additions and 408 deletions.
diff --git a/Examples/Algorithms/TruthTracking/ActsExamples/TruthTracking/TrackTruthMatcher.cpp b/Examples/Algorithms/TruthTracking/ActsExamples/TruthTracking/TrackTruthMatcher.cpp
@@ -101,7 +101,8 @@ ActsExamples::ProcessCode TrackTruthMatcher::execute(
 
     if ((!m_cfg.doubleMatching && recoMatched) ||
         (m_cfg.doubleMatching && recoMatched && truthMatched)) {
-      trackParticleMatching[track.index()] = {majorityParticleId};
+      trackParticleMatching[track.index()] = {majorityParticleId,
+                                              particleHitCounts};
 
       auto& particleTrackMatch = particleTrackMatching[majorityParticleId];
       if (!particleTrackMatch.track) {
@@ -120,7 +121,7 @@ ActsExamples::ProcessCode TrackTruthMatcher::execute(
         ++particleTrackMatch.duplicates;
       }
     } else {
-      trackParticleMatching[track.index()] = {std::nullopt};
+      trackParticleMatching[track.index()] = {std::nullopt, particleHitCounts};
 
       auto& particleTrackMatch = particleTrackMatching[majorityParticleId];
       ++particleTrackMatch.fakes;

diff --git a/Examples/Framework/include/ActsExamples/EventData/SimParticle.hpp b/Examples/Framework/include/ActsExamples/EventData/SimParticle.hpp
@@ -50,6 +50,7 @@ struct SecondaryVertexIdGetter {
 using SimBarcode = ::ActsFatras::Barcode;
 using SimParticle = ::ActsFatras::Particle;
 /// Store particles ordered by particle identifier.
+using SimBarcodeContainer = ::boost::container::flat_set<SimBarcode>;
 using SimParticleContainer =
     ::boost::container::flat_set<SimParticle, detail::CompareParticleId>;
 

diff --git a/Examples/Framework/include/ActsExamples/EventData/TruthMatching.hpp b/Examples/Framework/include/ActsExamples/EventData/TruthMatching.hpp
@@ -10,6 +10,7 @@
 
 #include "ActsExamples/EventData/SimParticle.hpp"
 #include "ActsExamples/EventData/Track.hpp"
+#include "ActsExamples/Validation/TrackClassification.hpp"
 
 #include <cstdint>
 #include <map>
@@ -19,6 +20,10 @@ namespace ActsExamples {
 
 struct TrackMatchEntry {
   std::optional<SimBarcode> particle;
+
+  /// Number of hits on the track that are associated to a particle
+  /// Sorted by decreasing number of hits
+  std::vector<ParticleHitCount> contributingParticles;
 };
 
 struct ParticleMatchEntry {

diff --git a/Examples/Framework/include/ActsExamples/Validation/TrackClassification.hpp b/Examples/Framework/include/ActsExamples/Validation/TrackClassification.hpp
@@ -10,7 +10,7 @@
 
 #include "ActsExamples/EventData/Index.hpp"
 #include "ActsExamples/EventData/ProtoTrack.hpp"
-#include "ActsExamples/EventData/Trajectories.hpp"
+#include "ActsExamples/EventData/Track.hpp"
 #include "ActsFatras/EventData/Barcode.hpp"
 
 #include <cstddef>

diff --git a/Examples/Io/Csv/src/CsvTrackWriter.cpp b/Examples/Io/Csv/src/CsvTrackWriter.cpp
@@ -14,6 +14,7 @@
 #include "Acts/EventData/VectorMultiTrajectory.hpp"
 #include "Acts/Utilities/Helpers.hpp"
 #include "Acts/Utilities/MultiIndex.hpp"
+#include "ActsExamples/EventData/IndexSourceLink.hpp"
 #include "ActsExamples/EventData/Track.hpp"
 #include "ActsExamples/Framework/AlgorithmContext.hpp"
 #include "ActsExamples/Utilities/Paths.hpp"