Add files via upload

Author: 15071832337

PWGLF/Tasks/Strangeness/lambdaJetpolarization.cxx

@@ -143,6 +143,9 @@ struct LambdaJetpolarization {
   Configurable<bool> hasTOF1Leg{"hasTOF1Leg", 0, "hasTOF1Leg"};
   Configurable<bool> isMC{"isMC", 1, "fill MC histograms"};
   Configurable<int> pdgLmbda{"pdgLmbda", 3122, "pdgLmbda"};
+  Configurable<int> pdgkZeroShort{"pdgkZeroShort", 310, "pdgkZeroShort"};
+  Configurable<int> pdgProton{"pdgProton", 2212, "pdgProton"};
+  Configurable<int> pdgPion{"pdgPion", 211, "pdgPion"};
 
   void init(InitContext const&)
   {
@@ -2326,16 +2329,16 @@ struct LambdaJetpolarization {
         int lPDG = 0;
         bool isDauLambda = false, isDauAntiLambda = false;
         bool isprimary = false;
-        if (std::abs(v0mcparticle.pdgCode()) == 310 || std::abs(v0mcparticle.pdgCode()) == pdgLmbda) { // K0S or Lambda
+        if (std::abs(v0mcparticle.pdgCode()) == pdgkZeroShort || std::abs(v0mcparticle.pdgCode()) == pdgLmbda) { // K0S or Lambda
           lPDG = v0mcparticle.pdgCode();
           isprimary = v0mcparticle.isPhysicalPrimary();
         }
         for (const auto& mcparticleDaughter0 : v0mcparticle.daughters_as<aod::McParticles>()) {
           for (const auto& mcparticleDaughter1 : v0mcparticle.daughters_as<aod::McParticles>()) {
-            if (mcparticleDaughter0.pdgCode() == -211 && mcparticleDaughter1.pdgCode() == 2212) { // proton + pion^{-}
+            if (mcparticleDaughter0.pdgCode() == -pdgPion && mcparticleDaughter1.pdgCode() == pdgProton) { // proton + pion^{-}
               isDauLambda = true;
             }
-            if (mcparticleDaughter0.pdgCode() == 211 && mcparticleDaughter1.pdgCode() == -2212) { // antiproton + pion^{+}
+            if (mcparticleDaughter0.pdgCode() == pdgPion && mcparticleDaughter1.pdgCode() == -pdgProton) { // antiproton + pion^{+}
               isDauAntiLambda = true;
             }
           }
@@ -2405,16 +2408,16 @@ struct LambdaJetpolarization {
         int lPDG = 0;
         bool isDauLambda = false, isDauAntiLambda = false;
         bool isprimary = false;
-        if (std::abs(v0mcparticle.pdgCode()) == 310 || std::abs(v0mcparticle.pdgCode()) == pdgLmbda) { // K0S or Lambda
+        if (std::abs(v0mcparticle.pdgCode()) == pdgkZeroShort || std::abs(v0mcparticle.pdgCode()) == pdgLmbda) { // K0S or Lambda
           lPDG = v0mcparticle.pdgCode();
           isprimary = v0mcparticle.isPhysicalPrimary();
         }
         for (const auto& mcparticleDaughter0 : v0mcparticle.daughters_as<aod::McParticles>()) {
           for (const auto& mcparticleDaughter1 : v0mcparticle.daughters_as<aod::McParticles>()) {
-            if (mcparticleDaughter0.pdgCode() == -211 && mcparticleDaughter1.pdgCode() == 2212) { // proton + pion^{-}
+            if (mcparticleDaughter0.pdgCode() == -pdgPion && mcparticleDaughter1.pdgCode() == pdgProton) { // proton + pion^{-}
               isDauLambda = true;
             }
-            if (mcparticleDaughter0.pdgCode() == 211 && mcparticleDaughter1.pdgCode() == -2212) { // antiproton + pion^{+}
+            if (mcparticleDaughter0.pdgCode() == pdgPion && mcparticleDaughter1.pdgCode() == -pdgProton) { // antiproton + pion^{+}
               isDauAntiLambda = true;
             }
           }
@@ -2564,12 +2567,12 @@ struct LambdaJetpolarizationMCcount {
           if (!mcparticle.has_daughters()) {
             continue;
           }
-          if (mcparticle.pdgCode() == 310) {
+          if (mcparticle.pdgCode() == pdgkZeroShort) {
             registry.fill(HIST("hK0ShortCount"), 0.5);
             registry.fill(HIST("hK0ShortCount_PtDiff"), mcparticle.pt());
             for (const auto& mcparticleDaughter0 : mcparticle.daughters_as<aod::McParticles>()) {
               for (const auto& mcparticleDaughter1 : mcparticle.daughters_as<aod::McParticles>()) {
-                if (mcparticleDaughter0.pdgCode() == 211 && mcparticleDaughter1.pdgCode() == -211) {
+                if (mcparticleDaughter0.pdgCode() == pdgPion && mcparticleDaughter1.pdgCode() == -pdgPion) {
                   registry.fill(HIST("hK0ShortCount"), 1.5);
                 }
               }
@@ -2580,7 +2583,7 @@ struct LambdaJetpolarizationMCcount {
             registry.fill(HIST("hLambdaCount_PtDiff"), mcparticle.pt());
             for (const auto& mcparticleDaughter0 : mcparticle.daughters_as<aod::McParticles>()) {
               for (const auto& mcparticleDaughter1 : mcparticle.daughters_as<aod::McParticles>()) {
-                if (mcparticleDaughter0.pdgCode() == -211 && mcparticleDaughter1.pdgCode() == 2212) {
+                if (mcparticleDaughter0.pdgCode() == -pdgPion && mcparticleDaughter1.pdgCode() == pdgProton) {
                   registry.fill(HIST("hLambdaCount"), 1.5);
                 }
               }
@@ -2591,7 +2594,7 @@ struct LambdaJetpolarizationMCcount {
             registry.fill(HIST("hAntiLambdaCount_PtDiff"), mcparticle.pt());
             for (const auto& mcparticleDaughter0 : mcparticle.daughters_as<aod::McParticles>()) {
               for (const auto& mcparticleDaughter1 : mcparticle.daughters_as<aod::McParticles>()) {
-                if (mcparticleDaughter0.pdgCode() == 211 && mcparticleDaughter1.pdgCode() == -2212) {
+                if (mcparticleDaughter0.pdgCode() == pdgPion && mcparticleDaughter1.pdgCode() == -pdgProton) {
                   registry.fill(HIST("hAntiLambdaCount"), 1.5);
                 }
               }