[PWGHF] Store LF tracks based on sign in B->JPsi X data creator (AliceO2Group#16928)

Author: fchinu

PWGHF/Core/HfHelper.h

@@ -1025,14 +1025,14 @@ struct HfHelper {
     }
 
     // d0 of K*0 daughters
-    if (std::abs(candB0.impactParameter2()) < cuts->get(binPt, "d0 K*0") ||
-        std::abs(candB0.impactParameter3()) < cuts->get(binPt, "d0 K*0")) {
+    if (std::abs(candB0.impactParameter2()) < cuts->get(binPt, "d0 K*0 daughters") ||
+        std::abs(candB0.impactParameter3()) < cuts->get(binPt, "d0 K*0 daughters")) {
       return false;
     }
 
     // d0 of J/Psi
-    if (std::abs(candB0.impactParameter0()) < cuts->get(binPt, "d0 J/Psi") ||
-        std::abs(candB0.impactParameter1()) < cuts->get(binPt, "d0 J/Psi")) {
+    if (std::abs(candB0.impactParameter0()) < cuts->get(binPt, "d0 J/Psi daughters") ||
+        std::abs(candB0.impactParameter1()) < cuts->get(binPt, "d0 J/Psi daughters")) {
       return false;
     }
 

PWGHF/D2H/DataModel/ReducedDataModel.h

@@ -327,8 +327,9 @@ DECLARE_SOA_TABLE(HfRedBach0Bases, "AOD", "HFREDBACH0BASE", //! Table with track
                   hf_track_pid_reduced::TPCTOFNSigmaPr<pidtpc::TPCNSigmaPr, pidtof::TOFNSigmaPr>,
                   aod::track::Px<aod::track::Signed1Pt, aod::track::Snp, aod::track::Alpha>,
                   aod::track::Py<aod::track::Signed1Pt, aod::track::Snp, aod::track::Alpha>,
-                  aod::track::Pz<aod::track::Signed1Pt, track::Tgl>,
-                  aod::track::PVector<aod::track::Signed1Pt, aod::track::Snp, aod::track::Alpha, aod::track::Tgl>);
+                  aod::track::Pz<aod::track::Signed1Pt, aod::track::Tgl>,
+                  aod::track::PVector<aod::track::Signed1Pt, aod::track::Snp, aod::track::Alpha, aod::track::Tgl>,
+                  aod::track::Sign<aod::track::Signed1Pt>);
 
 DECLARE_SOA_TABLE(HfRedBach0Cov, "AOD", "HFREDBACH0COV", //! Table with track covariance information for reduced workflow
                   soa::Index<>,
@@ -359,8 +360,9 @@ DECLARE_SOA_TABLE(HfRedBach1Bases, "AOD", "HFREDBACH1BASE", //! Table with track
                   hf_track_pid_reduced::TPCTOFNSigmaPr<pidtpc::TPCNSigmaPr, pidtof::TOFNSigmaPr>,
                   aod::track::Px<aod::track::Signed1Pt, aod::track::Snp, aod::track::Alpha>,
                   aod::track::Py<aod::track::Signed1Pt, aod::track::Snp, aod::track::Alpha>,
-                  aod::track::Pz<aod::track::Signed1Pt, track::Tgl>,
-                  aod::track::PVector<aod::track::Signed1Pt, aod::track::Snp, aod::track::Alpha, aod::track::Tgl>);
+                  aod::track::Pz<aod::track::Signed1Pt, aod::track::Tgl>,
+                  aod::track::PVector<aod::track::Signed1Pt, aod::track::Snp, aod::track::Alpha, aod::track::Tgl>,
+                  aod::track::Sign<aod::track::Signed1Pt>);
 
 DECLARE_SOA_TABLE(HfRedBach1Cov, "AOD", "HFREDBACH1COV", //! Table with track covariance information for reduced workflow
                   soa::Index<>,

PWGHF/D2H/TableProducer/candidateCreatorBToJpsiReduced.cxx

@@ -274,6 +274,9 @@ struct HfCandidateCreatorBToJpsiReduced {
           rowCandidateBpProngs(candJpsi.globalIndex(), trackLf0.globalIndex());
         } else if constexpr (DecChannel == DecayChannel::B0ToJpsiK0Star) {
           for (const auto& trackLf1 : tracksLfDau1ThisCollision) {
+            if (trackLf0.sign() == trackLf1.sign()) {
+              continue; // require opposite-sign tracks for K*0 reconstruction
+            }
             // this track is among daughters
             if (trackLf1.trackId() == candJpsi.prongPosId() || trackLf1.trackId() == candJpsi.prongNegId()) {
               continue;
@@ -351,6 +354,9 @@ struct HfCandidateCreatorBToJpsiReduced {
           }
         } else if constexpr (DecChannel == DecayChannel::BsToJpsiPhi) {
           for (const auto& trackLf1 : tracksLfDau1ThisCollision) {
+            if (trackLf0.sign() == trackLf1.sign()) {
+              continue; // require opposite-sign tracks for phi reconstruction
+            }
             // this track is among daughters
             if (trackLf1.trackId() == candJpsi.prongPosId() || trackLf1.trackId() == candJpsi.prongNegId()) {
               continue;