[PWGLF] Updated the Table producer and Task for uncertainty analysis. (#17204)

Author: 15071832337

PWGLF/DataModel/LFSpincorrelationTables.h

@@ -52,6 +52,8 @@ DECLARE_SOA_COLUMN(ProtonEta, protonEta, float);                   //! Proton Et
 DECLARE_SOA_COLUMN(ProtonPhi, protonPhi, float);                   //! Proton Phi
 DECLARE_SOA_COLUMN(ProtonIndex, protonIndex, int);                 //! Proton index
 DECLARE_SOA_COLUMN(PionIndex, pionIndex, int);                     //! Pion index
+DECLARE_SOA_COLUMN(DcaV0ToPV, dcaV0ToPV, float);                   //! DCA of V0 to primary vertex
+
 } // namespace lambdapair
 DECLARE_SOA_TABLE(LambdaPairs, "AOD", "LAMBDAPAIR",
                   o2::soa::Index<>,
@@ -71,7 +73,8 @@ DECLARE_SOA_TABLE(LambdaPairs, "AOD", "LAMBDAPAIR",
                   lambdapair::ProtonEta,
                   lambdapair::ProtonPhi,
                   lambdapair::ProtonIndex,
-                  lambdapair::PionIndex);
+                  lambdapair::PionIndex,
+                  lambdapair::DcaV0ToPV);
 
 using LambdaPair = LambdaPairs::iterator;
 
@@ -105,6 +108,7 @@ DECLARE_SOA_COLUMN(ProtonEtamc, protonEtamc, float);                   //! Proto
 DECLARE_SOA_COLUMN(ProtonPhimc, protonPhimc, float);                   //! Proton Phi in montecarlo
 DECLARE_SOA_COLUMN(ProtonIndexmc, protonIndexmc, int);                 //! Proton index in montecarlo
 DECLARE_SOA_COLUMN(PionIndexmc, pionIndexmc, int);                     //! Pion index in montecarlo
+DECLARE_SOA_COLUMN(DcaV0ToPVmc, dcaV0ToPVmc, float);                   //! DCA of V0 to primary vertex
 } // namespace lambdapairmc
 DECLARE_SOA_TABLE(LambdaPairmcs, "AOD", "LAMBDAPAIRMC",
                   o2::soa::Index<>,
@@ -124,7 +128,8 @@ DECLARE_SOA_TABLE(LambdaPairmcs, "AOD", "LAMBDAPAIRMC",
                   lambdapairmc::ProtonEtamc,
                   lambdapairmc::ProtonPhimc,
                   lambdapairmc::ProtonIndexmc,
-                  lambdapairmc::PionIndexmc);
+                  lambdapairmc::PionIndexmc,
+                  lambdapairmc::DcaV0ToPVmc);
 
 using LambdaPairmc = LambdaPairmcs::iterator;
 

PWGLF/TableProducer/Strangeness/lambdaspincorrelation.cxx

@@ -298,6 +298,7 @@ struct lambdaspincorrelation {
     std::vector<int> positiveIndex = {};
     std::vector<int> negativeIndex = {};
     std::vector<float> dcaBetweenDaughter = {};
+    std::vector<float> dcaV0ToPV = {};
     int numbV0 = 0;
     // LOGF(info, "event collisions: (%d)", collision.index());
     auto centrality = collision.centFT0C();
@@ -355,6 +356,7 @@ struct lambdaspincorrelation {
           positiveIndex.push_back(postrack1.globalIndex());
           negativeIndex.push_back(negtrack1.globalIndex());
           v0Cospa.push_back(v0.v0cosPA());
+          dcaV0ToPV.push_back(std::abs(v0.dcav0topv()));
           v0Radius.push_back(v0.v0radius());
           dcaPositive.push_back(std::abs(v0.dcapostopv()));
           dcaNegative.push_back(std::abs(v0.dcanegtopv()));
@@ -391,7 +393,7 @@ struct lambdaspincorrelation {
           lambdaDummy = lambdaMother.at(i5);
           protonDummy = protonDaughter.at(i5);
           pionDummy = pionDaughter.at(i5);
-          lambdaPair(indexEvent, v0Status.at(i5), doubleStatus.at(i5), v0Cospa.at(i5), v0Radius.at(i5), dcaPositive.at(i5), dcaNegative.at(i5), dcaBetweenDaughter.at(i5), lambdaDummy.Pt(), lambdaDummy.Eta(), lambdaDummy.Phi(), lambdaDummy.M(), protonDummy.Pt(), protonDummy.Eta(), protonDummy.Phi(), positiveIndex.at(i5), negativeIndex.at(i5));
+          lambdaPair(indexEvent, v0Status.at(i5), doubleStatus.at(i5), v0Cospa.at(i5), v0Radius.at(i5), dcaPositive.at(i5), dcaNegative.at(i5), dcaBetweenDaughter.at(i5), lambdaDummy.Pt(), lambdaDummy.Eta(), lambdaDummy.Phi(), lambdaDummy.M(), protonDummy.Pt(), protonDummy.Eta(), protonDummy.Phi(), positiveIndex.at(i5), negativeIndex.at(i5), dcaV0ToPV.at(i5));
         }
       }
     }
@@ -410,6 +412,7 @@ struct lambdaspincorrelation {
     std::vector<int> positiveIndex = {};
     std::vector<int> negativeIndex = {};
     std::vector<float> dcaBetweenDaughter = {};
+    std::vector<float> dcaV0ToPV = {};
     int numbV0 = 0;
     // LOGF(info, "event collisions: (%d)", collision.index());
     auto centrality = collision.centFT0C();
@@ -470,6 +473,7 @@ struct lambdaspincorrelation {
           dcaPositive.push_back(std::abs(v0.dcapostopv()));
           dcaNegative.push_back(std::abs(v0.dcanegtopv()));
           dcaBetweenDaughter.push_back(std::abs(v0.dcaV0daughters()));
+          dcaV0ToPV.push_back(std::abs(v0.dcav0topv()));
           if (lambdaTag) {
             v0Status.push_back(0);
             proton = ROOT::Math::PxPyPzMVector(v0.pxpos(), v0.pypos(), v0.pzpos(), o2::constants::physics::MassProton);
@@ -502,7 +506,7 @@ struct lambdaspincorrelation {
           lambdaDummy = lambdaMother.at(i5);
           protonDummy = protonDaughter.at(i5);
           pionDummy = pionDaughter.at(i5);
-          lambdaPairmc(indexEvent, v0Status.at(i5), doubleStatus.at(i5), v0Cospa.at(i5), v0Radius.at(i5), dcaPositive.at(i5), dcaNegative.at(i5), dcaBetweenDaughter.at(i5), lambdaDummy.Pt(), lambdaDummy.Eta(), lambdaDummy.Phi(), lambdaDummy.M(), protonDummy.Pt(), protonDummy.Eta(), protonDummy.Phi(), positiveIndex.at(i5), negativeIndex.at(i5));
+          lambdaPairmc(indexEvent, v0Status.at(i5), doubleStatus.at(i5), v0Cospa.at(i5), v0Radius.at(i5), dcaPositive.at(i5), dcaNegative.at(i5), dcaBetweenDaughter.at(i5), lambdaDummy.Pt(), lambdaDummy.Eta(), lambdaDummy.Phi(), lambdaDummy.M(), protonDummy.Pt(), protonDummy.Eta(), protonDummy.Phi(), positiveIndex.at(i5), negativeIndex.at(i5), dcaV0ToPV.at(i5));
         }
       }
     }
@@ -530,6 +534,7 @@ struct lambdaspincorrelation {
         std::vector<int> positiveIndex = {};
         std::vector<int> negativeIndex = {};
         std::vector<float> dcaBetweenDaughter = {};
+        std::vector<float> dcaV0ToPV = {};
         int numbV0 = 0;
 
         auto centrality = collision.centFT0C();
@@ -610,6 +615,7 @@ struct lambdaspincorrelation {
               negativeIndex.push_back(negtrack1.globalIndex());
 
               v0Cospa.push_back(v0.v0cosPA());
+              dcaV0ToPV.push_back(std::abs(v0.dcav0topv()));
               v0Radius.push_back(v0.v0radius());
               dcaPositive.push_back(std::abs(v0.dcapostopv()));
               dcaNegative.push_back(std::abs(v0.dcanegtopv()));
@@ -695,7 +701,8 @@ struct lambdaspincorrelation {
                            protonDummy.Eta(),
                            protonDummy.Phi(),
                            positiveIndex.at(i5),
-                           negativeIndex.at(i5));
+                           negativeIndex.at(i5),
+                           dcaV0ToPV.at(i5));
             }
           }
         }