Skip to content

Commit 99c17cd

Browse files
authored
[PWGCF] add centrality estimators (#14459)
1 parent c4ac606 commit 99c17cd

File tree

1 file changed

+15
-3
lines changed

1 file changed

+15
-3
lines changed

PWGCF/Flow/Tasks/flowTask.cxx

Lines changed: 15 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -64,7 +64,7 @@ struct FlowTask {
6464

6565
// Basic event&track selections
6666
O2_DEFINE_CONFIGURABLE(cfgCutVertex, float, 10.0f, "Accepted z-vertex range")
67-
O2_DEFINE_CONFIGURABLE(cfgCentEstimator, int, 0, "0:FT0C; 1:FT0CVariant1; 2:FT0M; 3:FT0A")
67+
O2_DEFINE_CONFIGURABLE(cfgCentEstimator, int, 0, "0:FT0C; 1:FT0CVariant1; 2:FT0M; 3:FV0A; 4:NTPV; 5:NGlobal; 6:MFT")
6868
O2_DEFINE_CONFIGURABLE(cfgCentFT0CMin, float, 0.0f, "Minimum centrality (FT0C) to cut events in filter")
6969
O2_DEFINE_CONFIGURABLE(cfgCentFT0CMax, float, 100.0f, "Maximum centrality (FT0C) to cut events in filter")
7070
O2_DEFINE_CONFIGURABLE(cfgCutPtPOIMin, float, 0.2f, "Minimal pT for poi tracks")
@@ -185,7 +185,7 @@ struct FlowTask {
185185

186186
Filter collisionFilter = (nabs(aod::collision::posZ) < cfgCutVertex) && (aod::cent::centFT0C > cfgCentFT0CMin) && (aod::cent::centFT0C < cfgCentFT0CMax);
187187
Filter trackFilter = ((requireGlobalTrackInFilter()) || (aod::track::isGlobalTrackSDD == (uint8_t) true)) && (nabs(aod::track::eta) < cfgCutEta) && (aod::track::pt > cfgCutPtMin) && (aod::track::pt < cfgCutPtMax) && (aod::track::tpcChi2NCl < cfgCutChi2prTPCcls) && (nabs(aod::track::dcaZ) < cfgCutDCAz);
188-
using FilteredCollisions = soa::Filtered<soa::Join<aod::Collisions, aod::EvSels, aod::CentFT0Cs, aod::CentFT0CVariant1s, aod::CentFT0Ms, aod::CentFV0As, aod::Mults>>;
188+
using FilteredCollisions = soa::Filtered<soa::Join<aod::Collisions, aod::EvSels, aod::CentFT0Cs, aod::CentFT0CVariant1s, aod::CentFT0Ms, aod::CentFV0As, aod::CentNTPVs, aod::CentNGlobals, aod::CentMFTs, aod::Mults>>;
189189
using FilteredTracks = soa::Filtered<soa::Join<aod::Tracks, aod::TrackSelection, aod::TracksExtra, aod::TracksDCA>>;
190190
// Filter for MCcollisions
191191
Filter mccollisionFilter = nabs(aod::mccollision::posZ) < cfgCutVertex;
@@ -194,7 +194,7 @@ struct FlowTask {
194194
Filter particleFilter = (nabs(aod::mcparticle::eta) < cfgCutEta) && (aod::mcparticle::pt > cfgCutPtMin) && (aod::mcparticle::pt < cfgCutPtMax);
195195
using FilteredMcParticles = soa::Filtered<aod::McParticles>;
196196

197-
using FilteredSmallGroupMcCollisions = soa::SmallGroups<soa::Join<aod::McCollisionLabels, aod::Collisions, aod::EvSel, aod::CentFT0Cs, aod::CentFT0CVariant1s, aod::CentFT0Ms, aod::CentFV0As, aod::Mults>>;
197+
using FilteredSmallGroupMcCollisions = soa::SmallGroups<soa::Join<aod::McCollisionLabels, aod::Collisions, aod::EvSel, aod::CentFT0Cs, aod::CentFT0CVariant1s, aod::CentFT0Ms, aod::CentFV0As, aod::CentNTPVs, aod::CentNGlobals, aod::CentMFTs, aod::Mults>>;
198198

199199
// Corrections
200200
TH1D* mEfficiency = nullptr;
@@ -230,6 +230,9 @@ struct FlowTask {
230230
kCentFT0CVariant1,
231231
kCentFT0M,
232232
kCentFV0A,
233+
kCentNTPV,
234+
kCentNGlobal,
235+
kCentMFT,
233236
// Count the total number of enum
234237
kCount_CentEstimators
235238
};
@@ -992,6 +995,15 @@ struct FlowTask {
992995
case kCentFV0A:
993996
cent = collision.centFV0A();
994997
break;
998+
case kCentNTPV:
999+
cent = collision.centNTPV();
1000+
break;
1001+
case kCentNGlobal:
1002+
cent = collision.centNGlobal();
1003+
break;
1004+
case kCentMFT:
1005+
cent = collision.centMFT();
1006+
break;
9951007
default:
9961008
cent = collision.centFT0C();
9971009
}

0 commit comments

Comments
 (0)