tcs_tests.cxx: plot theta in mrad for various FF detectors

benchmarks/tcs/analysis/tcs_tests.cxx

@@ -30,11 +30,21 @@ void tcs_tests(const char* fname = "rec_tcs.root"){
 
   auto df0 = df.Define("n_parts", "ReconstructedParticles.size()")
                .Define("isQ2gt1", "InclusiveKinematicsTruth.Q2 > 1.0")
-               .Define("n_Q2gt1", "isQ2gt1.size()");
+               .Define("n_Q2gt1", "isQ2gt1.size()")
+               .Define("theta_mrad", "1000. * ReconstructedFFParticles.direction.theta")
+               .Define("isFFB0", "ReconstructedFFParticles.status == 1")
+               .Define("isFFRP", "ReconstructedFFParticles.status == 2")
+               .Define("isFFOMD", "ReconstructedFFParticles.status == 3")
+               .Define("isFFZDC", "ReconstructedFFParticles.status == 4")
+               ;
 
   auto h_n_parts = df0.Histo1D({"h_n_parts", "; h_n_parts n", 10, 0, 10}, "n_parts");
   auto h_Q2      = df0.Histo1D({"h_Q2", "; Q^{2} [GeV^{2}/c^{2}]", 100, 0, 30}, "InclusiveKinematicsTruth.Q2");
-  auto h_ff_status = df0.Histo1D({"h_ff_status", "; FF status", 10, -0.5, 9.5}, "ReconstructedFFParticles.status");
+  auto h_FF      = df0.Histo1D({"h_FF", "; FF status", 10, -0.5, 9.5}, "ReconstructedFFParticles.status");
+  auto h_FF_B0   = df0.Filter("isFFB0").Histo1D({"h_FF_B0", "; FF B0 Theta [mrad]", 10, 0.0, 25.0}, "theta_mrad");
+  auto h_FF_RP   = df0.Filter("isFFRP").Histo1D({"h_FF_RP", "; FF RP Theta [mrad]", 10, 0.0, 25.0}, "theta_mrad");
+  auto h_FF_OMD  = df0.Filter("isFFOMD").Histo1D({"h_FF_OMD", "; FF OMD Theta [mrad]", 10, 0.0, 25.0}, "theta_mrad");
+  auto h_FF_ZDC  = df0.Filter("isFFZDC").Histo1D({"h_FF_ZDC", "; FF ZDC Theta [mrad]", 10, 0.0, 25.0}, "theta_mrad");
   auto n_Q2gt1   = df0.Mean("n_Q2gt1");
   auto n_parts   = df0.Mean("n_parts");
 
@@ -54,7 +64,19 @@ void tcs_tests(const char* fname = "rec_tcs.root"){
   c->SaveAs("results/tcs/n_parts.pdf");
 
   c = new TCanvas();
-  h_ff_status->DrawCopy();
-  c->SaveAs("results/tcs/ff_status.png");
-  c->SaveAs("results/tcs/ff_status.pdf");
+  h_FF->DrawCopy();
+  c->SaveAs("results/tcs/ff.png");
+  c->SaveAs("results/tcs/ff.pdf");
+
+  c = new TCanvas();
+  h_FF_B0->SetLineColor(1);
+  h_FF_B0->DrawCopy();
+  h_FF_RP->SetLineColor(2);
+  h_FF_RP->DrawCopy();
+  h_FF_OMD->SetLineColor(3);
+  h_FF_OMD->DrawCopy();
+  h_FF_ZDC->SetLineColor(4);
+  h_FF_ZDC->DrawCopy();
+  c->SaveAs("results/tcs/ff_theta.png");
+  c->SaveAs("results/tcs/ff_theta.pdf");
 }