PWGMM: Mult: reducer export eta-phi map (#5613)

Author: aalkin

PWGMM/Mult/DataModel/ReducedTables.h

@@ -255,7 +255,8 @@ DECLARE_SOA_TABLE(RFeatMins, "AOD", "RFEATMIN",
                   features::ReconstructedVertexZ,
                   features::TimeRes,
                   features::ReconstructedForwardMultiplicityA,
-                  features::ReconstructedForwardMultiplicityC);
+                  features::ReconstructedForwardMultiplicityC,
+                  rcol::MapEtaPhi);
 
 } // namespace o2::aod
 namespace o2::soa

PWGMM/Mult/TableProducer/reducerPostprocess.cxx

@@ -26,9 +26,18 @@ struct ReducerPostprocess {
   void processFull(MCFull const&,
                    soa::Join<aod::RCollisions, aod::RMCColLabels> const& cols)
   {
+    std::vector<int> spatialMap;
+    bool first = true;
     for (auto& col : cols) {
       auto mccol = col.rmccollision_as<MCFull>();
-      features(mccol.multMCNParticlesEta10(), col.multNTracksPVeta1(), mccol.processId(), mccol.impactParameter(), col.posX(), col.posY(), col.posZ(), col.collisionTimeRes(), col.multFT0A(), col.multFT0C());
+      if (first) {
+        spatialMap.resize(col.mapetaphi().size());
+        first = false;
+      }
+      for (auto i = 0U; i < col.mapetaphi().size(); ++i) {
+        spatialMap[i] = col.mapetaphi()[i];
+      }
+      features(mccol.multMCNParticlesEta10(), col.multNTracksPVeta1(), mccol.processId(), mccol.impactParameter(), col.posX(), col.posY(), col.posZ(), col.collisionTimeRes(), col.multFT0A(), col.multFT0C(), spatialMap);
     }
   }
 
@@ -38,9 +47,18 @@ struct ReducerPostprocess {
   void processLite(MCLite const&,
                    soa::Join<aod::RCollisions, aod::RMCColLabels> const& cols)
   {
+    std::vector<int> spatialMap;
+    bool first = true;
     for (auto& col : cols) {
       auto mccol = col.rmccollision_as<MCLite>();
-      features(mccol.multMCNParticlesEta10(), col.multNTracksPVeta1(), -1, -1.f, col.posX(), col.posY(), col.posZ(), col.collisionTimeRes(), col.multFT0A(), col.multFT0C());
+      if (first) {
+        spatialMap.resize(col.mapetaphi().size());
+        first = false;
+      }
+      for (auto i = 0U; i < col.mapetaphi().size(); ++i) {
+        spatialMap[i] = col.mapetaphi()[i];
+      }
+      features(mccol.multMCNParticlesEta10(), col.multNTracksPVeta1(), -1, -1.f, col.posX(), col.posY(), col.posZ(), col.collisionTimeRes(), col.multFT0A(), col.multFT0C(), spatialMap);
     }
   }