Resolve "pi0 resolution for ECal barrel"

Closes #19 (closed)

benchmarks/barrel_ecal/scripts/emcal_barrel_pions_analysis.cxx

@@ -8,18 +8,18 @@
 
 #include "dd4pod/Geant4ParticleCollection.h"
 #include "dd4pod/CalorimeterHitCollection.h"
-/*
+
 #include "benchmark.h"
 #include "mt.h"
 #include "util.h"
-*/
+
 #include "TCanvas.h"
 #include "TStyle.h"
 #include "TMath.h"
 #include "TH1.h"
 #include "TF1.h"
 #include "TH1D.h"
-#include <TFitResult.h>
+#include "TFitResult.h"
 
 using ROOT::RDataFrame;
 using namespace ROOT::VecOps;
@@ -43,7 +43,7 @@ void emcal_barrel_pions_analysis(const char* input_fname = "sim_output/sim_emcal
   std:string detector = "Barrel_emcal";
   double resolutionTarget = 0.1;
 
-  eic::util::Test pion0_Energy_resolution{
+  eic::util::Test pi0_energy_resolution{
       {{"name", fmt::format("{}_energy_resolution", test_tag)},
        {"title", "Pion0 Energy resolution"},
        {"description",
@@ -83,7 +83,7 @@ void emcal_barrel_pions_analysis(const char* input_fname = "sim_output/sim_emcal
   return result;
   };
 
-  // Layer Sampling fraction = Esampling / Ethrown
+  // Sampling fraction = Esampling / Ethrown
   auto fsam = [](const std::vector<double>& sampled, const std::vector<double>& thrown) {
     std::vector<double> result;
     auto it_sam = sampled.cbegin();
@@ -116,6 +116,11 @@ void emcal_barrel_pions_analysis(const char* input_fname = "sim_output/sim_emcal
     return result;
   };
 
+  // Relative Energy Resolution = (Esampling - Ethrown)/Ethrown
+  auto getpid = [](std::vector<dd4pod::Geant4ParticleData> const& input) {
+    return input[2].pdgID();
+  };
+
   // Define variables
   auto d1 = d0.Define("Ethr",  Ethr,  {"mcparticles"})
 	      .Define("nhits",  nhits,      {"EcalBarrelHits"})
@@ -123,13 +128,15 @@ void emcal_barrel_pions_analysis(const char* input_fname = "sim_output/sim_emcal
         .Define("fsam",   fsam,       {"Esim","Ethr"})
         .Define("dE",     eResol,     {"Esim","Ethr"})
         .Define("dE_rel", eResol_rel, {"Esim","Ethr"})
+        .Define("pid", getpid, {"mcparticles"})
 	      ;
 
   // Define Histograms
   auto hEthr  = d1.Histo1D({"hEthr",  "Thrown Energy; Thrown Energy [GeV]; Events",        100,  0.0,    7.5}, "Ethr");
   auto hNhits = d1.Histo1D({"hNhits", "Number of hits per events; Number of hits; Events", 100,  0.0, 2000.0}, "nhits");
   auto hEsim  = d1.Histo1D({"hEsim",  "Energy Deposit; Energy Deposit [GeV]; Events",      100,  0.0,    1.0}, "Esim");
-  auto hfsam  = d1.Histo1D({"hfsam",  "Layer Sampling Fraction; Layer Sampling Fraction; Events",      100,  0.0,    0.1}, "fsam");
+  auto hfsam  = d1.Histo1D({"hfsam",  "Sampling Fraction; Sampling Fraction; Events",      100,  0.0,    0.1}, "fsam");
+  auto hpid   = d1.Histo1D({"hpid",  "PID; PID; Count",      100,  -220,    220}, "pid");
 
   // Event Counts
   auto nevents_thrown      = d1.Count();
@@ -175,6 +182,15 @@ void emcal_barrel_pions_analysis(const char* input_fname = "sim_output/sim_emcal
   c4->SaveAs("results/emcal_barrel_pions_fsam.png");
   c4->SaveAs("results/emcal_barrel_pions_fsam.pdf");
 
+  TCanvas *c5 = new TCanvas("c5", "c5", 700, 500);
+  c5->SetLogy(1);
+  hpid->GetYaxis()->SetTitleOffset(1.4);
+  hpid->SetLineWidth(2);
+  hpid->SetLineColor(kBlue);
+  hpid->DrawClone();
+  c5->SaveAs("results/emcal_barrel_pions_pid.png");
+  c5->SaveAs("results/emcal_barrel_pions_pid.pdf");
+
   //Energy Resolution Work
   auto hdE          = d1.Histo1D({"hdE",      "dE; dE[GeV]; Events",              20, -7.5, 7.5}, "dE");//changed from 100
   auto hdE_rel      = d1.Histo1D({"hdE_rel",  "dE Relative; dE Relative; Events", 20, -2, 2}, "dE_rel");//changed from 100
@@ -182,13 +198,12 @@ void emcal_barrel_pions_analysis(const char* input_fname = "sim_output/sim_emcal
   double* res       = hdE_rel->GetFunction("gaus")->GetParameters();
 
   //Pass/Fail
-  /*
+  
   if (res[2] <= resolutionTarget) {
-    pion0_energy_resolution.pass(res[2]);
+    pi0_energy_resolution.pass(res[2]);
   } else {
-    pion0_energy_resolution.fail(res[2]);
+    pi0_energy_resolution.fail(res[2]);
   }
-  */
   //std::printf("Energy Resolution is %f\n", res[2]);
 
   //Energy Resolution Histogram Plotting
@@ -211,7 +226,7 @@ void emcal_barrel_pions_analysis(const char* input_fname = "sim_output/sim_emcal
   cdE_rel->SaveAs("results/emcal_barrel_pi0_dE_rel.png");
   cdE_rel->SaveAs("results/emcal_barrel_pi0_dE_rel.pdf");
 
-  //eic::util::write_test({pion0_energy_resolution}, fmt::format("{}_pions.json", detector));
+  eic::util::write_test({pi0_energy_resolution}, fmt::format("{}_pions.json", detector));
 
 
 }