Skip to content

Commit 0a1577a

Browse files
MaximVirtaMaxim Virta
andauthored
[PWGCF] Added safeguards for pt fractions (#17359)
Co-authored-by: Maxim Virta <maximus@Maxims-MacBook-Pro.local>
1 parent 055feae commit 0a1577a

1 file changed

Lines changed: 6 additions & 4 deletions

File tree

PWGCF/GenericFramework/Tasks/flowGfwV02.cxx

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -67,7 +67,6 @@
6767
#include <memory>
6868
#include <string>
6969
#include <string_view>
70-
#include <utility>
7170
#include <vector>
7271

7372
using namespace o2;
@@ -886,6 +885,9 @@ struct FlowGfwV02 {
886885
int bootstrap = fRndm->Integer(gfwMemberCache.nBootstrap);
887886
for (int pid = 0; pid < PidTotal; pid++) {
888887
int normIndex = (cfgNormalizeByCharged) ? PidCharged : pid;
888+
if (!(pidStates.hPtForward[normIndex]->Integral() > 0) || !(pidStates.hPtBackward[normIndex]->Integral() > 0)) {
889+
continue; // Forward or backward pT distribution is not defined
890+
}
889891
for (int i = 1; i <= fSecondAxis->GetNbins(); i++) {
890892
ptFractionForward = pidStates.hPtForward[pid]->GetBinContent(i) / pidStates.hPtForward[normIndex]->Integral();
891893
ptFractionBackward = pidStates.hPtBackward[pid]->GetBinContent(i) / pidStates.hPtBackward[normIndex]->Integral();
@@ -1268,7 +1270,7 @@ struct FlowGfwV02 {
12681270
registry.fill(HIST("eventQA/eventSel"), kSel8);
12691271
registry.fill(HIST("eventQA/eventSel"), kOccupancy); // Add occupancy selection later
12701272

1271-
const XAxis xaxis{.centrality = getCentrality(collision), .multiplicity = static_cast<int64_t>(tracks.size())};
1273+
const XAxis xaxis{.centrality = getCentrality(collision), .multiplicity = tracks.size()};
12721274
if (cfgFillQA) {
12731275
fillEventQA<kBefore>(collision, xaxis);
12741276
registry.fill(HIST("eventQA/before/centrality"), xaxis.centrality);
@@ -1293,7 +1295,7 @@ struct FlowGfwV02 {
12931295
LOGF(info, "run = %d", run);
12941296
}
12951297
loadCorrections(run);
1296-
const XAxis xaxis{.centrality = collision.multiplicity(), .multiplicity = static_cast<int64_t>(tracks.size())};
1298+
const XAxis xaxis{.centrality = collision.multiplicity(), .multiplicity = tracks.size()};
12971299

12981300
registry.fill(HIST("eventQA/after/centrality"), xaxis.centrality);
12991301
registry.fill(HIST("eventQA/after/multiplicity"), xaxis.multiplicity);
@@ -1308,7 +1310,7 @@ struct FlowGfwV02 {
13081310
lastRun = run;
13091311
LOGF(info, "run = %d", run);
13101312
}
1311-
const XAxis xaxis{.centrality = collision.multiplicity(), .multiplicity = static_cast<int64_t>(tracks.size())};
1313+
const XAxis xaxis{.centrality = collision.multiplicity(), .multiplicity = tracks.size()};
13121314
registry.fill(HIST("eventQA/after/centrality"), xaxis.centrality);
13131315
registry.fill(HIST("eventQA/after/multiplicity"), xaxis.multiplicity);
13141316
// processCollision<kReco>(collision, tracks, xaxis, run);

0 commit comments

Comments
 (0)