Skip to content

[PWGCF] Run 3 Magnetic Field Path and configurable added #17782

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
jaelpark merged 3 commits into from
Sep 5, 2026
+30 −15
Merged

[PWGCF] Run 3 Magnetic Field Path and configurable added #17782

Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
07c33c0
Run 3 Magnetic Field Path and configurable added
nmallick19 Sep 5, 2026
2e0d1a5
Please consider the following formatting changes
alibuild Sep 5, 2026
d088b09
Merge pull request #7 from alibuild/alibot-cleanup-17782
nmallick19 Sep 5, 2026
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
45 changes: 30 additions & 15 deletions PWGCF/Tasks/correlations.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
// Copyright 2019-2020 CERN and copyright holders of ALICE O2.

Check failure on line 1 in PWGCF/Tasks/correlations.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[name/workflow-file] 

Name of a workflow file must match the name of the main struct in it (without the PWG prefix). (Class implementation files should be in "Core" directories.)
// See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
// All rights not expressly granted are reserved.
//
Expand Down Expand Up @@ -26,6 +26,7 @@
#include <CCDB/BasicCCDBManager.h>
#include <CommonConstants/MathConstants.h>
#include <CommonConstants/PhysicsConstants.h>
#include <DataFormatsParameters/GRPMagField.h>
#include <DataFormatsParameters/GRPObject.h>
#include <Framework/AnalysisDataModel.h>
#include <Framework/AnalysisHelpers.h>
Expand Down Expand Up @@ -114,6 +115,7 @@

O2_DEFINE_CONFIGURABLE(cfgEfficiencyTrigger, std::string, "", "CCDB path to efficiency object for trigger particles")
O2_DEFINE_CONFIGURABLE(cfgEfficiencyAssociated, std::string, "", "CCDB path to efficiency object for associated particles")
O2_DEFINE_CONFIGURABLE(cfgUseRun3MagField, bool, false, "false: Run 2 GRPObject; true: Run 3 GRPMagField")

O2_DEFINE_CONFIGURABLE(cfgNoMixedEvents, int, 5, "Number of mixed events per event")
O2_DEFINE_CONFIGURABLE(cfgRejectMixedPhiProngEvents, bool, true, "Reject associated hadrons from either mixed-phi prong event")
Expand Down Expand Up @@ -186,6 +188,8 @@
PairCuts mPairCuts;

Service<o2::ccdb::BasicCCDBManager> ccdb;
int mCachedRunNumber{-1}; // cached run number for magnetic field -- to avoid re-fetching the magnetic field for the same run, assuming that the magnetic field remains the same for the same run
int mCachedMagField{0}; // cached magnetic field --reduces number of calls to the CCDB

using AodCollisions = soa::Filtered<soa::Join<aod::Collisions, aod::EvSels, aod::CentRun2V0Ms>>;
using AodTracks = soa::Filtered<soa::Join<aod::Tracks, aod::TrackSelection>>;
Expand Down Expand Up @@ -331,21 +335,32 @@
ccdb->setCreatedNotAfter(now); // TODO must become global parameter from the train creation time
}

int getMagneticField(uint64_t timestamp)
int getMagneticField(int runNumber, uint64_t timestamp)
{
// TODO done only once (and not per run). Will be replaced by CCDBConfigurable
static o2::parameters::GRPObject* grpo = nullptr;
// static o2::parameters::GRPMagField* grpo = nullptr;
if (grpo == nullptr) {
grpo = ccdb->getForTimeStamp<o2::parameters::GRPObject>("GLO/GRP/GRP", timestamp);
// grpo = ccdb->getForTimeStamp<o2::parameters::GRPMagField>("GLO/Config/GRPMagField", timestamp);
static constexpr const char* kGRPPathRun2 = "GLO/GRP/GRP"; // fixed path for now, can be made into a config
static constexpr const char* kGRPPathRun3 = "GLO/Config/GRPMagField"; // fixed path for now, can be made into a config

if (runNumber == mCachedRunNumber) {
return mCachedMagField;
}

if (cfgUseRun3MagField) {
auto* grpmag = ccdb->getForTimeStamp<o2::parameters::GRPMagField>(kGRPPathRun3, timestamp);
if (grpmag == nullptr) {
LOGF(fatal, "Run 3 GRPMagField not found at %s for run %d timestamp %llu", kGRPPathRun3, runNumber, timestamp);
}
mCachedMagField = grpmag->getNominalL3Field();
} else {
auto* grpo = ccdb->getForTimeStamp<o2::parameters::GRPObject>(kGRPPathRun2, timestamp);
if (grpo == nullptr) {
LOGF(fatal, "GRP object not found for timestamp %llu", timestamp);
return 0;
LOGF(fatal, "Run 2 GRPObject not found at %s for run %d timestamp %llu", kGRPPathRun2, runNumber, timestamp);
}
LOGF(info, "Retrieved GRP for timestamp %llu with magnetic field of %d kG", timestamp, grpo->getNominalL3Field());
mCachedMagField = grpo->getNominalL3Field();
}
return grpo->getNominalL3Field();

mCachedRunNumber = runNumber;
LOGF(info, "Run %d: magnetic field %d kG", runNumber, mCachedMagField);
return mCachedMagField;
}

template <class p2typeIterator>
Expand Down Expand Up @@ -439,7 +454,7 @@

if (cfgCorrelationMethod == 1 && track1.decay() != track2.decay())
continue;
if (cfgCorrelationMethod == 2 && track1.decay() == track2.decay())

Check failure on line 457 in PWGCF/Tasks/correlations.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.
continue;
registry.fill(HIST("invMassTwoPart"), track1.invMass(), track2.invMass(), track1.pt(), track2.pt(), multiplicity);
registry.fill(HIST("invMassTwoPartDPhi"), track1.invMass(), track2.invMass(), track1.pt(), track2.pt(), TVector2::Phi_0_2pi(track1.phi() - track2.phi() + TMath::Pi() / 2.0) - TMath::Pi() / 2.0);
Expand Down Expand Up @@ -733,7 +748,7 @@
if constexpr (std::experimental::is_detected<HasDecay, typename TTracks1::iterator>::value && std::experimental::is_detected<HasDecay, typename TTracks2::iterator>::value) {
if (cfgCorrelationMethod == 1 && track1.decay() != track2.decay())
continue;
if (cfgCorrelationMethod == 2 && track1.decay() == track2.decay())

Check failure on line 751 in PWGCF/Tasks/correlations.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.
continue;
}

Expand Down Expand Up @@ -894,7 +909,7 @@
}
registry.fill(HIST("eventcount_same"), -2);
fillQA(collision, multiplicity, tracks);
fillCorrelations<CorrelationContainer::kCFStepReconstructed>(same, tracks, tracks, multiplicity, collision.posZ(), getMagneticField(bc.timestamp()), 1.0f);
fillCorrelations<CorrelationContainer::kCFStepReconstructed>(same, tracks, tracks, multiplicity, collision.posZ(), getMagneticField(bc.runNumber(), bc.timestamp()), 1.0f);
}
PROCESS_SWITCH(CorrelationTask, processSameAOD, "Process same event on AOD", true);

Expand All @@ -911,7 +926,7 @@
const auto multiplicity = collision.multiplicity();
int field = 0;
if (cfgTwoTrackCut > 0) {
field = getMagneticField(collision.timestamp());
field = getMagneticField(collision.runNumber(), collision.timestamp());
}

int bin = configurableBinningDerived.getBin({collision.posZ(), collision.multiplicity()});
Expand Down Expand Up @@ -1011,7 +1026,7 @@

// LOGF(info, "Tracks: %d and %d entries", tracks1.size(), tracks2.size());

fillCorrelations<CorrelationContainer::kCFStepReconstructed>(mixed, tracks1, tracks2, collision1.centRun2V0M(), collision1.posZ(), getMagneticField(bc.timestamp()), 1.0f / it.currentWindowNeighbours());
fillCorrelations<CorrelationContainer::kCFStepReconstructed>(mixed, tracks1, tracks2, collision1.centRun2V0M(), collision1.posZ(), getMagneticField(bc.runNumber(), bc.timestamp()), 1.0f / it.currentWindowNeighbours());
}
}
PROCESS_SWITCH(CorrelationTask, processMixedAOD, "Process mixed events on AOD", false);
Expand Down Expand Up @@ -1045,7 +1060,7 @@
float eventWeight = 1.0f / it.currentWindowNeighbours();
int field = 0;
if (cfgTwoTrackCut > 0) {
field = getMagneticField(collision1.timestamp());
field = getMagneticField(collision1.runNumber(), collision1.timestamp());
}

if (cfgVerbosity > 0) {
Expand Down
Loading