Use FT0M centrality for pp and FT0C for other systems

Author: gianniliveraro

PWGLF/TableProducer/Strangeness/sigma0builder.cxx

@@ -503,7 +503,8 @@ struct sigma0builder {
     if (fillHists)
       histos.fill(HIST("hEventSelection"), 19 /* Above max IR */);
 
-    histos.fill(HIST("hEventCentrality"), collision.centFT0C());
+    float centrality = doPPAnalysis ? collision.centFT0M() : collision.centFT0C();
+    histos.fill(HIST("hEventCentrality"), centrality);
     return true;
   }
 
@@ -751,6 +752,7 @@ struct sigma0builder {
     float pi0Mass = RecoDecay::m(arrpi0, std::array{o2::constants::physics::MassPhoton, o2::constants::physics::MassPhoton});
     float pi0Pt = RecoDecay::pt(std::array{gamma1.px() + gamma2.px(), gamma1.py() + gamma2.py()});
     float pi0Y = RecoDecay::y(std::array{gamma1.px() + gamma2.px(), gamma1.py() + gamma2.py(), gamma1.pz() + gamma2.pz()}, o2::constants::physics::MassPi0);
+    float centrality = doPPAnalysis ? collision.centFT0M() : collision.centFT0C();
 
     // MC-specific variables
     bool fIsPi0 = false, fIsMC = false;
@@ -766,12 +768,12 @@ struct sigma0builder {
             gamma1MC.pdgCodeMother() == 111 && gamma2MC.pdgCodeMother() == 111 &&
             gamma1.motherMCPartId() == gamma2.motherMCPartId()) {
           fIsPi0 = true;
-          histos.fill(HIST("Pi0QA/h3dMassPi0BeforeSel_MCAssoc"), collision.centFT0C(), pi0Pt, pi0Mass);
+          histos.fill(HIST("Pi0QA/h3dMassPi0BeforeSel_MCAssoc"), centrality, pi0Pt, pi0Mass);
         }
       }
     }
 
-    histos.fill(HIST("Pi0QA/h3dMassPi0BeforeSel_Candidates"), collision.centFT0C(), pi0Pt, pi0Mass);
+    histos.fill(HIST("Pi0QA/h3dMassPi0BeforeSel_Candidates"), centrality, pi0Pt, pi0Mass);
 
     // Photon-specific selections
     auto posTrackGamma1 = gamma1.template posTrackExtra_as<dauTracks>();
@@ -780,7 +782,7 @@ struct sigma0builder {
     auto negTrackGamma2 = gamma2.template negTrackExtra_as<dauTracks>();
 
     // Gamma1 Selection
-    bool passedTPCGamma1 = (TMath::Abs(posTrackGamma1.tpcNSigmaEl()) < Pi0PhotonMaxTPCNSigmas) ||
+    bool passedTPCGamma1 = (TMath::Abs(posTrackGamma1.tpcNSigmaEl()) < Pi0PhotonMaxTPCNSigmas) || 
                            (TMath::Abs(negTrackGamma1.tpcNSigmaEl()) < Pi0PhotonMaxTPCNSigmas);
 
     if (TMath::Abs(gamma1.mGamma()) > Pi0PhotonMaxMass ||
@@ -827,9 +829,9 @@ struct sigma0builder {
     }
 
     // Fill histograms
-    histos.fill(HIST("Pi0QA/h3dMassPi0AfterSel_Candidates"), collision.centFT0C(), pi0Pt, pi0Mass);
+    histos.fill(HIST("Pi0QA/h3dMassPi0AfterSel_Candidates"), centrality, pi0Pt, pi0Mass);
     if (fIsMC && fIsPi0)
-      histos.fill(HIST("Pi0QA/h3dMassPi0AfterSel_MCAssoc"), collision.centFT0C(), pi0Pt, pi0Mass);
+      histos.fill(HIST("Pi0QA/h3dMassPi0AfterSel_MCAssoc"), centrality, pi0Pt, pi0Mass);
   }
 
   // Process photon candidate
@@ -873,7 +875,8 @@ struct sigma0builder {
         return false;
       histos.fill(HIST("PhotonSel/hSelectionStatistics"), 6.);
     }
-    histos.fill(HIST("PhotonSel/h3dPhotonMass"), collision.centFT0C(), gamma.pt(), gamma.mGamma());
+    float centrality = doPPAnalysis ? collision.centFT0M() : collision.centFT0C();
+    histos.fill(HIST("PhotonSel/h3dPhotonMass"), centrality, gamma.pt(), gamma.mGamma());
     return true;
   }
 
@@ -917,8 +920,9 @@ struct sigma0builder {
       histos.fill(HIST("LambdaSel/hSelectionStatistics"), 6.);
     }
 
-    histos.fill(HIST("LambdaSel/h3dLambdaMass"), collision.centFT0C(), lambda.pt(), lambda.mLambda());
-    histos.fill(HIST("LambdaSel/h3dALambdaMass"), collision.centFT0C(), lambda.pt(), lambda.mAntiLambda());
+    float centrality = doPPAnalysis ? collision.centFT0M() : collision.centFT0C();
+    histos.fill(HIST("LambdaSel/h3dLambdaMass"), centrality, lambda.pt(), lambda.mLambda());
+    histos.fill(HIST("LambdaSel/h3dALambdaMass"), centrality, lambda.pt(), lambda.mAntiLambda());
 
     return true;
   }
@@ -940,9 +944,10 @@ struct sigma0builder {
     float sigmaMass = RecoDecay::m(arrMom, std::array{o2::constants::physics::MassPhoton, o2::constants::physics::MassLambda0});
     float sigmaY = RecoDecay::y(std::array{gamma.px() + lambda.px(), gamma.py() + lambda.py(), gamma.pz() + lambda.pz()}, o2::constants::physics::MassSigma0);
     float SigmapT = RecoDecay::pt(array{gamma.px() + lambda.px(), gamma.py() + lambda.py()});
+    float centrality = doPPAnalysis ? collision.centFT0M() : collision.centFT0C();
 
     // Before any selection
-    histos.fill(HIST("SigmaSel/h3dMassSigma0BeforeSel"), collision.centFT0C(), SigmapT, sigmaMass);
+    histos.fill(HIST("SigmaSel/h3dMassSigma0BeforeSel"), centrality, SigmapT, sigmaMass);
 
     histos.fill(HIST("SigmaSel/hSelectionStatistics"), 1.);
     histos.fill(HIST("SigmaSel/hSigmaMass"), sigmaMass);
@@ -975,7 +980,7 @@ struct sigma0builder {
       histos.fill(HIST("GeneralQA/h2dV0XY"), gamma.x(), gamma.y());
     }
 
-    histos.fill(HIST("SigmaSel/h3dMassSigma0AfterSel"), collision.centFT0C(), SigmapT, sigmaMass);
+    histos.fill(HIST("SigmaSel/h3dMassSigma0AfterSel"), centrality, SigmapT, sigmaMass);
 
     return true;
   }
@@ -1141,7 +1146,7 @@ struct sigma0builder {
       // Do analysis with collision-grouped V0s, retain full collision information
       const uint64_t collIdx = coll.globalIndex();
       auto V0s = fullV0s.sliceBy(perCollisionMCDerived, collIdx);
-      float centrality = coll.centFT0C();
+      float centrality = doPPAnalysis ? coll.centFT0M() : coll.centFT0C();
 
       bool fhasMCColl = false;
       if (coll.has_straMCCollision())
@@ -1186,10 +1191,10 @@ struct sigma0builder {
             histos.fill(HIST("MC/h2dPtVsCentralityBeforeSel_MCAssocALambda"), centrality, v0.pt());
         }
 
-        if (processPhotonCandidate(v0, coll))          // selecting photons
+        if (processPhotonCandidate(v0, coll))                // selecting photons
           bestGammasArray.push_back(v0.globalIndex()); // Save indices of best gamma candidates
 
-        if (processLambdaCandidate(v0, coll))           // selecting lambdas
+        if (processLambdaCandidate(v0, coll))                 // selecting lambdas
           bestLambdasArray.push_back(v0.globalIndex()); // Save indices of best lambda candidates
       }
 
@@ -1279,7 +1284,7 @@ struct sigma0builder {
             if (fIsAntiSigma)
               histos.fill(HIST("MC/h2dPtVsCentralityBeforeSel_MCAssocASigma0"), centrality, SigmaMCpT);
           }
-
+          
           // Build sigma0 candidate, please
           if (!buildSigma0(lambda, gamma, coll))
             continue;
@@ -1300,7 +1305,7 @@ struct sigma0builder {
                         LambdaCandPDGCode, LambdaCandPDGCodeMother, fIsLambdaPrimary, LambdaMCpT, fIsLambdaCorrectlyAssign);
 
           // Filling tables with accepted candidates
-          fillTables(lambda, gamma, coll);
+          fillTables(lambda, gamma, coll); 
 
           nSigmaCandidates++;
           if (nSigmaCandidates % 10000 == 0)
@@ -1315,12 +1320,12 @@ struct sigma0builder {
     for (const auto& coll : collisions) {
 
       if (!IsEventAccepted(coll, true))
-        continue;
+       continue;
 
       // Do analysis with collision-grouped V0s, retain full collision information
       const uint64_t collIdx = coll.globalIndex();
       auto V0s = fullV0s.sliceBy(perCollisionSTDDerived, collIdx);
-      float centrality = coll.centFT0C();
+      float centrality = doPPAnalysis ? coll.centFT0M() : coll.centFT0C();
 
       //_______________________________________________
       // Retrieving IR info
@@ -1340,10 +1345,10 @@ struct sigma0builder {
       //_______________________________________________
       // V0s loop
       for (auto& v0 : V0s) {
-        if (processPhotonCandidate(v0, coll))          // selecting photons
+        if (processPhotonCandidate(v0, coll))                // selecting photons
           bestGammasArray.push_back(v0.globalIndex()); // Save indices of best gamma candidates
 
-        if (processLambdaCandidate(v0, coll))           // selecting lambdas
+        if (processLambdaCandidate(v0, coll))                 // selecting lambdas
           bestLambdasArray.push_back(v0.globalIndex()); // Save indices of best lambda candidates
       }
 
@@ -1362,17 +1367,17 @@ struct sigma0builder {
       //_______________________________________________
       // Sigma0 nested loop
       for (size_t i = 0; i < bestGammasArray.size(); ++i) {
-        auto gamma = fullV0s.rawIteratorAt(bestGammasArray[i]);
+        auto gamma = fullV0s.rawIteratorAt(bestGammasArray[i]);        
 
         for (size_t j = 0; j < bestLambdasArray.size(); ++j) {
           auto lambda = fullV0s.rawIteratorAt(bestLambdasArray[j]);
-
+          
           // Building sigma0 candidate
           if (!buildSigma0(lambda, gamma, coll))
-            continue;
+           continue;
 
           // Filling tables with accepted candidates
-          fillTables(lambda, gamma, coll);
+          fillTables(lambda, gamma, coll); 
 
           nSigmaCandidates++;
           if (nSigmaCandidates % 10000 == 0)