Skip to content

[ALICE3] A3: Update LUT fetching in TOF and RICH #13080

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
njacazio merged 4 commits into from
Sep 22, 2025
Merged

[ALICE3] A3: Update LUT fetching in TOF and RICH #13080

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
8cfdac6
A3: Update LUT fetching in TOF and RICH
njacazio Sep 22, 2025
5efe381
Fix
njacazio Sep 22, 2025
24d38a4
Remove cfg
njacazio Sep 22, 2025
5ee5335
U
njacazio Sep 22, 2025
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
114 changes: 44 additions & 70 deletions ALICE3/TableProducer/OTF/onTheFlyRichPid.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -38,31 +38,31 @@
#include "Common/Core/trackUtilities.h"
#include "Common/DataModel/TrackSelectionTables.h"

#include "CCDB/BasicCCDBManager.h"
#include "CCDB/CcdbApi.h"
#include "CommonConstants/GeomConstants.h"
#include "CommonConstants/MathConstants.h"
#include "CommonConstants/PhysicsConstants.h"
#include "CommonUtils/NameConf.h"
#include "DataFormatsCalibration/MeanVertexObject.h"
#include "DataFormatsParameters/GRPMagField.h"
#include "DetectorsBase/GeometryManager.h"
#include "DetectorsBase/Propagator.h"
#include "DetectorsVertexing/HelixHelper.h"
#include "Framework/ASoAHelpers.h"
#include "Framework/AnalysisDataModel.h"
#include "Framework/AnalysisTask.h"
#include "Framework/HistogramRegistry.h"
#include "Framework/O2DatabasePDGPlugin.h"
#include "Framework/RunningWorkflowInfo.h"
#include "Framework/runDataProcessing.h"
#include "ReconstructionDataFormats/DCA.h"
#include "ReconstructionDataFormats/PID.h"

#include "TRandom3.h"
#include "TString.h"
#include "TVector3.h"
#include <CCDB/BasicCCDBManager.h>
#include <CCDB/CcdbApi.h>
#include <CommonConstants/GeomConstants.h>
#include <CommonConstants/MathConstants.h>
#include <CommonConstants/PhysicsConstants.h>
#include <CommonUtils/NameConf.h>
#include <DataFormatsCalibration/MeanVertexObject.h>
#include <DataFormatsParameters/GRPMagField.h>
#include <DetectorsBase/GeometryManager.h>
#include <DetectorsBase/Propagator.h>
#include <DetectorsVertexing/HelixHelper.h>
#include <Framework/ASoAHelpers.h>
#include <Framework/AnalysisDataModel.h>
#include <Framework/AnalysisTask.h>
#include <Framework/HistogramRegistry.h>
#include <Framework/O2DatabasePDGPlugin.h>
#include <Framework/RunningWorkflowInfo.h>
#include <Framework/runDataProcessing.h>
#include <ReconstructionDataFormats/DCA.h>
#include <ReconstructionDataFormats/PID.h>

#include <TPDGCode.h>
#include <TRandom3.h>
#include <TString.h>
#include <TVector3.h>

#include <cmath>
#include <map>
Expand All @@ -80,6 +80,8 @@ struct OnTheFlyRichPid {

// necessary for particle charges
Service<o2::framework::O2DatabasePDG> pdg;
// Necessary for LUTs
Service<o2::ccdb::BasicCCDBManager> ccdb;

// master setting: magnetic field
Configurable<float> magneticField{"magneticField", 0, "magnetic field (kilogauss) if 0, taken from the tracker task"};
Expand Down Expand Up @@ -131,14 +133,6 @@ struct OnTheFlyRichPid {
Configurable<float> bRICHPixelSize{"bRICHPixelSize", 0.1, "barrel RICH pixel size (cm)"};
Configurable<float> bRichGapRefractiveIndex{"bRichGapRefractiveIndex", 1.000283, "barrel RICH gap refractive index"};

struct : ConfigurableGroup {
Configurable<std::string> lutEl{"lutEl", "inherit", "LUT for electrons (if inherit, inherits from otf tracker task)"};
Configurable<std::string> lutMu{"lutMu", "inherit", "LUT for muons (if inherit, inherits from otf tracker task)"};
Configurable<std::string> lutPi{"lutPi", "inherit", "LUT for pions (if inherit, inherits from otf tracker task)"};
Configurable<std::string> lutKa{"lutKa", "inherit", "LUT for kaons (if inherit, inherits from otf tracker task)"};
Configurable<std::string> lutPr{"lutPr", "inherit", "LUT for protons (if inherit, inherits from otf tracker task)"};
} simConfig;

o2::base::Propagator::MatCorrType matCorr = o2::base::Propagator::MatCorrType::USEMatCorrNONE;

// Track smearer (here used to get relative pt and eta uncertainties)
Expand Down Expand Up @@ -304,47 +298,27 @@ struct OnTheFlyRichPid {
LOG(info) << "Bz = " << magneticField.value << " T";
}

// Check if inheriting the LUT configuration
auto configLutPath = [&](Configurable<std::string>& lut) {
if (lut.value != "inherit") {
return;
}
if (!getTaskOptionValue(initContext, "on-the-fly-tracker", lut, false)) {
LOG(fatal) << "Could not get " << lut.name << " from on-the-fly-tracker task";
}
};
configLutPath(simConfig.lutEl);
configLutPath(simConfig.lutMu);
configLutPath(simConfig.lutPi);
configLutPath(simConfig.lutKa);
configLutPath(simConfig.lutPr);

// Load LUT for pt and eta smearing
if (flagIncludeTrackAngularRes && flagRICHLoadDelphesLUTs) {
std::map<int, const char*> mapPdgLut;
const char* lutElChar = simConfig.lutEl->c_str();
const char* lutMuChar = simConfig.lutMu->c_str();
const char* lutPiChar = simConfig.lutPi->c_str();
const char* lutKaChar = simConfig.lutKa->c_str();
const char* lutPrChar = simConfig.lutPr->c_str();

LOGF(info, "Will load electron lut file ..: %s for RICH PID", lutElChar);
LOGF(info, "Will load muon lut file ......: %s for RICH PID", lutMuChar);
LOGF(info, "Will load pion lut file ......: %s for RICH PID", lutPiChar);
LOGF(info, "Will load kaon lut file ......: %s for RICH PID", lutKaChar);
LOGF(info, "Will load proton lut file ....: %s for RICH PID", lutPrChar);

mapPdgLut.insert(std::make_pair(11, lutElChar));
mapPdgLut.insert(std::make_pair(13, lutMuChar));
mapPdgLut.insert(std::make_pair(211, lutPiChar));
mapPdgLut.insert(std::make_pair(321, lutKaChar));
mapPdgLut.insert(std::make_pair(2212, lutPrChar));

for (const auto& e : mapPdgLut) {
if (!mSmearer.loadTable(e.first, e.second)) {
LOG(fatal) << "Having issue with loading the LUT " << e.first << " " << e.second;
mSmearer.setCcdbManager(ccdb.operator->());
auto loadLUT = [&](int pdg, const std::string& cfgNameToInherit) {
std::string lut = "none";
if (!getTaskOptionValue(initContext, "on-the-fly-tracker", cfgNameToInherit, lut, false)) {
LOG(fatal) << "Could not get " << cfgNameToInherit << " from on-the-fly-tracker task";
}
}
bool success = mSmearer.loadTable(pdg, lut.c_str());
if (!success && !lut.empty()) {
LOG(fatal) << "Having issue with loading the LUT " << pdg << " " << lut;
}
};
loadLUT(11, "lutEl");
loadLUT(13, "lutMu");
loadLUT(211, "lutPi");
loadLUT(321, "lutKa");
loadLUT(2212, "lutPr");
loadLUT(1000010020, "lutDe");
loadLUT(1000010030, "lutTr");
loadLUT(1000020030, "lutHe3");
}

if (doQAplots) {
Expand Down
109 changes: 43 additions & 66 deletions ALICE3/TableProducer/OTF/onTheFlyTofPid.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -32,30 +32,30 @@
#include "Common/Core/trackUtilities.h"
#include "Common/DataModel/TrackSelectionTables.h"

#include "CCDB/BasicCCDBManager.h"
#include "CCDB/CcdbApi.h"
#include "CommonConstants/GeomConstants.h"
#include "CommonConstants/MathConstants.h"
#include "CommonConstants/PhysicsConstants.h"
#include "CommonUtils/NameConf.h"
#include "DataFormatsCalibration/MeanVertexObject.h"
#include "DataFormatsParameters/GRPMagField.h"
#include "DetectorsBase/GeometryManager.h"
#include "DetectorsBase/Propagator.h"
#include "DetectorsVertexing/HelixHelper.h"
#include "Framework/ASoAHelpers.h"
#include "Framework/AnalysisDataModel.h"
#include "Framework/AnalysisTask.h"
#include "Framework/HistogramRegistry.h"
#include "Framework/O2DatabasePDGPlugin.h"
#include "Framework/RunningWorkflowInfo.h"
#include "Framework/runDataProcessing.h"
#include "ReconstructionDataFormats/DCA.h"

#include "TEfficiency.h"
#include "THashList.h"
#include "TRandom3.h"
#include <CCDB/BasicCCDBManager.h>
#include <CCDB/CcdbApi.h>
#include <CommonConstants/GeomConstants.h>
#include <CommonConstants/MathConstants.h>
#include <CommonConstants/PhysicsConstants.h>
#include <CommonUtils/NameConf.h>
#include <DataFormatsCalibration/MeanVertexObject.h>
#include <DataFormatsParameters/GRPMagField.h>
#include <DetectorsBase/GeometryManager.h>
#include <DetectorsBase/Propagator.h>
#include <DetectorsVertexing/HelixHelper.h>
#include <Framework/ASoAHelpers.h>
#include <Framework/AnalysisDataModel.h>
#include <Framework/AnalysisTask.h>
#include <Framework/HistogramRegistry.h>
#include <Framework/O2DatabasePDGPlugin.h>
#include <Framework/RunningWorkflowInfo.h>
#include <Framework/runDataProcessing.h>
#include <ReconstructionDataFormats/DCA.h>

#include <TEfficiency.h>
#include <THashList.h>
#include <TPDGCode.h>
#include <TRandom3.h>

#include <map>
#include <string>
Expand All @@ -81,6 +81,8 @@ struct OnTheFlyTofPid {

// necessary for particle charges
Service<o2::framework::O2DatabasePDG> pdg;
// Necessary for LUTs
Service<o2::ccdb::BasicCCDBManager> ccdb;

// these are the settings governing the TOF layers to be used
// note that there are two layers foreseen for now: inner and outer TOF
Expand All @@ -97,11 +99,6 @@ struct OnTheFlyTofPid {
Configurable<float> multiplicityEtaRange{"multiplicityEtaRange", 0.800000012, "eta range to compute the multiplicity"};
Configurable<bool> flagIncludeTrackTimeRes{"flagIncludeTrackTimeRes", true, "flag to include or exclude track time resolution"};
Configurable<bool> flagTOFLoadDelphesLUTs{"flagTOFLoadDelphesLUTs", false, "flag to load Delphes LUTs for tracking correction (use recoTrack parameters if false)"};
Configurable<std::string> lutEl{"lutEl", "inherit", "LUT for electrons (if inherit, inherits from otf tracker task)"};
Configurable<std::string> lutMu{"lutMu", "inherit", "LUT for muons (if inherit, inherits from otf tracker task)"};
Configurable<std::string> lutPi{"lutPi", "inherit", "LUT for pions (if inherit, inherits from otf tracker task)"};
Configurable<std::string> lutKa{"lutKa", "inherit", "LUT for kaons (if inherit, inherits from otf tracker task)"};
Configurable<std::string> lutPr{"lutPr", "inherit", "LUT for protons (if inherit, inherits from otf tracker task)"};
} simConfig;

struct : ConfigurableGroup {
Expand Down Expand Up @@ -151,47 +148,27 @@ struct OnTheFlyTofPid {
LOG(info) << "Bz = " << simConfig.magneticField.value << " T";
}

// Check if inheriting the LUT configuration
auto configLutPath = [&](Configurable<std::string>& lut) {
if (lut.value != "inherit") {
return;
}
if (!getTaskOptionValue(initContext, "on-the-fly-tracker", lut, false)) {
LOG(fatal) << "Could not get " << lut.name << " from on-the-fly-tracker task";
}
};
configLutPath(simConfig.lutEl);
configLutPath(simConfig.lutMu);
configLutPath(simConfig.lutPi);
configLutPath(simConfig.lutKa);
configLutPath(simConfig.lutPr);

// Load LUT for pt and eta smearing
if (simConfig.flagIncludeTrackTimeRes && simConfig.flagTOFLoadDelphesLUTs) {
std::map<int, const char*> mapPdgLut;
const char* lutElChar = simConfig.lutEl->c_str();
const char* lutMuChar = simConfig.lutMu->c_str();
const char* lutPiChar = simConfig.lutPi->c_str();
const char* lutKaChar = simConfig.lutKa->c_str();
const char* lutPrChar = simConfig.lutPr->c_str();

LOGF(info, "Will load electron lut file ..: %s for TOF PID", lutElChar);
LOGF(info, "Will load muon lut file ......: %s for TOF PID", lutMuChar);
LOGF(info, "Will load pion lut file ......: %s for TOF PID", lutPiChar);
LOGF(info, "Will load kaon lut file ......: %s for TOF PID", lutKaChar);
LOGF(info, "Will load proton lut file ....: %s for TOF PID", lutPrChar);

mapPdgLut.insert(std::make_pair(11, lutElChar));
mapPdgLut.insert(std::make_pair(13, lutMuChar));
mapPdgLut.insert(std::make_pair(211, lutPiChar));
mapPdgLut.insert(std::make_pair(321, lutKaChar));
mapPdgLut.insert(std::make_pair(2212, lutPrChar));

for (const auto& e : mapPdgLut) {
if (!mSmearer.loadTable(e.first, e.second)) {
LOG(fatal) << "Having issue with loading the LUT " << e.first << " " << e.second;
mSmearer.setCcdbManager(ccdb.operator->());
auto loadLUT = [&](int pdg, const std::string& cfgNameToInherit) {
std::string lut = "none";
if (!getTaskOptionValue(initContext, "on-the-fly-tracker", cfgNameToInherit, lut, false)) {
LOG(fatal) << "Could not get " << cfgNameToInherit << " from on-the-fly-tracker task";
}
}
bool success = mSmearer.loadTable(pdg, lut.c_str());
if (!success && !lut.empty()) {
LOG(fatal) << "Having issue with loading the LUT " << pdg << " " << lut;
}
};
loadLUT(11, "lutEl");
loadLUT(13, "lutMu");
loadLUT(211, "lutPi");
loadLUT(321, "lutKa");
loadLUT(2212, "lutPr");
loadLUT(1000010020, "lutDe");
loadLUT(1000010030, "lutTr");
loadLUT(1000020030, "lutHe3");
}

if (plotsConfig.doQAplots) {
Expand Down
Loading