[PWGLF] Few additions for hypertriton 3-body decay reconstruction KFParticle (new) (#9963)

creetz16 · web-flow · commit 183b90530db0 · 2025-02-16T11:11:35.000+01:00
diff --git a/PWGLF/DataModel/Vtx3BodyTables.h b/PWGLF/DataModel/Vtx3BodyTables.h
@@ -388,6 +388,12 @@ DECLARE_SOA_COLUMN(Track2Sign, track2sign, float);                   //! sign of
 DECLARE_SOA_COLUMN(TPCInnerParamTrack0, tpcinnerparamtrack0, float); //! momentum at inner wall of TPC of proton daughter
 DECLARE_SOA_COLUMN(TPCInnerParamTrack1, tpcinnerparamtrack1, float); //! momentum at inner wall of TPC of pion daughter
 DECLARE_SOA_COLUMN(TPCInnerParamTrack2, tpcinnerparamtrack2, float); //! momentum at inner wall of TPC of deuteron daughter
+DECLARE_SOA_COLUMN(TPCNClTrack0, tpcncltrack0, int);                 //! Number of TPC clusters of proton daughter
+DECLARE_SOA_COLUMN(TPCNClTrack1, tpcncltrack1, int);                 //! Number of TPC clusters of pion daughter
+DECLARE_SOA_COLUMN(TPCNClTrack2, tpcncltrack2, int);                 //! Number of TPC clusters of deuteron daughter
+DECLARE_SOA_COLUMN(TPCChi2NClDeuteron, tpcchi2ncldeuteron, float);   //! TPC Chi2 / cluster of deuteron daughter
+DECLARE_SOA_COLUMN(DeltaPhiDeuteron, deltaphideuteron, float);       //! phi before track rotation - phi after track rotation for deuteron track
+DECLARE_SOA_COLUMN(DeltaPhiProton, deltaphiproton, float);           //! phi before track rotation - phi after track rotation for proton track
 // PID
 DECLARE_SOA_COLUMN(TPCNSigmaProton, tpcnsigmaproton, float);            //! nsigma proton of TPC PID of the proton daughter
 DECLARE_SOA_COLUMN(TPCNSigmaPion, tpcnsigmapion, float);                //! nsigma pion of TPC PID of the pion daughter
@@ -418,7 +424,10 @@ DECLARE_SOA_COLUMN(GenBachP, genbachp, float);          //! generated momentum b
 DECLARE_SOA_COLUMN(GenBachPt, genbachpt, float);        //! generated transverse momentum bachelor daughter particle
 DECLARE_SOA_COLUMN(IsTrueH3L, istrueh3l, bool);         //! flag for true hypertriton candidate
 DECLARE_SOA_COLUMN(IsTrueAntiH3L, istrueantih3l, bool); //! flag for true anti-hypertriton candidate
-DECLARE_SOA_COLUMN(PdgCode, pdgcode, int);              //! MC particle PDG code
+DECLARE_SOA_COLUMN(PdgCode, pdgcode, int);              //! MC particle mother PDG code
+DECLARE_SOA_COLUMN(PdgCodeDau0, pdgcodedau0, int);      //! MC particle daughter 0 PDG code
+DECLARE_SOA_COLUMN(PdgCodeDau1, pdgcodedau1, int);      //! MC particle daughter 1 PDG code
+DECLARE_SOA_COLUMN(PdgCodeDau2, pdgcodedau2, int);      //! MC particle daughter 2 PDG code
 DECLARE_SOA_COLUMN(SurvEvSel, survevsel, int);          //! flag if reco collision survived event selection
 DECLARE_SOA_COLUMN(IsReco, isreco, int);                //! flag if candidate was reconstructed
 
@@ -455,6 +464,9 @@ DECLARE_SOA_TABLE(KFVtx3BodyDatas, "AOD", "KFVTX3BODYDATA",
                   vtx3body::PxTrack1, vtx3body::PyTrack1, vtx3body::PzTrack1,                                        // pion
                   vtx3body::PxTrack2, vtx3body::PyTrack2, vtx3body::PzTrack2,                                        // deuteron
                   kfvtx3body::TPCInnerParamTrack0, kfvtx3body::TPCInnerParamTrack1, kfvtx3body::TPCInnerParamTrack2, // proton, pion, deuteron
+                  kfvtx3body::TPCNClTrack0, kfvtx3body::TPCNClTrack1, kfvtx3body::TPCNClTrack2,                      // proton, pion, deuteron
+                  kfvtx3body::TPCChi2NClDeuteron,
+                  kfvtx3body::DeltaPhiDeuteron, kfvtx3body::DeltaPhiProton,
                   kfvtx3body::DCATrack0ToPVKF, kfvtx3body::DCATrack1ToPVKF, kfvtx3body::DCATrack2ToPVKF, kfvtx3body::DCAxyTrack0ToPVKF, kfvtx3body::DCAxyTrack1ToPVKF, kfvtx3body::DCAxyTrack2ToPVKF,
                   kfvtx3body::DCAxyTrack0ToSVKF, kfvtx3body::DCAxyTrack1ToSVKF, kfvtx3body::DCAxyTrack2ToSVKF,
                   kfvtx3body::DCATrack0ToTrack1KF, kfvtx3body::DCATrack0ToTrack2KF, kfvtx3body::DCATrack1ToTrack2KF,
@@ -495,21 +507,23 @@ DECLARE_SOA_TABLE(KFVtx3BodyDatasLite, "AOD", "KF3BODYLITE",
                   kfvtx3body::CTauKFtopo, kfvtx3body::TrackedClSize,
 
                   // V0
-                  kfvtx3body::MassV0, kfvtx3body::Chi2MassV0,
+                  kfvtx3body::MassV0,
                   kfvtx3body::CosPAV0,
 
                   // daughters
                   vtx3body::PxTrack0, vtx3body::PyTrack0, vtx3body::PzTrack0,                                        // proton
                   vtx3body::PxTrack1, vtx3body::PyTrack1, vtx3body::PzTrack1,                                        // pion
                   vtx3body::PxTrack2, vtx3body::PyTrack2, vtx3body::PzTrack2,                                        // deuteron
                   kfvtx3body::TPCInnerParamTrack0, kfvtx3body::TPCInnerParamTrack1, kfvtx3body::TPCInnerParamTrack2, // proton, pion, deuteron
+                  kfvtx3body::TPCNClTrack0, kfvtx3body::TPCNClTrack1, kfvtx3body::TPCNClTrack2,                      // proton, pion, deuteron
+                  kfvtx3body::TPCChi2NClDeuteron,
+                  kfvtx3body::DeltaPhiDeuteron, kfvtx3body::DeltaPhiProton,
                   kfvtx3body::DCATrack0ToPVKF, kfvtx3body::DCATrack1ToPVKF, kfvtx3body::DCATrack2ToPVKF, kfvtx3body::DCAxyTrack0ToPVKF, kfvtx3body::DCAxyTrack1ToPVKF, kfvtx3body::DCAxyTrack2ToPVKF,
                   kfvtx3body::DCAxyTrack0ToSVKF, kfvtx3body::DCAxyTrack1ToSVKF, kfvtx3body::DCAxyTrack2ToSVKF,
                   kfvtx3body::DCATrack0ToTrack1KF, kfvtx3body::DCATrack0ToTrack2KF, kfvtx3body::DCATrack1ToTrack2KF,
                   kfvtx3body::DCAVtxDaughtersKF,
                   kfvtx3body::Track0Sign, kfvtx3body::Track1Sign, kfvtx3body::Track2Sign, // track sing: proton, pion, deuteron
                   kfvtx3body::TPCNSigmaProton, kfvtx3body::TPCNSigmaPion, kfvtx3body::TPCNSigmaDeuteron, kfvtx3body::TPCNSigmaPionBach,
-                  kfvtx3body::TPCdEdxProton, kfvtx3body::TPCdEdxPion, kfvtx3body::TPCdEdxDeuteron,
                   kfvtx3body::TOFNSigmaDeuteron,
                   kfvtx3body::ITSClusSizeDeuteron,
                   kfvtx3body::PIDTrackingDeuteron);
@@ -542,6 +556,9 @@ DECLARE_SOA_TABLE(McKFVtx3BodyDatas, "AOD", "MCKF3BODYDATAS",
                   vtx3body::PxTrack1, vtx3body::PyTrack1, vtx3body::PzTrack1,                                        // pion
                   vtx3body::PxTrack2, vtx3body::PyTrack2, vtx3body::PzTrack2,                                        // deuteron
                   kfvtx3body::TPCInnerParamTrack0, kfvtx3body::TPCInnerParamTrack1, kfvtx3body::TPCInnerParamTrack2, // proton, pion, deuteron
+                  kfvtx3body::TPCNClTrack0, kfvtx3body::TPCNClTrack1, kfvtx3body::TPCNClTrack2,                      // proton, pion, deuteron
+                  kfvtx3body::TPCChi2NClDeuteron,
+                  kfvtx3body::DeltaPhiDeuteron, kfvtx3body::DeltaPhiProton,
                   kfvtx3body::DCATrack0ToPVKF, kfvtx3body::DCATrack1ToPVKF, kfvtx3body::DCATrack2ToPVKF, kfvtx3body::DCAxyTrack0ToPVKF, kfvtx3body::DCAxyTrack1ToPVKF, kfvtx3body::DCAxyTrack2ToPVKF,
                   kfvtx3body::DCAxyTrack0ToSVKF, kfvtx3body::DCAxyTrack1ToSVKF, kfvtx3body::DCAxyTrack2ToSVKF,
                   kfvtx3body::DCATrack0ToTrack1KF, kfvtx3body::DCATrack0ToTrack2KF, kfvtx3body::DCATrack1ToTrack2KF,
@@ -567,7 +584,7 @@ DECLARE_SOA_TABLE(McKFVtx3BodyDatas, "AOD", "MCKF3BODYDATAS",
                   kfvtx3body::GenNegP, kfvtx3body::GenNegPt,
                   kfvtx3body::GenBachP, kfvtx3body::GenBachPt,
                   kfvtx3body::IsTrueH3L, kfvtx3body::IsTrueAntiH3L,
-                  kfvtx3body::PdgCode,
+                  kfvtx3body::PdgCode, kfvtx3body::PdgCodeDau0, kfvtx3body::PdgCodeDau1, kfvtx3body::PdgCodeDau2,
                   kfvtx3body::IsReco,
                   kfvtx3body::SurvEvSel);
 
diff --git a/PWGLF/TableProducer/Nuspex/decay3bodybuilder.cxx b/PWGLF/TableProducer/Nuspex/decay3bodybuilder.cxx
@@ -117,15 +117,20 @@ struct kfCandidate {
   float chi2topoNDF;
   float ctau;
   float trackedClSize;
+  float DeltaPhiRotDeuteron;
+  float DeltaPhiRotProton;
   // V0
   float massV0;
   float chi2massV0;
   float cpaV0ToPV;
-  // daughter momenta at vertex
+  // daughter momenta
   float protonMom[3];
   float pionMom[3];
   float deuteronMom[3];
   float tpcInnerParam[3]; // proton, pion, deuteron
+  // daughter track quality
+  int tpcNClDaughters[3]; // proton, pion, deuteron
+  float tpcChi2NClDeuteron;
   // daughter DCAs KF
   float DCAdaughterToPV[3];   // proton, pion, deuteron
   float DCAdaughterToPVxy[3]; // proton, pion, deuteron
@@ -816,13 +821,17 @@ struct decay3bodyBuilder {
       candidate.ctau,
       candidate.trackedClSize,
       // V0
-      candidate.massV0, candidate.chi2massV0,
+      candidate.massV0,
       candidate.cpaV0ToPV,
       // daughter momenta at vertex
       candidate.protonMom[0], candidate.protonMom[1], candidate.protonMom[2],
       candidate.pionMom[0], candidate.pionMom[1], candidate.pionMom[2],
       candidate.deuteronMom[0], candidate.deuteronMom[1], candidate.deuteronMom[2],
       candidate.tpcInnerParam[0], candidate.tpcInnerParam[1], candidate.tpcInnerParam[2], // proton, pion, deuteron
+      // daughter track quality
+      candidate.tpcNClDaughters[0], candidate.tpcNClDaughters[1], candidate.tpcNClDaughters[2], // proton, pion, deuteron
+      candidate.tpcChi2NClDeuteron,
+      candidate.DeltaPhiRotDeuteron, candidate.DeltaPhiRotProton,
       // daughter DCAs KF
       candidate.DCAdaughterToPV[0], candidate.DCAdaughterToPV[1], candidate.DCAdaughterToPV[2],       // proton, pion, deuteron
       candidate.DCAdaughterToPVxy[0], candidate.DCAdaughterToPVxy[1], candidate.DCAdaughterToPVxy[2], // proton, pion, deuteron
@@ -833,7 +842,6 @@ struct decay3bodyBuilder {
       candidate.daughterCharge[0], candidate.daughterCharge[1], candidate.daughterCharge[2], // proton, pion, deuteron
       // daughter PID
       candidate.tpcNsigma[0], candidate.tpcNsigma[1], candidate.tpcNsigma[2], candidate.tpcNsigma[3], // proton, pion, deuteron, bach with pion hyp
-      candidate.tpcdEdx[0], candidate.tpcdEdx[1], candidate.tpcdEdx[2],                               // proton, pion, deuteron
       candidate.tofNsigmaDeuteron,
       candidate.averageClusterSizeDeuteron,
       candidate.pidForTrackingDeuteron);
@@ -864,6 +872,10 @@ struct decay3bodyBuilder {
         candidate.pionMom[0], candidate.pionMom[1], candidate.pionMom[2],
         candidate.deuteronMom[0], candidate.deuteronMom[1], candidate.deuteronMom[2],
         candidate.tpcInnerParam[0], candidate.tpcInnerParam[1], candidate.tpcInnerParam[2], // proton, pion, deuteron
+        // daughter track quality
+        candidate.tpcNClDaughters[0], candidate.tpcNClDaughters[1], candidate.tpcNClDaughters[2], // proton, pion, deuteron
+        candidate.tpcChi2NClDeuteron,
+        candidate.DeltaPhiRotDeuteron, candidate.DeltaPhiRotProton,
         // daughter DCAs KF
         candidate.DCAdaughterToPV[0], candidate.DCAdaughterToPV[1], candidate.DCAdaughterToPV[2],       // proton, pion, deuteron
         candidate.DCAdaughterToPVxy[0], candidate.DCAdaughterToPVxy[1], candidate.DCAdaughterToPVxy[2], // proton, pion, deuteron
@@ -1011,13 +1023,25 @@ struct decay3bodyBuilder {
     registry.fill(HIST("Counters/hVtx3BodyCounterKFParticle"), kKfVtxEta);
 
     // number of TPC clusters
+    int tpcNClProton;
+    int tpcNClPion;
+    int tpcNClDeuteron = trackBach.tpcNClsFound();
+    float tpcChi2NCl = trackBach.tpcChi2NCl();
     if (trackBach.tpcNClsFound() <= kfparticleConfigurations.mintpcNClsBach) {
       return;
     }
-    if (isMatter && ((kfparticleConfigurations.useTPCforPion && trackNeg.tpcNClsFound() <= kfparticleConfigurations.mintpcNClsPion) || trackPos.tpcNClsFound() <= kfparticleConfigurations.mintpcNClsProton)) {
-      return;
-    } else if (!isMatter && ((kfparticleConfigurations.useTPCforPion && trackPos.tpcNClsFound() <= kfparticleConfigurations.mintpcNClsPion) || trackNeg.tpcNClsFound() <= kfparticleConfigurations.mintpcNClsProton)) {
-      return;
+    if (isMatter) {
+      tpcNClPion = trackNeg.tpcNClsFound();
+      tpcNClProton = trackPos.tpcNClsFound();
+      if ((kfparticleConfigurations.useTPCforPion && trackNeg.tpcNClsFound() <= kfparticleConfigurations.mintpcNClsPion) || trackPos.tpcNClsFound() <= kfparticleConfigurations.mintpcNClsProton) {
+        return;
+      }
+    } else if (!isMatter) {
+      tpcNClPion = trackPos.tpcNClsFound();
+      tpcNClProton = trackNeg.tpcNClsFound();
+      if ((kfparticleConfigurations.useTPCforPion && trackPos.tpcNClsFound() <= kfparticleConfigurations.mintpcNClsPion) || trackNeg.tpcNClsFound() <= kfparticleConfigurations.mintpcNClsProton) {
+        return;
+      }
     }
     registry.fill(HIST("Counters/hVtx3BodyCounterKFParticle"), kKfVtxTPCNcls);
 
@@ -1195,6 +1219,13 @@ struct decay3bodyBuilder {
     kfpDeuteron = createKFParticleFromTrackParCov(trackParCovBach, trackBach.sign() * bachelorcharge, constants::physics::MassDeuteron);
     LOG(debug) << "KFParticle created from deuteron track.";
     float kfpvPos[3] = {kfpv.GetX(), kfpv.GetY(), kfpv.GetZ()};
+    // Check phi angle of "default" candidate daughter tracks
+    float phiDeuteron_noRotation;
+    float phiDeuteronSigma_noRotation;
+    float phiProton_noRotation;
+    float phiProtonSigma_noRotation;
+    kfpDeuteron.GetPhi(phiDeuteron_noRotation, phiDeuteronSigma_noRotation);
+    kfpProton.GetPhi(phiProton_noRotation, phiProtonSigma_noRotation);
 
     // Construct vertex
     /// BRIEF: Case 1: fully uncorrelated bkg via rotation of proton track
@@ -1340,6 +1371,14 @@ struct decay3bodyBuilder {
         trackedClSize = !fTrackedClSizeVector.empty() ? fTrackedClSizeVector[decay3bodyID] : 0;
       }
 
+      // get deuteron and proton phi after rotation
+      float phiDeuteron_afterRotation;
+      float phiDeuteronSigma_afterRotation;
+      float phiProton_afterRotation;
+      float phiProtonSigma_afterRotation;
+      kfDeuteronDaughters[i].GetPhi(phiDeuteron_afterRotation, phiDeuteronSigma_afterRotation);
+      kfProtonDaughters[i].GetPhi(phiProton_afterRotation, phiProtonSigma_afterRotation);
+
       // candidate filling
       kfCandidate candidate;
       candidate.collisionID = collision.globalIndex();
@@ -1377,6 +1416,8 @@ struct decay3bodyBuilder {
       candidate.chi2topoNDF = chi2topoNDF;
       candidate.ctau = KFHtPV.GetLifeTime();
       candidate.trackedClSize = trackedClSize;
+      candidate.DeltaPhiRotDeuteron = phiDeuteron_noRotation - phiDeuteron_afterRotation;
+      candidate.DeltaPhiRotProton = phiProton_noRotation - phiProton_afterRotation;
       // V0
       candidate.massV0 = massV0;
       candidate.chi2massV0 = chi2massV0;
@@ -1394,6 +1435,11 @@ struct decay3bodyBuilder {
       candidate.tpcInnerParam[0] = tpcInnerParamProton;
       candidate.tpcInnerParam[1] = tpcInnerParamPion;
       candidate.tpcInnerParam[2] = tpcInnerParamDeuteron;
+      // daughter track quality
+      candidate.tpcNClDaughters[0] = tpcNClProton;
+      candidate.tpcNClDaughters[1] = tpcNClPion;
+      candidate.tpcNClDaughters[2] = tpcNClDeuteron;
+      candidate.tpcChi2NClDeuteron = tpcChi2NCl;
       // daughter DCAs with KF
       candidate.DCAdaughterToPV[0] = kfProtonDaughters[i].GetDistanceFromVertex(kfpv);
       candidate.DCAdaughterToPV[1] = kfPionDaughters[i].GetDistanceFromVertex(kfpv);
diff --git a/PWGLF/TableProducer/Nuspex/threebodyKFTask.cxx b/PWGLF/TableProducer/Nuspex/threebodyKFTask.cxx
