Skip to content

[ALICE3] Update multicharm table producer with lutConfigId at collision level #14650

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
njacazio merged 4 commits into from
Jan 27, 2026
+17 −45
Merged

[ALICE3] Update multicharm table producer with lutConfigId at collision level #14650

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
53761b4
Update multicharm table producer with lutConfigId at collision level
jesgum Jan 27, 2026
b63b905
Remove unused variables
jesgum Jan 27, 2026
6ffa333
Remove debug logging
jesgum Jan 27, 2026
6af3e8a
formatting
jesgum Jan 27, 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
57 changes: 16 additions & 41 deletions ALICE3/TableProducer/alice3MulticharmFinder.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -68,7 +68,8 @@ using namespace o2::framework::expressions;
#define BIT_CHECK(var, nbit) ((var) & (static_cast<uint32_t>(1) << (nbit)))
#define GET_HIST(type, name) std::get<std::shared_ptr<type>>(histPointers[name])

using Alice3Tracks = soa::Join<aod::Tracks, aod::TracksCov, aod::Alice3DecayMaps, aod::McTrackLabels, aod::TracksDCA, aod::TracksExtraA3, aod::UpgradeTofs, aod::UpgradeTofExpectedTimes, aod::UpgradeRichs, aod::UpgradeRichSignals, aod::OTFLUTConfigId>;
using Alice3Tracks = soa::Join<aod::Tracks, aod::TracksCov, aod::Alice3DecayMaps, aod::McTrackLabels, aod::TracksDCA, aod::TracksExtraA3, aod::UpgradeTofs, aod::UpgradeTofExpectedTimes, aod::UpgradeRichs, aod::UpgradeRichSignals>;
using Alice3Collision = soa::Join<aod::Collisions, aod::OTFLUTConfigId>;

struct Alice3MulticharmFinder {
SliceCache cache;
Expand Down Expand Up @@ -388,12 +389,6 @@ struct Alice3MulticharmFinder {
return returnValue;
}

template <typename TTrackType>
bool checkSameLUTConf(TTrackType const& track1, const int track2)
{
return track1.lutConfigId() == track2;
}

void init(InitContext&)
{
// initialize O2 2-prong fitter (only once)
Expand Down Expand Up @@ -489,7 +484,7 @@ struct Alice3MulticharmFinder {
}
}

void initConf(const int icfg)
void initDetectorConfiguration(const int icfg)
{
if (std::find(savedConfigs.begin(), savedConfigs.end(), icfg) != savedConfigs.end()) {
return;
Expand All @@ -498,7 +493,7 @@ struct Alice3MulticharmFinder {
savedConfigs.push_back(icfg);

// do more plots
std::string histPath = "Configuration_" + std::to_string(icfg) + "/";
const std::string histPath = "Configuration_" + std::to_string(icfg) + "/";
histPointers.insert({histPath + "hMassXiCC", histos.add((histPath + "hMassXiCC").c_str(), "hMassXiCC", {kTH1D, {{axisXiCCMass}}})});
histPointers.insert({histPath + "hNCollisions", histos.add((histPath + "hNCollisions").c_str(), "hNCollisions", {kTH1D, {{2, 0.5, 2.5}}})});
histPointers.insert({histPath + "hNTracks", histos.add((histPath + "hNTracks").c_str(), "hNTracks", {kTH1D, {{20000, 0, 20000}}})});
Expand All @@ -519,19 +514,24 @@ struct Alice3MulticharmFinder {
}

//*+-+*+-+*+-+*+-+*+-+*+-+*+-+*+-+*+-+*+-+*
void processFindXiCC(aod::Collision const& collision, Alice3Tracks const& tracks, aod::McParticles const&, aod::UpgradeCascades const& cascades)
void processFindXiCC(Alice3Collision::iterator const& collision, Alice3Tracks const& tracks, aod::McParticles const&, aod::UpgradeCascades const& cascades)
{
const std::string histPath = "Configuration_" + std::to_string(collision.lutConfigId()) + "/";
initDetectorConfiguration(collision.lutConfigId());

GET_HIST(TH1, histPath + "hNCollisions")->Fill(1);
GET_HIST(TH1, histPath + "hNTracks")->Fill(tracks.size());
if (tracks.size() < minNTracks.value[collision.lutConfigId()]) {
return;
}
GET_HIST(TH1, histPath + "hNCollisions")->Fill(2);

// group with this collision
// n.b. cascades do not need to be grouped, being used directly in iterator-grouping
auto picTracksGrouped = picTracks->sliceByCached(aod::track::collisionId, collision.globalIndex(), cache);
auto piccTracksGrouped = piccTracks->sliceByCached(aod::track::collisionId, collision.globalIndex(), cache);

static constexpr int MaxLUTConfigs = 20;
std::vector<int> nTracks(MaxLUTConfigs);
for (auto const& track : tracks) {
int lutConfigId = track.lutConfigId();
nTracks[lutConfigId]++;

if (BIT_CHECK(track.decayMap(), kTruePiFromXiC)) {
histos.fill(HIST("h2dDCAxyVsPtPiFromXiC"), track.pt(), track.dcaXY() * 1e+4);
}
Expand All @@ -542,19 +542,6 @@ struct Alice3MulticharmFinder {

for (auto const& xiCand : cascades) {
auto xi = xiCand.cascadeTrack_as<Alice3Tracks>(); // de-reference cascade track
int lutConfigId = xi.lutConfigId();
initConf(lutConfigId);
if (minNTracks.value.size() < static_cast<size_t>(lutConfigId)) {
if (nTracks[lutConfigId] < minNTracks.value.front()) {
continue; // fallback to first
}
} else {
if (nTracks[lutConfigId] < minNTracks.value[lutConfigId]) {
continue;
}
}

std::string histPath = "Configuration_" + std::to_string(lutConfigId) + "/";
histos.fill(HIST("hMassXi"), xiCand.mXi());
histos.fill(HIST("h2dDCAxyVsPtXiFromXiC"), xi.pt(), xi.dcaXY() * 1e+4);
if (std::fabs(xiCand.mXi() - o2::constants::physics::MassXiMinus) > xiMassWindow) {
Expand Down Expand Up @@ -583,10 +570,6 @@ struct Alice3MulticharmFinder {

histos.fill(HIST("hMinXiDecayRadius"), xiCand.cascRadius());
for (auto const& pi1c : picTracksGrouped) {
if (!checkSameLUTConf(pi1c, lutConfigId)) {
continue;
}

if (mcSameMotherCheck && !checkSameMother(xi, pi1c)) {
continue;
}
Expand All @@ -609,10 +592,6 @@ struct Alice3MulticharmFinder {
histos.fill(HIST("hInnerTOFTrackTimeRecoPi1c"), pi1cTOFDiffInner);
// second pion from XiC decay for starts here
for (auto const& pi2c : picTracksGrouped) {
if (!checkSameLUTConf(pi2c, lutConfigId)) {
continue;
}

if (mcSameMotherCheck && !checkSameMother(xi, pi2c)) {
continue; // keep only if same mother
}
Expand Down Expand Up @@ -691,10 +670,6 @@ struct Alice3MulticharmFinder {
// attempt XiCC finding
uint32_t nCombinationsCC = 0;
for (auto const& picc : piccTracksGrouped) {
if (!checkSameLUTConf(picc, lutConfigId)) {
continue;
}

if (mcSameMotherCheck && !checkSameMotherExtra(xi, picc)) {
continue;
}
Expand Down Expand Up @@ -823,7 +798,7 @@ struct Alice3MulticharmFinder {
xicDecayDistanceFromPV,
xiccProperLength,
pi1c.pt(), pi2c.pt(), picc.pt(),
lutConfigId);
collision.lutConfigId());

multiCharmPID(
pi1cTOFDiffInner, pi1c.nSigmaPionInnerTOF(),
Expand Down
5 changes: 1 addition & 4 deletions ALICE3/Tasks/alice3Multicharm.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -69,13 +69,9 @@ using MultiCharmTracksFull = soa::Join<aod::MCharmCores, aod::MCharmPID, aod::MC

struct Alice3Multicharm {
HistogramRegistry histos{"histos", {}, OutputObjHandlingPolicy::AnalysisObject};
std::map<std::string, HistPtr> histPointers;
std::vector<int> savedConfigs;
std::string histPath;

std::map<int, int> pdgToBin;
o2::ml::OnnxModel bdtMCharm;

std::map<std::string, std::string> metadata;
o2::ccdb::CcdbApi ccdbApi;
Service<o2::ccdb::BasicCCDBManager> ccdb;
Expand Down Expand Up @@ -131,6 +127,7 @@ struct Alice3Multicharm {
Configurable<float> xiccMinProperLength{"xiccMinProperLength", -1, "Minimum proper length for Xicc decay (cm)"};
Configurable<float> xiccMaxProperLength{"xiccMaxProperLength", 1e+4, "Minimum proper length for Xicc decay (cm)"};
Configurable<int> otfConfig{"otfConfig", 0, "OTF configuration flag"};

Filter configFilter = (aod::otfmulticharm::lutConfigId == otfConfig);

void init(InitContext&)
Expand Down
Loading