add plots in ATLAS coordinates to projection plots

scripts/plot_projection.cpp

@@ -54,24 +54,33 @@ void writePlots(SimulationConfig const& config) {
     "truthCentroids", RDFHelp::truth::getTruthCentroids, {"truthParticles"}
   );
   std::array<ROOT::RDF::RResultPtr<TH1D>, 2> nTruthParticles;
-  std::array<ROOT::RDF::RResultPtr<TH2D>, 2> truthCentroid;
+  std::array<ROOT::RDF::RResultPtr<TH2D>, 2> truthCentroidRPD;
+  std::array<ROOT::RDF::RResultPtr<TH2D>, 2> truthCentroidATLAS;
   std::array<ROOT::RDF::RResultPtr<TH2D>, 2> recoCentroid;
   std::array<ROOT::RDF::RResultPtr<TH2D>, 2> xTruthVsRecoCentroid;
   std::array<ROOT::RDF::RResultPtr<TH2D>, 2> yTruthVsRecoCentroid;
   for (auto const side : SIDES) {
     auto dfWithTruthCentroid = df.Define(
-      "truthCentroidX", RDFHelp::truth::getSideTruthXCentroid(side, true), {"truthCentroids"}
+      "truthCentroidXRPD", RDFHelp::truth::getSideTruthXCentroid(side, true), {"truthCentroids"}
     ).Define(
-      "truthCentroidY", RDFHelp::truth::getSideTruthYCentroid(side, true), {"truthCentroids"}
+      "truthCentroidYRPD", RDFHelp::truth::getSideTruthYCentroid(side, true), {"truthCentroids"}
+    ).Define(
+      "truthCentroidXATLAS", RDFHelp::truth::getSideTruthXCentroid(side, false), {"truthCentroids"}
+    ).Define(
+      "truthCentroidYATLAS", RDFHelp::truth::getSideTruthYCentroid(side, false), {"truthCentroids"}
     ).Define(
       "nTruthParticles", RDFHelp::truth::getSideNTruthParticles(side), {"truthCentroids"}
     );
     nTruthParticles.at(side) = dfWithTruthCentroid.Histo1D<unsigned int>(
       {Form("side%c_nTruthParticles", getSideLabel(side)), ";# Truth Particles;Count", BINS(config.getNTruthParticlesAxis())}, "nTruthParticles"
     );
-    truthCentroid.at(side) = dfWithTruthCentroid.Histo2D<float, float>(
-      {Form("side%c_truthCentroid", getSideLabel(side)), ";x [mm];y [mm];Count", BINS(axis::xAtRPD), BINS(axis::yAtRPD)},
-      "truthCentroidX", "truthCentroidY"
+    truthCentroidRPD.at(side) = dfWithTruthCentroid.Histo2D<float, float>(
+      {Form("side%c_truthCentroidRPD", getSideLabel(side)), ";x_{RPD} [mm];y_{RPD} [mm];Count", BINS(axis::xAtRPD), BINS(axis::yAtRPD)},
+      "truthCentroidXRPD", "truthCentroidYRPD"
+    );
+    truthCentroidATLAS.at(side) = dfWithTruthCentroid.Histo2D<float, float>(
+      {Form("side%c_truthCentroidATLAS", getSideLabel(side)), ";x_{ATLAS} [mm];y_{ATLAS} [mm];Count", BINS(axis::xAtRPDATLAS), BINS(axis::yAtRPDATLAS)},
+      "truthCentroidXATLAS", "truthCentroidYATLAS"
     );
     {
       // VALID CENTROID
@@ -88,17 +97,18 @@ void writePlots(SimulationConfig const& config) {
       );
       xTruthVsRecoCentroid.at(side) = dfCentroidValid.Histo2D<float, float>(
         {Form("side%c_xTruthVsRecoCentroid", getSideLabel(side)), ";Truth Particle Centroid x [mm];Reconstructed Centroid x [mm];Count", BINS(axis::xAtRPD), BINS(axis::xAtRPD)},
-        "truthCentroidX", "recoCentroidX"
+        "truthCentroidXRPD", "recoCentroidX"
       );
       yTruthVsRecoCentroid.at(side) = dfCentroidValid.Histo2D<float, float>(
         {Form("side%c_yTruthVsRecoCentroid", getSideLabel(side)), ";Truth Particle Centroid y [mm];Reconstructed Centroid y [mm];Count", BINS(axis::yAtRPD), BINS(axis::yAtRPD)},
-        "truthCentroidY", "recoCentroidY"
+        "truthCentroidYRPD", "recoCentroidY"
       );
     }
   }
   for (auto const side : SIDES) {
     write1D(nTruthParticles.at(side));
-    write2D(truthCentroid.at(side));
+    write2D(truthCentroidRPD.at(side));
+    write2D(truthCentroidATLAS.at(side));
     write2D(recoCentroid.at(side));
     write2D(xTruthVsRecoCentroid.at(side));
     write2D(yTruthVsRecoCentroid.at(side));