From 821111385adcf537bcf3ab81307cabddc2caf889 Mon Sep 17 00:00:00 2001
From: Luigi Dello Stritto <ldellost@alicecerno2.cern.ch>
Date: Tue, 24 Oct 2023 15:16:15 +0200
Subject: [PATCH] PWGHF: Adding multiplicity info in the Lc event tree

---
 PWGHF/TableProducer/treeCreatorLcToPKPi.cxx | 31 +++++++++++++++++----
 1 file changed, 26 insertions(+), 5 deletions(-)

diff --git a/PWGHF/TableProducer/treeCreatorLcToPKPi.cxx b/PWGHF/TableProducer/treeCreatorLcToPKPi.cxx
index 99f75a494b5..a7fff765d5b 100644
--- a/PWGHF/TableProducer/treeCreatorLcToPKPi.cxx
+++ b/PWGHF/TableProducer/treeCreatorLcToPKPi.cxx
@@ -22,6 +22,7 @@
 #include "PWGHF/Core/HfHelper.h"
 #include "PWGHF/DataModel/CandidateReconstructionTables.h"
 #include "PWGHF/DataModel/CandidateSelectionTables.h"
+#include "Common/DataModel/Multiplicity.h"
 
 using namespace o2;
 using namespace o2::framework;
@@ -83,6 +84,11 @@ DECLARE_SOA_INDEX_COLUMN_FULL(Candidate, candidate, int, HfCand3Prong, "_0");
 DECLARE_SOA_INDEX_COLUMN(McCollision, mcCollision);
 DECLARE_SOA_COLUMN(IsEventReject, isEventReject, int);
 DECLARE_SOA_COLUMN(RunNumber, runNumber, int);
+DECLARE_SOA_COLUMN(MultZeqFT0A, multZeqFT0A, float);
+DECLARE_SOA_COLUMN(MultZeqFT0C, multZeqFT0C, float);
+DECLARE_SOA_COLUMN(MultFT0M, multFT0M, float);
+DECLARE_SOA_COLUMN(MultZeqFV0A, multZeqFV0A, float);
+DECLARE_SOA_COLUMN(MultZeqNTracksPV, multZeqNTracksPV, float);
 } // namespace full
 
 DECLARE_SOA_TABLE(HfCandLcFulls, "AOD", "HFCANDLCFULL",
@@ -167,7 +173,12 @@ DECLARE_SOA_TABLE(HfCandLcFullEvs, "AOD", "HFCANDLCFULLEV",
                   collision::PosY,
                   collision::PosZ,
                   full::IsEventReject,
-                  full::RunNumber);
+                  full::RunNumber,
+                  full::MultZeqFT0A,
+                  full::MultZeqFT0C,
+                  full::MultFT0M,
+                  full::MultZeqFV0A,
+                  full::MultZeqNTracksPV);
 
 DECLARE_SOA_TABLE(HfCandLcFullPs, "AOD", "HFCANDLCFULLP",
                   full::McCollisionId,
@@ -197,7 +208,7 @@ struct HfTreeCreatorLcToPKPi {
   {
   }
 
-  void processMc(soa::Join<aod::Collisions, aod::McCollisionLabels> const& collisions,
+  void processMc(soa::Join<aod::Collisions, aod::McCollisionLabels, aod::Mults, aod::MultZeqs> const& collisions,
                  aod::McCollisions const& mcCollisions,
                  soa::Join<aod::HfCand3Prong, aod::HfCand3ProngMcRec, aod::HfSelLc> const& candidates,
                  soa::Join<aod::McParticles, aod::HfCand3ProngMcGen> const& particles,
@@ -215,7 +226,12 @@ struct HfTreeCreatorLcToPKPi {
         collision.posY(),
         collision.posZ(),
         0,
-        collision.bc().runNumber());
+        collision.bc().runNumber(),
+        collision.multZeqFT0A(),
+        collision.multZeqFT0C(),
+        collision.multFT0M(),
+        collision.multZeqFV0A(),
+        collision.multZeqNTracksPV());
     }
 
     // Filling candidate properties
@@ -330,7 +346,7 @@ struct HfTreeCreatorLcToPKPi {
   }
   PROCESS_SWITCH(HfTreeCreatorLcToPKPi, processMc, "Process MC tree writer", true);
 
-  void processData(aod::Collisions const& collisions,
+  void processData(soa::Join<aod::Collisions, aod::Mults, aod::MultZeqs> const& collisions,
                    soa::Join<aod::HfCand3Prong, aod::HfSelLc> const& candidates,
                    TracksWPid const& tracks, aod::BCs const&)
   {
@@ -346,7 +362,12 @@ struct HfTreeCreatorLcToPKPi {
         collision.posY(),
         collision.posZ(),
         0,
-        collision.bc().runNumber());
+        collision.bc().runNumber(),
+        collision.multZeqFT0A(),
+        collision.multZeqFT0C(),
+        collision.multFT0M(),
+        collision.multZeqFV0A(),
+        collision.multZeqNTracksPV());
     }
 
     // Filling candidate properties