Skip to content

Commit bd1aafc

Browse files
[PWGDQ] Adding charge information to minitree (#10236)
Co-authored-by: Lucamicheletti93 <luca.mike93@gmail.com>
1 parent eca755f commit bd1aafc

File tree

2 files changed

+23
-6
lines changed

2 files changed

+23
-6
lines changed

PWGDQ/Core/VarManager.h

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2746,6 +2746,22 @@ void VarManager::FillPair(T1 const& t1, T2 const& t2, float* values)
27462746
double Ptot2 = TMath::Sqrt(v2.Px() * v2.Px() + v2.Py() * v2.Py() + v2.Pz() * v2.Pz());
27472747
values[kDeltaPtotTracks] = Ptot1 - Ptot2;
27482748

2749+
if (t1.sign() > 0) {
2750+
values[kPt1] = t1.pt();
2751+
values[kEta1] = t1.eta();
2752+
values[kPhi1] = t1.phi();
2753+
values[kPt2] = t2.pt();
2754+
values[kEta2] = t2.eta();
2755+
values[kPhi2] = t2.phi();
2756+
} else {
2757+
values[kPt1] = t2.pt();
2758+
values[kEta1] = t2.eta();
2759+
values[kPhi1] = t2.phi();
2760+
values[kPt2] = t1.pt();
2761+
values[kEta2] = t1.eta();
2762+
values[kPhi2] = t1.phi();
2763+
}
2764+
27492765
if (fgUsedVars[kDeltaPhiPair2]) {
27502766
double phipair2 = v1.Phi() - v2.Phi();
27512767
if (phipair2 > 3 * TMath::Pi() / 2) {

PWGDQ/Tasks/tableReader.cxx

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1442,19 +1442,20 @@ struct AnalysisSameEventPairing {
14421442
fHistMan->FillHistClass(Form("%s_unambiguous", histNames[iCut][0].Data()), VarManager::fgValues);
14431443
}
14441444
if (useMiniTree.fConfigMiniTree) {
1445+
// By default (kPt1, kEta1, kPhi1) are for the positive charge
14451446
float dileptonMass = VarManager::fgValues[VarManager::kMass];
14461447
if (dileptonMass > useMiniTree.fConfigMiniTreeMinMass && dileptonMass < useMiniTree.fConfigMiniTreeMaxMass) {
14471448
dileptonMiniTree(VarManager::fgValues[VarManager::kMass],
14481449
VarManager::fgValues[VarManager::kPt],
14491450
VarManager::fgValues[VarManager::kRap],
14501451
VarManager::fgValues[VarManager::kCentFT0C],
14511452
VarManager::fgValues[VarManager::kCos2DeltaPhi],
1452-
t1.pt(),
1453-
t1.eta(),
1454-
t1.phi(),
1455-
t2.pt(),
1456-
t2.eta(),
1457-
t2.phi());
1453+
VarManager::fgValues[VarManager::kPt1],
1454+
VarManager::fgValues[VarManager::kEta1],
1455+
VarManager::fgValues[VarManager::kPhi1],
1456+
VarManager::fgValues[VarManager::kPt2],
1457+
VarManager::fgValues[VarManager::kEta2],
1458+
VarManager::fgValues[VarManager::kPhi2]);
14581459
}
14591460
}
14601461
} else {

0 commit comments

Comments
 (0)