Adding extra zdc tables for SP calibrations (#6973)

Co-authored-by: Lucamicheletti93 <luca.mike93@gmail.com>

Author: lucamicheletti93

PWGDQ/Core/HistogramsLibrary.cxx

@@ -154,6 +154,19 @@ void o2::aod::dqhistograms::DefineHistograms(HistogramManager* hm, const char* h
       hm->AddHistogram(histClass, "IntercalibZNA_CentFT0C", "", false, 18, 0.0, 90.0, VarManager::kCentFT0C, 500, -50.0, 50.0, VarManager::KIntercalibZNA);
       hm->AddHistogram(histClass, "IntercalibZNC_CentFT0C", "", false, 18, 0.0, 90.0, VarManager::kCentFT0C, 500, -50.0, 50.0, VarManager::KIntercalibZNC);
 
+      hm->AddHistogram(histClass, "EnergyCommonZNA", "", false, 90, 0.0, 90.0, VarManager::kCentFT0C, 2000, 0, 2000, VarManager::kEnergyCommonZNA);
+      hm->AddHistogram(histClass, "EnergyCommonZNC", "", false, 90, 0.0, 90.0, VarManager::kCentFT0C, 2000, 0, 2000, VarManager::kEnergyCommonZNC);
+
+      hm->AddHistogram(histClass, "EnergyZNA1", "", false, 90, 0.0, 90.0, VarManager::kCentFT0C, 2000, 0, 2000, VarManager::kEnergyZNA1);
+      hm->AddHistogram(histClass, "EnergyZNA2", "", false, 90, 0.0, 90.0, VarManager::kCentFT0C, 2000, 0, 2000, VarManager::kEnergyZNA2);
+      hm->AddHistogram(histClass, "EnergyZNA3", "", false, 90, 0.0, 90.0, VarManager::kCentFT0C, 2000, 0, 2000, VarManager::kEnergyZNA3);
+      hm->AddHistogram(histClass, "EnergyZNA4", "", false, 90, 0.0, 90.0, VarManager::kCentFT0C, 2000, 0, 2000, VarManager::kEnergyZNA4);
+
+      hm->AddHistogram(histClass, "EnergyZNC1", "", false, 90, 0.0, 90.0, VarManager::kCentFT0C, 2000, 0, 2000, VarManager::kEnergyZNC1);
+      hm->AddHistogram(histClass, "EnergyZNC2", "", false, 90, 0.0, 90.0, VarManager::kCentFT0C, 2000, 0, 2000, VarManager::kEnergyZNC2);
+      hm->AddHistogram(histClass, "EnergyZNC3", "", false, 90, 0.0, 90.0, VarManager::kCentFT0C, 2000, 0, 2000, VarManager::kEnergyZNC3);
+      hm->AddHistogram(histClass, "EnergyZNC4", "", false, 90, 0.0, 90.0, VarManager::kCentFT0C, 2000, 0, 2000, VarManager::kEnergyZNC4);
+
       hm->AddHistogram(histClass, "Q2X0A", "", false, 500, -10.0, 10.0, VarManager::kQ2X0A);
       hm->AddHistogram(histClass, "Q2Y0A", "", false, 500, -10.0, 10.0, VarManager::kQ2Y0A);
       hm->AddHistogram(histClass, "Q2X0B", "", false, 500, -10.0, 10.0, VarManager::kQ2X0B);

PWGDQ/Core/VarManager.h

@@ -343,6 +343,14 @@ class VarManager : public TObject
     kEnergyCommonZNC,
     kEnergyCommonZPA,
     kEnergyCommonZPC,
+    kEnergyZNA1,
+    kEnergyZNA2,
+    kEnergyZNA3,
+    kEnergyZNA4,
+    kEnergyZNC1,
+    kEnergyZNC2,
+    kEnergyZNC3,
+    kEnergyZNC4,
     kTimeZNA,
     kTimeZNC,
     kTimeZPA,
@@ -3732,6 +3740,26 @@ void VarManager::FillSpectatorPlane(C const& collision, float* values)
   }
   float znaCommon = collision.energyCommonZNA() < 0 ? -1.f : collision.energyCommonZNA();
   float zncCommon = collision.energyCommonZNC() < 0 ? -1.f : collision.energyCommonZNC();
+  float zpaCommon = collision.energyCommonZPA() < 0 ? -1.f : collision.energyCommonZPA();
+  float zpcCommon = collision.energyCommonZPC() < 0 ? -1.f : collision.energyCommonZPC();
+
+  // Store ZNA and ZNC energies for calibrations
+  values[kEnergyCommonZNA] = znaCommon;
+  values[kEnergyCommonZNC] = zncCommon;
+  values[kEnergyCommonZPA] = zpaCommon;
+  values[kEnergyCommonZPC] = zpcCommon;
+  values[kEnergyZNA1] = znaEnergy[0];
+  values[kEnergyZNA2] = znaEnergy[1];
+  values[kEnergyZNA3] = znaEnergy[2];
+  values[kEnergyZNA4] = znaEnergy[3];
+  values[kEnergyZNC1] = zncEnergy[0];
+  values[kEnergyZNC2] = zncEnergy[1];
+  values[kEnergyZNC3] = zncEnergy[2];
+  values[kEnergyZNC4] = zncEnergy[3];
+  values[kTimeZNA] = collision.timeZNA();
+  values[kTimeZNC] = collision.timeZNC();
+  values[kTimeZPA] = collision.timeZPA();
+  values[kTimeZPC] = collision.timeZPC();
 
   constexpr float beamEne = 5.36 * 0.5;
   constexpr float x[4] = {-1.75, 1.75, -1.75, 1.75};

PWGDQ/DataModel/ReducedInfoTables.h

@@ -192,6 +192,14 @@ DECLARE_SOA_COLUMN(EnergyCommonZNA, energyCommonZNA, float); //!
 DECLARE_SOA_COLUMN(EnergyCommonZNC, energyCommonZNC, float); //!
 DECLARE_SOA_COLUMN(EnergyCommonZPA, energyCommonZPA, float); //!
 DECLARE_SOA_COLUMN(EnergyCommonZPC, energyCommonZPC, float); //!
+DECLARE_SOA_COLUMN(EnergyZNA1, energyZNA1, float);           //!
+DECLARE_SOA_COLUMN(EnergyZNA2, energyZNA2, float);           //!
+DECLARE_SOA_COLUMN(EnergyZNA3, energyZNA3, float);           //!
+DECLARE_SOA_COLUMN(EnergyZNA4, energyZNA4, float);           //!
+DECLARE_SOA_COLUMN(EnergyZNC1, energyZNC1, float);           //!
+DECLARE_SOA_COLUMN(EnergyZNC2, energyZNC2, float);           //!
+DECLARE_SOA_COLUMN(EnergyZNC3, energyZNC3, float);           //!
+DECLARE_SOA_COLUMN(EnergyZNC4, energyZNC4, float);           //!
 DECLARE_SOA_COLUMN(TimeZNA, timeZNA, float);                 //!
 DECLARE_SOA_COLUMN(TimeZNC, timeZNC, float);                 //!
 DECLARE_SOA_COLUMN(TimeZPA, timeZPA, float);                 //!
@@ -204,7 +212,12 @@ DECLARE_SOA_TABLE(ReducedZdcs, "AOD", "REDUCEDZDC", //!   Event ZDC information
                   reducedzdc::TimeZNA, reducedzdc::TimeZNC,
                   reducedzdc::TimeZPA, reducedzdc::TimeZPC);
 
+DECLARE_SOA_TABLE(ReducedZdcsExtra, "AOD", "REDUCEDZDCEXTRA", //!   Event ZDC extra information
+                  reducedzdc::EnergyZNA1, reducedzdc::EnergyZNA2, reducedzdc::EnergyZNA3, reducedzdc::EnergyZNA4,
+                  reducedzdc::EnergyZNC1, reducedzdc::EnergyZNC2, reducedzdc::EnergyZNC3, reducedzdc::EnergyZNC4);
+
 using ReducedZdc = ReducedZdcs::iterator;
+using ReducedZdcExtra = ReducedZdcsExtra::iterator;
 
 namespace reducedtrack
 {

PWGDQ/Tasks/dqFlow.cxx

@@ -101,6 +101,8 @@ struct DQEventQvector {
   Produces<ReducedEventsQvectorCentr> eventQvectorCentr;
   Produces<ReducedEventsRefFlow> eventRefFlow;
   Produces<ReducedEventsQvectorZN> eventQvectorZN;
+  Produces<ReducedZdc> eventReducedZdc;
+  Produces<ReducedZdcExtra> eventReducedZdcExtra;
 
   Configurable<std::string> fConfigEventCuts{"cfgEventCuts", "eventStandard", "Event selection"};
   Configurable<bool> fConfigQA{"cfgQA", true, "If true, fill QA histograms"};
@@ -481,8 +483,14 @@ struct DQEventQvector {
                           collision.sumAmplFT0A(), collision.sumAmplFT0C(), collision.sumAmplFT0M(), collision.sumAmplFV0A(), collision.nTrkBPos(), collision.nTrkBNeg());
         if (bc.has_zdc()) {
           eventQvectorZN(VarManager::fgValues[VarManager::kQ1ZNAX], VarManager::fgValues[VarManager::kQ1ZNAY], VarManager::fgValues[VarManager::kQ1ZNCX], VarManager::fgValues[VarManager::kQ1ZNCY]);
+          eventReducedZdc(VarManager::fgValues[VarManager::kEnergyCommonZNA], VarManager::fgValues[VarManager::kEnergyCommonZNC], VarManager::fgValues[VarManager::kEnergyCommonZPA], VarManager::fgValues[VarManager::kEnergyCommonZPC],
+                          VarManager::fgValues[VarManager::kTimeZNA], VarManager::fgValues[VarManager::kTimeZNC], VarManager::fgValues[VarManager::kTimeZPA], VarManager::fgValues[VarManager::kTimeZPC]);
+          eventReducedZdcExtra(VarManager::fgValues[VarManager::kEnergyZNA1], VarManager::fgValues[VarManager::kEnergyZNA2], VarManager::fgValues[VarManager::kEnergyZNA3], VarManager::fgValues[VarManager::kEnergyZNA4],
+                               VarManager::fgValues[VarManager::kEnergyZNC1], VarManager::fgValues[VarManager::kEnergyZNC2], VarManager::fgValues[VarManager::kEnergyZNC3], VarManager::fgValues[VarManager::kEnergyZNC4]);
         } else {
           eventQvectorZN(-999, -999, -999, -999);
+          eventReducedZdc(-999, -999, -999, -999, -999, -999, -999, -999);
+          eventReducedZdcExtra(-999, -999, -999, -999, -999, -999, -999, -999);
         }
       }
     }