From cb7a4d82e1c72319cffce323e4a1b748c3308abb Mon Sep 17 00:00:00 2001
From: Whitney Armstrong <warmstrong@anl.gov>
Date: Thu, 26 Aug 2021 16:56:15 +0000
Subject: [PATCH] Added plots do debug hit positions
---
.../options/track_reconstruction.py | 14 ++++++++------
.../scripts/rec_single_tracks.cxx | 19 +++++++++++++++++++
benchmarks/track_fitting/single_tracks.sh | 3 +++
3 files changed, 30 insertions(+), 6 deletions(-)
diff --git a/benchmarks/track_fitting/options/track_reconstruction.py b/benchmarks/track_fitting/options/track_reconstruction.py
index 00ee10d8..89fc8fc7 100644
--- a/benchmarks/track_fitting/options/track_reconstruction.py
+++ b/benchmarks/track_fitting/options/track_reconstruction.py
@@ -119,10 +119,10 @@ algorithms.append(gem_ec_reco)
trk_hit_col = TrackingHitsCollector("trk_hit_col",
inputTrackingHits=[
- str(trk_b_reco.outputHitCollection),
- str(trk_ec_reco.outputHitCollection),
str(vtx_b_reco.outputHitCollection),
str(vtx_ec_reco.outputHitCollection),
+ str(trk_b_reco.outputHitCollection),
+ str(trk_ec_reco.outputHitCollection),
str(gem_ec_reco.outputHitCollection) ],
trackingHits="trackingHits")
algorithms.append( trk_hit_col )
@@ -131,13 +131,15 @@ algorithms.append( trk_hit_col )
sourcelinker = TrackerSourceLinker("sourcelinker",
inputHitCollection=trk_hit_col.trackingHits,
outputSourceLinks="TrackSourceLinks",
- outputMeasurements="TrackMeasurements")
+ outputMeasurements="TrackMeasurements",
+ OutputLevel=DEBUG)
algorithms.append( sourcelinker )
## Track param init
truth_trk_init = TrackParamTruthInit("truth_trk_init",
inputMCParticles="mcparticles",
- outputInitialTrackParameters="InitTrackParams")
+ outputInitialTrackParameters="InitTrackParams",
+ OutputLevel=DEBUG)
#OutputLevel=DEBUG)
algorithms.append( truth_trk_init )
@@ -146,8 +148,8 @@ trk_find_alg = TrackFindingAlgorithm("trk_find_alg",
inputSourceLinks = sourcelinker.outputSourceLinks,
inputMeasurements = sourcelinker.outputMeasurements,
inputInitialTrackParameters= "InitTrackParams",#"InitTrackParamsFromClusters",
- outputTrajectories="trajectories")
- #OutputLevel=DEBUG)
+ outputTrajectories="trajectories",
+ OutputLevel=DEBUG)
algorithms.append( trk_find_alg )
parts_from_fit = ParticlesFromTrackFit("parts_from_fit",
diff --git a/benchmarks/track_fitting/scripts/rec_single_tracks.cxx b/benchmarks/track_fitting/scripts/rec_single_tracks.cxx
index d7835c4f..20c42392 100644
--- a/benchmarks/track_fitting/scripts/rec_single_tracks.cxx
+++ b/benchmarks/track_fitting/scripts/rec_single_tracks.cxx
@@ -104,6 +104,13 @@ int rec_single_tracks(const char* fname = "topside/rec_single_tracks.root")
.Define("N_EndcapHits", [](std::vector<eic::TrackerHitData> hits) { return hits.size();}, {"TrackerEndcapRecHits"})
;
+ auto htrackingHits_x_vs_y = df0.Histo2D({"htrackingHits_x_vs_y", "; x [mm]; y [mm] ", 100, -600, 600, 100, -600, 600},
+ "trackingHits.position.x", "trackingHits.position.y");
+ auto htrackingHits_y_vs_z = df0.Histo2D({"htrackingHits_y_vs_z", "; z [mm]; y [mm]", 100, -1500, 1500, 100, -600, 600},
+ "trackingHits.position.z", "trackingHits.position.y");
+ auto htrackingHits_x_vs_z = df0.Histo2D({"htrackingHits_x_vs_z", "; z [mm]; x [mm]", 100, -1500, 1500, 100, -600, 600},
+ "trackingHits.position.z", "trackingHits.position.x");
+
auto h_nTracks_vs_theta = df0.Histo2D({"h_nTracks_vs_theta", "; #theta; N tracks ", 40,0,180,10, 0, 10}, "theta0","nTracks");
auto h_nTracks_vs_eta = df0.Histo2D({"h_nTracks_vs_eta", "; #eta; N tracks ", 50,-4,4,10, 0, 10}, "eta0","nTracks");
@@ -143,11 +150,23 @@ int rec_single_tracks(const char* fname = "topside/rec_single_tracks.root")
c->SaveAs("results/track_fitting/rec_single_tracks_nTracks.png");
c->SaveAs("results/track_fitting/rec_single_tracks_nTracks.pdf");
+
// -----------------------------------------------
h_pTracks->DrawCopy();
c->SaveAs("results/track_fitting/rec_single_tracks_pTracks.png");
c->SaveAs("results/track_fitting/rec_single_tracks_pTracks.pdf");
+ // -----------------------------------------------
+ c = new TCanvas();
+ htrackingHits_x_vs_y->Draw("colz");
+ c->SaveAs("results/track_fitting/rec_single_tracks_trackingHits_xy.png");
+
+ htrackingHits_x_vs_z->Draw("colz");
+ c->SaveAs("results/track_fitting/rec_single_tracks_trackingHits_xz.png");
+
+ htrackingHits_y_vs_z->Draw("colz");
+ c->SaveAs("results/track_fitting/rec_single_tracks_trackingHits_yz.png");
+
// -----------------------------------------------
c = new TCanvas();
THStack * hs = new THStack("hs_delta_p","; GeV/c ");
diff --git a/benchmarks/track_fitting/single_tracks.sh b/benchmarks/track_fitting/single_tracks.sh
index e10a71f1..0761afed 100644
--- a/benchmarks/track_fitting/single_tracks.sh
+++ b/benchmarks/track_fitting/single_tracks.sh
@@ -90,6 +90,9 @@ then
fi
fi
+
+rootls -t ${JUGGLER_REC_FILE}
+
mkdir -p results/track_fitting
root -b -q "benchmarks/track_fitting/scripts/rec_single_tracks.cxx(\"${JUGGLER_REC_FILE}\")"
--
GitLab