Skip to content

Fix subscription of reduced tables #19

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
creetz16 merged 2 commits into from
Feb 20, 2025
Merged

Fix subscription of reduced tables #19

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
303635c
Fix subscription of reduced tables
wang-yuanzhe Feb 20, 2025
d464fab
Fix code formattiing error
wang-yuanzhe Feb 20, 2025
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
37 changes: 18 additions & 19 deletions PWGLF/TableProducer/Nuspex/decay3bodybuilder.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -71,7 +71,6 @@ using FullTracksExtIU = soa::Join<aod::TracksIU, aod::TracksExtra, aod::TracksCo
using FullTracksExtPIDIU = soa::Join<FullTracksExtIU, aod::pidTPCFullPr, aod::pidTPCFullPi, aod::pidTPCFullDe>;

using ColwithEvTimes = o2::soa::Join<aod::Collisions, aod::EvSels, aod::EvTimeTOFFT0>;
using FullCols = o2::soa::Join<ColwithEvTimes, aod::CentFT0Cs>;
using ColwithEvTimesMults = o2::soa::Join<ColwithEvTimes, aod::Mults>;
using TrackExtIUwithEvTimes = soa::Join<FullTracksExtIU, aod::EvTimeTOFFT0ForTrack>;
using TrackExtPIDIUwithEvTimes = soa::Join<FullTracksExtPIDIU, aod::EvTimeTOFFT0ForTrack>;
Expand Down Expand Up @@ -676,7 +675,7 @@ struct decay3bodyBuilder {

//------------------------------------------------------------------
// 3body candidate builder
template <class TCollisionClass, typename TCollisionTable, typename TTrackTable>
template <typename TCollisionTable, typename TTrackTable>
void fillVtxCand(TCollisionTable const& collision, TTrackTable const& t0, TTrackTable const& t1, TTrackTable const& t2, int64_t decay3bodyId, int bachelorcharge = 1, double tofNSigmaBach = -999)
{

Expand Down Expand Up @@ -1497,7 +1496,7 @@ struct decay3bodyBuilder {
tofNSigmaBach = bachelorTOFPID.GetTOFNSigma(t2, originalcol, collision);
}

fillVtxCand<ColwithEvTimes>(collision, t0, t1, t2, d3body.globalIndex(), bachelorcharge, tofNSigmaBach);
fillVtxCand(collision, t0, t1, t2, d3body.globalIndex(), bachelorcharge, tofNSigmaBach);
}

for (const auto& candVtx : vtxCandidates) {
Expand All @@ -1516,17 +1515,17 @@ struct decay3bodyBuilder {
int lastRunNumber = -1;

for (const auto& d3body : decay3bodys) {
auto t0 = d3body.template track0_as<aod::RedIUTracks>();
auto t1 = d3body.template track1_as<aod::RedIUTracks>();
auto t2 = d3body.template track2_as<aod::RedIUTracks>();
auto collision = d3body.template collision_as<aod::RedCollisions>();
auto t0 = d3body.track0_as<aod::RedIUTracks>();
auto t1 = d3body.track1_as<aod::RedIUTracks>();
auto t2 = d3body.track2_as<aod::RedIUTracks>();
auto collision = d3body.collision_as<aod::RedCollisions>();

if (collision.runNumber() != lastRunNumber) {
initCCDBReduced(collision.runNumber());
lastRunNumber = collision.runNumber(); // Update the last run number
LOG(debug) << "CCDB initialized for run " << lastRunNumber;
}
fillVtxCand<aod::RedCollisions>(collision, t0, t1, t2, d3body.globalIndex(), bachelorcharge, t2.tofNSigmaDe());
fillVtxCand(collision, t0, t1, t2, d3body.globalIndex(), bachelorcharge, t2.tofNSigmaDe());
}

for (const auto& candVtx : vtxCandidates) {
Expand All @@ -1537,37 +1536,37 @@ struct decay3bodyBuilder {

//------------------------------------------------------------------
// Event-mixing background
void processRun3ReducedEM(ReducedCollisionsMultsCents const&, aod::ReducedDecay3Bodys const&, aod::ReducedTracksIU const&)
void processRun3ReducedEM(ReducedCollisionsMultsCents const&, aod::RedDecay3Bodys const&, aod::RedIUTracks const&)
{
vtxCandidates.clear();

BinningType binningEvent{{axisPosZ, axisCentrality}, true};
SameKindPair<ReducedCollisionsMultsCents, aod::ReducedDecay3Bodys, BinningType> pair{binningEvent, EMTrackSel.nUseMixedEvent, -1, &cache};
SameKindPair<ReducedCollisionsMultsCents, aod::RedDecay3Bodys, BinningType> pair{binningEvent, EMTrackSel.nUseMixedEvent, -1, &cache};

int lastRunNumber = -1;

for (const auto& [c0, decay3bodys0, c1, decay3bodys1] : pair) {
for (auto& [d3body0, d3body1] : combinations(soa::CombinationsFullIndexPolicy(decay3bodys0, decay3bodys1))) {
auto tpos0 = d3body0.track0_as<aod::ReducedTracksIU>();
auto tneg0 = d3body0.track1_as<aod::ReducedTracksIU>();
auto tbach0 = d3body0.track2_as<aod::ReducedTracksIU>();
auto tpos1 = d3body1.track0_as<aod::ReducedTracksIU>();
auto tneg1 = d3body1.track1_as<aod::ReducedTracksIU>();
auto tbach1 = d3body1.track2_as<aod::ReducedTracksIU>();
auto tpos0 = d3body0.track0_as<aod::RedIUTracks>();
auto tneg0 = d3body0.track1_as<aod::RedIUTracks>();
auto tbach0 = d3body0.track2_as<aod::RedIUTracks>();
auto tpos1 = d3body1.track0_as<aod::RedIUTracks>();
auto tneg1 = d3body1.track1_as<aod::RedIUTracks>();
auto tbach1 = d3body1.track2_as<aod::RedIUTracks>();

if (c0.runNumber() != lastRunNumber) {
initCCDBReduced(c0.runNumber());
lastRunNumber = c0.runNumber(); // Update the last run number
LOG(debug) << "CCDB initialized for run " << lastRunNumber;
}
fillVtxCand<FullCols>(c0, tpos0, tneg0, tbach1, -1, bachelorcharge, tbach1.tofNSigmaDe());
fillVtxCand(c0, tpos0, tneg0, tbach1, -1, bachelorcharge, tbach1.tofNSigmaDe());

if (c1.runNumber() != lastRunNumber) {
initCCDBReduced(c1.runNumber());
lastRunNumber = c1.runNumber(); // Update the last run number
LOG(debug) << "CCDB initialized for run " << lastRunNumber;
}
fillVtxCand<FullCols>(c1, tpos1, tneg1, tbach0, -1, bachelorcharge, tbach0.tofNSigmaDe());
fillVtxCand(c1, tpos1, tneg1, tbach0, -1, bachelorcharge, tbach0.tofNSigmaDe());
}
}

Expand Down
20 changes: 10 additions & 10 deletions PWGLF/TableProducer/Nuspex/threebodyRecoTask.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -43,7 +43,7 @@ using namespace o2;
using namespace o2::framework;
using namespace o2::framework::expressions;

using ReducedCols = soa::Join<aod::ReducedCollisions, aod::ReducedCentFT0Cs>;
using ReducedCols = soa::Join<aod::RedCollisions, aod::RedCentFT0Cs>;
using FullTracksExtIU = soa::Join<aod::TracksIU, aod::TracksExtra, aod::TracksCovIU, aod::pidTPCFullPr, aod::pidTPCFullPi, aod::pidTPCFullDe>;
using MCLabeledTracksIU = soa::Join<FullTracksExtIU, aod::McTrackLabels>;

Expand Down Expand Up @@ -562,9 +562,9 @@ struct ThreebodyRecoTask {
template <class TTrackClass, typename TCollisionTable, typename TCandTable, typename TTracks>
void reducedAnalysis(TCollisionTable const& collision, TCandTable const& candData, TTracks tracks, bool isTrueCand = false, int lLabel = -1, TLorentzVector lmother = {0, 0, 0, 0}, double MClifetime = -1)
{
auto track0 = tracks.iteratorAt(candData.track0Id());
auto track1 = tracks.iteratorAt(candData.track1Id());
auto track2 = tracks.iteratorAt(candData.track2Id());
auto track0 = tracks.rawIteratorAt(candData.track0Id());
auto track1 = tracks.rawIteratorAt(candData.track1Id());
auto track2 = tracks.rawIteratorAt(candData.track2Id());

bool isMatter = track2.sign() > 0; // true if the candidate is hypertriton (p pi- d)

Expand All @@ -582,9 +582,9 @@ struct ThreebodyRecoTask {
template <class TTrackClass, typename TCollisionTable, typename TCandTable, typename TTracks>
void reducedLikeSignAnalysis(TCollisionTable const& collision, TCandTable const& candData, TTracks tracks, bool isTrueCand = false, int lLabel = -1, TLorentzVector lmother = {0, 0, 0, 0}, double MClifetime = -1)
{
auto track0 = tracks.iteratorAt(candData.track0Id());
auto track1 = tracks.iteratorAt(candData.track1Id());
auto track2 = tracks.iteratorAt(candData.track2Id());
auto track0 = tracks.rawIteratorAt(candData.track0Id());
auto track1 = tracks.rawIteratorAt(candData.track1Id());
auto track2 = tracks.rawIteratorAt(candData.track2Id());

bool isMatter = track2.sign() < 0; // true if seach for background consists of (p pi- anti-d)

Expand Down Expand Up @@ -701,16 +701,16 @@ struct ThreebodyRecoTask {

//------------------------------------------------------------------
// process reduced data analysis
void processReducedData(ReducedCols const& collisions, aod::Vtx3BodyDatas const& vtx3bodydatas, aod::ReducedTracksIU const& tracks)
void processReducedData(ReducedCols const& collisions, aod::Vtx3BodyDatas const& vtx3bodydatas, aod::RedIUTracks const& tracks)
{
candidates3body.clear();

for (const auto& vtx : vtx3bodydatas) {
const auto& collision = collisions.iteratorAt(vtx.collisionId());
if (cfgLikeSignAnalysis) {
reducedLikeSignAnalysis<aod::ReducedTracksIU>(collision, vtx, tracks);
reducedLikeSignAnalysis<aod::RedIUTracks>(collision, vtx, tracks);
} else {
reducedAnalysis<aod::ReducedTracksIU>(collision, vtx, tracks);
reducedAnalysis<aod::RedIUTracks>(collision, vtx, tracks);
}
for (const auto& cand3body : candidates3body) {
fillOutputDataTable(cand3body);
Expand Down
Loading