New header for matching functions

Author: Marcellocosti

PWGHF/Core/DecayChannels.h

@@ -12,7 +12,6 @@
 /// \file DecayChannels.h
 /// \brief Definitions of constants for MC flagging of HF decay channels.
 /// \author Vít Kučera <vit.kucera@cern.ch>, Inha University
-/// \author Marcello Di Costanzo <marcello.di.costanzo@cern.ch>, Polytechnic University of Turin and INFN
 /// \note DecayChannelMain enums define unique combinations of the mother and the daughters for main channels.
 /// \note DecayChannelResonant enums define unique combinations of the mother and the daughters for resonant channels.
 /// \note Value 0 is reserved to indicate no match.
@@ -21,12 +20,7 @@
 #ifndef PWGHF_CORE_DECAYCHANNELS_H_
 #define PWGHF_CORE_DECAYCHANNELS_H_
 
-#include <array>
 #include <cstdint>
-#include <unordered_map>
-#include <vector>
-
-using namespace o2::constants::physics;
 
 namespace o2::hf_decay
 {
@@ -59,24 +53,6 @@ enum DecayChannelResonant : int8_t {
   //
   LastChannelResonant
 };
-
-std::unordered_map<DecayChannelMain, std::vector<int>> finalStates2Prongs =
-  {
-    {DecayChannelMain::D0ToPiK, std::vector<int>{+kKMinus, +kPiPlus}},
-    {DecayChannelMain::D0ToPiKPi0, std::vector<int>{+kKMinus, +kPiPlus, +kPi0}},
-    {DecayChannelMain::D0ToPiPi, std::vector<int>{+kPiMinus, +kPiPlus}},
-    {DecayChannelMain::D0ToPiPiPi0, std::vector<int>{+kPiMinus, +kPiPlus, +kPi0}},
-    {DecayChannelMain::D0ToKK, std::vector<int>{+kKMinus, +kKPlus}},
-};
-
-std::unordered_map<DecayChannelResonant, std::array<int, 2>> resoStatesD0 =
-  {
-    {DecayChannelResonant::D0ToRhoplusPi, std::array<int, 2>{213, +kPiMinus}},
-    {DecayChannelResonant::D0ToRhoplusK, std::array<int, 2>{213, +kKMinus}},
-    {DecayChannelResonant::D0ToKstar0Pi0, std::array<int, 2>{-kK0Star892, +kPi0}},
-    {DecayChannelResonant::D0ToKstarPi, std::array<int, 2>{-kKPlusStar892, +kPiPlus}},
-};
-
 } // namespace hf_cand_2prong
 
 namespace hf_cand_3prong
@@ -152,100 +128,6 @@ enum DecayChannelResonant : int8_t {
   //
   LastChannelResonant
 };
-
-std::unordered_map<DecayChannelResonant, std::array<int, 2>> resoStatesDPlus =
-  {
-    {DecayChannelResonant::DplusToPhiPi, std::array<int, 2>{+kPhi, +kPiPlus}},
-    {DecayChannelResonant::DplusToKstar0K, std::array<int, 2>{-kK0Star892, +kKPlus}},
-    {DecayChannelResonant::DplusToKstar1430_0K, std::array<int, 2>{+10311, +kKPlus}},
-    {DecayChannelResonant::DplusToRho0Pi, std::array<int, 2>{+113, +kPiPlus}},
-    {DecayChannelResonant::DplusToF2_1270Pi, std::array<int, 2>{+225, +kPiPlus}},
-};
-
-std::unordered_map<DecayChannelMain, std::vector<int>> finalStatesDPlus =
-  {
-    {DecayChannelMain::DplusToPiKPi, std::vector<int>{+kKMinus, +kKPlus, +kPiPlus}},
-    {DecayChannelMain::DplusToPiKK, std::vector<int>{+kKMinus, +kPiPlus, +kPiPlus}},
-    {DecayChannelMain::DplusToPiKPiPi0, std::vector<int>{+kKMinus, +kPiPlus, +kPiPlus, +kPi0}},
-    {DecayChannelMain::DplusToPiPiPi, std::vector<int>{+kPiMinus, +kPiPlus, +kPiPlus}},
-};
-
-// Ds± → K± K∓ π±
-std::unordered_map<DecayChannelResonant, std::array<int, 2>> resoStatesDs =
-  {
-    {DecayChannelResonant::DsToPhiPi, std::array<int, 2>{+kPhi, +kPiPlus}},
-    {DecayChannelResonant::DsToPhiRhoplus, std::array<int, 2>{+kPhi, +213}},
-    {DecayChannelResonant::DsToKstar0K, std::array<int, 2>{-kK0Star892, +kKPlus}},
-    {DecayChannelResonant::DsToKstar0Pi, std::array<int, 2>{+kK0Star892, +kPiPlus}},
-    {DecayChannelResonant::DsToRho0Pi, std::array<int, 2>{113, +kPiPlus}},
-    {DecayChannelResonant::DsToRho0K, std::array<int, 2>{113, +kKPlus}},
-    {DecayChannelResonant::DsToF2_1270Pi, std::array<int, 2>{225, +kPiPlus}},
-    {DecayChannelResonant::DsToF0_1370K, std::array<int, 2>{10221, +kKPlus}},
-    {DecayChannelResonant::DsToEtaPi, std::array<int, 2>{221, +kPiPlus}},
-};
-
-std::unordered_map<DecayChannelMain, std::vector<int>> finalStatesDs =
-  {
-    {DecayChannelMain::DsToPiKK, std::vector<int>{+kKMinus, +kKPlus, +kPiPlus}},
-    {DecayChannelMain::DsToPiKKPi0, std::vector<int>{+kKMinus, +kKPlus, +kPiPlus, +kPi0}},
-    {DecayChannelMain::DsToPiPiK, std::vector<int>{+kKPlus, +kPiPlus, +kPiMinus}},
-    {DecayChannelMain::DsToPiPiPi, std::vector<int>{+kPiMinus, +kPiPlus, +kPiPlus}},
-    {DecayChannelMain::DsToPiPiPiPi0, std::vector<int>{+kPiMinus, +kPiPlus, +kPiPlus, +kPi0}},
-};
-
-// Dstar → K± K∓ π±
-std::unordered_map<DecayChannelResonant, std::array<int, 2>> resoStatesDstar =
-  {
-    {DecayChannelResonant::DstarToD0ToRhoplusPi, std::array<int, 2>{213, +kPiMinus}},
-    {DecayChannelResonant::DstarToD0ToRhoplusK, std::array<int, 2>{213, +kKMinus}},
-    {DecayChannelResonant::DstarToD0ToKstar0Pi0, std::array<int, 2>{-kK0Star892, +kPi0}},
-    {DecayChannelResonant::DstarToD0ToKstarPi, std::array<int, 2>{-kKPlusStar892, +kPiPlus}},
-    {DecayChannelResonant::DstarToDplusToPhiPi, std::array<int, 2>{+kPhi, +kPiPlus}},
-    {DecayChannelResonant::DstarToDplusToKstar0K, std::array<int, 2>{-kK0Star892, +kKPlus}},
-    {DecayChannelResonant::DstarToDplusToKstar1430_0K, std::array<int, 2>{+10311, +kKPlus}},
-    {DecayChannelResonant::DstarToDplusToRho0Pi, std::array<int, 2>{+113, +kPiPlus}},
-    {DecayChannelResonant::DstarToDplusToF2_1270Pi, std::array<int, 2>{+225, +kPiPlus}},
-};
-
-std::unordered_map<DecayChannelMain, std::vector<int>> finalStatesDstar =
-  {
-    {DecayChannelMain::DstarToPiKPi, std::vector<int>{+kKMinus, +kPiPlus, +kPiPlus}},
-    {DecayChannelMain::DstarToPiKPiPi0, std::vector<int>{+kKMinus, +kPiPlus, +kPiPlus, +kPi0}},
-    {DecayChannelMain::DstarToPiKPiPi0Pi0, std::vector<int>{+kKMinus, +kPiPlus, +kPiPlus, +kPi0, +kPi0}},
-    {DecayChannelMain::DstarToPiKK, std::vector<int>{+kKMinus, +kKPlus, +kPiPlus}},
-    {DecayChannelMain::DstarToPiKKPi0, std::vector<int>{+kKMinus, +kKPlus, +kPiPlus, +kPi0}},
-    {DecayChannelMain::DstarToPiPiPi, std::vector<int>{+kPiMinus, +kPiPlus, +kPiPlus}},
-    {DecayChannelMain::DstarToPiPiPiPi0, std::vector<int>{+kPiMinus, +kPiPlus, +kPiPlus, +kPi0}},
-};
-
-// Lc → p K∓ π±
-std::unordered_map<DecayChannelResonant, std::array<int, 2>> resoStatesLambdaC =
-  {
-    {DecayChannelResonant::LcToPKstar0, std::array<int, 2>{+kK0Star892, +kProton}},
-    {DecayChannelResonant::LcToDeltaplusplusK, std::array<int, 2>{+2224, +kKMinus}},
-    {DecayChannelResonant::LcToL1520Pi, std::array<int, 2>{+102134, +kPiPlus}},
-};
-
-std::unordered_map<DecayChannelMain, std::vector<int>> finalStatesLc =
-  {
-    {DecayChannelMain::LcToPKPi, std::vector<int>{+kProton, +kKMinus, +kPiPlus}},
-    {DecayChannelMain::LcToPKPiPi0, std::vector<int>{+kProton, +kKMinus, +kPiPlus, +kPi0}},
-    {DecayChannelMain::LcToPPiPi, std::vector<int>{+kProton, +kPiMinus, +kPiPlus}},
-    {DecayChannelMain::LcToPKK, std::vector<int>{+kProton, +kKMinus, +kKPlus}}};
-
-// Xic → p K∓ π±
-std::unordered_map<DecayChannelResonant, std::array<int, 2>> resoStatesXiC =
-  {
-    {DecayChannelResonant::XicToPKstar0, std::array<int, 2>{-kK0Star892, +kProton}},
-    {DecayChannelResonant::XicToPPhi, std::array<int, 2>{+kProton, +kPhi}},
-};
-
-std::unordered_map<DecayChannelMain, std::vector<int>> finalStatesXic =
-  {
-    {DecayChannelMain::XicToPKPi, std::vector<int>{+kProton, +kKMinus, +kPiPlus}},
-    {DecayChannelMain::XicToPKK, std::vector<int>{+kProton, +kKMinus, +kKPlus}},
-    {DecayChannelMain::XicToSPiPi, std::vector<int>{+kSigmaPlus, +kPiMinus, +kPiPlus}},
-};
 } // namespace hf_cand_3prong
 
 namespace hf_cand_dstar
@@ -335,108 +217,4 @@ enum DecayChannelToJpsiResonant : int8_t {
 } // namespace hf_cand_beauty
 } // namespace o2::hf_decay
 
-namespace o2::hf_corrbkg
-{
-using namespace o2::hf_decay;
-
-/// Returns a map of the possible final states for a specific 3-prong particle specie
-/// \param pdgMother PDG code of the mother particle
-/// \return a map of final states with their corresponding PDG codes
-std::unordered_map<hf_cand_3prong::DecayChannelMain, std::vector<int>> getDecayChannel3Prong(int pdgMother)
-{
-  switch (pdgMother) {
-    case Pdg::kDPlus:
-      return hf_cand_3prong::finalStatesDPlus;
-    case Pdg::kDS:
-      return hf_cand_3prong::finalStatesDs;
-    case Pdg::kDStar:
-      return hf_cand_3prong::finalStatesDstar;
-    case Pdg::kLambdaCPlus:
-      return hf_cand_3prong::finalStatesLc;
-    case Pdg::kXiCPlus:
-      return hf_cand_3prong::finalStatesXic;
-    default:
-      LOG(error) << "Unknown PDG code for 3-prong final states: " << pdgMother;
-      return {};
-  }
-}
-
-/// Returns a map of the resonant decay channels for a specific 3-prong particle specie
-/// \param pdgMother PDG code of the mother particle
-/// \return a map of resonant decay channels with their corresponding PDG codes
-std::unordered_map<hf_cand_3prong::DecayChannelResonant, std::array<int, 2>> getResoChannels3Prong(int pdgMother)
-{
-  switch (pdgMother) {
-    case Pdg::kDPlus:
-      return hf_cand_3prong::resoStatesDPlus;
-    case Pdg::kDS:
-      return hf_cand_3prong::resoStatesDs;
-    case Pdg::kDStar:
-      return hf_cand_3prong::resoStatesDstar;
-    case Pdg::kLambdaCPlus:
-      return hf_cand_3prong::resoStatesLambdaC;
-    case Pdg::kXiCPlus:
-      return hf_cand_3prong::resoStatesXiC;
-    default:
-      LOG(error) << "Unknown PDG code for 3-prong final states: " << pdgMother;
-      return {};
-  }
-}
-
-/// Perform the matching for a single resonant channel
-/// \tparam N size of the array of daughter PDG codes
-/// \param arrPdgResoChn array of daughter indices
-/// \param arrPdgDaugs array of PDG codes for the resonant decay
-/// \return true if the resonant channel is matched, false otherwise
-template <std::size_t N>
-bool checkResonantDecay(std::array<int, N> arrPdgResoChn, std::array<int, N> arrPdgDaugs)
-{
-  for (size_t i = 0; i < N; i++) {
-    bool findDaug = false;
-    for (size_t j = 0; j < N; j++) {
-      if (std::abs(arrPdgResoChn[i]) == std::abs(arrPdgDaugs[j])) {
-        arrPdgDaugs[j] = -1; // Mark as found
-        findDaug = true;
-        break;
-      }
-    }
-    if (!findDaug) {
-      return false;
-    }
-  }
-  return true;
-}
-
-/// Flag the resonant decays
-/// Function documentation:
-/// \tparam is3Prong bool to specify if the mother decays with a 3-prong decay
-/// \tparam N size of the array of daughter PDG codes
-/// \param motherPdg PDG code of the mother particle
-/// \param channel decay channel flag to be set
-/// \param arrDaughPdgs array of daughter PDG codes
-template <bool is3Prong = false, std::size_t N>
-void flagResonantDecay(int motherPdg, int8_t* channel, std::array<int, N> arrDaughPdgs)
-{
-  if constexpr (is3Prong) {
-    std::unordered_map<hf_cand_3prong::DecayChannelResonant, std::array<int, 2>> resoStates = getResoChannels3Prong(motherPdg);
-    for (const auto& [flag, pdgCodes] : resoStates) {
-      if (checkResonantDecay(arrDaughPdgs, pdgCodes)) {
-        *channel = flag;
-        break;
-      }
-    }
-  } else {
-    if (motherPdg != Pdg::kD0) {
-      return;
-    }
-    for (const auto& [flag, pdgCodes] : hf_cand_2prong::resoStatesD0) {
-      if (checkResonantDecay(arrDaughPdgs, pdgCodes)) {
-        *channel = flag;
-        break;
-      }
-    }
-  }
-}
-} // namespace o2::hf_corrbkg
-
 #endif // PWGHF_CORE_DECAYCHANNELS_H_

PWGHF/TableProducer/candidateCreator2Prong.cxx

@@ -54,6 +54,7 @@
 #include "PWGHF/Utils/utilsMcGen.h"
 #include "PWGHF/Utils/utilsPid.h"
 #include "PWGHF/Utils/utilsTrkCandHf.h"
+#include "PWGHF/Utils/utilsMcMatching.h"
 
 using namespace o2;
 using namespace o2::analysis;

PWGHF/TableProducer/candidateCreator3Prong.cxx

@@ -53,6 +53,7 @@
 #include "PWGHF/Utils/utilsMcGen.h"
 #include "PWGHF/Utils/utilsPid.h"
 #include "PWGHF/Utils/utilsTrkCandHf.h"
+#include "PWGHF/Utils/utilsMcMatching.h"
 
 using namespace o2;
 using namespace o2::hf_evsel;

PWGHF/Utils/utilsMcGen.h

@@ -30,6 +30,7 @@
 
 #include "PWGHF/Core/DecayChannels.h"
 #include "PWGHF/DataModel/CandidateReconstructionTables.h"
+#include "PWGHF/Utils/utilsMcMatching.h"
 
 using namespace o2::hf_decay::hf_cand_2prong;
 using namespace o2::hf_decay::hf_cand_3prong;