[PWGLF] V0 analysis vs multiplicity, add check dau pdg code (AliceO2Group#9444)

Author: ercolessi

PWGLF/DataModel/v0qaanalysis.h

@@ -54,6 +54,9 @@ DECLARE_SOA_COLUMN(NTOFSigmaPosPi, ntofsigmapospi, float);
 DECLARE_SOA_COLUMN(PosHasTOF, poshastof, float);
 DECLARE_SOA_COLUMN(NegHasTOF, neghastof, float);
 DECLARE_SOA_COLUMN(PDGCode, pdgcode, int);
+DECLARE_SOA_COLUMN(IsDauK0Short, isdauk0short, bool);
+DECLARE_SOA_COLUMN(IsDauLambda, isdaulambda, bool);
+DECLARE_SOA_COLUMN(IsDauAntiLambda, isdauantilambda, bool);
 DECLARE_SOA_COLUMN(IsPhysicalPrimary, isphysprimary, bool);
 DECLARE_SOA_COLUMN(MultFT0M, multft0m, float);
 DECLARE_SOA_COLUMN(MultFV0A, multfv0a, float);
@@ -73,7 +76,7 @@ DECLARE_SOA_TABLE(MyV0Candidates, "AOD", "MYV0CANDIDATES", o2::soa::Index<>,
                   myv0candidates::NTPCSigmaNegPr, myv0candidates::NTPCSigmaPosPr, myv0candidates::NTPCSigmaNegPi, myv0candidates::NTPCSigmaPosPi,
                   myv0candidates::NTOFSigmaNegPr, myv0candidates::NTOFSigmaPosPr, myv0candidates::NTOFSigmaNegPi, myv0candidates::NTOFSigmaPosPi,
                   myv0candidates::PosHasTOF, myv0candidates::NegHasTOF,
-                  myv0candidates::PDGCode, myv0candidates::IsPhysicalPrimary,
+                  myv0candidates::PDGCode, myv0candidates::IsDauK0Short, myv0candidates::IsDauLambda, myv0candidates::IsDauAntiLambda, myv0candidates::IsPhysicalPrimary,
                   myv0candidates::MultFT0M, myv0candidates::MultFV0A,
                   myv0candidates::EvFlag, myv0candidates::Alpha, myv0candidates::QtArm);
 

PWGLF/TableProducer/Strangeness/v0qaanalysis.cxx

@@ -229,6 +229,7 @@ struct LfV0qaanalysis {
 
       int lPDG = 0;
       bool isPhysicalPrimary = isMC;
+      bool isDauK0Short = false, isDauLambda = false, isDauAntiLambda = false;
 
       if (v0.v0radius() > v0radius &&
           v0.v0cosPA() > v0cospa &&
@@ -247,7 +248,7 @@ struct LfV0qaanalysis {
               v0.negTrack_as<DauTracks>().tpcNSigmaPi(), v0.posTrack_as<DauTracks>().tpcNSigmaPi(),
               v0.negTrack_as<DauTracks>().tofNSigmaPr(), v0.posTrack_as<DauTracks>().tofNSigmaPr(),
               v0.negTrack_as<DauTracks>().tofNSigmaPi(), v0.posTrack_as<DauTracks>().tofNSigmaPi(),
-              v0.posTrack_as<DauTracks>().hasTOF(), v0.negTrack_as<DauTracks>().hasTOF(), lPDG, isPhysicalPrimary,
+              v0.posTrack_as<DauTracks>().hasTOF(), v0.negTrack_as<DauTracks>().hasTOF(), lPDG, isDauK0Short, isDauLambda, isDauAntiLambda, isPhysicalPrimary,
               collision.centFT0M(), collision.centFV0A(), evFlag, v0.alpha(), v0.qtarm());
       }
     }
@@ -321,11 +322,25 @@ struct LfV0qaanalysis {
         }
 
         int lPDG = 0;
+        bool isDauK0Short = false, isDauLambda = false, isDauAntiLambda = false;
         bool isprimary = false;
         if (TMath::Abs(v0mcparticle.pdgCode()) == 310 || TMath::Abs(v0mcparticle.pdgCode()) == 3122) {
           lPDG = v0mcparticle.pdgCode();
           isprimary = v0mcparticle.isPhysicalPrimary();
         }
+        for (auto& mcparticleDaughter0 : v0mcparticle.daughters_as<aod::McParticles>()) {
+          for (auto& mcparticleDaughter1 : v0mcparticle.daughters_as<aod::McParticles>()) {
+            if (mcparticleDaughter0.pdgCode() == 211 && mcparticleDaughter1.pdgCode() == -211) {
+              isDauK0Short = true;
+            }
+            if (mcparticleDaughter0.pdgCode() == -211 && mcparticleDaughter1.pdgCode() == 2212) {
+              isDauLambda = true;
+            }
+            if (mcparticleDaughter0.pdgCode() == 211 && mcparticleDaughter1.pdgCode() == -2212) {
+              isDauAntiLambda = true;
+            }
+          }
+        }
 
         int posITSNhits = 0, negITSNhits = 0;
         for (unsigned int i = 0; i < 7; i++) {
@@ -359,7 +374,7 @@ struct LfV0qaanalysis {
                 v0.negTrack_as<DauTracksMC>().tpcNSigmaPi(), v0.posTrack_as<DauTracksMC>().tpcNSigmaPi(),
                 v0.negTrack_as<DauTracksMC>().tofNSigmaPr(), v0.posTrack_as<DauTracksMC>().tofNSigmaPr(),
                 v0.negTrack_as<DauTracksMC>().tofNSigmaPi(), v0.posTrack_as<DauTracksMC>().tofNSigmaPi(),
-                v0.posTrack_as<DauTracksMC>().hasTOF(), v0.negTrack_as<DauTracksMC>().hasTOF(), lPDG, isprimary,
+                v0.posTrack_as<DauTracksMC>().hasTOF(), v0.negTrack_as<DauTracksMC>().hasTOF(), lPDG, isDauK0Short, isDauLambda, isDauAntiLambda, isprimary,
                 mcCollision.centFT0M(), cent, evFlag, v0.alpha(), v0.qtarm());
         }
       }