Skip to content

Commit 5eca288

Browse files
authored
Add configurable for split MC reco collisions
1 parent 0f3b1ea commit 5eca288

1 file changed

Lines changed: 20 additions & 11 deletions

File tree

PWGLF/Tasks/GlobalEventProperties/heavyionMultiplicity.cxx

Lines changed: 20 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -15,20 +15,20 @@
1515
/// \author Abhi Modak (abhi.modak@cern.ch)
1616
/// \since September 15, 2023
1717

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"
2320
#include "PWGMM/Mult/DataModel/bestCollisionTable.h"
24-
#include "CCDB/BasicCCDBManager.h"
25-
#include "Common/Core/trackUtilities.h"
21+
2622
#include "Common/CCDB/EventSelectionParams.h"
2723
#include "Common/Core/TrackSelection.h"
24+
#include "Common/Core/trackUtilities.h"
2825
#include "Common/DataModel/Centrality.h"
29-
#include "Common/DataModel/Multiplicity.h"
3026
#include "Common/DataModel/EventSelection.h"
27+
#include "Common/DataModel/Multiplicity.h"
28+
#include "Common/DataModel/PIDResponse.h"
3129
#include "Common/DataModel/TrackSelectionTables.h"
30+
31+
#include "CCDB/BasicCCDBManager.h"
3232
#include "CommonConstants/MathConstants.h"
3333
#include "Framework/ASoAHelpers.h"
3434
#include "Framework/AnalysisDataModel.h"
@@ -38,9 +38,12 @@
3838
#include "Framework/runDataProcessing.h"
3939
#include "ReconstructionDataFormats/GlobalTrackID.h"
4040
#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>
4447

4548
using namespace o2;
4649
using namespace o2::framework;
@@ -164,6 +167,7 @@ struct HeavyionMultiplicity {
164167
Configurable<bool> isApplyCentFT0M{"isApplyCentFT0M", false, "Centrality based on FT0A + FT0C"};
165168
Configurable<bool> isApplyCentNGlobal{"isApplyCentNGlobal", false, "Centrality based on global tracks"};
166169
Configurable<bool> isApplyCentMFT{"isApplyCentMFT", false, "Centrality based on MFT tracks"};
170+
Configurable<bool> isApplySplitRecCol{"isApplySplitRecCol", false, "Split MC reco collisions"};
167171

168172
void init(InitContext const&)
169173
{
@@ -434,6 +438,11 @@ struct HeavyionMultiplicity {
434438

435439
void processMonteCarlo(CollisionMCTrueTable::iterator const&, CollisionMCRecTable const& RecCols, TrackMCTrueTable const& GenParticles, FilTrackMCRecTable const& RecTracks)
436440
{
441+
442+
if (isApplySplitRecCol && (RecCols.size() == 0 || RecCols.size() > 1)) {
443+
return;
444+
}
445+
437446
for (const auto& RecCol : RecCols) {
438447
if (!isEventSelected(RecCol)) {
439448
continue;

0 commit comments

Comments
 (0)