Skip to content

[PWGUD] Implement luminosity calculation in SG candidate producer #11089

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
rolavick merged 3 commits into from
May 14, 2025
Merged

[PWGUD] Implement luminosity calculation in SG candidate producer #11089

Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
ee404b9
Implement luminosity calculation in SG candidate producer
nburmaso May 5, 2025
e0f5b02
Fixing const-ref-in-for-loop
rolavick May 14, 2025
362233a
Fix o2linter
rolavick May 14, 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: 107 additions & 7 deletions PWGUD/TableProducer/SGCandProducer.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,13 @@
// In applying this license CERN does not waive the privileges and immunities
// granted to it by virtue of its status as an Intergovernmental Organization
// or submit itself to any jurisdiction.
//
/// \file SGCandProducer.cxx
/// \brief Produces PWGUD derived table from standard tables
///
/// \author Alexander Bylinkin <roman.lavicka@cern.ch>, Uniersity of Bergen
/// \since 23.11.2023
//

#include <cmath>
#include <vector>
Expand All @@ -21,6 +28,8 @@
#include "DataFormatsFIT/Triggers.h"
#include "DataFormatsParameters/GRPMagField.h"
#include "DataFormatsParameters/GRPObject.h"
#include "DataFormatsParameters/GRPLHCIFData.h"
#include "DataFormatsParameters/AggregatedRunInfo.h"

#include "Framework/AnalysisTask.h"
#include "Framework/ASoAHelpers.h"
Expand All @@ -38,7 +47,7 @@
using namespace o2::framework::expressions;
using namespace o2::dataformats;

#define getHist(type, name) std::get<std::shared_ptr<type>>(histPointers[name])

Check failure on line 50 in PWGUD/TableProducer/SGCandProducer.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[name/macro] 

Use SCREAMING_SNAKE_CASE for names of macros. Leading and double underscores are not allowed.

struct SGCandProducer {
Service<o2::ccdb::BasicCCDBManager> ccdb;
Expand All @@ -60,20 +69,21 @@

// get an SGCutparHolder
SGCutParHolder sameCuts = SGCutParHolder(); // SGCutparHolder
Configurable<SGCutParHolder> SGCuts{"SGCuts", {}, "SG event cuts"};

Check failure on line 72 in PWGUD/TableProducer/SGCandProducer.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[name/function-variable] 

Use lowerCamelCase for names of functions and variables.
Configurable<bool> verboseInfo{"verboseInfo", false, "Print general info to terminal; default it false."};
Configurable<bool> saveAllTracks{"saveAllTracks", true, "save only PV contributors or all tracks associated to a collision"};
Configurable<bool> savenonPVCITSOnlyTracks{"savenonPVCITSOnlyTracks", false, "save non PV contributors with ITS only information"};
Configurable<bool> rejectAtTFBoundary{"rejectAtTFBoundary", true, "reject collisions at a TF boundary"};
Configurable<bool> noITSROFrameBorder{"noITSROFrameBorder", true, "reject ITS RO Frame Border"};
Configurable<bool> noSameBunchPileUp{"noSameBunchPileUp", true, "reject SameBunchPileUp"};
Configurable<bool> IsGoodVertex{"IsGoodVertex", false, "Select FT0 PV vertex matching"};

Check failure on line 79 in PWGUD/TableProducer/SGCandProducer.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[name/function-variable] 

Use lowerCamelCase for names of functions and variables.
Configurable<bool> ITSTPCVertex{"ITSTPCVertex", true, "reject ITS-only vertex"}; // if one wants to look at Single Gap pp events

Check failure on line 80 in PWGUD/TableProducer/SGCandProducer.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[name/function-variable] 

Use lowerCamelCase for names of functions and variables.
Configurable<std::vector<int>> generatorIds{"generatorIds", std::vector<int>{-1}, "MC generatorIds to process"};

// Configurables to decide which tables are filled
Configurable<bool> fillTrackTables{"fillTrackTables", true, "Fill track tables"};
Configurable<bool> fillFwdTrackTables{"fillFwdTrackTables", true, "Fill forward track tables"};

// SG selector
SGSelector sgSelector;
ctpRateFetcher mRateFetcher;
Expand Down Expand Up @@ -103,6 +113,8 @@
{}};
std::map<std::string, HistPtr> histPointers;

int runNumber = -1;

// function to update UDFwdTracks, UDFwdTracksExtra
template <typename TFwdTrack>
void updateUDFwdTrackTables(TFwdTrack const& fwdtrack, uint64_t const& bcnum)
Expand Down Expand Up @@ -180,6 +192,78 @@
outputTracksLabel(track.globalIndex());
}

// function to process trigger counters, accounting for BC selection bits
void processCountersTrg(BCs const& bcs, aod::FT0s const&, aod::Zdcs const&)
{
const auto& firstBc = bcs.iteratorAt(0);
if (runNumber != firstBc.runNumber())
runNumber = firstBc.runNumber();

auto hCountersTrg = getHist(TH1, "reco/hCountersTrg");
auto hCountersTrgBcSel = getHist(TH1, "reco/hCountersTrgBcSel");
auto hLumi = getHist(TH1, "reco/hLumi");
auto hLumiBcSel = getHist(TH1, "reco/hLumiBcSel");

// Cross sections in ub. Using dummy -1 if lumi estimator is not reliable
float csTCE = 10.36e6;
float csZEM = 415.2e6; // see AN: https://alice-notes.web.cern.ch/node/1515
float csZNC = 214.5e6; // see AN: https://alice-notes.web.cern.ch/node/1515
if (runNumber > 543437 && runNumber < 543514) {

Check failure on line 211 in PWGUD/TableProducer/SGCandProducer.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.
csTCE = 8.3e6;
}
if (runNumber >= 543514) {

Check failure on line 214 in PWGUD/TableProducer/SGCandProducer.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.
csTCE = 4.10e6; // see AN: https://alice-notes.web.cern.ch/node/1515
}

for (const auto& bc : bcs) {
bool hasFT0 = bc.has_foundFT0();
bool hasZDC = bc.has_foundZDC();
if (!hasFT0 && !hasZDC)
continue;
bool isSelectedBc = true;
if (rejectAtTFBoundary && !bc.selection_bit(aod::evsel::kNoTimeFrameBorder))
isSelectedBc = false;
if (noITSROFrameBorder && !bc.selection_bit(aod::evsel::kNoITSROFrameBorder))
isSelectedBc = false;
if (hasFT0) {
auto ft0TrgMask = bc.ft0().triggerMask();
if (TESTBIT(ft0TrgMask, o2::fit::Triggers::bitVertex)) {
hCountersTrg->Fill("TVX", 1);
if (isSelectedBc)
hCountersTrgBcSel->Fill("TVX", 1);
}
if (TESTBIT(ft0TrgMask, o2::fit::Triggers::bitVertex) && TESTBIT(ft0TrgMask, o2::fit::Triggers::bitCen)) {
hCountersTrg->Fill("TCE", 1);
hLumi->Fill("TCE", 1. / csTCE);
if (isSelectedBc) {
hCountersTrgBcSel->Fill("TCE", 1);
hLumiBcSel->Fill("TCE", 1. / csTCE);
}
}
}
if (hasZDC) {
if (bc.selection_bit(aod::evsel::kIsBBZNA) || bc.selection_bit(aod::evsel::kIsBBZNC)) {
hCountersTrg->Fill("ZEM", 1);
hLumi->Fill("ZEM", 1. / csZEM);
if (isSelectedBc) {
hCountersTrgBcSel->Fill("ZEM", 1);
hLumiBcSel->Fill("ZEM", 1. / csZEM);
}
}
if (bc.selection_bit(aod::evsel::kIsBBZNC)) {
hCountersTrg->Fill("ZNC", 1);
hLumi->Fill("ZNC", 1. / csZNC);
if (isSelectedBc) {
hCountersTrgBcSel->Fill("ZNC", 1);
hLumiBcSel->Fill("ZNC", 1. / csZNC);
}
}
}
}
}

PROCESS_SWITCH(SGCandProducer, processCountersTrg, "Produce trigger counters and luminosity histograms", true);

// function to process reconstructed data
template <typename TCol>
void processReco(std::string histdir, TCol const& collision, BCs const& bcs,
Expand Down Expand Up @@ -241,14 +325,14 @@
auto isSGEvent = sgSelector.IsSelected(sameCuts, collision, bcRange, bc);
// auto isSGEvent = sgSelector.IsSelected(sameCuts, collision, bcRange, tracks);
int issgevent = isSGEvent.value;
if (isSGEvent.bc && issgevent < 2) {

Check failure on line 328 in PWGUD/TableProducer/SGCandProducer.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.
newbc = *(isSGEvent.bc);
} else {
if (verboseInfo)
LOGF(info, "No Newbc %i", bc.globalBC());
}
getHist(TH1, histdir + "/Stat")->Fill(issgevent + 8, 1.);
if (issgevent <= 2) {

Check failure on line 335 in PWGUD/TableProducer/SGCandProducer.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.
if (verboseInfo)
LOGF(info, "Current BC: %i, %i, %i", bc.globalBC(), newbc.globalBC(), issgevent);
if (sameCuts.minRgtrwTOF()) {
Expand All @@ -263,7 +347,7 @@
uint8_t chFV0A = 0;
int occ = collision.trackOccupancyInTimeRange();
udhelpers::getFITinfo(fitInfo, newbc, bcs, ft0s, fv0as, fdds);
int upc_flag = (collision.flags() & dataformats::Vertex<o2::dataformats::TimeStamp<int>>::Flags::UPCMode) ? 1 : 0;

Check failure on line 350 in PWGUD/TableProducer/SGCandProducer.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[name/function-variable] 

Use lowerCamelCase for names of functions and variables.
// update SG candidates tables
outputCollisions(bc.globalBC(), bc.runNumber(),
collision.posX(), collision.posY(), collision.posZ(), upc_flag,
Expand All @@ -289,12 +373,12 @@
}
// update SGTracks tables
if (fillTrackTables) {
for (auto& track : tracks) {
for (const auto& track : tracks) {
if (track.pt() > sameCuts.minPt() && track.eta() > sameCuts.minEta() && track.eta() < sameCuts.maxEta()) {
if (track.isPVContributor()) {
updateUDTrackTables(outputCollisions.lastIndex(), track, bc.globalBC());
} else if (saveAllTracks) {
if (track.itsClusterSizes() && track.itsChi2NCl() > 0 && ((track.tpcNClsFindable() == 0 && savenonPVCITSOnlyTracks) || track.tpcNClsFindable() > 50))

Check failure on line 381 in PWGUD/TableProducer/SGCandProducer.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.
updateUDTrackTables(outputCollisions.lastIndex(), track, bc.globalBC());
// if (track.isPVContributor()) updateUDTrackTables(outputCollisions.lastIndex(), track, bc.globalBC());
}
Expand All @@ -304,7 +388,7 @@
// update SGFwdTracks tables
if (fillFwdTrackTables) {
if (sameCuts.withFwdTracks()) {
for (auto& fwdtrack : fwdtracks) {
for (const auto& fwdtrack : fwdtracks) {
if (!sgSelector.FwdTrkSelector(fwdtrack))
updateUDFwdTrackTables(fwdtrack, bc.globalBC());
}
Expand All @@ -324,6 +408,22 @@
histPointers.clear();
if (context.mOptions.get<bool>("processData")) {
histPointers.insert({"reco/Stat", registry.add("reco/Stat", "Cut statistics; Selection criterion; Collisions", {HistType::kTH1F, {{14, -0.5, 13.5}}})});

const AxisSpec axisCountersTrg{10, 0.5, 10.5, ""};
histPointers.insert({"reco/hCountersTrg", registry.add("reco/hCountersTrg", "Trigger counts before selections; Trigger; Counts", {HistType::kTH1F, {axisCountersTrg}})});
histPointers.insert({"reco/hCountersTrgBcSel", registry.add("reco/hCountersTrgSel", "Trigger counts after BC selections; Trigger; Counts", {HistType::kTH1F, {axisCountersTrg}})});
histPointers.insert({"reco/hLumi", registry.add("reco/hLumi", "Integrated luminosity before selections; Trigger; Luminosity, 1/#mub", {HistType::kTH1F, {axisCountersTrg}})});
histPointers.insert({"reco/hLumiBcSel", registry.add("reco/hLumiBcSel", "Integrated luminosity before selections; Trigger; Luminosity, 1/#mub", {HistType::kTH1F, {axisCountersTrg}})});
auto hCountersTrg = getHist(TH1, "reco/hCountersTrg");
auto hCountersTrgBcSel = getHist(TH1, "reco/hCountersTrgBcSel");
auto hLumi = getHist(TH1, "reco/hLumi");
auto hLumiBcSel = getHist(TH1, "reco/hLumiBcSel");
for (const auto& h : {hCountersTrg, hCountersTrgBcSel, hLumi, hLumiBcSel}) {
h->GetXaxis()->SetBinLabel(1, "TVX");
h->GetXaxis()->SetBinLabel(2, "TCE");
h->GetXaxis()->SetBinLabel(3, "ZEM");
h->GetXaxis()->SetBinLabel(4, "ZNC");
}
}
if (context.mOptions.get<bool>("processMcData")) {
histPointers.insert({"MCreco/Stat", registry.add("MCreco/Stat", "Cut statistics; Selection criterion; Collisions", {HistType::kTH1F, {{14, -0.5, 13.5}}})});
Expand Down Expand Up @@ -441,7 +541,7 @@
// This is needed to be able to assign the new daughter indices
std::map<int64_t, int64_t> oldnew;
auto lastId = outputMcParticles.lastIndex();
for (auto mcpart : McParts) {
for (const auto& mcpart : McParts) {
auto oldId = mcpart.globalIndex();
if (mcPartIsSaved.find(oldId) != mcPartIsSaved.end()) {
oldnew[oldId] = mcPartIsSaved[oldId];
Expand All @@ -452,12 +552,12 @@
}

// all particles of the McCollision are saved
for (auto mcpart : McParts) {
for (const auto& mcpart : McParts) {
if (mcPartIsSaved.find(mcpart.globalIndex()) == mcPartIsSaved.end()) {
// mothers
newmids.clear();
auto oldmids = mcpart.mothersIds();
for (auto oldmid : oldmids) {
for (const auto& oldmid : oldmids) {
auto m = McParts.rawIteratorAt(oldmid);
if (verboseInfoMC)
LOGF(debug, " m %d", m.globalIndex());
Expand Down Expand Up @@ -524,7 +624,7 @@
void updateUDMcTrackLabels(TTrack const& udtracks, std::map<int64_t, int64_t>& mcPartIsSaved)
{
// loop over all tracks
for (auto udtrack : udtracks) {
for (const auto& udtrack : udtracks) {
// udtrack (UDTCs) -> track (TCs) -> mcTrack (McParticles) -> udMcTrack (UDMcParticles)
auto trackId = udtrack.trackId();
if (trackId >= 0) {
Expand Down Expand Up @@ -634,7 +734,7 @@

// update UDMcParticles and UDMcTrackLabels (for each UDTrack -> UDMcParticles)
// loop over tracks of dgcand
for (auto sgtrack : sgTracks) {
for (const auto& sgtrack : sgTracks) {
if (sgtrack.has_track()) {
auto track = sgtrack.track_as<TCs>();
if (track.has_mcParticle()) {
Expand Down
Loading