|
15 | 15 | /// \author Abhi Modak (abhi.modak@cern.ch) |
16 | 16 | /// \since September 15, 2023 |
17 | 17 |
|
18 | | -#include <cmath> |
19 | | -#include <cstdlib> |
20 | | -#include <TPDGCode.h> |
21 | | -#include <vector> |
22 | | - |
| 18 | +#include "PWGLF/DataModel/LFStrangenessTables.h" |
| 19 | +#include "PWGMM/Mult/DataModel/Index.h" |
23 | 20 | #include "PWGMM/Mult/DataModel/bestCollisionTable.h" |
24 | | -#include "CCDB/BasicCCDBManager.h" |
25 | | -#include "Common/Core/trackUtilities.h" |
| 21 | + |
26 | 22 | #include "Common/CCDB/EventSelectionParams.h" |
27 | 23 | #include "Common/Core/TrackSelection.h" |
| 24 | +#include "Common/Core/trackUtilities.h" |
28 | 25 | #include "Common/DataModel/Centrality.h" |
29 | | -#include "Common/DataModel/Multiplicity.h" |
30 | 26 | #include "Common/DataModel/EventSelection.h" |
| 27 | +#include "Common/DataModel/Multiplicity.h" |
| 28 | +#include "Common/DataModel/PIDResponse.h" |
31 | 29 | #include "Common/DataModel/TrackSelectionTables.h" |
| 30 | + |
| 31 | +#include "CCDB/BasicCCDBManager.h" |
32 | 32 | #include "CommonConstants/MathConstants.h" |
33 | 33 | #include "Framework/ASoAHelpers.h" |
34 | 34 | #include "Framework/AnalysisDataModel.h" |
|
38 | 38 | #include "Framework/runDataProcessing.h" |
39 | 39 | #include "ReconstructionDataFormats/GlobalTrackID.h" |
40 | 40 | #include "ReconstructionDataFormats/Track.h" |
41 | | -#include "PWGMM/Mult/DataModel/Index.h" |
42 | | -#include "Common/DataModel/PIDResponse.h" |
43 | | -#include "PWGLF/DataModel/LFStrangenessTables.h" |
| 41 | + |
| 42 | +#include <TPDGCode.h> |
| 43 | + |
| 44 | +#include <cmath> |
| 45 | +#include <cstdlib> |
| 46 | +#include <vector> |
44 | 47 |
|
45 | 48 | using namespace o2; |
46 | 49 | using namespace o2::framework; |
@@ -164,6 +167,7 @@ struct HeavyionMultiplicity { |
164 | 167 | Configurable<bool> isApplyCentFT0M{"isApplyCentFT0M", false, "Centrality based on FT0A + FT0C"}; |
165 | 168 | Configurable<bool> isApplyCentNGlobal{"isApplyCentNGlobal", false, "Centrality based on global tracks"}; |
166 | 169 | Configurable<bool> isApplyCentMFT{"isApplyCentMFT", false, "Centrality based on MFT tracks"}; |
| 170 | + Configurable<bool> isApplySplitRecCol{"isApplySplitRecCol", false, "Split MC reco collisions"}; |
167 | 171 |
|
168 | 172 | void init(InitContext const&) |
169 | 173 | { |
@@ -434,6 +438,11 @@ struct HeavyionMultiplicity { |
434 | 438 |
|
435 | 439 | void processMonteCarlo(CollisionMCTrueTable::iterator const&, CollisionMCRecTable const& RecCols, TrackMCTrueTable const& GenParticles, FilTrackMCRecTable const& RecTracks) |
436 | 440 | { |
| 441 | + |
| 442 | + if (isApplySplitRecCol && (RecCols.size() == 0 || RecCols.size() > 1)) { |
| 443 | + return; |
| 444 | + } |
| 445 | + |
437 | 446 | for (const auto& RecCol : RecCols) { |
438 | 447 | if (!isEventSelected(RecCol)) { |
439 | 448 | continue; |
|
0 commit comments