Skip to content

Commit 8ce975b

Browse files
authored
[PWGCF] Fix MC only processing (#16888)
1 parent bf46df0 commit 8ce975b

2 files changed

Lines changed: 9 additions & 6 deletions

File tree

PWGCF/Femto/Core/mcBuilder.h

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -184,16 +184,19 @@ class McBuilder
184184
void fillMcCollision(T1 const& mcCol, T2& mcProducts)
185185
{
186186
float centrality = -1;
187+
float multiplicity = -1;
187188
if constexpr (modes::isFlagSet(system, modes::System::kPP)) {
188189
centrality = mcCol.centFT0M();
190+
multiplicity = mcCol.multMCNParticlesEta08();
189191
}
190192
if constexpr (modes::isFlagSet(system, modes::System::kPbPb)) {
191193
centrality = mcCol.centFT0C();
194+
multiplicity = mcCol.multMCNParticlesEta08();
192195
}
193196

194197
mcProducts.producedMcCollisions(
195198
mcCol.posZ(),
196-
mcCol.multMCNParticlesEta08(),
199+
multiplicity,
197200
centrality);
198201
mCollisionMap.emplace(mcCol.globalIndex(), mcProducts.producedMcCollisions.lastIndex());
199202
}

PWGCF/Femto/TableProducer/femtoProducer.cxx

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -166,7 +166,7 @@ struct FemtoProducer {
166166
(!doprocessTracksV0sRun3ppMc && !doprocessTracksKinksRun3ppMc &&
167167
!doprocessTracksV0sCascadesRun3ppMc && !doprocessTracksV0sKinksRun3ppMc &&
168168
!doprocessTracksRun3ppMc && !doprocessTracksRun3PbPbMc && !doprocessTracksV0sRun3PbPbMc &&
169-
!doprocessRun3ppMcOnly)) {
169+
!doprocessMcOnly)) {
170170
LOG(fatal) << "At least one mc table is enabled, but wrong process function is enabled. Breaking...";
171171
}
172172

@@ -531,18 +531,18 @@ struct FemtoProducer {
531531

532532
// process generator level only input (for MCGEN datasets)
533533
// do not preslice mcParticles tables for each collision, otherwise the finding of the partonic mother can fail
534-
void processRun3ppMcOnly(consumeddata::Run3PpMcGenCollisions const& mcCols, consumeddata::Run3McGenParticles const& mcParticles)
534+
void processMcOnly(o2::aod::McCollisions const& mcCols, o2::aod::McParticles const& mcParticles)
535535
{
536536
mcBuilder.reset(mcParticles);
537537
for (const auto& mcCol : mcCols) {
538-
mcBuilder.fillMcCollision<modes::System::kPP_Run3_MC>(mcCol, mcProducts);
538+
mcBuilder.fillMcCollision<modes::System::kMC>(mcCol, mcProducts);
539539
auto particlesThisCollision = mcParticles.sliceBy(perMcCollision, mcCol.globalIndex());
540540
for (const auto& mcParticle : particlesThisCollision) {
541-
mcBuilder.fillMcParticle<modes::System::kPP_Run3_MC>(mcParticle, mcParticles, mcCol, mcProducts);
541+
mcBuilder.fillMcParticle<modes::System::kMC>(mcParticle, mcParticles, mcCol, mcProducts);
542542
}
543543
}
544544
}
545-
PROCESS_SWITCH(FemtoProducer, processRun3ppMcOnly, "Provide generated particles", false);
545+
PROCESS_SWITCH(FemtoProducer, processMcOnly, "Provide generated particles", false);
546546
};
547547

548548
o2::framework::WorkflowSpec defineDataProcessing(o2::framework::ConfigContext const& cfgc)

0 commit comments

Comments
 (0)