From 011487d85e64116c987f1b3d793a0084bff380a9 Mon Sep 17 00:00:00 2001 From: Sylvester Joosten <sjoosten@anl.gov> Date: Tue, 10 Aug 2021 17:58:39 +0000 Subject: [PATCH] update analysis scripts for minor dd4pod changes --- .../scripts/emcal_barrel_energy_scan_analysis.cxx | 2 +- .../scripts/emcal_barrel_particles_analysis.cxx | 2 +- .../barrel_ecal/scripts/emcal_barrel_pi0_analysis.cxx | 2 +- .../scripts/emcal_barrel_pion_rejection_analysis.cxx | 10 +++++----- .../scripts/emcal_barrel_pions_analysis.cxx | 2 +- .../scripts/hcal_barrel_energy_scan_analysis.cxx | 2 +- .../scripts/hcal_barrel_particles_analysis.cxx | 2 +- benchmarks/zdc/scripts/zdc_neutrons_analysis.cxx | 2 +- 8 files changed, 12 insertions(+), 12 deletions(-) diff --git a/benchmarks/barrel_ecal/scripts/emcal_barrel_energy_scan_analysis.cxx b/benchmarks/barrel_ecal/scripts/emcal_barrel_energy_scan_analysis.cxx index 61041363..91e35341 100644 --- a/benchmarks/barrel_ecal/scripts/emcal_barrel_energy_scan_analysis.cxx +++ b/benchmarks/barrel_ecal/scripts/emcal_barrel_energy_scan_analysis.cxx @@ -57,7 +57,7 @@ std::tuple <double, double, double, double> extract_sampling_fraction_parameters // Thrown Energy [GeV] auto Ethr = [](std::vector<dd4pod::Geant4ParticleData> const& input) { auto p = input[2]; - auto energy = TMath::Sqrt(p.psx * p.psx + p.psy * p.psy + p.psz * p.psz + p.mass * p.mass); + auto energy = TMath::Sqrt(p.ps.x * p.ps.x + p.ps.y * p.ps.y + p.ps.z * p.ps.z + p.mass * p.mass); return energy; }; diff --git a/benchmarks/barrel_ecal/scripts/emcal_barrel_particles_analysis.cxx b/benchmarks/barrel_ecal/scripts/emcal_barrel_particles_analysis.cxx index a56ececd..61edb15d 100644 --- a/benchmarks/barrel_ecal/scripts/emcal_barrel_particles_analysis.cxx +++ b/benchmarks/barrel_ecal/scripts/emcal_barrel_particles_analysis.cxx @@ -73,7 +73,7 @@ void emcal_barrel_particles_analysis(std::string particle_name = "electron", boo // Thrown Energy [GeV] auto Ethr = [](std::vector<dd4pod::Geant4ParticleData> const& input) { auto p = input[2]; - auto energy = TMath::Sqrt(p.psx * p.psx + p.psy * p.psy + p.psz * p.psz + p.mass * p.mass); + auto energy = TMath::Sqrt(p.ps.x * p.ps.x + p.ps.y * p.ps.y + p.ps.z * p.ps.z + p.mass * p.mass); return energy; }; diff --git a/benchmarks/barrel_ecal/scripts/emcal_barrel_pi0_analysis.cxx b/benchmarks/barrel_ecal/scripts/emcal_barrel_pi0_analysis.cxx index f56d9d78..83be1e2b 100644 --- a/benchmarks/barrel_ecal/scripts/emcal_barrel_pi0_analysis.cxx +++ b/benchmarks/barrel_ecal/scripts/emcal_barrel_pi0_analysis.cxx @@ -56,7 +56,7 @@ void emcal_barrel_pi0_analysis(const char* input_fname = "sim_output/sim_emcal_b // Thrown Energy [GeV] //double meanE = 5; // Calculated later auto Ethr = [](std::vector<dd4pod::Geant4ParticleData> const& input) { - return TMath::Sqrt(input[2].psx*input[2].psx + input[2].psy*input[2].psy + input[2].psz*input[2].psz + 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); }; // Number of hits diff --git a/benchmarks/barrel_ecal/scripts/emcal_barrel_pion_rejection_analysis.cxx b/benchmarks/barrel_ecal/scripts/emcal_barrel_pion_rejection_analysis.cxx index 63661655..2e0d41cd 100644 --- a/benchmarks/barrel_ecal/scripts/emcal_barrel_pion_rejection_analysis.cxx +++ b/benchmarks/barrel_ecal/scripts/emcal_barrel_pion_rejection_analysis.cxx @@ -106,23 +106,23 @@ void emcal_barrel_pion_rejection_analysis( // Thrown Energy [GeV] auto Ethr = [](std::vector<dd4pod::Geant4ParticleData> const& input) { - return TMath::Sqrt(input[2].psx*input[2].psx + input[2].psy*input[2].psy + input[2].psz*input[2].psz + 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); }; // Thrown Momentum [GeV] auto Pthr = [](std::vector<dd4pod::Geant4ParticleData> const& input) { - return TMath::Sqrt(input[2].psx*input[2].psx + input[2].psy*input[2].psy + input[2].psz*input[2].psz); + 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); }; // Thrown Eta auto Eta = [](std::vector<dd4pod::Geant4ParticleData> const& input) { - double E = TMath::Sqrt(input[2].psx*input[2].psx + input[2].psy*input[2].psy + input[2].psz*input[2].psz + input[2].mass*input[2].mass); - return 0.5*TMath::Log((E + input[2].psz) / (E - input[2].psz)); + double E = 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 0.5*TMath::Log((E + input[2].ps.z) / (E - input[2].ps.z)); }; // Thrown pT [GeV] auto pT = [](std::vector<dd4pod::Geant4ParticleData> const& input) { - return TMath::Sqrt(input[2].psx*input[2].psx + input[2].psy*input[2].psy); + return TMath::Sqrt(input[2].ps.x*input[2].ps.x + input[2].ps.y*input[2].ps.y); }; // Number of hits diff --git a/benchmarks/barrel_ecal/scripts/emcal_barrel_pions_analysis.cxx b/benchmarks/barrel_ecal/scripts/emcal_barrel_pions_analysis.cxx index 92c83bc8..24eaafdc 100644 --- a/benchmarks/barrel_ecal/scripts/emcal_barrel_pions_analysis.cxx +++ b/benchmarks/barrel_ecal/scripts/emcal_barrel_pions_analysis.cxx @@ -46,7 +46,7 @@ void emcal_barrel_pions_analysis(const char* input_fname = "sim_output/sim_emcal // Thrown Energy [GeV] auto Ethr = [](std::vector<dd4pod::Geant4ParticleData> const& input) { auto p = input[2]; - auto energy = TMath::Sqrt(p.psx * p.psx + p.psy * p.psy + p.psz * p.psz + p.mass * p.mass); + auto energy = TMath::Sqrt(p.ps.x * p.ps.x + p.ps.y * p.ps.y + p.ps.z * p.ps.z + p.mass * p.mass); return energy; }; diff --git a/benchmarks/barrel_hcal/scripts/hcal_barrel_energy_scan_analysis.cxx b/benchmarks/barrel_hcal/scripts/hcal_barrel_energy_scan_analysis.cxx index 8ca904ef..8ac13ba1 100644 --- a/benchmarks/barrel_hcal/scripts/hcal_barrel_energy_scan_analysis.cxx +++ b/benchmarks/barrel_hcal/scripts/hcal_barrel_energy_scan_analysis.cxx @@ -53,7 +53,7 @@ std::tuple <double, double, double, double> extract_sampling_fraction_parameters // Thrown Energy [GeV] auto Ethr = [](std::vector<dd4pod::Geant4ParticleData> const& input) { auto p = input[2]; - auto energy = TMath::Sqrt(p.psx * p.psx + p.psy * p.psy + p.psz * p.psz + p.mass * p.mass); + auto energy = TMath::Sqrt(p.ps.x * p.ps.x + p.ps.y * p.ps.y + p.ps.z * p.ps.z + p.mass * p.mass); return energy; }; diff --git a/benchmarks/barrel_hcal/scripts/hcal_barrel_particles_analysis.cxx b/benchmarks/barrel_hcal/scripts/hcal_barrel_particles_analysis.cxx index 45559e2a..1ae105d3 100644 --- a/benchmarks/barrel_hcal/scripts/hcal_barrel_particles_analysis.cxx +++ b/benchmarks/barrel_hcal/scripts/hcal_barrel_particles_analysis.cxx @@ -54,7 +54,7 @@ void hcal_barrel_particles_analysis(std::string particle_name = "electron") // Thrown Energy [GeV] auto Ethr = [](std::vector<dd4pod::Geant4ParticleData> const& input) { auto p = input[2]; - auto energy = TMath::Sqrt(p.psx * p.psx + p.psy * p.psy + p.psz * p.psz + p.mass * p.mass); + auto energy = TMath::Sqrt(p.ps.x * p.ps.x + p.ps.y * p.ps.y + p.ps.z * p.ps.z + p.mass * p.mass); return energy; }; diff --git a/benchmarks/zdc/scripts/zdc_neutrons_analysis.cxx b/benchmarks/zdc/scripts/zdc_neutrons_analysis.cxx index be218bb2..3c8bcd63 100644 --- a/benchmarks/zdc/scripts/zdc_neutrons_analysis.cxx +++ b/benchmarks/zdc/scripts/zdc_neutrons_analysis.cxx @@ -38,7 +38,7 @@ void zdc_neutrons_analysis(const char* input_fname = "sim_output/sim_zdc_uniform // Thrown Energy [GeV] auto Ethr = [](std::vector<dd4pod::Geant4ParticleData> const& input) { auto p = input[2]; - auto energy = TMath::Sqrt(p.psx * p.psx + p.psy * p.psy + p.psz * p.psz + p.mass * p.mass); + auto energy = TMath::Sqrt(p.ps.x * p.ps.x + p.ps.y * p.ps.y + p.ps.z * p.ps.z + p.mass * p.mass); return energy; }; -- GitLab