Skip to content

Commit 103a6c3

Browse files
committed
Remove centrality from MC + cleanup table subscriptions
1 parent bd4749c commit 103a6c3

File tree

1 file changed

+16
-35
lines changed

1 file changed

+16
-35
lines changed

PWGHF/D2H/Tasks/taskOmegac0ToOmegapi.cxx

Lines changed: 16 additions & 35 deletions
Original file line numberDiff line numberDiff line change
@@ -146,7 +146,6 @@ struct HfTaskOmegac0ToOmegapi {
146146
axes.push_back(thnAxisPtB);
147147
axes.push_back(thnAxisOrigin);
148148
axes.push_back(thnAxisMatchFlag);
149-
axes.push_back(thnAxisNumPvContr);
150149
}
151150

152151
if (doprocessDataWithKFParticleMl || doprocessDataWithKFParticleMlFT0C || doprocessDataWithKFParticleMlFT0M || doprocessMcWithKFParticleMl) {
@@ -166,8 +165,8 @@ struct HfTaskOmegac0ToOmegapi {
166165
return o2::hf_centrality::getCentralityColl<Coll>(collision);
167166
}
168167

169-
template <bool applyMl, typename CandType, typename CollType>
170-
void processData(const CandType& candidates, CollType const&)
168+
template <bool applyMl, typename CandType>
169+
void processData(const CandType& candidates)
171170
{
172171
for (const auto& candidate : candidates) {
173172
if (!(candidate.resultSelections() == true || (candidate.resultSelections() == false && !selectionFlagOmegac0))) {
@@ -190,7 +189,6 @@ struct HfTaskOmegac0ToOmegapi {
190189
void processDataCent(const CandType& candidates, CollType const& collisions)
191190
{
192191
for (const auto& collision : collisions) {
193-
194192
auto thisCollId = collision.globalIndex();
195193
auto groupedOmegacCandidates = applyMl ? candidates.sliceBy(candOmegacKFMlPerCollision, thisCollId) : candidates.sliceBy(candOmegacKFPerCollision, thisCollId);
196194
auto numPvContributors = collision.numContrib();
@@ -220,9 +218,8 @@ struct HfTaskOmegac0ToOmegapi {
220218
}
221219
}
222220

223-
template <bool applyMl, typename CandType, typename CollType>
224-
void processMc(const CandType& candidates, Omegac0Gen const& mcParticles, TracksMc const&,
225-
CollType const& collisions, aod::McCollisions const&)
221+
template <bool applyMl, typename CandType>
222+
void processMc(const CandType& candidates, Omegac0Gen const& mcParticles)
226223
{
227224
// MC rec.
228225
for (const auto& candidate : candidates) {
@@ -233,13 +230,11 @@ struct HfTaskOmegac0ToOmegapi {
233230
continue;
234231
}
235232

236-
auto numPvContributors = candidate.template collision_as<CollType>().numContrib();
237-
238233
if constexpr (applyMl) {
239-
registry.fill(HIST("hReco"), candidate.invMassCharmBaryon(), candidate.ptCharmBaryon(), candidate.kfRapOmegac(), candidate.ptBhadMotherPart(), candidate.originMcRec(), candidate.flagMcMatchRec(), numPvContributors, candidate.mlProbOmegac()[0]);
234+
registry.fill(HIST("hReco"), candidate.invMassCharmBaryon(), candidate.ptCharmBaryon(), candidate.kfRapOmegac(), candidate.ptBhadMotherPart(), candidate.originMcRec(), candidate.flagMcMatchRec(), candidate.mlProbOmegac()[0]);
240235

241236
} else {
242-
registry.fill(HIST("hReco"), candidate.invMassCharmBaryon(), candidate.ptCharmBaryon(), candidate.kfRapOmegac(), candidate.ptBhadMotherPart(), candidate.originMcRec(), candidate.flagMcMatchRec(), numPvContributors);
237+
registry.fill(HIST("hReco"), candidate.invMassCharmBaryon(), candidate.ptCharmBaryon(), candidate.kfRapOmegac(), candidate.ptBhadMotherPart(), candidate.originMcRec(), candidate.flagMcMatchRec());
243238
}
244239
}
245240

@@ -252,32 +247,24 @@ struct HfTaskOmegac0ToOmegapi {
252247
auto ptGen = particle.pt();
253248
auto yGen = particle.rapidityCharmBaryonGen();
254249

255-
int maxNumContrib = 0;
256-
const auto& recoCollsPerMcColl = collisions.sliceBy(colPerMcCollision, particle.mcCollision().globalIndex());
257-
for (const auto& recCol : recoCollsPerMcColl) {
258-
maxNumContrib = recCol.numContrib() > maxNumContrib ? recCol.numContrib() : maxNumContrib;
259-
}
260-
261250
if (particle.originMcGen() == RecoDecay::OriginType::Prompt) {
262-
registry.fill(HIST("hMcGen"), ptGen, -1., yGen, RecoDecay::OriginType::Prompt, maxNumContrib);
251+
registry.fill(HIST("hMcGen"), ptGen, -1., yGen, RecoDecay::OriginType::Prompt);
263252
} else {
264253
float ptGenB = mcParticles.rawIteratorAt(particle.idxBhadMotherPart()).pt();
265-
registry.fill(HIST("hMcGen"), ptGen, ptGenB, yGen, RecoDecay::OriginType::NonPrompt, maxNumContrib);
254+
registry.fill(HIST("hMcGen"), ptGen, ptGenB, yGen, RecoDecay::OriginType::NonPrompt);
266255
}
267256
}
268257
}
269258

270-
void processDataWithKFParticle(Omegac0CandsKF const& candidates,
271-
Collisions const& collisions)
259+
void processDataWithKFParticle(Omegac0CandsKF const& candidates)
272260
{
273-
processData<false>(candidates, collisions);
261+
processData<false>(candidates);
274262
}
275263
PROCESS_SWITCH(HfTaskOmegac0ToOmegapi, processDataWithKFParticle, "process HfTaskOmegac0ToOmegapi with KFParticle", false);
276264

277-
void processDataWithKFParticleMl(Omegac0CandsMlKF const& candidates,
278-
Collisions const& collisions)
265+
void processDataWithKFParticleMl(Omegac0CandsMlKF const& candidates)
279266
{
280-
processData<true>(candidates, collisions);
267+
processData<true>(candidates);
281268
}
282269
PROCESS_SWITCH(HfTaskOmegac0ToOmegapi, processDataWithKFParticleMl, "process HfTaskOmegac0ToOmegapi with KFParticle and ML selections", false);
283270

@@ -310,22 +297,16 @@ struct HfTaskOmegac0ToOmegapi {
310297
PROCESS_SWITCH(HfTaskOmegac0ToOmegapi, processDataWithKFParticleMlFT0M, "process HfTaskOmegac0ToOmegapi with KFParticle and ML selections and with FT0M centrality", false);
311298

312299
void processMcWithKFParticle(OmegaC0CandsMcKF const& omegaC0CandidatesMcKF,
313-
Omegac0Gen const& mcParticles,
314-
TracksMc const& tracks,
315-
CollisionsWithMcLabels const& collisions,
316-
aod::McCollisions const& mcCollisions)
300+
Omegac0Gen const& mcParticles)
317301
{
318-
processMc<false>(omegaC0CandidatesMcKF, mcParticles, tracks, collisions, mcCollisions);
302+
processMc<false>(omegaC0CandidatesMcKF, mcParticles);
319303
}
320304
PROCESS_SWITCH(HfTaskOmegac0ToOmegapi, processMcWithKFParticle, "Process MC with KFParticle", false);
321305

322306
void processMcWithKFParticleMl(Omegac0CandsMlMcKF const& omegac0CandidatesMlMcKF,
323-
Omegac0Gen const& mcParticles,
324-
TracksMc const& tracks,
325-
CollisionsWithMcLabels const& collisions,
326-
aod::McCollisions const& mcCollisions)
307+
Omegac0Gen const& mcParticles)
327308
{
328-
processMc<true>(omegac0CandidatesMlMcKF, mcParticles, tracks, collisions, mcCollisions);
309+
processMc<true>(omegac0CandidatesMlMcKF, mcParticles);
329310
}
330311
PROCESS_SWITCH(HfTaskOmegac0ToOmegapi, processMcWithKFParticleMl, "Process MC with KFParticle and ML selections", false);
331312
};

0 commit comments

Comments
 (0)