Skip to content

PWGHF: added BDT scores to Charm resonances analysis #5689

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

Merged
fgrosa merged 14 commits into from
Apr 22, 2024
Merged

PWGHF: added BDT scores to Charm resonances analysis #5689

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
14 commits
Select commit Hold shift + click to select a range
81f66fa
merged changes to Dplus task on master
Apr 18, 2024
a4ec2e6
marginal change
Apr 3, 2024
024f2fe
Please consider the following formatting changes
alibuild Apr 3, 2024
e00cbe2
Please consider the following formatting changes
alibuild Apr 3, 2024
440a998
Please consider the following formatting changes
alibuild Apr 3, 2024
e797a81
try2
Apr 18, 2024
b04f0df
first draft for addition of ML tables to Charm Reso analysis
Apr 10, 2024
3556328
first draft of data creator
Apr 12, 2024
fad0e5a
sync Dplus Task
Apr 12, 2024
c3e78b7
added ML scores table to ReducedDataModel.h
Apr 15, 2024
a0e583e
added BDT scores to Reso candidate creator and added Z impact paramet…
Apr 18, 2024
9444916
Please consider the following formatting changes
alibuild Apr 18, 2024
631da59
Merge pull request #18 from alibuild/alibot-cleanup-5689
Luca610 Apr 18, 2024
cb54635
Added new line at the end of taskDplus.cxx
Luca610 Apr 22, 2024
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
10 changes: 9 additions & 1 deletion PWGHF/D2H/DataModel/ReducedDataModel.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -399,7 +399,9 @@ DECLARE_SOA_COLUMN(PtProng0, ptProng0, float); //! Pt of D daughter in
DECLARE_SOA_COLUMN(PtProng1, ptProng1, float); //! Pt of V0 daughter in GeV/c
DECLARE_SOA_COLUMN(InvMassProng0, invMassProng0, float); //! Invariant Mass of D daughter in GeV/c
DECLARE_SOA_COLUMN(InvMassProng1, invMassProng1, float); //! Invariant Mass of V0 daughter in GeV/c

DECLARE_SOA_COLUMN(MlScoreBkgProng0, mlScoreBkgProng0, float); //! Bkg ML score of the D daughter
DECLARE_SOA_COLUMN(MlScorePromptProng0, mlScorePromptProng0, float); //! Prompt ML score of the D daughter
DECLARE_SOA_COLUMN(MlScoreNonpromptProng0, mlScoreNonpromptProng0, float); //! Nonprompt ML score of the D daughter
} // namespace hf_reso_cand_reduced

DECLARE_SOA_TABLE(HfRedVzeros, "AOD", "HFREDVZERO", //! Table with V0 candidate information for resonances reduced workflow
Expand Down Expand Up @@ -439,6 +441,12 @@ DECLARE_SOA_TABLE(HfCandCharmReso, "AOD", "HFCANDCHARMRESO", //! Table with Reso
hf_reso_cand_reduced::Cpa,
hf_reso_cand_reduced::Dca,
hf_reso_cand_reduced::Radius);

DECLARE_SOA_TABLE(HfCharmResoMLs, "AOD", "HFCHARMRESOML", //! Table with ML scores for the D daughter
hf_reso_cand_reduced::MlScoreBkgProng0,
hf_reso_cand_reduced::MlScorePromptProng0,
hf_reso_cand_reduced::MlScoreNonpromptProng0,
o2::soa::Marker<1>);
} // namespace aod

namespace soa
Expand Down
109 changes: 90 additions & 19 deletions PWGHF/D2H/TableProducer/candidateCreatorCharmResoReduced.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -39,7 +39,8 @@ enum Selections : uint8_t {
enum DecayChannel : uint8_t {
Ds1ToDstarK0s = 0,
Ds2StarToDplusK0s,
XcToDplusLambda
XcToDplusLambda,
LambdaDminus
};
enum V0Type : uint8_t {
K0s = 0,
Expand All @@ -61,6 +62,8 @@ auto vecBins = std::vector<double>{binsPt, binsPt + nBins + 1};
struct HfCandidateCreatorCharmResoReduced {
// Produces: Tables with resonance info
Produces<aod::HfCandCharmReso> rowCandidateReso;
// Optional D daughter ML scores table
Produces<aod::HfCharmResoMLs> mlScores;

// Configurables
Configurable<double> invMassWindowD{"invMassWindowD", 0.5, "invariant-mass window for D candidates (GeV/c2)"};
Expand All @@ -70,6 +73,8 @@ struct HfCandidateCreatorCharmResoReduced {
// Hist Axis
Configurable<std::vector<double>> binsPt{"binsPt", std::vector<double>{vecBins}, "pT bin limits"};

using reducedDWithMl = soa::Join<aod::HfRed3PrNoTrks, aod::HfRed3ProngsMl>;

// Partition of V0 candidates based on v0Type
Partition<aod::HfRedVzeros> candidatesK0s = aod::hf_reso_cand_reduced::v0Type == (uint8_t)1 || aod::hf_reso_cand_reduced::v0Type == (uint8_t)3 || aod::hf_reso_cand_reduced::v0Type == (uint8_t)5;
Partition<aod::HfRedVzeros> candidatesLambda = aod::hf_reso_cand_reduced::v0Type == (uint8_t)2 || aod::hf_reso_cand_reduced::v0Type == (uint8_t)4;
Expand All @@ -85,14 +90,17 @@ struct HfCandidateCreatorCharmResoReduced {

void init(InitContext const&)
{
for (const auto& value : vecBins) {
LOGF(info, "bin limit %f", value);
// check that only one process function is enabled
std::array<bool, 8> doprocess{doprocessDs2StarToDplusK0s, doprocessDs2StarToDplusK0sWithMl, doprocessDs1ToDstarK0s, doprocessDs1ToDstarK0sWithMl, doprocessXcToDplusLambda, doprocessXcToDplusLambdaWithMl, doprocessLambdaDminus, doprocessLambdaDminusWithMl};
if ((std::accumulate(doprocess.begin(), doprocess.end(), 0)) != 1) {
LOGP(fatal, "Only one process function should be enabled! Please check your configuration!");
}
const AxisSpec axisPt{(std::vector<double>)vecBins, "#it{p}_{T} (GeV/#it{c})"};
// histograms
registry.add("hMassDs1", "Ds1 candidates;m_{Ds1} - m_{D^{*}} (GeV/#it{c}^{2});entries", {HistType::kTH2F, {{100, 2.4, 2.7}, {(std::vector<double>)binsPt, "#it{p}_{T} (GeV/#it{c})"}}});
registry.add("hMassDs2Star", "Ds^{*}2 candidates; Ds^{*}2 - m_{D^{#plus}} (GeV/#it{c}^{2}) ;entries", {HistType::kTH2F, {{100, 2.4, 2.7}, {(std::vector<double>)binsPt, "#it{p}_{T} (GeV/#it{c})"}}});
registry.add("hMassXcRes", "XcRes candidates; XcRes - m_{D^{#plus}} (GeV/#it{c}^{2}) ;entries", {HistType::kTH2F, {{100, 2.9, 3.3}, {(std::vector<double>)binsPt, "#it{p}_{T} (GeV/#it{c})"}}});
const AxisSpec axisPt{(std::vector<double>)vecBins, "#it{p}_{T} (GeV/#it{c})"};
registry.add("hMassDs1", "Ds1 candidates;m_{Ds1} (GeV/#it{c}^{2});entries", {HistType::kTH2F, {{100, 2.4, 2.7}, {(std::vector<double>)binsPt, "#it{p}_{T} (GeV/#it{c})"}}});
registry.add("hMassDs2Star", "Ds^{*}2 candidates; m_Ds^{*}2 (GeV/#it{c}^{2}) ;entries", {HistType::kTH2F, {{100, 2.4, 2.7}, {(std::vector<double>)binsPt, "#it{p}_{T} (GeV/#it{c})"}}});
registry.add("hMassXcRes", "XcRes candidates; m_XcRes (GeV/#it{c}^{2}) ;entries", {HistType::kTH2F, {{100, 2.9, 3.3}, {(std::vector<double>)binsPt, "#it{p}_{T} (GeV/#it{c})"}}});
registry.add("hMassLambdaDminus", "LambdaDminus candidates; m_LambdaDminus (GeV/#it{c}^{2}) ;entries", {HistType::kTH2F, {{100, 2.9, 3.3}, {(std::vector<double>)binsPt, "#it{p}_{T} (GeV/#it{c})"}}});
if (activateQA) {
constexpr int kNBinsSelections = Selections::NSelSteps;
std::string labels[kNBinsSelections];
Expand All @@ -105,19 +113,21 @@ struct HfCandidateCreatorCharmResoReduced {
registry.get<TH1>(HIST("hSelections"))->GetXaxis()->SetBinLabel(iBin + 1, labels[iBin].data());
}
}

// mass constants
massK0 = o2::constants::physics::MassK0Short;
massLambda = o2::constants::physics::MassLambda;
massDplus = o2::constants::physics::MassDPlus;
massDstar = o2::constants::physics::MassDStar;
}

/// Basic selection of D candidates
/// \param candD is the reduced D meson candidate
/// \return true if selections are passed
template <DecayChannel channel, typename DRedTable>
bool isDSelected(DRedTable const& candD)
{
float massD{0.};
// slection on D candidate mass
if (channel == DecayChannel::Ds2StarToDplusK0s || channel == DecayChannel::XcToDplusLambda) {
if (channel == DecayChannel::Ds2StarToDplusK0s || channel == DecayChannel::XcToDplusLambda || channel == DecayChannel::LambdaDminus) {
massD = massDplus;
} else if (channel == DecayChannel::Ds1ToDstarK0s) {
massD = massDstar;
Expand All @@ -128,19 +138,27 @@ struct HfCandidateCreatorCharmResoReduced {
return true;
}

/// Basic selection of V0 candidates
/// \param candV0 is the reduced V0 candidate
/// \param candD is the reduced D meson candidate
/// \return true if selections are passed
template <DecayChannel channel, typename DRedTable, typename V0RedTable>
bool isV0Selected(V0RedTable const& candV0, DRedTable const& candD)
{
float massV0{0.};
float invMassV0{0.};

// slection on V0 candidate mass
if (channel == DecayChannel::Ds2StarToDplusK0s || channel == DecayChannel::Ds1ToDstarK0s) {
massV0 = massK0;
invMassV0 = candV0.invMassK0s();
} else if (channel == DecayChannel::XcToDplusLambda) {
} else if (channel == DecayChannel::XcToDplusLambda || channel == DecayChannel::LambdaDminus) {
massV0 = massLambda;
int wsFact{1};
if (channel == DecayChannel::LambdaDminus)
wsFact = -1;
uint8_t targetV0Type{0};
if (candD.dType() > 0) {
if (wsFact * candD.dType() > 0) {
invMassV0 = candV0.invMassLambda();
targetV0Type = V0Type::Lambda;
} else {
Expand All @@ -157,7 +175,7 @@ struct HfCandidateCreatorCharmResoReduced {
return true;
}

template <DecayChannel channel, typename Coll, typename DRedTable, typename V0RedTable>
template <bool fillMl, DecayChannel channel, typename Coll, typename DRedTable, typename V0RedTable>
void runCandidateCreation(Coll const& collisions,
DRedTable const& candsD,
V0RedTable const& candsV0)
Expand Down Expand Up @@ -213,6 +231,15 @@ struct HfCandidateCreatorCharmResoReduced {
invMassReso = RecoDecay::m(std::array{pVecD, pVecV0}, std::array{massDplus, massLambda});
registry.fill(HIST("hMassXcRes"), invMassReso, ptReso);
break;
case DecayChannel::LambdaDminus:
if (candD.dType() < 0) {
invMassV0 = candV0.invMassLambda();
} else {
invMassV0 = candV0.invMassAntiLambda();
}
invMassReso = RecoDecay::m(std::array{pVecD, pVecV0}, std::array{massDplus, massLambda});
registry.fill(HIST("hMassLambdaDminus"), invMassReso, ptReso);
break;
default:
break;
}
Expand All @@ -227,6 +254,9 @@ struct HfCandidateCreatorCharmResoReduced {
candV0.cpa(),
candV0.dca(),
candV0.radius());
if constexpr (fillMl) {
mlScores(candD.mlScoreBkgMassHypo0(), candD.mlScorePromptMassHypo0(), candD.mlScoreNonpromptMassHypo0());
}
}
}
} // main function
Expand All @@ -235,25 +265,66 @@ struct HfCandidateCreatorCharmResoReduced {
aod::HfRed3PrNoTrks const& candsD,
aod::HfRedVzeros const&)
{
runCandidateCreation<DecayChannel::Ds2StarToDplusK0s>(collision, candsD, candidatesK0s);
runCandidateCreation<false, DecayChannel::Ds2StarToDplusK0s>(collision, candsD, candidatesK0s);
}
PROCESS_SWITCH(HfCandidateCreatorCharmResoReduced, processDs2StarToDplusK0s, "Process Ds2* candidates without MC info and without ML info", true);
PROCESS_SWITCH(HfCandidateCreatorCharmResoReduced, processDs2StarToDplusK0s, "Process Ds2* candidates without ML info", true);

void processDs2StarToDplusK0sWithMl(aod::HfRedCollisions::iterator const& collision,
soa::Join<aod::HfRed3PrNoTrks, aod::HfRed3ProngsMl> const& candsD,
aod::HfRedVzeros const&)
{
runCandidateCreation<true, DecayChannel::Ds2StarToDplusK0s>(collision, candsD, candidatesK0s);
}
PROCESS_SWITCH(HfCandidateCreatorCharmResoReduced, processDs2StarToDplusK0sWithMl, "Process Ds2* candidates with Ml info", false);

void processDs1ToDstarK0s(aod::HfRedCollisions::iterator const& collision,
aod::HfRed3PrNoTrks const& candsD,
aod::HfRedVzeros const&)
{
runCandidateCreation<DecayChannel::Ds1ToDstarK0s>(collision, candsD, candidatesK0s);
runCandidateCreation<false, DecayChannel::Ds1ToDstarK0s>(collision, candsD, candidatesK0s);
}
PROCESS_SWITCH(HfCandidateCreatorCharmResoReduced, processDs1ToDstarK0s, "Process Ds1 candidates without Ml info", false);

void processDs1ToDstarK0sWithMl(aod::HfRedCollisions::iterator const& collision,
soa::Join<aod::HfRed3PrNoTrks, aod::HfRed3ProngsMl> const& candsD,
aod::HfRedVzeros const&)
{
runCandidateCreation<true, DecayChannel::Ds1ToDstarK0s>(collision, candsD, candidatesK0s);
}
PROCESS_SWITCH(HfCandidateCreatorCharmResoReduced, processDs1ToDstarK0s, "Process Ds1 candidates without MC info and without ML info", false);
PROCESS_SWITCH(HfCandidateCreatorCharmResoReduced, processDs1ToDstarK0sWithMl, "Process Ds1 candidates with Ml info", false);

void processXcToDplusLambda(aod::HfRedCollisions::iterator const& collision,
aod::HfRed3PrNoTrks const& candsD,
aod::HfRedVzeros const&)
{
runCandidateCreation<DecayChannel::XcToDplusLambda>(collision, candsD, candidatesLambda);
runCandidateCreation<false, DecayChannel::XcToDplusLambda>(collision, candsD, candidatesLambda);
}
PROCESS_SWITCH(HfCandidateCreatorCharmResoReduced, processXcToDplusLambda, "Process Xc candidates without Ml info", false);

void processXcToDplusLambdaWithMl(aod::HfRedCollisions::iterator const& collision,
soa::Join<aod::HfRed3PrNoTrks, aod::HfRed3ProngsMl> const& candsD,
aod::HfRedVzeros const&)
{
runCandidateCreation<true, DecayChannel::XcToDplusLambda>(collision, candsD, candidatesLambda);
}
PROCESS_SWITCH(HfCandidateCreatorCharmResoReduced, processXcToDplusLambda, "Process Xc candidates without MC info and without ML info", false);
PROCESS_SWITCH(HfCandidateCreatorCharmResoReduced, processXcToDplusLambdaWithMl, "Process Xc candidates with Ml info", false);

void processLambdaDminus(aod::HfRedCollisions::iterator const& collision,
aod::HfRed3PrNoTrks const& candsD,
aod::HfRedVzeros const&)
{
runCandidateCreation<false, DecayChannel::LambdaDminus>(collision, candsD, candidatesLambda);
}
PROCESS_SWITCH(HfCandidateCreatorCharmResoReduced, processLambdaDminus, "Process LambdaDminus candidates without Ml info", false);

void processLambdaDminusWithMl(aod::HfRedCollisions::iterator const& collision,
soa::Join<aod::HfRed3PrNoTrks, aod::HfRed3ProngsMl> const& candsD,
aod::HfRedVzeros const&)
{
runCandidateCreation<true, DecayChannel::LambdaDminus>(collision, candsD, candidatesLambda);
}
PROCESS_SWITCH(HfCandidateCreatorCharmResoReduced, processLambdaDminusWithMl, "Process LambdaDminus candidates with Ml info", false);

}; // struct

WorkflowSpec defineDataProcessing(ConfigContext const& cfgc)
Expand Down
Loading