[PWGHF] Add Ds prong info and offline processes (#10286)

scattaru · web-flow · commit dcf318975815 · 2025-03-03T13:53:13.000+01:00
diff --git a/PWGHF/HFC/DataModel/DerivedDataCorrelationTables.h b/PWGHF/HFC/DataModel/DerivedDataCorrelationTables.h
@@ -40,6 +40,9 @@ using HfcRedCollision = HfcRedCollisions::iterator;
 namespace hf_candidate_reduced
 {
 DECLARE_SOA_INDEX_COLUMN(HfcRedCollision, hfcRedCollision); //! ReducedCollision index
+DECLARE_SOA_COLUMN(Prong0Id, prong0Id, int);                //! Prong 0 index
+DECLARE_SOA_COLUMN(Prong1Id, prong1Id, int);                //! Prong 1 index
+DECLARE_SOA_COLUMN(Prong2Id, prong2Id, int);                //! Prong2 index
 DECLARE_SOA_COLUMN(PhiCand, phiCand, float);                //! Phi of the candidate
 DECLARE_SOA_COLUMN(EtaCand, etaCand, float);                //! Eta of the candidate
 DECLARE_SOA_COLUMN(PtCand, ptCand, float);                  //! Pt of the candidate
@@ -53,7 +56,10 @@ DECLARE_SOA_TABLE(DsCandReduceds, "AOD", "DSCANDREDUCED", //! Table with Ds cand
                   aod::hf_candidate_reduced::PhiCand,
                   aod::hf_candidate_reduced::EtaCand,
                   aod::hf_candidate_reduced::PtCand,
-                  aod::hf_candidate_reduced::InvMassDs);
+                  aod::hf_candidate_reduced::InvMassDs,
+                  aod::hf_candidate_reduced::Prong0Id,
+                  aod::hf_candidate_reduced::Prong1Id,
+                  aod::hf_candidate_reduced::Prong2Id);
 
 DECLARE_SOA_TABLE(DsCandSelInfos, "AOD", "DSCANDSELINFO", //! Table with Ds candidate selection info
                   soa::Index<>,
@@ -63,7 +69,7 @@ DECLARE_SOA_TABLE(DsCandSelInfos, "AOD", "DSCANDSELINFO", //! Table with Ds cand
 
 namespace hf_assoc_track_reduced
 {
-DECLARE_SOA_COLUMN(TrackId, trackId, int);                 //! Original track index
+DECLARE_SOA_COLUMN(OriginTrackId, originTrackId, int);     //! Original track index
 DECLARE_SOA_COLUMN(NTpcCrossedRows, nTpcCrossedRows, int); //! Number of crossed TPC Rows
 DECLARE_SOA_COLUMN(ItsClusterMap, itsClusterMap, int);     //! ITS cluster map, one bit per a layer, starting from the innermost
 DECLARE_SOA_COLUMN(ItsNCls, itsNCls, int);                 //! Number of ITS clusters
@@ -76,11 +82,12 @@ DECLARE_SOA_COLUMN(DcaZ, dcaZ, float);                     //! Impact parameter
 DECLARE_SOA_TABLE(AssocTrackReds, "AOD", "ASSOCTRACKRED", //! Table with associated track info
                   soa::Index<>,
                   aod::hf_candidate_reduced::HfcRedCollisionId,
+                  aod::hf_assoc_track_reduced::OriginTrackId,
                   aod::hf_assoc_track_reduced::PhiAssocTrack,
                   aod::hf_assoc_track_reduced::EtaAssocTrack,
                   aod::hf_assoc_track_reduced::PtAssocTrack);
 
-DECLARE_SOA_TABLE(AssocTrackSelInfos, "AOD", "ASSOCTRACKSELINFO", //! Table with associated track info
+DECLARE_SOA_TABLE(AssocTrackSels, "AOD", "ASSOCTRACKSEL", //! Table with associated track info
                   soa::Index<>,
                   aod::hf_candidate_reduced::HfcRedCollisionId,
                   aod::hf_assoc_track_reduced::NTpcCrossedRows,
diff --git a/PWGHF/HFC/TableProducer/correlatorDsHadrons.cxx b/PWGHF/HFC/TableProducer/correlatorDsHadrons.cxx
@@ -145,7 +145,7 @@ struct HfCorrelatorDsHadrons {
   Produces<aod::DsCandReduceds> candReduced;
   Produces<aod::DsCandSelInfos> candSelInfo;
   Produces<aod::AssocTrackReds> assocTrackReduced;
-  Produces<aod::AssocTrackSelInfos> assocTrackSelInfo;
+  Produces<aod::AssocTrackSels> assocTrackSelInfo;
 
   Configurable<bool> fillHistoData{"fillHistoData", true, "Flag for filling histograms in data processes"};
   Configurable<bool> fillHistoMcRec{"fillHistoMcRec", true, "Flag for filling histograms in MC Rec processes"};
@@ -717,13 +717,13 @@ struct HfCorrelatorDsHadrons {
           for (unsigned int iclass = 0; iclass < classMl->size(); iclass++) {
             outputMl[iclass] = candidate.mlProbDsToKKPi()[classMl->at(iclass)];
           }
-          candReduced(indexHfcReducedCollision, candidate.phi(), candidate.eta(), candidate.pt(), hfHelper.invMassDsToKKPi(candidate));
+          candReduced(indexHfcReducedCollision, candidate.phi(), candidate.eta(), candidate.pt(), hfHelper.invMassDsToKKPi(candidate), candidate.prong0Id(), candidate.prong1Id(), candidate.prong2Id());
           candSelInfo(indexHfcReducedCollision, outputMl[0], outputMl[2]);
         } else if (candidate.isSelDsToPiKK() >= selectionFlagDs) {
           for (unsigned int iclass = 0; iclass < classMl->size(); iclass++) {
             outputMl[iclass] = candidate.mlProbDsToPiKK()[classMl->at(iclass)];
           }
-          candReduced(indexHfcReducedCollision, candidate.phi(), candidate.eta(), candidate.pt(), hfHelper.invMassDsToPiKK(candidate));
+          candReduced(indexHfcReducedCollision, candidate.phi(), candidate.eta(), candidate.pt(), hfHelper.invMassDsToPiKK(candidate), candidate.prong0Id(), candidate.prong1Id(), candidate.prong2Id());
           candSelInfo(indexHfcReducedCollision, outputMl[0], outputMl[2]);
         }
       }
@@ -733,7 +733,7 @@ struct HfCorrelatorDsHadrons {
         if (!track.isGlobalTrackWoDCA()) {
           continue;
         }
-        assocTrackReduced(indexHfcReducedCollision, track.phi(), track.eta(), track.pt());
+        assocTrackReduced(indexHfcReducedCollision, track.globalIndex(), track.phi(), track.eta(), track.pt());
         assocTrackSelInfo(indexHfcReducedCollision, track.tpcNClsCrossedRows(), track.itsClusterMap(), track.itsNCls(), track.dcaXY(), track.dcaZ());
       }
 
diff --git a/PWGHF/HFC/TableProducer/correlatorDsHadronsReduced.cxx b/PWGHF/HFC/TableProducer/correlatorDsHadronsReduced.cxx
@@ -10,7 +10,7 @@
 // or submit itself to any jurisdiction.
 
 /// \file correlatorDsHadronsReduced.cxx
-/// \brief Ds-Hadrons correlator task for ME offline
+/// \brief Ds-Hadrons correlator task for offline analysis
 /// \author Samuele Cattaruzzi <samuele.cattaruzzi@cern.ch>
 
 #include <vector>
@@ -43,10 +43,15 @@ using BinningTypeDerived = ColumnBinningPolicy<aod::hf_collisions_reduced::PosZ,
 struct HfCorrelatorDsHadronsReduced {
   Produces<aod::DsHadronPair> entryDsHadronPair;
   Produces<aod::DsHadronRecoInfo> entryDsHadronRecoInfo;
+  Produces<aod::DsHadronMlInfo> entryDsHadronMlInfo;
+  Produces<aod::DsCandRecoInfo> entryDsCandRecoInfo;
+  Produces<aod::TrackRecoInfo> entryTrackRecoInfo;
   // Produces<aod::DsHadronGenInfo> entryDsHadronGenInfo;
 
   Configurable<bool> fillHistoData{"fillHistoData", true, "Flag for filling histograms in data processes"};
   Configurable<int> numberEventsMixed{"numberEventsMixed", 5, "Number of events mixed in ME process"};
+  Configurable<std::vector<double>> binsPtD{"binsPtD", std::vector<double>{1., 3., 5., 8., 16., 36.}, "pT bin limits for candidate mass plots"};
+  Configurable<std::vector<double>> binsPtHadron{"binsPtHadron", std::vector<double>{0.3, 1., 2., 50.}, "pT bin limits for assoc particle"};
 
   SliceCache cache;
 
@@ -56,9 +61,11 @@ struct HfCorrelatorDsHadronsReduced {
 
   ConfigurableAxis zPoolBins{"zPoolBins", {VARIABLE_WIDTH, -10.0, -2.5, 2.5, 10.0}, "z vertex position pools"};
   ConfigurableAxis multPoolBins{"multPoolBins", {VARIABLE_WIDTH, 0., 900., 1800., 6000.}, "event multiplicity pools (FT0M)"};
-  ConfigurableAxis binsMultFT0M{"binsMultFT0M", {600, 0., 6000.}, "Multiplicity as FT0M signal amplitude"};
+  ConfigurableAxis binsMultFT0M{"binsMultFT0M", {100, 0., 10000.}, "Multiplicity as FT0M signal amplitude"};
   ConfigurableAxis binsPosZ{"binsPosZ", {100, -10., 10.}, "primary vertex z coordinate"};
   ConfigurableAxis binsPoolBin{"binsPoolBin", {9, 0., 9.}, "PoolBin"};
+  ConfigurableAxis binsEta{"binsEta", {50, -2., 2.}, "#it{#eta}"};
+  ConfigurableAxis binsPhi{"binsPhi", {64, -PIHalf, 3. * PIHalf}, "#it{#varphi}"};
 
   HistogramRegistry registry{"registry", {}, OutputObjHandlingPolicy::AnalysisObject};
 
@@ -67,23 +74,95 @@ struct HfCorrelatorDsHadronsReduced {
     AxisSpec axisMultFT0M = {binsMultFT0M, "MultiplicityFT0M"};
     AxisSpec axisPosZ = {binsPosZ, "PosZ"};
     AxisSpec axisPoolBin = {binsPoolBin, "PoolBin"};
+    AxisSpec axisEta = {binsEta, "#it{#eta}"};
+    AxisSpec axisPhi = {binsPhi, "#it{#varphi}"};
+    AxisSpec axisPtD = {(std::vector<double>)binsPtD, "#it{p}_{T} (GeV/#it{c})"};
+    AxisSpec axisPtHadron = {(std::vector<double>)binsPtHadron, "#it{p}_{T} Hadron (GeV/#it{c})"};
 
     // Histograms for data analysis
     if (fillHistoData) {
       registry.add("hMultFT0M", "Multiplicity FT0M", {HistType::kTH1F, {axisMultFT0M}});
       registry.add("hZVtx", "z vertex", {HistType::kTH1F, {axisPosZ}});
+      registry.add("hCollisionPoolBin", "Collision pool bin", {HistType::kTH1F, {axisPoolBin}});
       registry.add("hDsPoolBin", "Ds candidates pool bin", {HistType::kTH1F, {axisPoolBin}});
+      registry.add("hPhiVsPtCand", "Ds candidates phiVsPt", {HistType::kTH2F, {{axisPhi}, {axisPtD}}});
+      registry.add("hPhiVsPtPartAssoc", "Particles associated phiVsPt", {HistType::kTH3F, {{axisPhi}, {axisPtD}, {axisPtHadron}}});
+      registry.add("hEtaVsPtCand", "Ds candidates etaVsPt", {HistType::kTH2F, {{axisEta}, {axisPtD}}});
+      registry.add("hEtaVsPtPartAssoc", "Particles associated etaVsPt", {HistType::kTH3F, {{axisEta}, {axisPtD}, {axisPtHadron}}});
       registry.add("hTracksPoolBin", "Particles associated pool bin", {HistType::kTH1F, {axisPoolBin}});
     }
   }
 
+  void processDerivedData(aod::HfcRedCollisions const& collisions,
+                          soa::Join<aod::DsCandReduceds, aod::DsCandSelInfos> const& candidates,
+                          soa::Join<aod::AssocTrackReds, aod::AssocTrackSels> const& tracks)
+  {
+
+    BinningTypeDerived corrBinning{{zPoolBins, multPoolBins}, true};
+
+    for (const auto& collision : collisions) {
+      int poolBin = corrBinning.getBin(std::make_tuple(collision.posZ(), collision.multiplicity()));
+      registry.fill(HIST("hCollisionPoolBin"), poolBin);
+      registry.fill(HIST("hMultFT0M"), collision.multiplicity());
+      registry.fill(HIST("hZVtx"), collision.posZ());
+
+      auto thisCollId = collision.globalIndex();
+      auto candsDsThisColl = candidates.sliceBy(candPerCol, thisCollId);
+      auto tracksThisColl = tracks.sliceBy(tracksPerCol, thisCollId);
+
+      for (const auto& candidate : candsDsThisColl) {
+        registry.fill(HIST("hDsPoolBin"), poolBin);
+        registry.fill(HIST("hPhiVsPtCand"), RecoDecay::constrainAngle(candidate.phiCand(), -PIHalf), candidate.ptCand());
+        registry.fill(HIST("hEtaVsPtCand"), candidate.etaCand(), candidate.ptCand());
+        entryDsCandRecoInfo(candidate.invMassDs(), candidate.ptCand(), candidate.bdtScorePrompt(), candidate.bdtScoreBkg());
+        for (const auto& track : tracksThisColl) {
+          registry.fill(HIST("hTracksPoolBin"), poolBin);
+          registry.fill(HIST("hPhiVsPtPartAssoc"), RecoDecay::constrainAngle(track.phiAssocTrack(), -PIHalf), candidate.ptCand(), track.ptAssocTrack());
+          registry.fill(HIST("hEtaVsPtPartAssoc"), track.etaAssocTrack(), candidate.ptCand(), track.ptAssocTrack());
+
+          entryDsHadronPair(getDeltaPhi(track.phiAssocTrack(), candidate.phiCand()),
+                            track.etaAssocTrack() - candidate.etaCand(),
+                            candidate.ptCand(),
+                            track.ptAssocTrack(),
+                            poolBin);
+          entryDsHadronRecoInfo(candidate.invMassDs(), false, false);
+          entryDsHadronMlInfo(candidate.bdtScorePrompt(), candidate.bdtScoreBkg());
+          entryTrackRecoInfo(track.dcaXY(), track.dcaZ(), track.nTpcCrossedRows());
+        }
+      }
+    }
+  }
+  PROCESS_SWITCH(HfCorrelatorDsHadronsReduced, processDerivedData, "Process Derived Data", true);
+
   void processDerivedDataME(aod::HfcRedCollisions const& collisions,
                             aod::DsCandReduceds const& candidates,
                             aod::AssocTrackReds const& tracks)
   {
 
     BinningTypeDerived corrBinning{{zPoolBins, multPoolBins}, true};
 
+    for (const auto& collision : collisions) {
+      int poolBin = corrBinning.getBin(std::make_tuple(collision.posZ(), collision.multiplicity()));
+      registry.fill(HIST("hCollisionPoolBin"), poolBin);
+      registry.fill(HIST("hMultFT0M"), collision.multiplicity());
+      registry.fill(HIST("hZVtx"), collision.posZ());
+
+      auto thisCollId = collision.globalIndex();
+      auto candsDsThisColl = candidates.sliceBy(candPerCol, thisCollId);
+      auto tracksThisColl = tracks.sliceBy(tracksPerCol, thisCollId);
+
+      for (const auto& candidate : candsDsThisColl) {
+        registry.fill(HIST("hDsPoolBin"), poolBin);
+        registry.fill(HIST("hPhiVsPtCand"), RecoDecay::constrainAngle(candidate.phiCand(), -PIHalf), candidate.ptCand());
+        registry.fill(HIST("hEtaVsPtCand"), candidate.etaCand(), candidate.ptCand());
+        for (const auto& track : tracksThisColl) {
+          registry.fill(HIST("hTracksPoolBin"), poolBin);
+          registry.fill(HIST("hPhiVsPtPartAssoc"), RecoDecay::constrainAngle(track.phiAssocTrack(), -PIHalf), candidate.ptCand(), track.ptAssocTrack());
+          registry.fill(HIST("hEtaVsPtPartAssoc"), track.etaAssocTrack(), candidate.ptCand(), track.ptAssocTrack());
+        }
+      }
+    }
+
     auto tracksTuple = std::make_tuple(candidates, tracks);
 
     Pair<aod::HfcRedCollisions, aod::DsCandReduceds, aod::AssocTrackReds, BinningTypeDerived> pairData{corrBinning, numberEventsMixed, -1, collisions, tracksTuple, &cache};
@@ -95,13 +174,13 @@ struct HfCorrelatorDsHadronsReduced {
 
       int poolBin = corrBinning.getBin(std::make_tuple(c2.posZ(), c2.multiplicity()));
       int poolBinDs = corrBinning.getBin(std::make_tuple(c1.posZ(), c1.multiplicity()));
-      registry.fill(HIST("hMultFT0M"), c1.multiplicity());
-      registry.fill(HIST("hZVtx"), c1.posZ());
-      registry.fill(HIST("hTracksPoolBin"), poolBin);
-      registry.fill(HIST("hDsPoolBin"), poolBinDs);
+
+      if (poolBin != poolBinDs) {
+        LOGF(info, "Error, poolBins are diffrent");
+      }
 
       for (const auto& [cand, pAssoc] : o2::soa::combinations(o2::soa::CombinationsFullIndexPolicy(tracks1, tracks2))) {
-        LOGF(info, "Mixed event tracks pair: (%d, %d) from events (%d, %d), track event: (%d, %d)", cand.index(), pAssoc.index(), c1.index(), c2.index(), cand.hfcRedCollisionId(), pAssoc.hfcRedCollisionId());
+        // LOGF(info, "Mixed event tracks pair: (%d, %d) from events (%d, %d), track event: (%d, %d)", cand.index(), pAssoc.index(), c1.index(), c2.index(), cand.hfcRedCollisionId(), pAssoc.hfcRedCollisionId());
 
         entryDsHadronPair(getDeltaPhi(pAssoc.phiAssocTrack(), cand.phiCand()),
                           pAssoc.etaAssocTrack() - cand.etaCand(),
@@ -113,7 +192,7 @@ struct HfCorrelatorDsHadronsReduced {
       }
     }
   }
-  PROCESS_SWITCH(HfCorrelatorDsHadronsReduced, processDerivedDataME, "Process Mixed Event Derived Data", true);
+  PROCESS_SWITCH(HfCorrelatorDsHadronsReduced, processDerivedDataME, "Process Mixed Event Derived Data", false);
 };
 
 WorkflowSpec defineDataProcessing(ConfigContext const& cfgc)

Original file line number	Diff line number	Diff line change
`@@ -145,7 +145,7 @@ struct HfCorrelatorDsHadrons {`
`145`	`145`	`Produces<aod::DsCandReduceds> candReduced;`
`146`	`146`	`Produces<aod::DsCandSelInfos> candSelInfo;`
`147`	`147`	`Produces<aod::AssocTrackReds> assocTrackReduced;`
`148`		`- Produces<aod::AssocTrackSelInfos> assocTrackSelInfo;`
	`148`	`+ Produces<aod::AssocTrackSels> assocTrackSelInfo;`
`149`	`149`
`150`	`150`	`Configurable<bool> fillHistoData{"fillHistoData", true, "Flag for filling histograms in data processes"};`
`151`	`151`	`Configurable<bool> fillHistoMcRec{"fillHistoMcRec", true, "Flag for filling histograms in MC Rec processes"};`
`@@ -717,13 +717,13 @@ struct HfCorrelatorDsHadrons {`
`717`	`717`	`for (unsigned int iclass = 0; iclass < classMl->size(); iclass++) {`
`718`	`718`	`outputMl[iclass] = candidate.mlProbDsToKKPi()[classMl->at(iclass)];`
`719`	`719`	`}`
`720`		`- candReduced(indexHfcReducedCollision, candidate.phi(), candidate.eta(), candidate.pt(), hfHelper.invMassDsToKKPi(candidate));`
	`720`	`+ candReduced(indexHfcReducedCollision, candidate.phi(), candidate.eta(), candidate.pt(), hfHelper.invMassDsToKKPi(candidate), candidate.prong0Id(), candidate.prong1Id(), candidate.prong2Id());`
`721`	`721`	`candSelInfo(indexHfcReducedCollision, outputMl[0], outputMl[2]);`
`722`	`722`	`} else if (candidate.isSelDsToPiKK() >= selectionFlagDs) {`
`723`	`723`	`for (unsigned int iclass = 0; iclass < classMl->size(); iclass++) {`
`724`	`724`	`outputMl[iclass] = candidate.mlProbDsToPiKK()[classMl->at(iclass)];`
`725`	`725`	`}`
`726`		`- candReduced(indexHfcReducedCollision, candidate.phi(), candidate.eta(), candidate.pt(), hfHelper.invMassDsToPiKK(candidate));`
	`726`	`+ candReduced(indexHfcReducedCollision, candidate.phi(), candidate.eta(), candidate.pt(), hfHelper.invMassDsToPiKK(candidate), candidate.prong0Id(), candidate.prong1Id(), candidate.prong2Id());`
`727`	`727`	`candSelInfo(indexHfcReducedCollision, outputMl[0], outputMl[2]);`
`728`	`728`	`}`
`729`	`729`	`}`
`@@ -733,7 +733,7 @@ struct HfCorrelatorDsHadrons {`
`733`	`733`	`if (!track.isGlobalTrackWoDCA()) {`
`734`	`734`	`continue;`
`735`	`735`	`}`
`736`		`- assocTrackReduced(indexHfcReducedCollision, track.phi(), track.eta(), track.pt());`
	`736`	`+ assocTrackReduced(indexHfcReducedCollision, track.globalIndex(), track.phi(), track.eta(), track.pt());`
`737`	`737`	`assocTrackSelInfo(indexHfcReducedCollision, track.tpcNClsCrossedRows(), track.itsClusterMap(), track.itsNCls(), track.dcaXY(), track.dcaZ());`
`738`	`738`	`}`
`739`	`739`