Skip to content

Commit 3aae9c5

Browse files
authored
Merge pull request #1173 from niermann999/fix-finding-writer
fix: Number of truth particles in finding performance writer
2 parents 078a74b + 7eab8ab commit 3aae9c5

File tree

2 files changed

+12
-1
lines changed

2 files changed

+12
-1
lines changed

examples/run/cpu/truth_finding_example.cpp

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -149,6 +149,8 @@ int seq_run(const traccc::opts::track_finding& finding_opts,
149149
seeds.push_back(truth_track_candidates.tracks.at(i_trk).params());
150150
}
151151

152+
std::cout << "Number of seeds: " << seeds.size() << std::endl;
153+
152154
// Read measurements
153155
traccc::measurement_collection_types::host measurements_per_event{
154156
&host_mr};

performance/src/efficiency/finding_performance_writer.cpp

Lines changed: 10 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -241,9 +241,16 @@ void finding_performance_writer::write_common(
241241

242242
// For each truth particle...
243243
for (auto const& [pid, ptc] : evt_data.m_particle_map) {
244-
std::size_t num_measurements = 0;
244+
245+
auto ptc_particle =
246+
detail::particle_from_pdg_number<scalar>(ptc.particle_type);
247+
if (ptc_particle.pdg_num() == 0) {
248+
// TODO: Add some debug logging here.
249+
continue;
250+
}
245251

246252
// Find the number of measurements belonging to this track
253+
std::size_t num_measurements = 0;
247254
if (auto it = evt_data.m_ptc_to_meas_map.find(ptc);
248255
it != evt_data.m_ptc_to_meas_map.end()) {
249256
num_measurements = it->second.size();
@@ -273,6 +280,8 @@ void finding_performance_writer::write_common(
273280
total_matched_truth_particles++;
274281
n_matched_seeds_for_particle = it->second;
275282
total_duplicate_tracks += n_matched_seeds_for_particle - 1;
283+
} else {
284+
TRACCC_DEBUG("Not matched: " << pid);
276285
}
277286

278287
// Finds how many (fake) tracks were made with at least one hit from the

0 commit comments

Comments
 (0)