1313// / \brief Task to generate a table of dNdEta, pt and PID dependent weigths for MC particles to reflect the measured particle abundances
1414// / \author Mario Krüger <mario.kruger@cern.ch>
1515
16+ #include < PWGLF/DataModel/particleCompositionCorrectionTable.h>
17+
1618#include " Common/Core/MetadataHelper.h"
1719#include " Common/Core/TrackSelection.h"
1820#include " Common/Core/TrackSelectionDefaults.h"
1921#include " Common/DataModel/Centrality.h"
2022#include " Common/DataModel/EventSelection.h"
2123#include " Common/DataModel/TrackSelectionTables.h"
24+ #include " Tools/ML/model.h"
2225
2326#include < Framework/AnalysisTask.h>
2427#include < Framework/HistogramRegistry.h>
2528#include < Framework/O2DatabasePDGPlugin.h>
2629#include < Framework/runDataProcessing.h>
2730#include < ReconstructionDataFormats/Track.h>
2831
29- #include < PWGLF/DataModel/particleCompositionCorrectionTable .h>
30- #include " Tools/ML/model.h "
32+ #include < TMCProcess .h>
33+ #include < TPDGCode.h >
3134
3235#include < unordered_set>
3336#include < vector>
34- #include < TPDGCode.h>
35- #include < TMCProcess.h>
3637
3738using namespace o2 ;
3839using namespace o2 ::framework;
@@ -116,10 +117,13 @@ void ParticleCompositionCorrection::process(aod::McCollisions::iterator const&,
116117 // determine dNdEta of the collision
117118 float dNdEta = 0 .f ;
118119 for (const auto & particle : particles) {
119- if (!particle.isPhysicalPrimary ()) continue ;
120+ if (!particle.isPhysicalPrimary ())
121+ continue ;
120122 auto pdgParticle = pdg->GetParticle (particle.pdgCode ());
121- if (!pdgParticle || pdgParticle->Charge () == 0 .) continue ;
122- if (std::abs (particle.eta ()) >= 0.5 ) continue ;
123+ if (!pdgParticle || pdgParticle->Charge () == 0 .)
124+ continue ;
125+ if (std::abs (particle.eta ()) >= 0.5 )
126+ continue ;
123127 ++dNdEta;
124128 }
125129
@@ -133,7 +137,8 @@ void ParticleCompositionCorrection::process(aod::McCollisions::iterator const&,
133137
134138 // find initial particle of secondaries from decays or interactions with material
135139 while (!particles.iteratorAt (refParticleID).producedByGenerator () && particles.iteratorAt (refParticleID).has_mothers ()) {
136- if (skipSec) break ;
140+ if (skipSec)
141+ break ;
137142 const auto & mother = particle.mothers_first_as <aod::McParticles>();
138143 auto motherID = mother.globalIndex () - particles.offset ();
139144 if (motherID == static_cast <long unsigned int >(refParticleID)) { // FIXME: this needs to be investigated!
@@ -171,7 +176,8 @@ void ParticleCompositionCorrection::process(aod::McCollisions::iterator const&,
171176 std::vector<std::vector<float >> input = {{dNdEta}, {pt}, {iterMapPID->second }};
172177 float fracData = particleFractionsData.evalModel (input)[0 ];
173178 float fracMC = particleFractionsMC.evalModel (input)[0 ];
174- if (fracMC) weight = fracData / fracMC;
179+ if (fracMC)
180+ weight = fracData / fracMC;
175181
176182 if (enableQAHistos && std::abs (particle.eta ()) < 0.8 ) {
177183 if (particle.isPhysicalPrimary ()) {
0 commit comments