Skip to content

[PWGHF] Update tree creator for Xic0Omegac0 #16421

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 9 commits into from
May 29, 2026
Merged

[PWGHF] Update tree creator for Xic0Omegac0 #16421

Changes from all commits
Commits
Show all changes
9 commits
Select commit Hold shift + click to select a range
0f8ebed
First commit to update treeCreatorToXiPiQa.cxx
May 27, 2026
585a74f
Update output table & apply appropriate mass hypothesis for Omegac0
May 27, 2026
c87394a
Please consider the following formatting changes
alibuild May 27, 2026
6912c7e
Merge pull request #24 from alibuild/alibot-cleanup-16421
jinhyunni May 27, 2026
9953b75
Added helper function to fill in table for particle info
May 27, 2026
c923aed
Merge branch 'updateworkflow' of https://github.com/jinhyunni/O2Physi…
May 27, 2026
85c75c3
Merge branch 'master' of https://github.com/AliceO2Group/O2Physics in…
May 28, 2026
c3f3995
Adjust helper function arguement
May 28, 2026
2aeda8f
Merge branch 'master' of https://github.com/AliceO2Group/O2Physics in…
May 29, 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
188 changes: 166 additions & 22 deletions PWGHF/TableProducer/treeCreatorToXiPiQa.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,7 @@
/// \author Krista Smith <krista.lizbeth.smith@cern.ch>, Pusan National University

#include "PWGHF/Core/CentralityEstimation.h"
#include "PWGHF/Core/DecayChannelsLegacy.h"
Comment thread
vkucera marked this conversation as resolved.
#include "PWGHF/DataModel/CandidateReconstructionTables.h"
#include "PWGHF/DataModel/CandidateSelectionTables.h"

Expand All @@ -38,6 +39,7 @@

using namespace o2;
using namespace o2::framework;
using namespace o2::framework::expressions;

// SV Reco method
enum {
Expand Down Expand Up @@ -124,6 +126,8 @@ DECLARE_SOA_COLUMN(EtaPiFromCasc, etaPiFromCasc, float);
DECLARE_SOA_COLUMN(EtaPiFromCharmBaryon, etaPiFromCharmBaryon, float);
DECLARE_SOA_COLUMN(EtaCharmBaryon, etaCharmBaryon, float);
DECLARE_SOA_COLUMN(EtaCascade, etaCascade, float);
DECLARE_SOA_COLUMN(PhiCharmBaryon, phiCharmBaryon, float);
DECLARE_SOA_COLUMN(YCharmBaryon, yCharmBaryon, float);
DECLARE_SOA_COLUMN(EtaV0, etaV0, float);
DECLARE_SOA_COLUMN(DcaXYToPvV0Dau0, dcaXYToPvV0Dau0, float);
DECLARE_SOA_COLUMN(DcaXYToPvV0Dau1, dcaXYToPvV0Dau1, float);
Expand Down Expand Up @@ -212,6 +216,8 @@ DECLARE_SOA_COLUMN(CascChi2OverNdf, cascChi2OverNdf, float);
DECLARE_SOA_COLUMN(XicChi2OverNdf, xicChi2OverNdf, float);
DECLARE_SOA_COLUMN(MassV0Chi2OverNdf, massV0Chi2OverNdf, float);
DECLARE_SOA_COLUMN(MassCascChi2OverNdf, massCascChi2OverNdf, float);
// MC
DECLARE_SOA_COLUMN(ParticlePdg, particlePdg, int);
} // namespace full

DECLARE_SOA_TABLE(HfToXiPiEvs, "AOD", "HFTOXIPIEV",
Expand Down Expand Up @@ -303,6 +309,15 @@ DECLARE_SOA_TABLE(HfKfXicFulls, "AOD", "HFKFXICFULL",
full::ResultSelections,
full::FlagMcMatchRec, full::DebugMcRec, full::OriginRec, full::CollisionMatched);

DECLARE_SOA_TABLE(HfCandToXiPiGen, "AOD", "HFCANDTOXIPIGEN",
full::InvMassCharmBaryon,
full::PtCharmBaryon,
full::EtaCharmBaryon,
full::PhiCharmBaryon,
full::YCharmBaryon,
full::FlagMcMatchRec,
full::OriginRec,
full::ParticlePdg)
} // namespace o2::aod

/// Writes the full information in an output TTree
Expand All @@ -311,15 +326,21 @@ struct HfTreeCreatorToXiPiQa {
Produces<o2::aod::HfToXiPiFulls> rowCandidateFull;
Produces<o2::aod::HfToXiPiLites> rowCandidateLite;
Produces<o2::aod::HfKfXicFulls> rowKfCandidate;
Produces<o2::aod::HfCandToXiPiGen> rowCandidateParticles;
Produces<o2::aod::HfToXiPiEvs> rowEv;

Configurable<float> zPvCut{"zPvCut", 10., "Cut on absolute value of primary vertex z coordinate"};
Configurable<int8_t> genSelection{"genSelection", o2::aod::hf_cand_xic0_omegac0::DecayType::XiczeroToXiPi, "Decay channel to be used to match particle information"};
Configurable<bool> fillGenTable{"fillGenTable", true, "Fill generated MC information if requested"};

using MyTrackTable = soa::Join<aod::Tracks, aod::TrackSelection, aod::TracksExtra>;
using MyEventTable = soa::Join<aod::Collisions, aod::EvSels>;
using MyEventTableWithFT0C = soa::Join<aod::Collisions, aod::EvSels, aod::CentFT0Cs>;
using MyEventTableWithFT0M = soa::Join<aod::Collisions, aod::EvSels, aod::CentFT0Ms>;
using MyEventTableWithNTracksPV = soa::Join<aod::Collisions, aod::EvSels, aod::CentNTPVs>;
using MatchedGenXiPi = soa::Filtered<soa::Join<aod::McParticles, aod::HfXicToXiPiMCGen>>;

Filter filterGenXiPi = nabs(aod::hf_cand_mc_flag::flagMcMatchGen) == static_cast<int8_t>(BIT(genSelection));

void init(InitContext const&)
{
Expand Down Expand Up @@ -611,6 +632,19 @@ struct HfTreeCreatorToXiPiQa {
}
}

template <typename T>
void fillParticle(const T& particle, double massCharmBaryon)
{
rowCandidateParticles(RecoDecay::m(particle.pVector(), particle.e()),
particle.pt(),
particle.eta(),
particle.phi(),
RecoDecay::y(particle.pVector(), massCharmBaryon),
particle.flagMcMatchGen(),
particle.originMcGen(),
particle.pdgCode());
}

////////////////////////////////////
// //
// Process functions //
Expand Down Expand Up @@ -783,8 +817,10 @@ struct HfTreeCreatorToXiPiQa {
//*~~~~~~~MC with DCAFitter~~~~~~~~*//
//*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*//

void processMcFullXic0(MyEventTable const& collisions, MyTrackTable const&,
soa::Join<aod::HfCandToXiPi, aod::HfSelToXiPi, aod::HfXicToXiPiMCRec> const& candidates)
void processMcFullXic0(MyEventTable const& collisions,
MyTrackTable const&,
soa::Join<aod::HfCandToXiPi, aod::HfSelToXiPi, aod::HfXicToXiPiMCRec> const& candidates,
MatchedGenXiPi const& mcParticles)
{
// Filling event properties
rowEv.reserve(collisions.size());
Expand All @@ -797,10 +833,20 @@ struct HfTreeCreatorToXiPiQa {
for (const auto& candidate : candidates) {
fillCandidate<DCAFITTER, FULL, false, MyEventTable>(candidate, candidate.flagMcMatchRec(), candidate.debugMcRec(), candidate.originMcRec(), candidate.collisionMatched());
}

// Filling particle properties if requested
if (fillGenTable) {
rowCandidateParticles.reserve(mcParticles.size());
for (const auto& particle : mcParticles) {
fillParticle(particle, o2::constants::physics::MassXiC0);
}
}
}

void processMcFullOmegac0(MyEventTable const& collisions, MyTrackTable const&,
soa::Join<aod::HfCandToXiPi, aod::HfSelToXiPi, aod::HfOmegacToXiPiMCRec> const& candidates)
void processMcFullOmegac0(MyEventTable const& collisions,
MyTrackTable const&,
soa::Join<aod::HfCandToXiPi, aod::HfSelToXiPi, aod::HfOmegacToXiPiMCRec> const& candidates,
MatchedGenXiPi const& mcParticles)
{
// Filling event properties
rowEv.reserve(collisions.size());
Expand All @@ -813,10 +859,20 @@ struct HfTreeCreatorToXiPiQa {
for (const auto& candidate : candidates) {
fillCandidate<DCAFITTER, FULL, false, MyEventTable>(candidate, candidate.flagMcMatchRec(), candidate.debugMcRec(), candidate.originMcRec(), candidate.collisionMatched());
}

// Filling particle properties if requested
if (fillGenTable) {
rowCandidateParticles.reserve(mcParticles.size());
for (const auto& particle : mcParticles) {
fillParticle(particle, o2::constants::physics::MassOmegaC0);
}
}
}

void processMcLiteXic0(MyEventTable const& collisions, MyTrackTable const&,
soa::Join<aod::HfCandToXiPi, aod::HfSelToXiPi, aod::HfXicToXiPiMCRec> const& candidates)
void processMcLiteXic0(MyEventTable const& collisions,
MyTrackTable const&,
soa::Join<aod::HfCandToXiPi, aod::HfSelToXiPi, aod::HfXicToXiPiMCRec> const& candidates,
MatchedGenXiPi const& mcParticles)
{
// Filling event properties
rowEv.reserve(collisions.size());
Expand All @@ -829,10 +885,20 @@ struct HfTreeCreatorToXiPiQa {
for (const auto& candidate : candidates) {
fillCandidate<DCAFITTER, LITE, false, MyEventTable>(candidate, candidate.flagMcMatchRec(), -7, candidate.originMcRec(), candidate.collisionMatched());
}

// Filling particle properties if requested
if (fillGenTable) {
rowCandidateParticles.reserve(mcParticles.size());
for (const auto& particle : mcParticles) {
fillParticle(particle, o2::constants::physics::MassXiC0);
}
}
}

void processMcLiteXic0WithFT0C(MyEventTableWithFT0C const& collisions, MyTrackTable const&,
soa::Join<aod::HfCandToXiPi, aod::HfSelToXiPi, aod::HfXicToXiPiMCRec> const& candidates)
void processMcLiteXic0WithFT0C(MyEventTableWithFT0C const& collisions,
MyTrackTable const&,
soa::Join<aod::HfCandToXiPi, aod::HfSelToXiPi, aod::HfXicToXiPiMCRec> const& candidates,
MatchedGenXiPi const& mcParticles)
{
// Filling event properties
rowEv.reserve(collisions.size());
Expand All @@ -845,10 +911,20 @@ struct HfTreeCreatorToXiPiQa {
for (const auto& candidate : candidates) {
fillCandidate<DCAFITTER, LITE, true, MyEventTableWithFT0C>(candidate, candidate.flagMcMatchRec(), -7, candidate.originMcRec(), candidate.collisionMatched());
}

// Filling particle properties if requested
if (fillGenTable) {
rowCandidateParticles.reserve(mcParticles.size());
for (const auto& particle : mcParticles) {
fillParticle(particle, o2::constants::physics::MassXiC0);
}
}
}

void processMcLiteXic0WithFT0M(MyEventTableWithFT0M const& collisions, MyTrackTable const&,
soa::Join<aod::HfCandToXiPi, aod::HfSelToXiPi, aod::HfXicToXiPiMCRec> const& candidates)
void processMcLiteXic0WithFT0M(MyEventTableWithFT0M const& collisions,
MyTrackTable const&,
soa::Join<aod::HfCandToXiPi, aod::HfSelToXiPi, aod::HfXicToXiPiMCRec> const& candidates,
MatchedGenXiPi const& mcParticles)
{
// Filling event properties
rowEv.reserve(collisions.size());
Expand All @@ -861,10 +937,20 @@ struct HfTreeCreatorToXiPiQa {
for (const auto& candidate : candidates) {
fillCandidate<DCAFITTER, LITE, true, MyEventTableWithFT0M>(candidate, candidate.flagMcMatchRec(), -7, candidate.originMcRec(), candidate.collisionMatched());
}

// Filling particle properties if requested
if (fillGenTable) {
rowCandidateParticles.reserve(mcParticles.size());
for (const auto& particle : mcParticles) {
fillParticle(particle, o2::constants::physics::MassXiC0);
}
}
}

void processMcLiteXic0WithNTracksPV(MyEventTableWithNTracksPV const& collisions, MyTrackTable const&,
soa::Join<aod::HfCandToXiPi, aod::HfSelToXiPi, aod::HfXicToXiPiMCRec> const& candidates)
void processMcLiteXic0WithNTracksPV(MyEventTableWithNTracksPV const& collisions,
MyTrackTable const&,
soa::Join<aod::HfCandToXiPi, aod::HfSelToXiPi, aod::HfXicToXiPiMCRec> const& candidates,
MatchedGenXiPi const& mcParticles)
{
// Filling event properties
rowEv.reserve(collisions.size());
Expand All @@ -877,10 +963,20 @@ struct HfTreeCreatorToXiPiQa {
for (const auto& candidate : candidates) {
fillCandidate<DCAFITTER, LITE, true, MyEventTableWithNTracksPV>(candidate, candidate.flagMcMatchRec(), -7, candidate.originMcRec(), candidate.collisionMatched());
}

// Filling particle properties if requested
if (fillGenTable) {
rowCandidateParticles.reserve(mcParticles.size());
for (const auto& particle : mcParticles) {
fillParticle(particle, o2::constants::physics::MassXiC0);
}
}
}

void processMcLiteOmegac0(MyEventTable const& collisions, MyTrackTable const&,
soa::Join<aod::HfCandToXiPi, aod::HfSelToXiPi, aod::HfOmegacToXiPiMCRec> const& candidates)
void processMcLiteOmegac0(MyEventTable const& collisions,
MyTrackTable const&,
soa::Join<aod::HfCandToXiPi, aod::HfSelToXiPi, aod::HfOmegacToXiPiMCRec> const& candidates,
MatchedGenXiPi const& mcParticles)
{
// Filling event properties
rowEv.reserve(collisions.size());
Expand All @@ -893,6 +989,14 @@ struct HfTreeCreatorToXiPiQa {
for (const auto& candidate : candidates) {
fillCandidate<DCAFITTER, LITE, false, MyEventTable>(candidate, candidate.flagMcMatchRec(), -7, candidate.originMcRec(), candidate.collisionMatched());
}

// Filling particle properties if requested
if (fillGenTable) {
rowCandidateParticles.reserve(mcParticles.size());
for (const auto& particle : mcParticles) {
fillParticle(particle, o2::constants::physics::MassOmegaC0);
}
}
}

PROCESS_SWITCH(HfTreeCreatorToXiPiQa, processMcFullXic0, "Process MC with full information for xic0 w/o centrality", false);
Expand All @@ -906,8 +1010,10 @@ struct HfTreeCreatorToXiPiQa {
//*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*//
//*~~~~~~~MC with KFParticle~~~~~~~~*//
//*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*//
void processKfMcXic0(MyEventTable const& collisions, MyTrackTable const&,
soa::Join<aod::HfCandToXiPiKf, aod::HfSelToXiPiKf, aod::HfXicToXiPiMCRec> const& candidates)
void processKfMcXic0(MyEventTable const& collisions,
MyTrackTable const&,
soa::Join<aod::HfCandToXiPiKf, aod::HfSelToXiPiKf, aod::HfXicToXiPiMCRec> const& candidates,
MatchedGenXiPi const& mcParticles)
{
// Filling event properties
rowEv.reserve(collisions.size());
Expand All @@ -920,10 +1026,20 @@ struct HfTreeCreatorToXiPiQa {
for (const auto& candidate : candidates) {
fillCandidate<KFPARTICLE, FULL, false, MyEventTable>(candidate, candidate.flagMcMatchRec(), candidate.debugMcRec(), candidate.originMcRec(), candidate.collisionMatched());
}

// Filling particle properties if requested
if (fillGenTable) {
rowCandidateParticles.reserve(mcParticles.size());
for (const auto& particle : mcParticles) {
fillParticle(particle, o2::constants::physics::MassXiC0);
}
}
}

void processKfMcXic0WithFT0C(MyEventTableWithFT0C const& collisions, MyTrackTable const&,
soa::Join<aod::HfCandToXiPiKf, aod::HfSelToXiPiKf, aod::HfXicToXiPiMCRec> const& candidates)
void processKfMcXic0WithFT0C(MyEventTableWithFT0C const& collisions,
MyTrackTable const&,
soa::Join<aod::HfCandToXiPiKf, aod::HfSelToXiPiKf, aod::HfXicToXiPiMCRec> const& candidates,
MatchedGenXiPi const& mcParticles)
{
// Filling event properties
rowEv.reserve(collisions.size());
Expand All @@ -936,10 +1052,20 @@ struct HfTreeCreatorToXiPiQa {
for (const auto& candidate : candidates) {
fillCandidate<KFPARTICLE, FULL, true, MyEventTableWithFT0C>(candidate, candidate.flagMcMatchRec(), candidate.debugMcRec(), candidate.originMcRec(), candidate.collisionMatched());
}

// Filling particle properties if requested
if (fillGenTable) {
rowCandidateParticles.reserve(mcParticles.size());
for (const auto& particle : mcParticles) {
fillParticle(particle, o2::constants::physics::MassXiC0);
}
}
}

void processKfMcXic0WithFT0M(MyEventTableWithFT0M const& collisions, MyTrackTable const&,
soa::Join<aod::HfCandToXiPiKf, aod::HfSelToXiPiKf, aod::HfXicToXiPiMCRec> const& candidates)
void processKfMcXic0WithFT0M(MyEventTableWithFT0M const& collisions,
MyTrackTable const&,
soa::Join<aod::HfCandToXiPiKf, aod::HfSelToXiPiKf, aod::HfXicToXiPiMCRec> const& candidates,
MatchedGenXiPi const& mcParticles)
{
// Filling event properties
rowEv.reserve(collisions.size());
Expand All @@ -952,10 +1078,20 @@ struct HfTreeCreatorToXiPiQa {
for (const auto& candidate : candidates) {
fillCandidate<KFPARTICLE, FULL, true, MyEventTableWithFT0M>(candidate, candidate.flagMcMatchRec(), candidate.debugMcRec(), candidate.originMcRec(), candidate.collisionMatched());
}

// Filling particle properties if requested
if (fillGenTable) {
rowCandidateParticles.reserve(mcParticles.size());
for (const auto& particle : mcParticles) {
fillParticle(particle, o2::constants::physics::MassXiC0);
}
}
}

void processKfMcXic0WithNTracksPV(MyEventTableWithNTracksPV const& collisions, MyTrackTable const&,
soa::Join<aod::HfCandToXiPiKf, aod::HfSelToXiPiKf, aod::HfXicToXiPiMCRec> const& candidates)
void processKfMcXic0WithNTracksPV(MyEventTableWithNTracksPV const& collisions,
MyTrackTable const&,
soa::Join<aod::HfCandToXiPiKf, aod::HfSelToXiPiKf, aod::HfXicToXiPiMCRec> const& candidates,
MatchedGenXiPi const& mcParticles)
{
// Filling event properties
rowEv.reserve(collisions.size());
Expand All @@ -968,6 +1104,14 @@ struct HfTreeCreatorToXiPiQa {
for (const auto& candidate : candidates) {
fillCandidate<KFPARTICLE, FULL, true, MyEventTableWithNTracksPV>(candidate, candidate.flagMcMatchRec(), candidate.debugMcRec(), candidate.originMcRec(), candidate.collisionMatched());
}

// Filling particle properties if requested
if (fillGenTable) {
rowCandidateParticles.reserve(mcParticles.size());
for (const auto& particle : mcParticles) {
fillParticle(particle, o2::constants::physics::MassXiC0);
}
}
}

PROCESS_SWITCH(HfTreeCreatorToXiPiQa, processKfMcXic0, "Process MC with information for xic0", false);
Expand Down
Loading