1515#include " Common/DataModel/PIDResponse.h"
1616#include " PWGLF/DataModel/LFStrangenessTables.h"
1717
18+ #include " TPDGCode.h"
1819#include " TLorentzVector.h"
1920
2021using 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