Skip to content

Commit 9e37a6b

Browse files
committed
Now using named PDG codes
1 parent 77a58dd commit 9e37a6b

File tree

1 file changed

+13
-12
lines changed

1 file changed

+13
-12
lines changed

PWGCF/MultiparticleCorrelations/Tasks/ThreeParticleCorrelations.cxx

Lines changed: 13 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
#include "Common/DataModel/PIDResponse.h"
1616
#include "PWGLF/DataModel/LFStrangenessTables.h"
1717

18+
#include "TPDGCode.h"
1819
#include "TLorentzVector.h"
1920

2021
using namespace o2;
@@ -278,17 +279,17 @@ struct ThreePartCorr {
278279
for (const auto& particle : particles) {
279280
if (particle.isPhysicalPrimary()) {
280281

281-
if (particle.pdgCode() == 211) { // Pos pions
282+
if (particle.pdgCode() == kPiPlus) { // Pos pions
282283
MCRegistry.fill(HIST("hGenPionP"), particle.pt());
283-
} else if (particle.pdgCode() == -211) { // Neg pions
284+
} else if (particle.pdgCode() == kPiMinus) { // Neg pions
284285
MCRegistry.fill(HIST("hGenPionN"), particle.pt());
285-
} else if (particle.pdgCode() == 321) { // Pos kaons
286+
} else if (particle.pdgCode() == kKPlus) { // Pos kaons
286287
MCRegistry.fill(HIST("hGenKaonP"), particle.pt());
287-
} else if (particle.pdgCode() == -321) { // Neg kaons
288+
} else if (particle.pdgCode() == kKMinus) { // Neg kaons
288289
MCRegistry.fill(HIST("hGenKaonN"), particle.pt());
289-
} else if (particle.pdgCode() == 2212) { // Pos protons
290+
} else if (particle.pdgCode() == kProton) { // Pos protons
290291
MCRegistry.fill(HIST("hGenProtonP"), particle.pt());
291-
} else if (particle.pdgCode() == -2212) { // Neg protons
292+
} else if (particle.pdgCode() == kProtonBar) { // Neg protons
292293
MCRegistry.fill(HIST("hGenProtonN"), particle.pt());
293294
}
294295
}
@@ -312,17 +313,17 @@ struct ThreePartCorr {
312313
auto particle = track.mcParticle();
313314
if (particle.isPhysicalPrimary()) {
314315

315-
if (particle.pdgCode() == 211) { // Pos pions
316+
if (particle.pdgCode() == kPiPlus) { // Pos pions
316317
MCRegistry.fill(HIST("hRecPionP"), track.pt());
317-
} else if (particle.pdgCode() == -211) { // Neg pions
318+
} else if (particle.pdgCode() == kPiMinus) { // Neg pions
318319
MCRegistry.fill(HIST("hRecPionN"), track.pt());
319-
} else if (particle.pdgCode() == 321) { // Pos kaons
320+
} else if (particle.pdgCode() == kKPlus) { // Pos kaons
320321
MCRegistry.fill(HIST("hRecKaonP"), track.pt());
321-
} else if (particle.pdgCode() == -321) { // Neg kaons
322+
} else if (particle.pdgCode() == kKMinus) { // Neg kaons
322323
MCRegistry.fill(HIST("hRecKaonN"), track.pt());
323-
} else if (particle.pdgCode() == 2212) { // Pos protons
324+
} else if (particle.pdgCode() == kProton) { // Pos protons
324325
MCRegistry.fill(HIST("hRecProtonP"), track.pt());
325-
} else if (particle.pdgCode() == -2212) { // Neg protons
326+
} else if (particle.pdgCode() == kProtonBar) { // Neg protons
326327
MCRegistry.fill(HIST("hRecProtonN"), track.pt());
327328
}
328329
}

0 commit comments

Comments
 (0)