Skip to content

[PWGDQ] Use RCT Flags in tableReader_withAssoc and dqEfficiency_withAssoc #18468

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

Open
axti98 wants to merge 2 commits into
base: master
Choose a base branch
from
+134 −3
Open

[PWGDQ] Use RCT Flags in tableReader_withAssoc and dqEfficiency_withAssoc #18468

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
abd09b6
Implement possibility to select RCT Flags in tableReader_withAssoc an…
axti98 Oct 26, 2026
3fd8b0b
Add configurable for passName and check validity of input flag
axti98 Oct 27, 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
68 changes: 67 additions & 1 deletion PWGDQ/Tasks/dqEfficiency_withAssoc.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -24,6 +24,7 @@
#include "PWGDQ/Core/VarManager.h"
#include "PWGDQ/DataModel/ReducedInfoTables.h"

#include "Common/CCDB/RCTSelectionFlags.h"
#include "Common/Core/TableHelper.h"

#include <CCDB/BasicCCDBManager.h>
Expand Down Expand Up @@ -309,6 +310,12 @@ struct AnalysisEventSelection {
Configurable<std::string> fConfigCcdbUrl{"ccdb-url", "http://alice-ccdb.cern.ch", "url of the ccdb repository"};
Configurable<int64_t> fConfigNoLaterThan{"ccdb-no-later-than", std::chrono::duration_cast<std::chrono::milliseconds>(std::chrono::system_clock::now().time_since_epoch()).count(), "latest acceptable timestamp of creation for the object"};

o2::framework::Configurable<bool> cfgUseRCT{"cfgUseRCT", false, "Apply RCT quality selection"};
o2::framework::Configurable<std::string> cfgRCTPath{"cfgRCTPath", "RCT/Flags/RunFlags", "CCDB path to the RCT run flags"};
o2::framework::Configurable<std::string> cfgRCTLabel{"cfgRCTLabel", "CBT", "RCT selection label"};
o2::framework::Configurable<bool> cfgCheckZDC{"cfgCheckZDC", false, "Also check ZDC quality"};
o2::framework::Configurable<std::string> cfgRCTPassName{"cfgRCTPassName", "", "Reconstruction pass; leave empty for run-only lookup"};

HistogramManager* fHistMan = nullptr;
MixingHandler* fMixHandler = nullptr;

Expand All @@ -322,6 +329,10 @@ struct AnalysisEventSelection {
std::map<std::string, std::string> fMetadataRCT, fHeader;
int fCurrentRun = -1;

o2::aod::rctsel::RCTFlagsChecker rctChecker{"CBT"};
std::map<uint64_t, uint32_t> rctFlags;
int rctRun = -1;

void init(o2::framework::InitContext& context)
{
if (context.mOptions.get<bool>("processDummy")) {
Expand Down Expand Up @@ -378,6 +389,61 @@ struct AnalysisEventSelection {
fCCDB->setLocalObjectValidityChecking();
fCCDB->setCreatedNotAfter(fConfigNoLaterThan.value);
fCCDBApi.init(fConfigCcdbUrl.value);

if (cfgUseRCT.value) {
if (cfgRCTLabel.value != "CBT" && cfgRCTLabel.value != "CBT_hadronPID" && cfgRCTLabel.value != "CBT_electronPID" && cfgRCTLabel.value != "CBT_calo" && cfgRCTLabel.value != "CBT_muon" && cfgRCTLabel.value != "CBT_muon_glo") {
LOGF(fatal, "RCT Label \"%s\" is unknown.", cfgRCTLabel.value.c_str());
}

rctChecker.init(cfgRCTLabel.value, cfgCheckZDC.value);

if (!rctChecker.any()) {
LOGF(fatal, "Empty RCT mask: check cfgRCTLabel.");
}
}
}

void loadRCT(int run, uint64_t timestamp)
{
std::map<std::string, std::string> metadata{{"run", std::to_string(run)}};

LOG(info) << "Requesting " << cfgRCTPath.value << " run=" << run << " timestamp=" << timestamp;

if (!cfgRCTPassName.value.empty()) {
metadata["passName"] = cfgRCTPassName.value;
}

auto* flags = fCCDB->getSpecific<std::map<uint64_t, uint32_t>>(cfgRCTPath.value, timestamp, metadata);
if (flags == nullptr && !cfgRCTPassName.value.empty()) {
metadata.erase("passName");
flags = fCCDB->getSpecific<std::map<uint64_t, uint32_t>>(cfgRCTPath.value, timestamp, metadata);
}

if (flags == nullptr || flags->empty()) {
LOGF(fatal, "Missing or empty RCT flags for run %d.", run);
}

rctFlags = *flags;
rctRun = run;
}

bool passesRCT(int run, uint64_t timestamp)
{
if (!cfgUseRCT.value) {
return true;
}

if (run != rctRun) {
loadRCT(run, timestamp);
}

auto iterator = rctFlags.upper_bound(timestamp);
if (iterator == rctFlags.begin()) {
return false;
}

const uint32_t flags = std::prev(iterator)->second;
return (static_cast<uint64_t>(flags) & rctChecker.value()) == 0;
}

template <uint32_t TEventFillMap, typename TEvents, typename TEventsMC>
Expand Down Expand Up @@ -406,7 +472,7 @@ struct AnalysisEventSelection {
if (fConfigQA) {
fHistMan->FillHistClass("Event_BeforeCuts", dqefficiency_helpers::varValues()); // automatically fill all the histograms in the class Event
}
if (fEventCut->IsSelected(dqefficiency_helpers::varValues())) {
if (fEventCut->IsSelected(dqefficiency_helpers::varValues()) && passesRCT(event.runNumber(), event.timestamp())) {
if (fConfigQA) {
fHistMan->FillHistClass("Event_AfterCuts", dqefficiency_helpers::varValues());
}
Expand Down
69 changes: 67 additions & 2 deletions PWGDQ/Tasks/tableReader_withAssoc.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -30,6 +30,7 @@
#include "PWGDQ/DataModel/ReducedInfoTables.h"

#include "Common/CCDB/EventSelectionParams.h"
#include "Common/CCDB/RCTSelectionFlags.h"
#include "Common/CCDB/ctpRateFetcher.h"
#include "Common/Core/RecoDecay.h"
#include "Common/Core/TableHelper.h"
Expand Down Expand Up @@ -339,6 +340,12 @@ struct AnalysisEventSelection {
o2::framework::Configurable<bool> fConfigFetchInteractionRate{"cfgFetchInteractionRate", false, "Fetch event-wise interaction rate from the CCDB"};
o2::framework::Configurable<std::string> fConfigIRSource{"cfgIRSource", "ZNC hadronic", "Estimator of the interaction rate (Recommended: pp --> T0VTX, Pb-Pb --> ZNC hadronic)"};

o2::framework::Configurable<bool> cfgUseRCT{"cfgUseRCT", false, "Apply RCT quality selection"};
o2::framework::Configurable<std::string> cfgRCTPath{"cfgRCTPath", "RCT/Flags/RunFlags", "CCDB path to the RCT run flags"};
o2::framework::Configurable<std::string> cfgRCTLabel{"cfgRCTLabel", "CBT", "RCT selection label"};
o2::framework::Configurable<bool> cfgCheckZDC{"cfgCheckZDC", false, "Also check ZDC quality"};
o2::framework::Configurable<std::string> cfgRCTPassName{"cfgRCTPassName", "", "Reconstruction pass; leave empty for run-only lookup"};

HistogramManager* fHistMan = nullptr;
MixingHandler* fMixHandler = nullptr;
AnalysisCompositeCut* fEventCut = nullptr;
Expand All @@ -351,9 +358,12 @@ struct AnalysisEventSelection {
std::map<uint64_t, std::vector<int64_t>> fBCCollMap; // key: global BC, value: vector of reduced event global indices
int fCurrentRun = -1;

o2::aod::rctsel::RCTFlagsChecker rctChecker{"CBT"};
std::map<uint64_t, uint32_t> rctFlags;
int rctRun = -1;

void init(o2::framework::InitContext& context)
{

bool isAnyProcessEnabled = context.mOptions.get<bool>("processSkimmed") || context.mOptions.get<bool>("processSkimmedBasic") || context.mOptions.get<bool>("processSkimmedWithZdc") || context.mOptions.get<bool>("processSkimmedWithMultExtra") || context.mOptions.get<bool>("processSkimmedWithMultExtraNoQvector") || context.mOptions.get<bool>("processSkimmedWithMultExtraZdc") || context.mOptions.get<bool>("processSkimmedWithMultExtraZdcFit") || context.mOptions.get<bool>("processSkimmedWithQvectorCentr") || context.mOptions.get<bool>("processSkimmedWithQvectorCentrMerge");
bool isDummyEnabled = context.mOptions.get<bool>("processDummy");

Expand Down Expand Up @@ -424,6 +434,61 @@ struct AnalysisEventSelection {
fCCDB->setLocalObjectValidityChecking();
fCCDB->setCreatedNotAfter(fConfigNoLaterThan.value);
fCCDBApi.init(fConfigCcdbUrl.value);

if (cfgUseRCT.value) {
if (cfgRCTLabel.value != "CBT" && cfgRCTLabel.value != "CBT_hadronPID" && cfgRCTLabel.value != "CBT_electronPID" && cfgRCTLabel.value != "CBT_calo" && cfgRCTLabel.value != "CBT_muon" && cfgRCTLabel.value != "CBT_muon_glo") {
LOGF(fatal, "RCT Label \"%s\" is unknown.", cfgRCTLabel.value.c_str());
}

rctChecker.init(cfgRCTLabel.value, cfgCheckZDC.value);

if (!rctChecker.any()) {
LOGF(fatal, "Empty RCT mask: check cfgRCTLabel.");
}
}
}

void loadRCT(int run, uint64_t timestamp)
{
std::map<std::string, std::string> metadata{{"run", std::to_string(run)}};

LOG(info) << "Requesting " << cfgRCTPath.value << " run=" << run << " timestamp=" << timestamp;

if (!cfgRCTPassName.value.empty()) {
metadata["passName"] = cfgRCTPassName.value;
}

auto* flags = fCCDB->getSpecific<std::map<uint64_t, uint32_t>>(cfgRCTPath.value, timestamp, metadata);
if (flags == nullptr && !cfgRCTPassName.value.empty()) {
metadata.erase("passName");
flags = fCCDB->getSpecific<std::map<uint64_t, uint32_t>>(cfgRCTPath.value, timestamp, metadata);
}

if (flags == nullptr || flags->empty()) {
LOGF(fatal, "Missing or empty RCT flags for run %d.", run);
}

rctFlags = *flags;
rctRun = run;
}

bool passesRCT(int run, uint64_t timestamp)
{
if (!cfgUseRCT.value) {
return true;
}

if (run != rctRun) {
loadRCT(run, timestamp);
}

auto iterator = rctFlags.upper_bound(timestamp);
if (iterator == rctFlags.begin()) {
return false;
}

const uint32_t flags = std::prev(iterator)->second;
return (static_cast<uint64_t>(flags) & rctChecker.value()) == 0;
}

template <uint32_t TEventFillMap, typename TEvents>
Expand Down Expand Up @@ -463,7 +528,7 @@ struct AnalysisEventSelection {
}

// Apply event cuts and fill histograms after selection
if (fEventCut->IsSelected(dqtablereader_helpers::varValues())) {
if (fEventCut->IsSelected(dqtablereader_helpers::varValues()) && passesRCT(event.runNumber(), event.timestamp())) {
if (fConfigRunZorro) {
if (event.tag_bit(56)) { // This is the bit used for the software trigger event selections [TO BE DONE: find a more clear way to use it]
if (fConfigQA) {
Expand Down
Loading