Improvements for corr bkgs

Author: Marcellocosti

PWGHF/Core/DecayChannels.h

@@ -127,6 +127,7 @@ enum DecayChannelResonant : HfDecayChannel {
   LcToPKstar0 = 24,        // p anti-K*0(892)
   LcToDeltaplusplusK = 25, // Δ++ K−
   LcToL1520Pi = 26,        // Λ(1520) π+
+  LcToPPhi = 29,           // p φ
   // Ξc+
   XicToPKstar0 = 27, // p anti-K*0(892)
   XicToPPhi = 28,    // p φ

PWGHF/HFC/TableProducer/correlatorFlowCharmHadronsReduced.cxx

@@ -89,6 +89,42 @@ double getPhi(const TTrack& track)
   }
 }
 
+/// Get charm candidate or hadron track pT
+/// \param track is the candidate
+template <typename TTrack>
+double getPt(const TTrack& track)
+{
+  if constexpr (requires { track.ptAssoc(); }) {
+    return track.ptAssoc();
+  } else {
+    return track.ptTrig();
+  }
+}
+
+/// Get charm candidate or hadron track eta
+/// \param track is the candidate
+template <typename TTrack>
+double getEta(const TTrack& track)
+{
+  if constexpr (requires { track.etaAssoc(); }) {
+    return track.etaAssoc();
+  } else {
+    return track.etaTrig();
+  }
+}
+
+/// Get charm candidate or hadron track phi
+/// \param track is the candidate
+template <typename TTrack>
+double getPhi(const TTrack& track)
+{
+  if constexpr (requires { track.phiAssoc(); }) {
+    return track.phiAssoc();
+  } else {
+    return track.phiTrig();
+  }
+}
+
 struct HfCorrelatorFlowCharmHadronsReduced {
   Produces<aod::HfcRedSEChHads> rowPairSECharmHads; //! Correlation pairs information Same Event
   Produces<aod::HfcRedMEChHads> rowPairMECharmHads; //! Correlation pairs information Mixed Event
@@ -128,6 +164,13 @@ struct HfCorrelatorFlowCharmHadronsReduced {
   Preslice<AssocTracks> assocTracksPerCol = aod::hf_correl_charm_had_reduced::hfcRedCorrCollId;
   Preslice<TrigCharmCands> trigCharmCandsPerCol = aod::hf_correl_charm_had_reduced::hfcRedCorrCollId;
 
+  Filter filterAssocTracks = (nabs(aod::hf_correl_charm_had_reduced::dcaXYAssoc) < dcaXYTrackMax) && (nabs(aod::hf_correl_charm_had_reduced::dcaZAssoc) < dcaZTrackMax) && (aod::hf_correl_charm_had_reduced::nTpcCrossedRowsAssoc > tpcCrossedRowsMin) && (aod::hf_correl_charm_had_reduced::itsNClsAssoc > itsNClsMin);
+  Filter filterTrigTracks = (nabs(aod::hf_correl_charm_had_reduced::dcaXYTrig) < dcaXYTrackMax) && (nabs(aod::hf_correl_charm_had_reduced::dcaZTrig) < dcaZTrackMax) && (aod::hf_correl_charm_had_reduced::nTpcCrossedRowsTrig > tpcCrossedRowsMin) && (aod::hf_correl_charm_had_reduced::itsNClsTrig > itsNClsMin);
+  Filter filterSameEvtPairs = (nabs(aod::hf_correl_charm_had_reduced::deltaEta) > deltaEtaAbsMin) && (nabs(aod::hf_correl_charm_had_reduced::deltaEta) < deltaEtaAbsMax);
+
+  Preslice<AssocTracks> assocTracksPerCol = aod::hf_correl_charm_had_reduced::hfcRedCorrCollId;
+  Preslice<TrigCharmCands> trigCharmCandsPerCol = aod::hf_correl_charm_had_reduced::hfcRedCorrCollId;
+
   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 centPoolBins{"centPoolBins", {VARIABLE_WIDTH, 0., 10., 20., 30., 40., 50., 60., 70., 80., 90., 100}, "Event centrality pools"};
@@ -168,6 +211,12 @@ struct HfCorrelatorFlowCharmHadronsReduced {
       poolBins = (multPoolBins->size() - 1) * (zPoolBins->size() - 1);
     }
 
+    if (doprocessSameEventCharmHadWCentMix || doprocessSameEventHadHadWCentMix || doprocessMixedEventCharmHadWCentMix || doprocessMixedEventHadHadWCentMix) {
+      poolBins = (centPoolBins->size() - 1) * (zPoolBins->size() - 1);
+    } else {
+      poolBins = (multPoolBins->size() - 1) * (zPoolBins->size() - 1);
+    }
+
     const AxisSpec axisInvMass{binsInvMass, "Inv. mass (GeV/#it{c}^{2})"};
     const AxisSpec axisCent = {binsCent, "Centrality"};
     const AxisSpec axisMultFT0M = {binsMultFT0M, "MultiplicityFT0M"};

PWGHF/HFC/TableProducer/derivedDataCreatorCorrelationsReduced.cxx

@@ -407,6 +407,7 @@ struct HfDerivedDataCreatorCorrelationsReduced {
       rowAssocBases(rowCollisions.lastIndex(), assTrk.phi(), assTrk.eta(), assTrkPt);
       rowAssocTrkSels(assTrk.tpcNClsCrossedRows(), assTrk.itsClusterMap(), assTrk.itsNCls(), assTrk.dcaXY(), assTrk.dcaZ());
     }
+    return poolBin;
   }
 
   // Dplus with ML selections

PWGHF/TableProducer/treeCreatorD0ToKPi.cxx

@@ -536,7 +536,10 @@ struct HfTreeCreatorD0ToKPi {
         }
       }
       if constexpr (onlySig) {
-        if ((std::abs(candidate.flagMcMatchRec()) != o2::hf_decay::hf_cand_2prong::DecayChannelMain::D0ToPiK) || (fillCorrBkgs && (candidate.flagMcMatchRec() != 0))) {
+        if (fillCorrBkgs && candidate.flagMcMatchRec() == 0) {
+          continue;
+        }
+        if (!fillCorrBkgs && std::abs(candidate.flagMcMatchRec()) != o2::hf_decay::hf_cand_2prong::DecayChannelMain::D0ToPiK) {
           continue;
         }
       }

PWGHF/TableProducer/treeCreatorDplusToPiKPi.cxx

@@ -371,8 +371,8 @@ struct HfTreeCreatorDplusToPiKPi {
         candidate.eta(),
         candidate.phi(),
         hfHelper.yDplus(candidate),
-        coll.numContrib(),
         cent,
+        coll.numContrib(),
         flagMc,
         originMc,
         channelMc);

PWGHF/Utils/utilsMcMatching.h

@@ -138,6 +138,7 @@ static const std::unordered_map<DecayChannelResonant, const std::array<int, 2>>
   {DecayChannelResonant::LcToPKstar0, {-o2::constants::physics::Pdg::kK0Star892, +PDG_t::kProton}},
   {DecayChannelResonant::LcToDeltaplusplusK, {+2224, +PDG_t::kKMinus}},
   {DecayChannelResonant::LcToL1520Pi, {+102134, +PDG_t::kPiPlus}},
+  {DecayChannelResonant::LcToPPhi, {+PDG_t::kProton, +o2::constants::physics::Pdg::kPhi}},
 };
 
 // Ξc+