PWGHF: Add new columns to tree creator for candidate indexing for HF studies (#2728)

Author: nzardosh

PWGHF/TableProducer/treeCreatorBplusToD0Pi.cxx

@@ -94,6 +94,7 @@ DECLARE_SOA_COLUMN(NSigmaTPCTrk1Pi, nSigmaTPCTrk1Pi, float);
 // Events
 DECLARE_SOA_COLUMN(IsEventReject, isEventReject, int);
 DECLARE_SOA_COLUMN(RunNumber, runNumber, int);
+DECLARE_SOA_COLUMN(GlobalIndex, globalIndex, int);
 } // namespace full
 
 // put the arguments into the table
@@ -158,7 +159,8 @@ DECLARE_SOA_TABLE(HfCandBplusFull, "AOD", "HFCANDBPFull",
                   full::NSigmaTOFTrk1Pi,
                   full::NSigmaTOFTrk1Ka,
                   full::NSigmaTPCTrk1Pi,
-                  full::NSigmaTPCTrk1Ka);
+                  full::NSigmaTPCTrk1Ka,
+                  full::GlobalIndex);
 
 DECLARE_SOA_TABLE(HfCandBplusFullEvents, "AOD", "HFCANDBPFullE",
                   collision::BCId,
@@ -175,7 +177,8 @@ DECLARE_SOA_TABLE(HfCandBplusFullParticles, "AOD", "HFCANDBPFullP",
                   full::Eta,
                   full::Phi,
                   full::Y,
-                  full::MCflag);
+                  full::MCflag,
+                  full::GlobalIndex);
 
 } // namespace o2::aod
 
@@ -297,7 +300,8 @@ struct HfTreeCreatorBplusToD0Pi {
             d0Daughter1.tofNSigmaPi(),
             d0Daughter1.tofNSigmaKa(),
             d0Daughter1.tpcNSigmaPi(),
-            d0Daughter1.tpcNSigmaKa());
+            d0Daughter1.tpcNSigmaKa(),
+            candidate.globalIndex());
         }
       };
 
@@ -315,7 +319,8 @@ struct HfTreeCreatorBplusToD0Pi {
           particle.eta(),
           particle.phi(),
           RecoDecay::y(array{particle.px(), particle.py(), particle.pz()}, RecoDecay::getMassPDG(particle.pdgCode())),
-          particle.flagMcMatchGen());
+          particle.flagMcMatchGen(),
+          particle.globalIndex());
       }
     }
   }

PWGHF/TableProducer/treeCreatorD0ToKPi.cxx

@@ -70,6 +70,7 @@ DECLARE_SOA_COLUMN(IsEventReject, isEventReject, int);
 DECLARE_SOA_COLUMN(RunNumber, runNumber, int);
 DECLARE_SOA_COLUMN(OriginMcRec, originMcRec, int8_t); // is prompt or non-prompt, reco level
 DECLARE_SOA_COLUMN(OriginMcGen, originMcGen, int8_t); // is prompt or non-prompt, Gen level
+DECLARE_SOA_COLUMN(GlobalIndex, globalIndex, int);
 } // namespace full
 
 DECLARE_SOA_TABLE(HfCand2ProngFull, "AOD", "HFCAND2PFull",
@@ -127,7 +128,8 @@ DECLARE_SOA_TABLE(HfCand2ProngFull, "AOD", "HFCAND2PFull",
                   full::Y,
                   full::E,
                   full::FlagMc,
-                  full::OriginMcRec);
+                  full::OriginMcRec,
+                  full::GlobalIndex);
 
 DECLARE_SOA_TABLE(HfCand2ProngFullEvents, "AOD", "HFCAND2PFullE",
                   collision::BCId,
@@ -145,7 +147,8 @@ DECLARE_SOA_TABLE(HfCand2ProngFullParticles, "AOD", "HFCAND2PFullP",
                   full::Phi,
                   full::Y,
                   full::FlagMc,
-                  full::OriginMcGen);
+                  full::OriginMcGen,
+                  full::GlobalIndex);
 
 } // namespace o2::aod
 
@@ -232,7 +235,8 @@ struct HfTreeCreatorD0ToKPi {
         y,
         e,
         flagMc,
-        origin);
+        origin,
+        candidate.globalIndex());
     }
   }
 
@@ -296,7 +300,8 @@ struct HfTreeCreatorD0ToKPi {
           particle.phi(),
           RecoDecay::y(array{particle.px(), particle.py(), particle.pz()}, RecoDecay::getMassPDG(particle.pdgCode())),
           particle.flagMcMatchGen(),
-          particle.originMcGen());
+          particle.originMcGen(),
+          particle.globalIndex());
       }
     }
   }

PWGHF/TableProducer/treeCreatorLcToPKPi.cxx

@@ -81,6 +81,7 @@ DECLARE_SOA_COLUMN(IsCandidateSwapped, isCandidateSwapped, int8_t);
 // Events
 DECLARE_SOA_COLUMN(IsEventReject, isEventReject, int);
 DECLARE_SOA_COLUMN(RunNumber, runNumber, int);
+DECLARE_SOA_COLUMN(GlobalIndex, globalIndex, int);
 } // namespace full
 
 DECLARE_SOA_TABLE(HfCand3ProngFull, "AOD", "HFCAND3PFull",
@@ -155,7 +156,8 @@ DECLARE_SOA_TABLE(HfCand3ProngFull, "AOD", "HFCAND3PFull",
                   full::E,
                   full::MCflag,
                   full::OriginMcRec,
-                  full::IsCandidateSwapped);
+                  full::IsCandidateSwapped,
+                  full::GlobalIndex);
 
 DECLARE_SOA_TABLE(HfCand3ProngFullEvents, "AOD", "HFCAND3PFullE",
                   collision::BCId,
@@ -173,7 +175,8 @@ DECLARE_SOA_TABLE(HfCand3ProngFullParticles, "AOD", "HFCAND3PFullP",
                   full::Phi,
                   full::Y,
                   full::MCflag,
-                  full::OriginMcGen);
+                  full::OriginMcGen,
+                  full::GlobalIndex);
 
 } // namespace o2::aod
 
@@ -295,7 +298,8 @@ struct HfTreeCreatorLcToPKPi {
             FunctionE,
             candidate.flagMcMatchRec(),
             candidate.originMcRec(),
-            candidate.isCandidateSwapped());
+            candidate.isCandidateSwapped(),
+            candidate.globalIndex());
         }
       };
 
@@ -314,7 +318,8 @@ struct HfTreeCreatorLcToPKPi {
           particle.phi(),
           RecoDecay::y(array{particle.px(), particle.py(), particle.pz()}, RecoDecay::getMassPDG(particle.pdgCode())),
           particle.flagMcMatchGen(),
-          particle.originMcGen());
+          particle.originMcGen(),
+          particle.globalIndex());
       }
     }
   }
@@ -424,7 +429,8 @@ struct HfTreeCreatorLcToPKPi {
             FunctionE,
             0.,
             0.,
-            0.);
+            0.,
+            candidate.globalIndex());
         }
       };