Skip to content
Snippets Groups Projects

Resolve "Broken pi0 ecal barrel analysis script"

Merged Marshall Scott requested to merge 47-broken-pi0-ecal-barrel-analysis-script into master
1 file
+ 95
139
Compare changes
  • Side-by-side
  • Inline
@@ -29,9 +29,11 @@ using ROOT::RDataFrame;
@@ -29,9 +29,11 @@ using ROOT::RDataFrame;
using namespace ROOT::VecOps;
using namespace ROOT::VecOps;
using json = nlohmann::json;
using json = nlohmann::json;
void emcal_barrel_pi0_analysis(const char* input_fname = "sim_output/sim_emcal_barrel_pi0.root")
void emcal_barrel_pi0_analysis(
 
const char* input_fname = "sim_output/sim_emcal_barrel_pi0.root"
 
//const char* input_fname = "../sim_output/sim_emcal_barrel_uniform_pi0.root"
 
)
{
{
//input_fname = "../sim_output/sim_emcal_barrel_pi0.root";
// Setting for graphs
// Setting for graphs
gROOT->SetStyle("Plain");
gROOT->SetStyle("Plain");
gStyle->SetOptFit(1);
gStyle->SetOptFit(1);
@@ -54,7 +56,6 @@ void emcal_barrel_pi0_analysis(const char* input_fname = "sim_output/sim_emcal_b
@@ -54,7 +56,6 @@ void emcal_barrel_pi0_analysis(const char* input_fname = "sim_output/sim_emcal_b
double samp_frac = j["electron"]["sampling_fraction"];
double samp_frac = j["electron"]["sampling_fraction"];
// Thrown Energy [GeV]
// Thrown Energy [GeV]
//double meanE = 5; // Calculated later
auto Ethr = [](std::vector<dd4pod::Geant4ParticleData> const& input) {
auto Ethr = [](std::vector<dd4pod::Geant4ParticleData> const& input) {
return TMath::Sqrt(input[2].ps.x*input[2].ps.x + input[2].ps.y*input[2].ps.y + input[2].ps.z*input[2].ps.z + input[2].mass*input[2].mass);
return TMath::Sqrt(input[2].ps.x*input[2].ps.x + input[2].ps.y*input[2].ps.y + input[2].ps.z*input[2].ps.z + input[2].mass*input[2].mass);
};
};
@@ -97,146 +98,101 @@ void emcal_barrel_pi0_analysis(const char* input_fname = "sim_output/sim_emcal_b
@@ -97,146 +98,101 @@ void emcal_barrel_pi0_analysis(const char* input_fname = "sim_output/sim_emcal_b
};
};
// Define variables
// Define variables
auto d1 = d0.Define("Ethr", Ethr, {"mcparticles"})
auto d1 = d0.Define("Ethr", Ethr, {"mcparticles"})
.Define("nhits", nhits, {"EcalBarrelHits"})
.Define("nhits", nhits, {"EcalBarrelHits"})
.Define("EsimImg", Esim, {"EcalBarrelHits"})
.Define("EsimImg", Esim, {"EcalBarrelHits"})
.Define("EsimScFi", Esim, {"EcalBarrelScFiHits"})
.Define("EsimScFi", Esim, {"EcalBarrelScFiHits"})
.Define("Esim", "EsimImg+EsimScFi")
.Define("Esim", "EsimImg+EsimScFi")
.Define("fsam", fsam, {"Esim","Ethr"})
.Define("fsam", fsam, {"Esim","Ethr"})
.Define("pid", getpid, {"mcparticles"})
.Define("pid", getpid, {"mcparticles"})
.Define("dau", getdau, {"mcparticles"})
.Define("dau", getdau, {"mcparticles"})
.Define("dE", eResol, {"Esim","Ethr"})
.Define("dE", eResol, {"Esim","Ethr"})
.Define("dE_rel", eResol_rel, {"Esim","Ethr"})
.Define("dE_rel", eResol_rel, {"Esim","Ethr"})
;
;
// Define Histograms
// Define Histograms
auto hEthr = d1.Histo1D({"hEthr", "Thrown Energy; Thrown Energy [GeV]; Events", 100, 0.0, 7.5}, "Ethr");
std::vector <std::string> titleStr = {
auto hNhits = d1.Histo1D({"hNhits", "Number of hits per events; Number of hits; Events", 100, 0.0, 2000.0}, "nhits");
"Thrown Energy; Thrown Energy [GeV]; Events",
auto hEsim = d1.Histo1D({"hEsim", "Energy Deposit; Energy Deposit [GeV]; Events", 100, 0.0, 2.0}, "Esim");
"Number of hits per events; Number of hits; Events",
auto hfsam = d1.Histo1D({"hfsam", "Sampling Fraction; Sampling Fraction; Events", 150, 0.0, 0.15}, "fsam");
"Energy Deposit; Energy Deposit [GeV]; Events",
auto hpid = d1.Histo1D({"hpid", "PID; PID; Count", 100, -220, 220}, "pid");
"dE Relative; dE Relative; Events"
auto hdau = d1.Histo1D({"hdau", "Number of Daughters; Number of Daughters; Count", 10, 0, 10}, "dau");
};
// put this line above GetMean below so it can be lazily evaluated with the histgorams.
std::vector<std::vector<double>> range = {{0, 7.5}, {0, 2000}, {0, 2}, {-3, 3}};
auto nevents_thrown = d1.Count();
std::vector<std::string> col = {"Ethr", "nhits", "Esim", "dE_rel"};
const double meanE = hEthr->GetMean();
double meanE = 5;
int nCol = range.size();
// Event Counts
for (int i = 0; i < nCol; i++){
std::cout << "Number of Thrown Events: " << (*nevents_thrown) << "\n";
int binNum = 100;
auto h = d1.Histo1D({"hist", titleStr[i].c_str(), binNum, range[i][0], range[i][1]}, col[i].c_str());
// Draw Histograms
if (col[i] == "Ethr"){
TCanvas *c1 = new TCanvas("c1", "c1", 700, 500);
meanE = h->GetMean();
c1->SetLogy(1);
}
fmt::print("0\n");
auto *c = new TCanvas("c", "c", 700, 500);
auto h1 = hEthr->DrawCopy();
c->SetLogy(1);
//h1->GetYaxis()->SetTitleOffset(1.4);
auto h1 = h->DrawCopy();
h1->SetLineWidth(2);
h1->GetYaxis()->SetTitleOffset(1.4);
h1->SetLineColor(kBlue);
h1->SetLineWidth(2);
c1->SaveAs("results/emcal_barrel_pi0_Ethr.png");
h1->SetLineColor(kBlue);
c1->SaveAs("results/emcal_barrel_pi0_Ethr.pdf");
c->SaveAs((fmt::format("results/emcal_barrel_pi0_{}.png", col[i])).c_str());
c->SaveAs((fmt::format("results/emcal_barrel_pi0_{}.pdf", col[i])).c_str());
fmt::print("1\n");
std::printf("Generic %d\n", i);
}
//TCanvas *c2 = new TCanvas("c2", "c2", 700, 500);
//c2->SetLogy(1);
// Resolution Plots
//h1 = hNhits->DrawCopy();
titleStr = {
//h1->GetYaxis()->SetTitleOffset(1.4);
"Sampling Fraction; Sampling Fraction; Events",
//h1->SetLineWidth(2);
"dE; dE[GeV]; Events"
//h1->SetLineColor(kBlue);
};
//c2->SaveAs("results/emcal_barrel_pi0_nhits.png");
range = {{0,0.15}, {-3, 3}};
//c2->SaveAs("results/emcal_barrel_pi0_nhits.pdf");
col = {"fsam", "dE"};
nCol = range.size();
//TCanvas *c3 = new TCanvas("c3", "c3", 700, 500);
//c3->SetLogy(1);
std::vector<std::vector<double>> fitRange = {{0.005, 0.1},{-3, 3}};
//h1 = hEsim->DrawCopy();
double sigmaOverE = 0;
//h1->GetYaxis()->SetTitleOffset(1.4);
for (int i = 0; i < nCol; i++){
//h1->SetLineWidth(2);
int binNum = 100;
//h1->SetLineColor(kBlue);
if (col[i] == "fsam"){binNum = 150;}
//c3->SaveAs("results/emcal_barrel_pi0_Esim.png");
auto h = d1.Histo1D({"hist", titleStr[i].c_str(), binNum, range[i][0], range[i][1]}, col[i].c_str());
//c3->SaveAs("results/emcal_barrel_pi0_Esim.pdf");
auto *c = new TCanvas("c", "c", 700, 500);
c->SetLogy(1);
//TCanvas *c4 = new TCanvas("c4", "c4", 700, 500);
auto h1 = h->DrawCopy();
//c4->SetLogy(1);
h1->GetYaxis()->SetTitleOffset(1.4);
//h1 = hfsam->DrawCopy();
h1->SetLineWidth(2);
//h1->GetYaxis()->SetTitleOffset(1.4);
h1->SetLineColor(kBlue);
//h1->SetLineWidth(2);
h1->Fit("gaus","","", fitRange[i][0], fitRange[i][1]);
//h1->SetLineColor(kBlue);
h1->GetFunction("gaus")->SetLineWidth(2);
//h1->Fit("gaus","","",0.005,0.1);
h1->GetFunction("gaus")->SetLineColor(kRed);
//h1->GetFunction("gaus")->SetLineWidth(2);
double* res = h1->GetFunction("gaus")->GetParameters();
//h1->GetFunction("gaus")->SetLineColor(kRed);
sigmaOverE = res[2] / meanE;
//c4->SaveAs("results/emcal_barrel_pi0_fsam.png");
//c4->SaveAs("results/emcal_barrel_pi0_fsam.pdf");
c->SaveAs((fmt::format("results/emcal_barrel_pi0_{}.png", col[i])).c_str());
c->SaveAs((fmt::format("results/emcal_barrel_pi0_{}.pdf", col[i])).c_str());
//TCanvas *c5 = new TCanvas("c5", "c5", 700, 500);
std::printf("Resolution %d\n", i);
//c5->SetLogy(1);
}
//h1 = hpid->DrawCopy();
//h1->GetYaxis()->SetTitleOffset(1.4);
// Energy Resolution Calculation
//h1->SetLineWidth(2);
std::string test_tag = "Barrel_emcal_pi0";// TODO: Change test_tag to something else
//h1->SetLineColor(kBlue);
std:string detEle = "Barrel_emcal";
//c5->SaveAs("results/emcal_barrel_pi0_pid.png");
//c5->SaveAs("results/emcal_barrel_pi0_pid.pdf");
// Energy resolution in the barrel region (-1 < eta < 1)
// Taken from : Initial considerations for EMCal of the EIC detector by A. Bazilevsky
//TCanvas *c6 = new TCanvas("c6", "c6", 700, 500);
// sigma_E / E = 12% / E^0.5 convoluted with 2%
//c5->SetLogy(1);
// sigma_E / E = [ (0.12/E^0.5)^2 + 0.02^2]^0.5, with E in [GeV]
//h1 = hdau->DrawCopy();
//h1->GetYaxis()->SetTitleOffset(1.4);
//h1->SetLineWidth(2);
//h1->SetLineColor(kBlue);
//c6->SaveAs("results/emcal_barrel_pi0_dau.png");
//c6->SaveAs("results/emcal_barrel_pi0_dau.pdf");
//// Energy Resolution Calculation
//std::string test_tag = "Barrel_emcal_pi0";// TODO: Change test_tag to something else
//std:string detEle = "Barrel_emcal";
//// Energy resolution in the barrel region (-1 < eta < 1)
//// Taken from : Initial considerations for EMCal of the EIC detector by A. Bazilevsky
//// sigma_E / E = 12% / E^0.5 convoluted with 2%
//// sigma_E / E = [ (0.12/E^0.5)^2 + 0.02^2]^0.5, with E in [GeV]
//
//
//double resolutionTarget = TMath::Sqrt(0.12 * 0.12 / meanE + 0.02 * 0.02);
double resolutionTarget = TMath::Sqrt(0.12 * 0.12 / meanE + 0.02 * 0.02);
//common_bench::Test pi0_energy_resolution{
common_bench::Test pi0_energy_resolution{
// {{"name", fmt::format("{}_energy_resolution", test_tag)},
{{"name", fmt::format("{}_energy_resolution", test_tag)},
// {"title", "Pi0 Energy resolution"},
{"title", "Pi0 Energy resolution"},
// {"description",
{"description",
// fmt::format("Pi0 energy resolution for {}, estimated using a Gaussian fit.", detEle)},
fmt::format("Pi0 energy resolution for {}, estimated using a Gaussian fit.", detEle)},
// {"quantity", "resolution (in %)"},
{"quantity", "resolution (in %)"},
// {"target", std::to_string(resolutionTarget)}}
{"target", std::to_string(resolutionTarget)}}
//};
};
//// Histograms and Fitting
//auto hdE = d1.Histo1D({"hdE", "dE; dE[GeV]; Events", 100, -3.0, 3.0}, "dE");
//auto hdE_rel = d1.Histo1D({"hdE_rel", "dE Relative; dE Relative; Events", 100, -3.0, 3.0}, "dE_rel");
//auto hdEcopy = hdE->DrawCopy();
//hdEcopy->Fit("gaus", "", "", -3.0, 3.0);
//double* res = hdEcopy->GetFunction("gaus")->GetParameters();
//double sigmaOverE = res[2] / meanE;
//// Pass/Fail
//// Pass/Fail
//sigmaOverE <= resolutionTarget ? pi0_energy_resolution.pass(sigmaOverE) : pi0_energy_resolution.fail(sigmaOverE);
sigmaOverE <= resolutionTarget ? pi0_energy_resolution.pass(sigmaOverE) : pi0_energy_resolution.fail(sigmaOverE);
//std::printf("Energy Resolution is %f\n", res[2]);
common_bench::write_test({pi0_energy_resolution}, fmt::format("results/{}_pi0.json", detEle));
//// Energy Resolution Histogram Plotting
//auto *cdE = new TCanvas("cdE", "cdE", 700, 500);
//cdE->SetLogy(1);
//h1 = hdEcopy->DrawCopy();
//hdEcopy->GetYaxis()->SetTitleOffset(1.4);
//hdEcopy->SetLineWidth(2);
//hdEcopy->SetLineColor(kBlue);
//hdEcopy->GetFunction("gaus")->SetLineWidth(2);
//hdEcopy->GetFunction("gaus")->SetLineColor(kRed);
//cdE->SaveAs("results/emcal_barrel_pi0_dE.png");
//cdE->SaveAs("results/emcal_barrel_pi0_dE.pdf");
//auto *cdE_rel = new TCanvas("cdE_rel", "cdE_rel", 700, 500);
//h1 = hdE_rel->DrawCopy();
//hdE_rel->GetYaxis()->SetTitleOffset(1.4);
//hdE_rel->SetLineWidth(2);
//hdE_rel->SetLineColor(kBlue);
//cdE_rel->SaveAs("results/emcal_barrel_pi0_dE_rel.png");
//cdE_rel->SaveAs("results/emcal_barrel_pi0_dE_rel.pdf");
//common_bench::write_test({pi0_energy_resolution}, fmt::format("results/{}_pi0.json", detEle));
}
}
Loading