Skip to content

[PWGCF] improved storing of multiplet profiles with unique keys based on corr… #17148

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
wenyaCern merged 1 commit into from
Jul 21, 2026
+96 −15
Merged

[PWGCF] improved storing of multiplet profiles with unique keys based on corr… #17148

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
111 changes: 96 additions & 15 deletions PWGCF/GenericFramework/Tasks/flowGfwNonflow.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -60,6 +60,7 @@
#include <cstdlib>
#include <map>
#include <memory>
#include <set>
#include <string>
#include <string_view>
#include <utility>
Expand Down Expand Up @@ -184,6 +185,14 @@ struct FlowGfwNonflow {
bool correctionsLoaded = false;
} correctionsConfig;

struct MultipletConfig {
std::size_t representativeConfigIndex = 0;
int order = 0;
std::shared_ptr<TProfile> profile;
};

std::map<std::string, MultipletConfig> effectiveMultiplets;

// Outputs
OutputObj<FlowContainer> fFC{FlowContainer("FlowContainer")};
OutputObj<FlowContainer> fFCgen{FlowContainer("FlowContainer_gen")};
Expand Down Expand Up @@ -234,11 +243,11 @@ struct FlowGfwNonflow {

// Generic Framework
GFW* fGFW = new GFW();
std::vector<GFW::CorrConfig> corrconfigs;
std::vector<GFW::CorrConfig> corrconfigs{};
TRandom3* fRndm = new TRandom3(0);
TAxis* fPtAxis = nullptr;
int lastRun = -1;
std::map<std::string, std::shared_ptr<TProfile>> recipNHistograms;
std::vector<std::string> multipletKeys{};

// Track selection - DCA functions
TF1* fPtDepDCAxy = nullptr;
Expand Down Expand Up @@ -395,10 +404,6 @@ struct FlowGfwNonflow {
}
for (auto i = 0; i < gfwMemberCache.configs.GetSize(); ++i) {
corrconfigs.push_back(fGFW->GetCorrelatorConfig(gfwMemberCache.configs.GetCorrs()[i], gfwMemberCache.configs.GetHeads()[i], gfwMemberCache.configs.GetpTDifs()[i] != 0));
const auto& head = gfwMemberCache.configs.GetHeads()[i];
std::string histName = "inverseN";
histName += head;
recipNHistograms[histName] = registry.add<TProfile>(histName, " centrality (%); ", HistType::kTProfile, {centAxis});
}
if (corrconfigs.empty()) {
LOGF(error, "Configuration contains vectors of different size - check the GFWCorrConfig configurable");
Expand All @@ -414,6 +419,13 @@ struct FlowGfwNonflow {
fFCgen->Initialize(oba, multAxis, cfgNbootstrap);
delete oba;

// Identify unique multiplet setups and populate with identifying keys
identifyUniqueMultipletKeys(multipletKeys, corrconfigs, centAxis); // For now keep centrality axis hardcoded
LOGF(info, "Unique multiplet keys for current configuration setup");
for (const auto& key : multipletKeys) {
LOGF(info, key);
}

const auto& ptPtGaps = cfgPtPtGaps->getData();
for (uint32_t i = 0; i < ptPtGaps.rows; ++i) {
const auto etaMin = ptPtGaps(i, 0);
Expand Down Expand Up @@ -512,6 +524,72 @@ struct FlowGfwNonflow {
}
}

void identifyUniqueMultipletKeys(std::vector<std::string>& keys, const std::vector<GFW::CorrConfig>& configs, AxisSpec xAxis)
{
std::set<std::string> uniqueKeys(keys.begin(), keys.end());
const auto& regionNames = gfwMemberCache.regions.GetNames();

for (std::size_t configIndex = 0; configIndex < configs.size(); ++configIndex) {
const auto& config = configs[configIndex];
// TODO: Support pT-differential and explicit-overlap configurations.
if (config.pTDif) {
continue;
}

if (config.Regs.size() != config.Hars.size()) {
LOGF(error, "Cannot construct multiplet key for %s: %zu region groups but %zu harmonic groups", config.Head.c_str(), config.Regs.size(), config.Hars.size());
continue;
}

std::string key;
int totalOrder = 0;
bool valid = true;

for (std::size_t group = 0; group < config.Regs.size(); ++group) {
const auto& regionIndices = config.Regs[group];
const auto& harmonics = config.Hars[group];

if (regionIndices.size() != 1 || harmonics.empty()) {
LOGF(error, "Cannot construct simple multiplet key for %s, group %zu: expected one region and at least one harmonic", config.Head.c_str(), group);
valid = false;
break;
}

const int regionIndex = regionIndices.front();
if (regionIndex < 0 ||
static_cast<std::size_t>(regionIndex) >= regionNames.size()) {
LOGF(error, "Invalid region index %d in configuration %s", regionIndex, config.Head.c_str());
valid = false;
break;
}

const int regionOrder = static_cast<int>(harmonics.size());
totalOrder += regionOrder;

if (!key.empty()) {
key += "_";
}

key += regionNames[regionIndex];
key += "_";
key += std::to_string(regionOrder);
}

if (!valid || key.empty()) {
continue;
}

key += "_";
key += std::to_string(totalOrder);

if (uniqueKeys.insert(key).second) {
keys.push_back(std::move(key));
MultipletConfig currentMultipletConfig{configIndex, totalOrder, registry.add<TProfile>(Form("%s", keys.back().c_str()), Form("; centrality ; N_{%d}", totalOrder), {HistType::kTProfile, {xAxis}})};
effectiveMultiplets[keys.back()] = currentMultipletConfig;
}
}
}

int getMagneticField(uint64_t timestamp)
{
// TODO done only once (and not per run). Will be replaced by CCDBConfigurable
Expand Down Expand Up @@ -745,21 +823,24 @@ struct FlowGfwNonflow {
fFCpt->fillCMProfiles(centmult, rndm);
fFCpt->fillCMSubeventProfiles(centmult, rndm);

for (const auto& [key, multiplet] : effectiveMultiplets) {
const auto& config = corrconfigs.at(multiplet.representativeConfigIndex);
const double dnx = fGFW->Calculate(config, 0, true).real();

if (dnx > 0.) {
const double multPower = std::pow(multiplicity, multiplet.order - 1);
if (multPower > 0.) {
multiplet.profile->Fill(centmult, 1. / multPower, dnx);
}
}
}

for (std::size_t l_ind = 0; l_ind < corrconfigs.size(); ++l_ind) {
if (!corrconfigs.at(l_ind).pTDif) {
auto dnx = fGFW->Calculate(corrconfigs.at(l_ind), 0, true).real();
if (dnx == 0) {
continue;
}
const auto corrOrder = gfwMemberCache.configs.GetHeads()[l_ind].back();
const auto& head = gfwMemberCache.configs.GetHeads()[l_ind];
std::string histName = "inverseN";
histName += head;
const int m = corrOrder - '0';
double multPower = std::pow(multiplicity, m - 1);
if (multPower != 0) {
recipNHistograms.at(histName)->Fill(centmult, 1. / multPower, dnx);
}
auto val = fGFW->Calculate(corrconfigs.at(l_ind), 0, false).real() / dnx;
if (std::abs(val) < 1) {
fFC->FillProfile(corrconfigs.at(l_ind).Head.c_str(), centmult, val, cfgUseMultiplicityFlowWeights ? dnx : 1.0, rndm);
Expand Down
Loading