Add information on swapped candidates and which mass hypotesis is con…

…sidered in Ds tree creator
AliceO2Group · Oct 11, 2024 · 53a0304 · 53a0304
1 parent 46f2490
commit 53a0304
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/PWGHF/TableProducer/treeCreatorDsToKKPi.cxx b/PWGHF/TableProducer/treeCreatorDsToKKPi.cxx
@@ -133,6 +133,7 @@ DECLARE_SOA_TABLE(HfCandDsLites, "AOD", "HFCANDDSLITE",
                   hf_cand_3prong::FlagMcMatchRec,
                   hf_cand_3prong::OriginMcRec,
                   hf_cand_3prong::FlagMcDecayChanRec,
+                  hf_cand_3prong::IsCandidateSwapped,
                   full::Sign);
 
 DECLARE_SOA_TABLE(HfCandDsFulls, "AOD", "HFCANDDSFULL",
@@ -205,6 +206,7 @@ DECLARE_SOA_TABLE(HfCandDsFulls, "AOD", "HFCANDDSFULL",
                   hf_cand_3prong::FlagMcMatchRec,
                   hf_cand_3prong::OriginMcRec,
                   hf_cand_3prong::FlagMcDecayChanRec,
+                  hf_cand_3prong::IsCandidateSwapped,
                   full::Sign);
 
 DECLARE_SOA_TABLE(HfCandDsFullEvs, "AOD", "HFCANDDSFULLEV",
@@ -294,13 +296,15 @@ struct HfTreeCreatorDsToKKPi {
     int8_t flagMc = 0;
     int8_t originMc = 0;
     int8_t channelMc = 0;
+    int8_t isSwapped = 1 - massHypo; // 0 if PiKK, 1 if KKPi
     float yCand = 0;
     float eCand = 0;
     float ctCand = 0;
     if constexpr (doMc) {
       flagMc = candidate.flagMcMatchRec();
       originMc = candidate.originMcRec();
       channelMc = candidate.flagMcDecayChanRec();
+      isSwapped = candidate.isCandidateSwapped();
       if (fillDplusMc && candidate.flagMcDecayChanRec() == (decayChannel + offsetDplusDecayChannel)) {
         yCand = hfHelper.yDplus(candidate);
         eCand = hfHelper.eDplus(candidate);
@@ -358,8 +362,8 @@ struct HfTreeCreatorDsToKKPi {
         prong2.tofNSigmaKa(),
         prong2.tpcTofNSigmaPi(),
         prong2.tpcTofNSigmaKa(),
-        candidate.isSelDsToKKPi(),
-        candidate.isSelDsToPiKK(),
+        massHypo == 0 ? candidate.isSelDsToKKPi() : -1,
+        massHypo == 1 ? candidate.isSelDsToPiKK() : -1,
         invMassDs,
         candidate.pt(),
         candidate.eta(),
@@ -381,6 +385,7 @@ struct HfTreeCreatorDsToKKPi {
         flagMc,
         originMc,
         channelMc,
+        isSwapped,
         prong0.sign() + prong1.sign() + prong2.sign());
     } else {
       rowCandidateFull(
@@ -425,8 +430,8 @@ struct HfTreeCreatorDsToKKPi {
         prong2.tofNSigmaKa(),
         prong2.tpcTofNSigmaPi(),
         prong2.tpcTofNSigmaKa(),
-        candidate.isSelDsToKKPi(),
-        candidate.isSelDsToPiKK(),
+        massHypo == 0 ? candidate.isSelDsToKKPi() : -1,
+        massHypo == 1 ? candidate.isSelDsToPiKK() : -1,
         candidate.xSecondaryVertex(),
         candidate.ySecondaryVertex(),
         candidate.zSecondaryVertex(),
@@ -453,6 +458,7 @@ struct HfTreeCreatorDsToKKPi {
         flagMc,
         originMc,
         channelMc,
+        isSwapped,
         prong0.sign() + prong1.sign() + prong2.sign());
     }
   }