Resolve "pi0 resolution for ECal barrel"

Closes #19 (closed)

benchmarks/barrel_ecal/scripts/emcal_barrel_pions_analysis.cxx

@@ -20,6 +20,8 @@
 #include "TF1.h"
 #include "TH1D.h"
 #include <TFitResult.h>
+#include "TSystem.h"
+#include <fstream>
 
 using ROOT::RDataFrame;
 using namespace ROOT::VecOps;
@@ -51,9 +53,17 @@ void emcal_barrel_pions_analysis(const char* input_fname = "sim_output/sim_emcal
        {"quantity", "resolution (in %)"},
        {"target", std::to_string(resolutionTarget)}}};
 */
-  gSystem->Exec("cp sim_output/sim_emcal_barrel_uniform_pions.root results/sim_emcal_barrel_uniform_pions.root");
+
   ROOT::EnableImplicitMT();
   ROOT::RDataFrame d0("events", input_fname);
+/*
+  auto colNames = d0.GetColumnNames();
+
+  ofstream out;
+  out.open("results/column_list.txt");
+  for (auto &&col : colNames){out << col << endl; cout << col << endl;}
+  out.close();
+  */
 
   // Thrown Energy [GeV]
   auto Ethr = [](std::vector<dd4pod::Geant4ParticleData> const& input) {
@@ -118,6 +128,13 @@ void emcal_barrel_pions_analysis(const char* input_fname = "sim_output/sim_emcal
     return result;
   };
 
+  // Thrown Energy [GeV]
+  auto Ethr = [](std::vector<dd4pod::Geant4ParticleData> const& input) {
+    std::vector<double> result;
+    result.push_back(input[2].charge);
+  return result;
+  };
+
   // Define variables
   auto d1 = d0.Define("Ethr",  Ethr,  {"mcparticles"})
 	      .Define("nhits",  nhits,      {"EcalBarrelHits"})
@@ -126,6 +143,7 @@ void emcal_barrel_pions_analysis(const char* input_fname = "sim_output/sim_emcal
         .Define("fsam",   fsam2,      {"Esim","Ethr"})
         .Define("dE",     eResol,     {"Esim", "Ethr"})
         .Define("dE_rel", eResol_rel, {"Esim", "Ethr"})
+        .Define("charge", {"mcparticles"})
 	      ;
 
   // Define Histograms
@@ -133,6 +151,7 @@ void emcal_barrel_pions_analysis(const char* input_fname = "sim_output/sim_emcal
   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",  "Sampling Fraction; Sampling Fraction; Events",      100,  0.0,    0.1}, "fsam");
+  auto hcharge = d1.Histo1D({"hcharge", "charge; charge; Count", 10, -2, 2}, "charge");
 
   // Event Counts
   auto nevents_thrown      = d1.Count();
@@ -178,6 +197,14 @@ 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);
+  hcharge->GetYaxis()->SetTitleOffset(1.4);
+  hcharge->SetLineWidth(2);
+  hcharge->SetLineColor(kBlue);
+  hcharge->DrawClone();
+  c5->SaveAs("results/emcal_barrel_pions_charge.png");
+  c5->SaveAs("results/emcal_barrel_pions_chage.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, -7.5, 7.5}, "dE_rel");//changed from 100