diff --git a/PWGDQ/Core/VarManager.h b/PWGDQ/Core/VarManager.h index 050b0a5c9bf..4f3d57282e3 100644 --- a/PWGDQ/Core/VarManager.h +++ b/PWGDQ/Core/VarManager.h @@ -2746,6 +2746,22 @@ void VarManager::FillPair(T1 const& t1, T2 const& t2, float* values) double Ptot2 = TMath::Sqrt(v2.Px() * v2.Px() + v2.Py() * v2.Py() + v2.Pz() * v2.Pz()); values[kDeltaPtotTracks] = Ptot1 - Ptot2; + if (t1.sign() > 0) { + values[kPt1] = t1.pt(); + values[kEta1] = t1.eta(); + values[kPhi1] = t1.phi(); + values[kPt2] = t2.pt(); + values[kEta2] = t2.eta(); + values[kPhi2] = t2.phi(); + } else { + values[kPt1] = t2.pt(); + values[kEta1] = t2.eta(); + values[kPhi1] = t2.phi(); + values[kPt2] = t1.pt(); + values[kEta2] = t1.eta(); + values[kPhi2] = t1.phi(); + } + if (fgUsedVars[kDeltaPhiPair2]) { double phipair2 = v1.Phi() - v2.Phi(); if (phipair2 > 3 * TMath::Pi() / 2) { diff --git a/PWGDQ/Tasks/tableReader.cxx b/PWGDQ/Tasks/tableReader.cxx index 3502825be02..f3fb74552be 100644 --- a/PWGDQ/Tasks/tableReader.cxx +++ b/PWGDQ/Tasks/tableReader.cxx @@ -1442,6 +1442,7 @@ struct AnalysisSameEventPairing { fHistMan->FillHistClass(Form("%s_unambiguous", histNames[iCut][0].Data()), VarManager::fgValues); } if (useMiniTree.fConfigMiniTree) { + // By default (kPt1, kEta1, kPhi1) are for the positive charge float dileptonMass = VarManager::fgValues[VarManager::kMass]; if (dileptonMass > useMiniTree.fConfigMiniTreeMinMass && dileptonMass < useMiniTree.fConfigMiniTreeMaxMass) { dileptonMiniTree(VarManager::fgValues[VarManager::kMass], @@ -1449,12 +1450,12 @@ struct AnalysisSameEventPairing { VarManager::fgValues[VarManager::kRap], VarManager::fgValues[VarManager::kCentFT0C], VarManager::fgValues[VarManager::kCos2DeltaPhi], - t1.pt(), - t1.eta(), - t1.phi(), - t2.pt(), - t2.eta(), - t2.phi()); + VarManager::fgValues[VarManager::kPt1], + VarManager::fgValues[VarManager::kEta1], + VarManager::fgValues[VarManager::kPhi1], + VarManager::fgValues[VarManager::kPt2], + VarManager::fgValues[VarManager::kEta2], + VarManager::fgValues[VarManager::kPhi2]); } } } else {