Skip to content

[PWGHF] XicToXiPiPi: enable BDT selection in KFParticle mode #18598

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
lubynets wants to merge 4 commits into
base: master
Choose a base branch
from
+140 −26
Open

[PWGHF] XicToXiPiPi: enable BDT selection in KFParticle mode #18598

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
c9a7cd3
add KF-specific features to HfMlResponseXicToXiPiPi
lubynets Nov 4, 2026
27f8d14
candidateSelectorXicToXiPiPi split process[Data,Mc] to DCAFitter and KF
lubynets Nov 4, 2026
67dc6ae
candidateSelectorXicToXiPiPi: enable BDT eval in KF mode
lubynets Nov 4, 2026
a4bba22
include header with VertexerType
lubynets Nov 4, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
83 changes: 77 additions & 6 deletions PWGHF/Core/HfMlResponseXicToXiPiPi.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -17,19 +17,21 @@
#define PWGHF_CORE_HFMLRESPONSEXICTOXIPIPI_H_

#include "PWGHF/Core/HfMlResponse.h"
#include "PWGHF/DataModel/CandidateReconstructionTables.h"

#include "Tools/ML/MlResponse.h"

#include <cstdint>
#include <map>
#include <string>
#include <vector>

// Fill the map of available input features
// the key is the feature's name (std::string)
// the value is the corresponding value in EnumInputFeatures
#define FILL_MAP_XICTOXIPIPI(FEATURE) \
{ \
#FEATURE, static_cast<uint8_t>(InputFeaturesXicToXiPiPi::FEATURE) \
}
#define FILL_MAP_XICTOXIPIPI(FEATURE) \
{ \
#FEATURE, static_cast<uint8_t>(InputFeaturesXicToXiPiPi::FEATURE)}

// Check if the index of mCachedIndices (index associated to a FEATURE)
// matches the entry in EnumInputFeatures associated to this FEATURE
Expand Down Expand Up @@ -88,10 +90,31 @@ enum class InputFeaturesXicToXiPiPi : uint8_t {
nSigTofPiFromXicPlus1,
nSigTofBachelorPi,
nSigTofPiFromLambda,
nSigTofPrFromLambda
nSigTofPrFromLambda,
// kf-specific variables
kfDecayLength,
kfDecayLengthNormalised,
kfDecayLengthXY,
kfDecayLengthXYNormalised,
kfCascadeChi2,
kfV0Chi2,
chi2XicPlusToPV,
chi2TopoXicPlusToPV,
chi2XiToPV,
chi2Pi0ToPV,
chi2Pi1ToPV,
chi2DevPi0Pi1,
chi2DevPi0Xi,
chi2DevPi1Xi,
dcaPi0Pi1,
dcaPi0Xi,
dcaPi1Xi,
dcaXYPi0Pi1,
dcaXYPi0Xi,
dcaXYPi1Xi
};

template <typename TypeOutputScore = float>
template <typename TypeOutputScore = float, aod::hf_cand::VertexerType ReconstructionType = aod::hf_cand::VertexerType::DCAFitter>
class HfMlResponseXicToXiPiPi : public HfMlResponse<TypeOutputScore>
{
public:
Expand Down Expand Up @@ -147,6 +170,30 @@ class HfMlResponseXicToXiPiPi : public HfMlResponse<TypeOutputScore>
CHECK_AND_FILL_VEC_XICTOXIPIPI(nSigTofPiFromLambda);
CHECK_AND_FILL_VEC_XICTOXIPIPI(nSigTofPrFromLambda);
}
if constexpr (ReconstructionType == aod::hf_cand::VertexerType::KfParticle) {
switch (idx) {
CHECK_AND_FILL_VEC_XICTOXIPIPI(kfDecayLength);
CHECK_AND_FILL_VEC_XICTOXIPIPI(kfDecayLengthNormalised);
CHECK_AND_FILL_VEC_XICTOXIPIPI(kfDecayLengthXY);
CHECK_AND_FILL_VEC_XICTOXIPIPI(kfDecayLengthXYNormalised);
CHECK_AND_FILL_VEC_XICTOXIPIPI(kfCascadeChi2);
CHECK_AND_FILL_VEC_XICTOXIPIPI(kfV0Chi2);
CHECK_AND_FILL_VEC_XICTOXIPIPI(chi2XicPlusToPV);
CHECK_AND_FILL_VEC_XICTOXIPIPI(chi2TopoXicPlusToPV);
CHECK_AND_FILL_VEC_XICTOXIPIPI(chi2XiToPV);
CHECK_AND_FILL_VEC_XICTOXIPIPI(chi2Pi0ToPV);
CHECK_AND_FILL_VEC_XICTOXIPIPI(chi2Pi1ToPV);
CHECK_AND_FILL_VEC_XICTOXIPIPI(chi2DevPi0Pi1);
CHECK_AND_FILL_VEC_XICTOXIPIPI(chi2DevPi0Xi);
CHECK_AND_FILL_VEC_XICTOXIPIPI(chi2DevPi1Xi);
CHECK_AND_FILL_VEC_XICTOXIPIPI(dcaPi0Pi1);
CHECK_AND_FILL_VEC_XICTOXIPIPI(dcaPi0Xi);
CHECK_AND_FILL_VEC_XICTOXIPIPI(dcaPi1Xi);
CHECK_AND_FILL_VEC_XICTOXIPIPI(dcaXYPi0Pi1);
CHECK_AND_FILL_VEC_XICTOXIPIPI(dcaXYPi0Xi);
CHECK_AND_FILL_VEC_XICTOXIPIPI(dcaXYPi1Xi);
}
}
}

return inputFeatures;
Expand Down Expand Up @@ -193,6 +240,30 @@ class HfMlResponseXicToXiPiPi : public HfMlResponse<TypeOutputScore>
FILL_MAP_XICTOXIPIPI(nSigTofBachelorPi),
FILL_MAP_XICTOXIPIPI(nSigTofPiFromLambda),
FILL_MAP_XICTOXIPIPI(nSigTofPrFromLambda)};
if constexpr (ReconstructionType == aod::hf_cand::VertexerType::KfParticle) {
std::map<std::string, uint8_t> mapKfFeatures{
FILL_MAP_XICTOXIPIPI(kfDecayLength),
FILL_MAP_XICTOXIPIPI(kfDecayLengthNormalised),
FILL_MAP_XICTOXIPIPI(kfDecayLengthXY),
FILL_MAP_XICTOXIPIPI(kfDecayLengthXYNormalised),
FILL_MAP_XICTOXIPIPI(kfCascadeChi2),
FILL_MAP_XICTOXIPIPI(kfV0Chi2),
FILL_MAP_XICTOXIPIPI(chi2XicPlusToPV),
FILL_MAP_XICTOXIPIPI(chi2TopoXicPlusToPV),
FILL_MAP_XICTOXIPIPI(chi2XiToPV),
FILL_MAP_XICTOXIPIPI(chi2Pi0ToPV),
FILL_MAP_XICTOXIPIPI(chi2Pi1ToPV),
FILL_MAP_XICTOXIPIPI(chi2DevPi0Pi1),
FILL_MAP_XICTOXIPIPI(chi2DevPi0Xi),
FILL_MAP_XICTOXIPIPI(chi2DevPi1Xi),
FILL_MAP_XICTOXIPIPI(dcaPi0Pi1),
FILL_MAP_XICTOXIPIPI(dcaPi0Xi),
FILL_MAP_XICTOXIPIPI(dcaPi1Xi),
FILL_MAP_XICTOXIPIPI(dcaXYPi0Pi1),
FILL_MAP_XICTOXIPIPI(dcaXYPi0Xi),
FILL_MAP_XICTOXIPIPI(dcaXYPi1Xi)};
MlResponse<TypeOutputScore>::mAvailableInputFeatures.insert(mapKfFeatures.begin(), mapKfFeatures.end());
}
}
};

Expand Down
83 changes: 63 additions & 20 deletions PWGHF/TableProducer/candidateSelectorXicToXiPiPi.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -92,7 +92,8 @@ struct HfCandidateSelectorXicToXiPiPi {
Configurable<int64_t> timestampCCDB{"timestampCCDB", -1, "timestamp of the ONNX file for ML model used to query in CCDB"};
Configurable<bool> loadModelsFromCCDB{"loadModelsFromCCDB", false, "Flag to enable or disable the loading of models from CCDB"};

o2::analysis::HfMlResponseXicToXiPiPi<float> hfMlResponse;
o2::analysis::HfMlResponseXicToXiPiPi<float, aod::hf_cand::VertexerType::DCAFitter> hfMlResponseDca;
o2::analysis::HfMlResponseXicToXiPiPi<float, aod::hf_cand::VertexerType::KfParticle> hfMlResponseKf;
std::vector<float> outputMlXicToXiPiPi;
o2::ccdb::CcdbApi ccdbApi;
TrackSelectorPi selectorPion;
Expand Down Expand Up @@ -121,9 +122,13 @@ struct HfCandidateSelectorXicToXiPiPi {

void init(InitContext&)
{
if ((doprocessData + doprocessMc) != 1) {
if ((doprocessDataWithDCAFitter + doprocessDataWithKFParticle + doprocessMcWithDCAFitter + doprocessMcWithKFParticle) != 1) {
LOGP(fatal, "Enable exactly one process function at a time.");
}
const bool doprocessData = doprocessDataWithDCAFitter || doprocessDataWithKFParticle;
const bool doprocessMc = doprocessMcWithDCAFitter || doprocessMcWithKFParticle;
const bool doprocessDCAFitter = doprocessDataWithDCAFitter || doprocessMcWithDCAFitter;
[[maybe_unused]] const bool doprocessKFParticle = doprocessDataWithKFParticle || doprocessMcWithKFParticle;

if (usePid) {
// pion
Expand Down Expand Up @@ -251,15 +256,22 @@ struct HfCandidateSelectorXicToXiPiPi {
}

if (applyMl) {
hfMlResponse.configure(binsPtMl, cutsMl, cutDirMl, nClassesMl);
if (loadModelsFromCCDB) {
ccdbApi.init(ccdbUrl);
hfMlResponse.setModelPathsCCDB(onnxFileNames, ccdbApi, modelPathsCCDB, timestampCCDB);
auto initMlResponse = [&](auto& hfMlResponseGeneric) {
hfMlResponseGeneric.configure(binsPtMl, cutsMl, cutDirMl, nClassesMl);
if (loadModelsFromCCDB) {
ccdbApi.init(ccdbUrl);
hfMlResponseGeneric.setModelPathsCCDB(onnxFileNames, ccdbApi, modelPathsCCDB, timestampCCDB);
} else {
hfMlResponseGeneric.setModelPathsLocal(onnxFileNames);
}
hfMlResponseGeneric.cacheInputFeaturesIndices(namesInputFeatures);
hfMlResponseGeneric.init();
};
if (doprocessDCAFitter) {
initMlResponse(hfMlResponseDca);
} else {
hfMlResponse.setModelPathsLocal(onnxFileNames);
initMlResponse(hfMlResponseKf);
}
hfMlResponse.cacheInputFeaturesIndices(namesInputFeatures);
hfMlResponse.init();
}
}

Expand Down Expand Up @@ -669,16 +681,21 @@ struct HfCandidateSelectorXicToXiPiPi {
/// \param hfCandXic Xic candidate
/// \param statusXicToXiPiPi Flag to store selection status as defined in hf_sel_candidate_xic::XicToXiPiPiSelectionStep
/// \param isMatchedSignal Flag to indicate if the candidate is matched to a genereated XiCplus MC particle
template <bool IsMc, typename XicCandidate>
template <bool IsMc, aod::hf_cand::VertexerType ReconstructionType, typename XicCandidate>
void isBdtSelected(XicCandidate const& hfCandXic,
int& statusXicToXiPiPi,
bool const isMatchedSignal = false)
{
bool isSelectedMlXicToXiPiPi = false;
float const ptCandXic = hfCandXic.pt();

std::vector<float> inputFeaturesXicToXiPiPi = hfMlResponse.getInputFeatures(hfCandXic);
isSelectedMlXicToXiPiPi = hfMlResponse.isSelectedMl(inputFeaturesXicToXiPiPi, ptCandXic, outputMlXicToXiPiPi);
if constexpr (ReconstructionType == aod::hf_cand::VertexerType::DCAFitter) {
std::vector<float> inputFeaturesXicToXiPiPi = hfMlResponseDca.getInputFeatures(hfCandXic);
isSelectedMlXicToXiPiPi = hfMlResponseDca.isSelectedMl(inputFeaturesXicToXiPiPi, ptCandXic, outputMlXicToXiPiPi);
} else {
std::vector<float> inputFeaturesXicToXiPiPi = hfMlResponseKf.getInputFeatures(hfCandXic);
isSelectedMlXicToXiPiPi = hfMlResponseKf.isSelectedMl(inputFeaturesXicToXiPiPi, ptCandXic, outputMlXicToXiPiPi);
}

hfMlXicToXiPiPiCandidate(outputMlXicToXiPiPi);

Expand All @@ -699,8 +716,8 @@ struct HfCandidateSelectorXicToXiPiPi {
}
}

void processData(aod::HfCandXic const& hfCandsXic,
TracksExtraWPid const& tracks)
template <aod::hf_cand::VertexerType ReconstructionType, typename CandType>
void runData(CandType const& hfCandsXic, TracksExtraWPid const& tracks)
{
for (const auto& hfCandXic : hfCandsXic) {
int statusXicToXiPiPi = 0;
Expand All @@ -719,18 +736,17 @@ struct HfCandidateSelectorXicToXiPiPi {

// ML selection
if (applyMl) {
isBdtSelected<false>(hfCandXic, statusXicToXiPiPi);
isBdtSelected<false, ReconstructionType>(hfCandXic, statusXicToXiPiPi);
} else {
registry.fill(HIST("hSelCandidates"), BdtSelected);
}

hfSelXicToXiPiPiCandidate(statusXicToXiPiPi);
}
}
PROCESS_SWITCH(HfCandidateSelectorXicToXiPiPi, processData, "Select candidates without MC matching information", true);

void processMc(soa::Join<aod::HfCandXic, aod::HfCandXicMcRec> const& hfCandsXic,
TracksExtraWPid const& tracks)
template <aod::hf_cand::VertexerType ReconstructionType, typename CandType>
void runMc(CandType const& hfCandsXic, TracksExtraWPid const& tracks)
{
for (const auto& hfCandXic : hfCandsXic) {
int statusXicToXiPiPi = 0;
Expand All @@ -754,7 +770,7 @@ struct HfCandidateSelectorXicToXiPiPi {

// ML selection
if (applyMl) {
isBdtSelected<true>(hfCandXic, statusXicToXiPiPi, isMatchedCandidate);
isBdtSelected<true, ReconstructionType>(hfCandXic, statusXicToXiPiPi, isMatchedCandidate);
} else if (isMatchedCandidate) {
registry.fill(HIST("hSelCandidatesRecSig"), BdtSelected);
} else {
Expand All @@ -764,7 +780,34 @@ struct HfCandidateSelectorXicToXiPiPi {
hfSelXicToXiPiPiCandidate(statusXicToXiPiPi);
}
}
PROCESS_SWITCH(HfCandidateSelectorXicToXiPiPi, processMc, "Select candidates with MC matching information", false);

void processDataWithDCAFitter(aod::HfCandXic const& hfCandsXic,
TracksExtraWPid const& tracks)
{
runData<aod::hf_cand::VertexerType::DCAFitter>(hfCandsXic, tracks);
}
PROCESS_SWITCH(HfCandidateSelectorXicToXiPiPi, processDataWithDCAFitter, "Select candidates without MC matching information using DCAFitter", true);

void processDataWithKFParticle(soa::Join<aod::HfCandXic, aod::HfCandXicKF> const& hfCandsXic,
TracksExtraWPid const& tracks)
{
runData<aod::hf_cand::VertexerType::KfParticle>(hfCandsXic, tracks);
}
PROCESS_SWITCH(HfCandidateSelectorXicToXiPiPi, processDataWithKFParticle, "Select candidates without MC matching information using KFParticle", false);

void processMcWithDCAFitter(soa::Join<aod::HfCandXic, aod::HfCandXicMcRec> const& hfCandsXic,
TracksExtraWPid const& tracks)
{
runMc<aod::hf_cand::VertexerType::DCAFitter>(hfCandsXic, tracks);
}
PROCESS_SWITCH(HfCandidateSelectorXicToXiPiPi, processMcWithDCAFitter, "Select candidates with MC matching information using DCAFitter", false);

void processMcWithKFParticle(soa::Join<aod::HfCandXic, aod::HfCandXicMcRec, aod::HfCandXicKF> const& hfCandsXic,
TracksExtraWPid const& tracks)
{
runMc<aod::hf_cand::VertexerType::KfParticle>(hfCandsXic, tracks);
}
PROCESS_SWITCH(HfCandidateSelectorXicToXiPiPi, processMcWithKFParticle, "Select candidates with MC matching information using KFParticle", false);
};

WorkflowSpec defineDataProcessing(ConfigContext const& cfgc)
Expand Down
Loading