Skip to content
Merged
Changes from all commits
Commits
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
8 changes: 8 additions & 0 deletions PWGHF/TableProducer/candidateSelectorLc.cxx
Original file line number Diff line number Diff line change
Expand Up @@ -24,6 +24,7 @@
#include "PWGHF/DataModel/CandidateReconstructionTables.h"
#include "PWGHF/DataModel/CandidateSelectionTables.h"
#include "PWGHF/DataModel/TrackIndexSkimmingTables.h"
#include "PWGHF/Utils/utilsAnalysis.h"

#include "Common/Core/TrackSelectorPID.h"
#include "Common/DataModel/PIDResponseCombined.h"
Expand Down Expand Up @@ -108,6 +109,8 @@ struct HfCandidateSelectorLc {
Configurable<std::vector<std::string>> onnxFileNames{"onnxFileNames", std::vector<std::string>{"ModelHandler_onnx_LcToPKPi.onnx"}, "ONNX file names for each pT bin (if not from CCDB full path)"};
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"};
// Mass Cut for trigger analysis
Configurable<bool> useTriggerMassCut{"useTriggerMassCut", false, "Flag to enable parametrize pT differential mass cut for triggered data"};

o2::analysis::HfMlResponseLcToPKPi<float, aod::hf_cand::VertexerType::DCAFitter> hfMlResponseDCA;
o2::analysis::HfMlResponseLcToPKPi<float, aod::hf_cand::VertexerType::KfParticle> hfMlResponseKF;
Expand All @@ -117,6 +120,7 @@ struct HfCandidateSelectorLc {
TrackSelectorPi selectorPion;
TrackSelectorKa selectorKaon;
TrackSelectorPr selectorProton;
HfTrigger3ProngCuts hfTriggerCuts;

using TracksSel = soa::Join<aod::TracksWExtra,
aod::TracksPidPi, aod::PidTpcTofFullPi, aod::TracksPidKa, aod::PidTpcTofFullKa, aod::TracksPidPr, aod::PidTpcTofFullPr>;
Expand Down Expand Up @@ -327,6 +331,10 @@ struct HfCandidateSelectorLc {
return false;
}

if (useTriggerMassCut && !isCandidateInMassRange(massLc, o2::constants::physics::MassLambdaCPlus, candpT, hfTriggerCuts)) {
return false;
}

/// cut on the Kpi pair invariant mass, to study Lc->pK*(->Kpi)
const double cutMassKPi = cuts->get(pTBin, "mass (Kpi)");
if (cutMassKPi > 0 && std::abs(massKPi - massK0Star892) > cutMassKPi) {
Expand Down
Loading