Skip to content

[PWGLF] Add Model Prediction Process in H-SCorrelation #16045

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
romainschotter merged 1 commit into from
Apr 30, 2026
+207 −6
Merged

[PWGLF] Add Model Prediction Process in H-SCorrelation #16045

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
213 changes: 207 additions & 6 deletions PWGLF/Tasks/Strangeness/hStrangeCorrelation.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,8 @@

#include "PWGLF/DataModel/LFHStrangeCorrelationTables.h"
#include "PWGLF/DataModel/LFStrangenessTables.h"
#include "PWGLF/DataModel/mcCentrality.h"
#include "PWGLF/Utils/inelGt.h"

#include "Common/CCDB/EventSelectionParams.h"
#include "Common/Core/RecoDecay.h"
Expand Down Expand Up @@ -87,6 +89,7 @@
Service<o2::ccdb::BasicCCDBManager> ccdb;

Service<o2::framework::O2DatabasePDG> pdgDB;
o2::pwglf::ParticleCounter<o2::framework::O2DatabasePDG> mCounter;

HistogramRegistry histos{"Histos", {}, OutputObjHandlingPolicy::AnalysisObject};

Expand Down Expand Up @@ -121,6 +124,8 @@
Configurable<bool> doMCassociation{"doMCassociation", false, "fill everything only for MC associated"};
Configurable<bool> doTriggPhysicalPrimary{"doTriggPhysicalPrimary", false, "require physical primary for trigger particles"};
Configurable<bool> applyNewMCSelection{"applyNewMCSelection", false, "apply new MC Generated selection"};
Configurable<bool> doSeparateFT0Prediction{"doSeparateFT0Prediction", false, "separate FT0M to FT0A and FT0C in prediction process"};
Configurable<bool> useCentralityinPrediction{"useCentralityinPrediction", false, "if true, use centrality instead of multiplisity"};
} masterConfigurations;

// master analysis switches
Expand Down Expand Up @@ -154,8 +159,9 @@
ConfigurableAxis axisPtAssoc{"axisPtAssoc", {VARIABLE_WIDTH, 0.5, 1.0, 1.5, 2.0, 3.0, 4.0, 6.0, 10.0}, "pt associated axis for histograms"};
ConfigurableAxis axisPtTrigger{"axisPtTrigger", {VARIABLE_WIDTH, 0.0, 1.0, 2.0, 3.0, 100}, "pt associated axis for histograms"};
ConfigurableAxis axisPtQA{"axisPtQA", {VARIABLE_WIDTH, 0.0f, 0.1f, 0.2f, 0.3f, 0.4f, 0.5f, 0.6f, 0.7f, 0.8f, 0.9f, 1.0f, 1.1f, 1.2f, 1.3f, 1.4f, 1.5f, 1.6f, 1.7f, 1.8f, 1.9f, 2.0f, 2.2f, 2.4f, 2.6f, 2.8f, 3.0f, 3.2f, 3.4f, 3.6f, 3.8f, 4.0f, 4.4f, 4.8f, 5.2f, 5.6f, 6.0f, 6.5f, 7.0f, 7.5f, 8.0f, 9.0f, 10.0f, 11.0f, 12.0f, 13.0f, 14.0f, 15.0f, 17.0f, 19.0f, 21.0f, 23.0f, 25.0f, 30.0f, 35.0f, 40.0f, 50.0f}, "pt axis for QA histograms"};
ConfigurableAxis axisMultCount{"axisMultCount", {VARIABLE_WIDTH, 0, 200, 400, 600, 800, 1000, 1400, 1800, 2300, 2800, 3300, 4000, 5000, 6000}, "Mixing bins - multiplicity"};
ConfigurableAxis axisMassNSigma{"axisMassNSigma", {40, -2, 2}, "Axis for mass Nsigma"};
ConfigurableAxis axisMultiplicity{"axisMultiplicity", {VARIABLE_WIDTH, 0, 20, 40, 60, 100, 120, 140, 160, 180, 200, 220, 240, 260, 280, 300}, "Binning of the Multiplicity axis in model prediction process"};

} axesConfigurations;

// for topo var QA
Expand Down Expand Up @@ -253,7 +259,7 @@
} cascadeSelections;

struct : ConfigurableGroup {
std::string prefix = "`";
std::string prefix = "checks";
// cascade selections
// more cascade selections in PbPb
// Configurable<float> bachBaryonCosPA{"bachBaryonCosPA", 0.9999, "Bachelor baryon CosPA"};
Expand Down Expand Up @@ -992,9 +998,9 @@
int binnumb = binMult * nBinsVtxZ + binVtxZ;
int hastirgorassoc = masterConfigurations.collisionHasTriggOrAssoc;
if ((hastirgorassoc == 1 && currentCollision.trigParticles.empty()) ||
(hastirgorassoc == 2 && currentCollision.assocParticles.empty()) ||

Check failure on line 1001 in PWGLF/Tasks/Strangeness/hStrangeCorrelation.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[magic-number] 

Avoid magic numbers in expressions. Assign the value to a clearly named variable or constant.
(hastirgorassoc == 3 && currentCollision.trigParticles.empty() && currentCollision.assocParticles.empty()) ||

Check failure on line 1002 in PWGLF/Tasks/Strangeness/hStrangeCorrelation.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[magic-number] 

Avoid magic numbers in expressions. Assign the value to a clearly named variable or constant.
(hastirgorassoc == 4 && (currentCollision.trigParticles.empty() || currentCollision.assocParticles.empty())))

Check failure on line 1003 in PWGLF/Tasks/Strangeness/hStrangeCorrelation.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[magic-number] 

Avoid magic numbers in expressions. Assign the value to a clearly named variable or constant.
return;
for (const auto& collision : validCollisions[binnumb]) {
BinningTypePP colBinning{{axesConfigurations.axisVtxZ, axesConfigurations.axisMult}, true};
Expand Down Expand Up @@ -1027,7 +1033,7 @@
if (Index == assoc.type && assoc.region == 1) {
fillCorrelationHistogram(histos.get<THn>(HIST("mixedEvent/Signal/") + HIST(V0names[Index])), binFillThn, 1, efficiencyTrigg * efficiencyAssoc, totalEffUncert, 1., 0.);
}
if (Index == assoc.type && assoc.region == 2) {

Check failure on line 1036 in PWGLF/Tasks/Strangeness/hStrangeCorrelation.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[magic-number] 

Avoid magic numbers in expressions. Assign the value to a clearly named variable or constant.
fillCorrelationHistogram(histos.get<THn>(HIST("mixedEvent/RightBg/") + HIST(V0names[Index])), binFillThn, 1, efficiencyTrigg * efficiencyAssoc, totalEffUncert, 1., 0.);
}
});
Expand Down Expand Up @@ -1295,9 +1301,9 @@
int binnumb = binMult * nBinsVtxZ + binVtxZ;
int hastirgorassoc = masterConfigurations.collisionHasTriggOrAssoc;
if ((hastirgorassoc == 1 && currentCollision.trigParticles.empty()) ||
(hastirgorassoc == 2 && currentCollision.assocParticles.empty()) ||

Check failure on line 1304 in PWGLF/Tasks/Strangeness/hStrangeCorrelation.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[magic-number] 

Avoid magic numbers in expressions. Assign the value to a clearly named variable or constant.
(hastirgorassoc == 3 && currentCollision.trigParticles.empty() && currentCollision.assocParticles.empty()) ||

Check failure on line 1305 in PWGLF/Tasks/Strangeness/hStrangeCorrelation.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[magic-number] 

Avoid magic numbers in expressions. Assign the value to a clearly named variable or constant.
(hastirgorassoc == 4 && (currentCollision.trigParticles.empty() || currentCollision.assocParticles.empty())))

Check failure on line 1306 in PWGLF/Tasks/Strangeness/hStrangeCorrelation.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[magic-number] 

Avoid magic numbers in expressions. Assign the value to a clearly named variable or constant.
return;
for (const auto& collision : validCollisions[binnumb]) {
BinningTypePP colBinning{{axesConfigurations.axisVtxZ, axesConfigurations.axisMult}, true};
Expand Down Expand Up @@ -1325,7 +1331,7 @@
if (Index == assoc.type && assoc.region == 1) {
fillCorrelationHistogram(histos.get<THn>(HIST("mixedEvent/Signal/") + HIST(Cascadenames[Index])), binFillThn, 1, efficiencyTrigg * efficiencyAssoc, totalEffUncert, 1., 0.);
}
if (Index == assoc.type && assoc.region == 2) {

Check failure on line 1334 in PWGLF/Tasks/Strangeness/hStrangeCorrelation.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[magic-number] 

Avoid magic numbers in expressions. Assign the value to a clearly named variable or constant.
fillCorrelationHistogram(histos.get<THn>(HIST("mixedEvent/RightBg/") + HIST(Cascadenames[Index])), binFillThn, 1, efficiencyTrigg * efficiencyAssoc, totalEffUncert, 1., 0.);
}
});
Expand Down Expand Up @@ -1568,6 +1574,7 @@
const AxisSpec preAxisMult{axesConfigurations.axisMult, "mult percentile"};
const AxisSpec axisPtLambda{axesConfigurations.axisPtAssoc, "#it{p}_{T}^{#Lambda} (GeV/c)"};
const AxisSpec axisPtCascade{axesConfigurations.axisPtAssoc, "#it{p}_{T}^{Mother} (GeV/c)"};
const AxisSpec preAxisMultiplicity{axesConfigurations.axisMultiplicity, "multiplicity"};

// store the original axes in specific TH1Cs for completeness
histos.add("axes/hDeltaPhiAxis", "", kTH1C, {preAxisDeltaPhi});
Expand All @@ -1576,34 +1583,39 @@
histos.add("axes/hPtTriggerAxis", "", kTH1C, {preAxisPtTrigger});
histos.add("axes/hVertexZAxis", "", kTH1C, {preAxisVtxZ});
histos.add("axes/hMultAxis", "", kTH1C, {preAxisMult});
histos.add("axes/hMultiplicityAxis", "", kTH1C, {preAxisMultiplicity});

std::vector<double> edgesDeltaPhiOrig = preAxisDeltaPhi.binEdges;
std::vector<double> edgesDeltaEtaOrig = preAxisDeltaEta.binEdges;
std::vector<double> edgesPtAssocOrig = preAxisPtAssoc.binEdges;
std::vector<double> edgesPtTriggerOrig = preAxisPtTrigger.binEdges;
std::vector<double> edgesVtxZOrig = preAxisVtxZ.binEdges;
std::vector<double> edgesMultOrig = preAxisMult.binEdges;
std::vector<double> edgesMultiplicityOrig = preAxisMultiplicity.binEdges;

std::vector<float> rangesDeltaPhi = {static_cast<float>(edgesDeltaPhiOrig[0]), static_cast<float>(edgesDeltaPhiOrig[edgesDeltaPhiOrig.size() - 1])};
std::vector<float> rangesDeltaEta = {static_cast<float>(edgesDeltaEtaOrig[0]), static_cast<float>(edgesDeltaEtaOrig[edgesDeltaEtaOrig.size() - 1])};
std::vector<float> rangesPtAssoc = {static_cast<float>(edgesPtAssocOrig[0]), static_cast<float>(edgesPtAssocOrig[edgesPtAssocOrig.size() - 1])};
std::vector<float> rangesPtTrigger = {static_cast<float>(edgesPtTriggerOrig[0]), static_cast<float>(edgesPtTriggerOrig[edgesPtTriggerOrig.size() - 1])};
std::vector<float> rangesVtxZ = {static_cast<float>(edgesVtxZOrig[0]), static_cast<float>(edgesVtxZOrig[edgesVtxZOrig.size() - 1])};
std::vector<float> rangesMult = {static_cast<float>(edgesMultOrig[0]), static_cast<float>(edgesMultOrig[edgesMultOrig.size() - 1])};
std::vector<float> rangesMultiplicity = {static_cast<float>(edgesMultiplicityOrig[0]), static_cast<float>(edgesMultiplicityOrig[edgesMultiplicityOrig.size() - 1])};

axisRanges.emplace_back(rangesDeltaPhi);
axisRanges.emplace_back(rangesDeltaEta);
axisRanges.emplace_back(rangesPtAssoc);
axisRanges.emplace_back(rangesPtTrigger);
axisRanges.emplace_back(rangesVtxZ);
axisRanges.emplace_back(rangesMult);
axisRanges.emplace_back(rangesMultiplicity);

std::vector<double> edgesDeltaPhi;
std::vector<double> edgesDeltaEta;
std::vector<double> edgesPtAssoc;
std::vector<double> edgesPtTrigger;
std::vector<double> edgesVtxZ;
std::vector<double> edgesMult;
std::vector<double> edgesMultiplicity;

// v--- skipUnderOverflowInTHn ---v
//
Expand Down Expand Up @@ -1688,25 +1700,39 @@
for (int i = offset; i < preAxisMult.nBins.value() + 1 - offset; i++)
edgesMult.emplace_back(min + static_cast<double>(i) * delta);
}
// ===] multiplicity count [===
if (!preAxisMultiplicity.nBins.has_value()) {
// variable binning, use bins provided
for (int i = offset; i < static_cast<int>(edgesMultiplicityOrig.size()) - offset; i++)
edgesMultiplicity.emplace_back(edgesMultiplicityOrig[i]);
} else {
// fixed binning, generate the bin edges on-the-spot
double min = edgesMultiplicityOrig[0];
double delta = (edgesMultiplicityOrig[1] - edgesMultiplicityOrig[0]) / preAxisMultiplicity.nBins.value();
for (int i = offset; i < preAxisMultiplicity.nBins.value() + 1 - offset; i++)
edgesMultiplicity.emplace_back(min + static_cast<double>(i) * delta);
}

LOGF(info, "Initialized THnF axis delta-phi with %i bins.", edgesDeltaPhi.size() - 1);
LOGF(info, "Initialized THnF axis delta-eta with %i bins.", edgesDeltaEta.size() - 1);
LOGF(info, "Initialized THnF axis pTassoc with %i bins.", edgesPtAssoc.size() - 1);
LOGF(info, "Initialized THnF axis pTtrigger with %i bins.", edgesPtTrigger.size() - 1);
LOGF(info, "Initialized THnF axis vertex-Z with %i bins.", edgesVtxZ.size() - 1);
LOGF(info, "Initialized THnF axis multiplicity with %i bins.", edgesMult.size() - 1);
LOGF(info, "Initialized THnF axis mult cent with %i bins.", edgesMult.size() - 1);
LOGF(info, "Initialized THnF axis multiplicity with %i bins.", edgesMultiplicity.size() - 1);

const AxisSpec axisDeltaPhiNDim{edgesDeltaPhi, "#Delta#varphi"};
const AxisSpec axisDeltaEtaNDim{edgesDeltaEta, "#Delta#eta"};
const AxisSpec axisPtAssocNDim{edgesPtAssoc, "#it{p}_{T}^{assoc} (GeV/c)"};
const AxisSpec axisPtTriggerNDim{edgesPtTrigger, "#it{p}_{T}^{trigger} (GeV/c)"};
const AxisSpec axisVtxZNDim{edgesVtxZ, "vertex Z (cm)"};
const AxisSpec axisMultNDim{edgesMult, "mult percentile"};
const AxisSpec axisMultiplicityNDim{edgesMultiplicity, "Multiplicity"};

if (doprocessMixedEventHV0sInBuffer || doprocessMixedEventHCascadesInBuffer) {
validCollisions.resize(histos.get<TH1>(HIST("axes/hMultAxis"))->GetNbinsX() * histos.get<TH1>(HIST("axes/hVertexZAxis"))->GetNbinsX());
for (std::vector<ValidCollision>& inner_vec : validCollisions) {
inner_vec.reserve(masterConfigurations.mixingParameter);
for (size_t i = 0; i < validCollisions.size(); ++i) {
validCollisions[i].reserve(masterConfigurations.mixingParameter);
}
}
if (!masterConfigurations.doPPAnalysis) {
Expand Down Expand Up @@ -1882,7 +1908,31 @@
histos.add("GeneratedWithPV/hAntiLambdaFromXiZero", "", kTH2F, {axesConfigurations.axisPtQA, axesConfigurations.axisEta});
histos.add("GeneratedWithPV/hAntiLambdaFromXiPlus", "", kTH2F, {axesConfigurations.axisPtQA, axesConfigurations.axisEta});
}

if (doprocessPrediction) {
mCounter.mPdgDatabase = pdgDB.service;
mCounter.mSelectPrimaries = doAssocPhysicalPrimary.value;
histos.add("Prediction/hEventSelection", "hEventSelection", kTH1F, {{3, 0, 3}});
TString eventSelLabel[] = {"Read", "INELgt0", "|Z|<10"};
for (int i = 1; i <= histos.get<TH1>(HIST("Prediction/hEventSelection"))->GetNbinsX(); i++) {
histos.get<TH1>(HIST("Prediction/hEventSelection"))->GetXaxis()->SetBinLabel(i, eventSelLabel[i - 1]);
}
histos.add("Prediction/hTrigger", "Trigger Tracks", kTH3F, {axesConfigurations.axisPtQA, axesConfigurations.axisEta, axesConfigurations.axisPhi});
for (int i = 0; i < AssocParticleTypes; i++) {
if (TESTBIT(doCorrelation, i))
histos.add(fmt::format("Prediction/h{}", Particlenames[i]).c_str(), "", kTH3F, {axesConfigurations.axisPtQA, axesConfigurations.axisEta, axesConfigurations.axisPhi});
if (masterConfigurations.useCentralityinPrediction) {
if (TESTBIT(doCorrelation, i))
histos.add(fmt::format("Prediction/sameEvent/{}", Particlenames[i]).c_str(), "", kTHnF, {axisDeltaPhiNDim, axisDeltaEtaNDim, axisPtAssocNDim, axisPtTriggerNDim, axisVtxZNDim, axisMultNDim});
} else {
if (TESTBIT(doCorrelation, i))
histos.add(fmt::format("Prediction/sameEvent/{}", Particlenames[i]).c_str(), "", kTHnF, {axisDeltaPhiNDim, axisDeltaEtaNDim, axisPtAssocNDim, axisPtTriggerNDim, axisVtxZNDim, axisMultiplicityNDim});
}
}
if (masterConfigurations.doSeparateFT0Prediction) {
histos.addClone("Prediction/sameEvent/", "Prediction/sameEventFT0A/");
histos.addClone("Prediction/sameEvent/", "Prediction/sameEventFT0C/");
}
}
// visual inspection of sizes
histos.print();

Expand Down Expand Up @@ -3226,6 +3276,156 @@
if (masterConfigurations.doFullCorrelationStudy)
fillCorrelationsCascade(triggerTracks, associatedCascades, true, true, collision.posX(), collision.posY(), collision.posZ(), cent, bField);
}
void processPrediction(soa::Join<aod::McCollisions, aod::McCentFT0Ms, aod::McCentFT0Cs, aod::McCentFT0As>::iterator const& mcCollision, aod::McParticles const& mcParticles)
{
std::vector<uint32_t> triggerIndices;
std::vector<std::vector<uint32_t>> associatedIndices;
std::vector<uint32_t> assocHadronIndices;
std::vector<uint32_t> piIndices;
std::vector<uint32_t> k0ShortIndices;
std::vector<uint32_t> lambdaIndices;
std::vector<uint32_t> antiLambdaIndices;
std::vector<uint32_t> xiMinusIndices;
std::vector<uint32_t> xiPlusIndices;
std::vector<uint32_t> omegaMinusIndices;
std::vector<uint32_t> omegaPlusIndices;
float centMultFT0M = -1;
float centMultFT0A = -1;
float centMultFT0C = -1;
float multFT0M = -1;
float multFT0A = -1;
float multFT0C = -1;
histos.fill(HIST("Prediction/hEventSelection"), 0.5);
if (masterConfigurations.selectINELgtZERO && !o2::pwglf::isINELgt0mc(mcParticles, pdgDB)) {
return;
}
histos.fill(HIST("Prediction/hEventSelection"), 1.5);
if (std::abs(mcCollision.posZ()) > masterConfigurations.zVertexCut) {
return;
}
histos.fill(HIST("Prediction/hEventSelection"), 2.5);
if (masterConfigurations.useCentralityinPrediction) {
centMultFT0M = mcCollision.centFT0M();
centMultFT0A = mcCollision.centFT0A();
centMultFT0C = mcCollision.centFT0C();
} else {
multFT0M = mCounter.countFT0A(mcParticles) + mCounter.countFT0C(mcParticles);
multFT0A = mCounter.countFT0A(mcParticles);
multFT0C = mCounter.countFT0C(mcParticles);
}
int iteratorNum = -1;
for (auto const& mcParticle : mcParticles) {
iteratorNum = iteratorNum + 1;
double geta = mcParticle.eta();
double gpt = mcParticle.pt();
double gphi = mcParticle.phi();
if (std::abs(geta) > etaSel) {
continue;
}
if (std::abs(mcParticle.pdgCode()) == PDG_t::kPiPlus || std::abs(mcParticle.pdgCode()) == PDG_t::kKPlus || std::abs(mcParticle.pdgCode()) == PDG_t::kProton || std::abs(mcParticle.pdgCode()) == PDG_t::kElectron || std::abs(mcParticle.pdgCode()) == PDG_t::kMuonMinus) {
if (!masterConfigurations.doTriggPhysicalPrimary || mcParticle.isPhysicalPrimary()) {
triggerIndices.emplace_back(iteratorNum);
histos.fill(HIST("Prediction/hTrigger"), gpt, geta, gphi);
}
if (masterConfigurations.doCorrelationHadron) {
if (!doAssocPhysicalPrimary || mcParticle.isPhysicalPrimary()) {
assocHadronIndices.emplace_back(iteratorNum);
histos.fill(HIST("Prediction/hHadron"), gpt, geta, gphi);
}
}
}
if (!doAssocPhysicalPrimary || mcParticle.isPhysicalPrimary()) {
if (std::abs(mcParticle.pdgCode()) == PDG_t::kPiPlus && masterConfigurations.doCorrelationPion) {
piIndices.emplace_back(iteratorNum);
histos.fill(HIST("Prediction/hPion"), gpt, geta, gphi);
}
if (mcParticle.pdgCode() == PDG_t::kK0Short && masterConfigurations.doCorrelationK0Short) {
k0ShortIndices.emplace_back(iteratorNum);
histos.fill(HIST("Prediction/hK0Short"), gpt, geta, gphi);
}
if (mcParticle.pdgCode() == PDG_t::kLambda0 && masterConfigurations.doCorrelationLambda) {
lambdaIndices.emplace_back(iteratorNum);
histos.fill(HIST("Prediction/hLambda"), gpt, geta, gphi);
}
if (mcParticle.pdgCode() == PDG_t::kLambda0Bar && masterConfigurations.doCorrelationAntiLambda) {
antiLambdaIndices.emplace_back(iteratorNum);
histos.fill(HIST("Prediction/hAntiLambda"), gpt, geta, gphi);
}
if (mcParticle.pdgCode() == PDG_t::kXiMinus && masterConfigurations.doCorrelationXiMinus) {
xiMinusIndices.emplace_back(iteratorNum);
histos.fill(HIST("Prediction/hXiMinus"), gpt, geta, gphi);
}
if (mcParticle.pdgCode() == PDG_t::kXiPlusBar && masterConfigurations.doCorrelationXiPlus) {
xiPlusIndices.emplace_back(iteratorNum);
histos.fill(HIST("Prediction/hXiPlus"), gpt, geta, gphi);
}
if (mcParticle.pdgCode() == PDG_t::kOmegaMinus && masterConfigurations.doCorrelationOmegaMinus) {
omegaMinusIndices.emplace_back(iteratorNum);
histos.fill(HIST("Prediction/hOmegaMinus"), gpt, geta, gphi);
}
if (mcParticle.pdgCode() == PDG_t::kOmegaPlusBar && masterConfigurations.doCorrelationOmegaPlus) {
omegaPlusIndices.emplace_back(iteratorNum);
histos.fill(HIST("Prediction/hOmegaPlus"), gpt, geta, gphi);
}
}
}

associatedIndices.emplace_back(k0ShortIndices);
associatedIndices.emplace_back(lambdaIndices);
associatedIndices.emplace_back(antiLambdaIndices);
associatedIndices.emplace_back(xiMinusIndices);
associatedIndices.emplace_back(xiPlusIndices);
associatedIndices.emplace_back(omegaMinusIndices);
associatedIndices.emplace_back(omegaPlusIndices);
associatedIndices.emplace_back(piIndices);
associatedIndices.emplace_back(assocHadronIndices);
for (std::size_t iTrigger = 0; iTrigger < triggerIndices.size(); iTrigger++) {
auto triggerParticle = mcParticles.iteratorAt(triggerIndices[iTrigger]);
// check range of trigger particle
if (triggerParticle.pt() > axisRanges[3][1] || triggerParticle.pt() < axisRanges[3][0]) {
continue;
}
double getatrigger = triggerParticle.eta();
double gphitrigger = triggerParticle.phi();
double pttrigger = triggerParticle.pt();
auto const& mother = triggerParticle.mothers_first_as<aod::McParticles>();
auto globalIndex = mother.globalIndex();
static_for<0, 8>([&](auto i) { // associated loop
constexpr int Index = i.value;
for (std::size_t iassoc = 0; iassoc < associatedIndices[Index].size(); iassoc++) {
auto assocParticle = mcParticles.iteratorAt(associatedIndices[Index][iassoc]);
if (triggerIndices[iTrigger] != associatedIndices[Index][iassoc] && globalIndex != assocParticle.globalIndex()) { // avoid self
double getaassoc = assocParticle.eta();
double gphiassoc = assocParticle.phi();
double ptassoc = assocParticle.pt();
double deltaphi = computeDeltaPhi(gphitrigger, gphiassoc);
double deltaeta = getatrigger - getaassoc;

// skip if basic ranges not met
if (deltaphi < axisRanges[0][0] || deltaphi > axisRanges[0][1])
continue;
if (deltaeta < axisRanges[1][0] || deltaeta > axisRanges[1][1])
continue;
if (ptassoc < axisRanges[2][0] || ptassoc > axisRanges[2][1])
continue;
if (TESTBIT(doCorrelation, i)) {
if (masterConfigurations.useCentralityinPrediction) {
histos.fill(HIST("Prediction/sameEvent/") + HIST(Particlenames[Index]), computeDeltaPhi(gphitrigger, gphiassoc), deltaeta, ptassoc, pttrigger, mcCollision.posZ(), centMultFT0M);
if (masterConfigurations.doSeparateFT0Prediction)
histos.fill(HIST("Prediction/sameEventFT0A/") + HIST(Particlenames[Index]), computeDeltaPhi(gphitrigger, gphiassoc), deltaeta, ptassoc, pttrigger, mcCollision.posZ(), centMultFT0A);
histos.fill(HIST("Prediction/sameEventFT0C/") + HIST(Particlenames[Index]), computeDeltaPhi(gphitrigger, gphiassoc), deltaeta, ptassoc, pttrigger, mcCollision.posZ(), centMultFT0C);
} else {
histos.fill(HIST("Prediction/sameEvent/") + HIST(Particlenames[Index]), computeDeltaPhi(gphitrigger, gphiassoc), deltaeta, ptassoc, pttrigger, mcCollision.posZ(), multFT0M);
if (masterConfigurations.doSeparateFT0Prediction)
histos.fill(HIST("Prediction/sameEventFT0A/") + HIST(Particlenames[Index]), computeDeltaPhi(gphitrigger, gphiassoc), deltaeta, ptassoc, pttrigger, mcCollision.posZ(), multFT0A);
histos.fill(HIST("Prediction/sameEventFT0C/") + HIST(Particlenames[Index]), computeDeltaPhi(gphitrigger, gphiassoc), deltaeta, ptassoc, pttrigger, mcCollision.posZ(), multFT0C);
}
}
}
}
});
}
}
PROCESS_SWITCH(HStrangeCorrelation, processSelectEventWithTrigger, "Select events with trigger only", true);
PROCESS_SWITCH(HStrangeCorrelation, processSameEventHV0s, "Process same events, h-V0s", true);
PROCESS_SWITCH(HStrangeCorrelation, processSameEventHCascades, "Process same events, h-Cascades", true);
Expand All @@ -3242,6 +3442,7 @@
PROCESS_SWITCH(HStrangeCorrelation, processMCGenerated, "Process MC generated", false);
PROCESS_SWITCH(HStrangeCorrelation, processClosureTest, "Process Closure Test", false);
PROCESS_SWITCH(HStrangeCorrelation, processFeedDown, "process Feed Down", false);
PROCESS_SWITCH(HStrangeCorrelation, processPrediction, "process model prediction", false);
};

WorkflowSpec defineDataProcessing(ConfigContext const& cfgc)
Expand Down
Loading