Skip to content

[PWGHF] Add generated MC outputs to UPC Lambda_c task #18010

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
zhangbiao-phy merged 2 commits into from
Sep 22, 2026
Merged

[PWGHF] Add generated MC outputs to UPC Lambda_c task #18010

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
aa6659c
add MCGen TTree
Rrantu Sep 22, 2026
0fa19dd
adjust default value
Rrantu Sep 22, 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
33 changes: 28 additions & 5 deletions PWGHF/D2H/Tasks/taskUpcLc.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,7 @@
/// \author Ran Tu <ran.tu@cern.ch>, Fudan University

#include "PWGHF/Core/CentralityEstimation.h"
#include "PWGHF/Core/DecayChannels.h"
#include "PWGHF/Core/HfHelper.h"
#include "PWGHF/Core/SelectorCuts.h"
#include "PWGHF/DataModel/AliasTables.h"
Expand All @@ -28,10 +29,12 @@
#include "PWGUD/Core/SGSelector.h"
#include "PWGUD/Core/UPCHelpers.h"

#include "Common/Core/RecoDecay.h"
#include "Common/DataModel/EventSelection.h"
#include "Common/DataModel/Multiplicity.h"

#include <CCDB/BasicCCDBManager.h>
#include <CommonConstants/PhysicsConstants.h>
#include <CommonDataFormat/TimeStamp.h>
#include <Framework/ASoA.h>
#include <Framework/AnalysisDataModel.h>
Expand Down Expand Up @@ -126,6 +129,10 @@ DECLARE_SOA_TABLE(HfUpcLcMcInfos, "AOD", "HFUPCLCMCINFO",
full::FlagMcMatchRec,
full::OriginMcRec,
full::PtBhadMotherPart);
DECLARE_SOA_TABLE(HfUpcLcMcGen, "AOD", "HFUPCLCMCGEN",
full::Pt,
hf_cand_mc_flag::FlagMcMatchGen,
hf_cand_mc_flag::OriginMcGen);
} // namespace o2::aod

/// Λc± → p± K∓ π± analysis task
Expand All @@ -134,12 +141,14 @@ struct HfTaskUpcLc {
Produces<o2::aod::HfUpcLcInfos> rowCandUpc;
Produces<o2::aod::HfUpcLcMcBdtInfos> rowCandUpcMcBdt;
Produces<o2::aod::HfUpcLcMcInfos> rowCandUpcMc;
Produces<o2::aod::HfUpcLcMcGen> rowLcMcGen;
Produces<o2::aod::HfUpcQa> rowUpcQa;

Configurable<int> selectionFlagLc{"selectionFlagLc", 1, "Selection Flag for Lc"};
Configurable<double> yCandRecoMax{"yCandRecoMax", 0.8, "max. cand. rapidity"};
Configurable<std::vector<double>> binsPt{"binsPt", std::vector<double>{hf_cuts_lc_to_p_k_pi::vecBinsPt}, "pT bin limits"};
Configurable<bool> fillTreeOnlySingleGap{"fillTreeOnlySingleGap", false, "Only fill the tree for candidates that pass the single-gap UPC events"};
Configurable<bool> fillMcGenLcTree{"fillMcGenLcTree", false, "Fill the generated Lc to p K pi tree"};
Configurable<bool> fillTreeUpcQa{"fillTreeUpcQa", false, "Fill Tree for UPC QA"};
Configurable<bool> fillHistQa{"fillHistQa", false, "Fill histograms for UPC detector QA"};
Configurable<bool> verticesWithUpc{"verticesWithUpc", false, "Consider vertices with UPC settings"};
Expand Down Expand Up @@ -211,10 +220,22 @@ struct HfTaskUpcLc {
BCsType const& bcs,
aod::FT0s const& ft0s,
aod::FV0As const& fv0as,
aod::FDDs const& fdds

)
aod::FDDs const& fdds,
soa::Join<aod::McParticles, aod::HfCand3ProngMcGen> const* mcParticles = nullptr)
{
if constexpr (IsMc) {
if (fillMcGenLcTree) {
for (const auto& particle : *mcParticles) {
if (std::abs(particle.flagMcMatchGen()) != hf_decay::hf_cand_3prong::DecayChannelMain::LcToPKPi) {
continue;
}
if (yCandRecoMax >= 0. && std::abs(RecoDecay::y(particle.pVector(), o2::constants::physics::MassLambdaCPlus)) > yCandRecoMax) {
continue;
}
rowLcMcGen(particle.pt(), particle.flagMcMatchGen(), particle.originMcGen());
}
}
}
for (const auto& collision : collisions) {
float centrality{-1.f};
const auto rejectionMask = hfEvSel.getHfCollisionRejectionMaskWithUpc<true, CentralityEstimator::None, BCsType>(collision, centrality, ccdb, registry, bcs);
Expand Down Expand Up @@ -397,27 +418,29 @@ struct HfTaskUpcLc {
aod::BcFullInfos const& bcs,
LcCandidatesMlMc const& selectedLcCandidatesMlMc,
aod::McCollisions const&,
soa::Join<aod::McParticles, aod::HfCand3ProngMcGen> const& mcParticles,
aod::TracksWMc const&,
aod::FT0s const& ft0s,
aod::FV0As const& fv0as,
aod::FDDs const& fdds,
aod::Zdcs const& /*zdcs*/)
{
runAnalysisPerCollisionWithUpc<true, true>(collisions, selectedLcCandidatesMlMc, bcs, ft0s, fv0as, fdds);
runAnalysisPerCollisionWithUpc<true, true>(collisions, selectedLcCandidatesMlMc, bcs, ft0s, fv0as, fdds, &mcParticles);
}
PROCESS_SWITCH(HfTaskUpcLc, processMcWithMlWithUpc, "Process MC with the ML method with UPC", false);

void processMcStdWithUpc(soa::Join<aod::Collisions, aod::McCollisionLabels, aod::EvSels, aod::Mults> const& collisions,
aod::BcFullInfos const& bcs,
LcCandidatesMc const& selectedLcCandidatesMc,
aod::McCollisions const&,
soa::Join<aod::McParticles, aod::HfCand3ProngMcGen> const& mcParticles,
aod::TracksWMc const&,
aod::FT0s const& ft0s,
aod::FV0As const& fv0as,
aod::FDDs const& fdds,
aod::Zdcs const& /*zdcs*/)
{
runAnalysisPerCollisionWithUpc<false, true>(collisions, selectedLcCandidatesMc, bcs, ft0s, fv0as, fdds);
runAnalysisPerCollisionWithUpc<false, true>(collisions, selectedLcCandidatesMc, bcs, ft0s, fv0as, fdds, &mcParticles);
}
PROCESS_SWITCH(HfTaskUpcLc, processMcStdWithUpc, "Process MC with the standard method with UPC", false);
};
Expand Down
Loading