ZDC Reduced tables added (#4627)

* ZDC Reduced tables added

* Please consider the following formatting changes

---------

Co-authored-by: ALICE Action Bot <alibuild@cern.ch>

Author: abylinkin

PWGUD/TableProducer/SGCandProducer.cxx

@@ -9,6 +9,12 @@
 // granted to it by virtue of its status as an Intergovernmental Organization
 // or submit itself to any jurisdiction.
 
+#include "Framework/AnalysisDataModel.h"
+#include "Common/CCDB/EventSelectionParams.h"
+#include "Common/DataModel/EventSelection.h"
+#include "CommonConstants/LHCConstants.h"
+#include "DataFormatsFIT/Triggers.h"
+
 #include "Framework/runDataProcessing.h"
 #include "Framework/AnalysisTask.h"
 #include "PWGUD/DataModel/UDTables.h"
@@ -29,10 +35,12 @@ struct SGCandProducer {
 
   // data tables
   Produces<aod::SGCollisions> outputSGCollisions;
+  //  Produces<aod::ZDCCollisions> outputZDCCollisions;
   Produces<aod::UDCollisions> outputCollisions;
   Produces<aod::UDCollisionsSels> outputCollisionsSels;
   Produces<aod::UDCollsLabels> outputCollsLabels;
   Produces<aod::UDZdcs> outputZdcs;
+  Produces<o2::aod::UDZdcsReduced> udZdcsReduced;
   Produces<aod::UDTracks> outputTracks;
   Produces<aod::UDTracksCov> outputTracksCov;
   Produces<aod::UDTracksDCA> outputTracksDCA;
@@ -51,7 +59,8 @@ struct SGCandProducer {
   // data inputs
   using CCs = soa::Join<aod::Collisions, aod::EvSels>;
   using CC = CCs::iterator;
-  using BCs = soa::Join<aod::BCsWithTimestamps, aod::BcSels, aod::Run3MatchedToBCSparse>;
+  //  using BCs = soa::Join<aod::BCsWithTimestamps, aod::BcSels, aod::Run3MatchedToBCSparse>;
+  using BCs = soa::Join<aod::BCs, aod::Timestamps, aod::BcSels, aod::Run3MatchedToBCSparse>;
   using BC = BCs::iterator;
   using TCs = soa::Join<aod::Tracks, /*aod::TracksCov,*/ aod::TracksExtra, aod::TracksDCA, aod::TrackSelection,
                         aod::pidTPCFullEl, aod::pidTPCFullMu, aod::pidTPCFullPi, aod::pidTPCFullKa, aod::pidTPCFullPr,
@@ -105,7 +114,7 @@ struct SGCandProducer {
                     track.tofNSigmaKa(),
                     track.tofNSigmaPr());
     outputTracksExtra(track.tpcInnerParam(),
-                      track.itsClusterSizes(),
+                      track.itsClusterMap(),
                       track.tpcNClsFindable(),
                       track.tpcNClsFindableMinusFound(),
                       track.tpcNClsFindableMinusCrossedRows(),
@@ -140,6 +149,7 @@ struct SGCandProducer {
 
   // process function for real data
   void process(CC const& collision, BCs const& bcs, TCs& tracks, FWs& fwdtracks,
+               // aod::Zdcs_000& zdcs, aod::FT0s& ft0s, aod::FV0As& fv0as, aod::FDDs& fdds)
                aod::Zdcs& zdcs, aod::FT0s& ft0s, aod::FV0As& fv0as, aod::FDDs& fdds)
   {
     LOGF(debug, "<SGCandProducer>  collision %d", collision.globalIndex());
@@ -167,6 +177,9 @@ struct SGCandProducer {
       // fill FITInfo
       upchelpers::FITInfo fitInfo{};
       udhelpers::getFITinfo(fitInfo, bc.globalBC(), bcs, ft0s, fv0as, fdds);
+      //  upchelpers::ZDCInfo zdcInfo{};
+      // udhelpers::getZDCinfo(zdcInfo, bc.globalBC(), bcs, zdcs);
+      // udhelpers::getZDCinfo(zdcInfo, bc.globalBC(), zdcs);
 
       // update SG candidates tables
       auto rtrwTOF = udhelpers::rPVtrwTOF<true>(tracks, collision.numContrib());
@@ -196,16 +209,34 @@ struct SGCandProducer {
       }
 
       // fill UDZdcs
-      if (bc.has_zdc()) {
-        auto zdc = bc.zdc();
-        auto enes = std::vector(zdc.energy().begin(), zdc.energy().end());
-        auto chEs = std::vector(zdc.channelE().begin(), zdc.channelE().end());
-        auto amps = std::vector(zdc.amplitude().begin(), zdc.amplitude().end());
-        auto times = std::vector(zdc.time().begin(), zdc.time().end());
-        auto chTs = std::vector(zdc.channelT().begin(), zdc.channelT().end());
-        outputZdcs(outputCollisions.lastIndex(), enes, chEs, amps, times, chTs);
+      // outputZDCCollisions(zdcInfo.timeZNA,zdcInfo.timeZNC,zdcInfo.multZNA,zdcInfo.multZNC);
+      std::map<uint64_t, int32_t> mapGlobalBcWithZdc{};
+      auto globalBC = bc.globalBC();
+      for (const auto& zdc : zdcs) {
+        // auto globalBC = zdc.bc_as<o2::aod::BCs>().globalBC();
+        // auto bc = zdc.bc_as<o2::aod::BCs>().globalBC();
+        if (std::abs(zdc.timeZNA()) > 2.f && std::abs(zdc.timeZNC()) > 2.f)
+          continue;
+        // mapGlobalBcWithZdc[bc.globalBC()] = zdc.globalIndex();
+        mapGlobalBcWithZdc[globalBC] = zdc.globalIndex();
+        // mapGlobalBcWithZdc[bc] = zdc.globalIndex();
       }
-
+      int candID = 0;
+      auto nZdcs = mapGlobalBcWithZdc.size();
+      if (nZdcs > 0) {
+        // auto itZDC = mapGlobalBcWithZdc.find(bc.globalBC());
+        auto itZDC = mapGlobalBcWithZdc.find(globalBC);
+        // auto itZDC = mapGlobalBcWithZdc.find(bc);
+        if (itZDC != mapGlobalBcWithZdc.end()) {
+          const auto& zdc = zdcs.iteratorAt(itZDC->second);
+          float timeZNA = zdc.timeZNA();
+          float timeZNC = zdc.timeZNC();
+          float eComZNA = zdc.energyCommonZNA();
+          float eComZNC = zdc.energyCommonZNC();
+          udZdcsReduced(candID, timeZNA, timeZNC, eComZNA, eComZNC);
+        }
+      }
+      candID++;
       // produce TPC signal histograms for 2-track events
       LOGF(debug, "SG candidate: number of PV tracks %d", collision.numContrib());
       if (collision.numContrib() == 2) {