Skip to content
Snippets Groups Projects

Cleaning up single track fitting

Closed Whitney Armstrong requested to merge track_fitting into master
Files
3
@@ -6,6 +6,10 @@
#include <iostream>
R__LOAD_LIBRARY(libfmt.so)
#include "fmt/core.h"
#include "fmt/color.h"
R__LOAD_LIBRARY(libeicd.so)
R__LOAD_LIBRARY(libDD4pod.so)
#include "dd4pod/Geant4ParticleCollection.h"
@@ -14,6 +18,7 @@ R__LOAD_LIBRARY(libDD4pod.so)
#include "eicd/ClusterData.h"
#include "eicd/TrackerHitCollection.h"
#include "common_bench/particles.h"
#include "common_bench/util.h"
namespace cb = common_bench;
@@ -104,6 +109,15 @@ 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 n_events = df.Count();
auto n_tracks = df0.Filter("nTracks>0").Count();
auto n_events_fwd = df0.Filter("eta0>1.4").Count();
auto n_events_bwd = df0.Filter("eta0<-1.4").Count();
auto n_tracks_fwd = df0.Filter("eta0>1.4").Filter("nTracks>0").Count();
auto n_tracks_bwd = df0.Filter("eta0<-1.4").Filter("nTracks>0").Count();
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},
@@ -144,6 +158,19 @@ int rec_single_tracks(const char* fname = "topside/rec_single_tracks.root")
auto hBarrel_Neta = df0.Histo1D({"hBarrel_Neta", "; #eta [deg.]", 50, -4, 4 }, "eta0");
auto hEndcap_Neta = df0.Histo1D({"hEndcap_Neta", "; #eta [deg.]", 50, -4, 4 }, "eta0");
// -----------------------------------------------
fmt::print(fmt::emphasis::bold | fg(fmt::color::forest_green), " - {} events thrown\n", *n_events);
fmt::print(fmt::emphasis::bold | fg(fmt::color::forest_green), " - {} events with track\n", *n_tracks);
fmt::print(fmt::emphasis::bold | fg(fmt::color::forest_green), " --> {} \n", double(*n_tracks)/double(*n_events));
fmt::print(fmt::emphasis::bold | fg(fmt::color::forest_green), " - {} events thrown\n", *n_events_fwd);
fmt::print(fmt::emphasis::bold | fg(fmt::color::forest_green), " - {} events with track\n", *n_tracks_fwd);
fmt::print(fmt::emphasis::bold | fg(fmt::color::forest_green), " --> {} \n", double(*n_tracks_fwd)/double(*n_events_fwd));
fmt::print(fmt::emphasis::bold | fg(fmt::color::forest_green), " - {} events thrown\n", *n_events_bwd);
fmt::print(fmt::emphasis::bold | fg(fmt::color::forest_green), " - {} events with track\n", *n_tracks_bwd);
fmt::print(fmt::emphasis::bold | fg(fmt::color::forest_green), " --> {} \n", double(*n_tracks_bwd)/double(*n_events_bwd));
auto c = new TCanvas();
h_nTracks->DrawCopy();
Loading