Skip to content

[PWGLF] Add INEL flag #13215

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

Closed
jmunozme wants to merge 1 commit into from
Closed

[PWGLF] Add INEL flag #13215

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
124 changes: 92 additions & 32 deletions PWGMM/UE/Tasks/uecharged.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,8 @@
/// \file uecharged.cxx
/// \brief Underlying event analysis task
/// \since November 2021
/// \last update: September 2025
/// \last update: October 2025


#include "Common/Core/TrackSelection.h"
#include "Common/Core/TrackSelectionDefaults.h"
Expand All @@ -29,6 +30,7 @@
#include "Framework/StaticFor.h"
#include "Framework/runDataProcessing.h"
#include "ReconstructionDataFormats/Track.h"
#include "PWGLF/Utils/inelGt.h"

#include "TDatabasePDG.h"
#include <TF1.h>
Expand All @@ -45,8 +47,7 @@
using namespace o2;
using namespace o2::framework;
using namespace o2::framework::expressions;
using BCsRun3 = soa::Join<aod::BCs, aod::Timestamps, aod::BcSels,
aod::Run3MatchedToBCSparse>;
using BCsRun3 = soa::Join<aod::BCs, aod::Timestamps, aod::BcSels, aod::Run3MatchedToBCSparse>;

struct ueCharged {

Expand Down Expand Up @@ -80,8 +81,7 @@
selectedTracks.SetMaxChi2PerClusterTPC(4.f);
selectedTracks.SetRequireHitsInITSLayers(1, {0, 1}); // one hit in any SPD layer
selectedTracks.SetMaxChi2PerClusterITS(36.f);
// selectedTracks.SetMaxDcaXYPtDep([](float pt) { return 0.0105f + 0.0350f /
// pow(pt, 1.1f); });
// selectedTracks.SetMaxDcaXYPtDep([](float pt) { return 0.0105f + 0.0350f / pow(pt, 1.1f); });
selectedTracks.SetMaxDcaZ(2.f);
return selectedTracks;
}
Expand All @@ -91,14 +91,15 @@

Service<o2::framework::O2DatabasePDG> pdg;
float deltaPhi(float phia, float phib, float rangeMin, float rangeMax);

// Configurable for event selection
Configurable<bool> isRun3{"isRun3", true, "is Run3 dataset"};
Configurable<bool> piluprejection{"piluprejection", true, "Pileup rejection"};
Configurable<bool> goodzvertex{"goodzvertex", true, "removes collisions with large differences between z of PV by tracks and z of PV from FT0 A-C time difference"};
Configurable<bool> sel8{"sel8", true, "Apply the sel8 event selection"};
Configurable<bool> removeITSROFBorder{"removeITSROFBorder", false, "Remove ITS Read-Out Frame border and only apply kIsTriggerTVX & kNoTimeFrameBorder (recommended for MC)"};
Configurable<bool> analyzeEvandTracksel{"analyzeEvandTracksel", true, "Analyze the event and track selection"};

Configurable<int> cfgINELCut{"cfgINELCut", 0, "INEL event selection: 0 no sel, 1 INEL>0, 2 INEL>1"};
// acceptance cuts
Configurable<float> cfgTrkEtaCut{"cfgTrkEtaCut", 0.8f, "Eta range for tracks"};
Configurable<float> cfgTrkLowPtCut{"cfgTrkLowPtCut", 0.15f, "Minimum constituent pT"};
Expand Down Expand Up @@ -130,6 +131,9 @@
static constexpr std::string_view hPtVsPtLeadingTrue[3] = {
"hPtVsPtLeadingTrue_NS", "hPtVsPtLeadingTrue_AS",
"hPtVsPtLeadingTrue_TS"};
static constexpr std::string_view hPtVsPtLeadingTruePS[3] = {
"hPtVsPtLeadingTruePS_NS", "hPtVsPtLeadingTruePS_AS",
"hPtVsPtLeadingTruePS_TS"};
// all wo detector effects
static constexpr std::string_view pNumDenTrueAll[3] = {
"pNumDenTrueAll_NS", "pNumDenTrueAll_AS", "pNumDenTrueAll_TS"};
Expand All @@ -140,7 +144,6 @@
"pNumDenTrue_NS", "pNumDenTrue_AS", "pNumDenTrue_TS"};
static constexpr std::string_view pSumPtTrue[3] = {
"pSumPtTrue_NS", "pSumPtTrue_AS", "pSumPtTrue_TS"};

// this must have all event selection effects, but it has not been implemented
// 50%
static constexpr std::string_view pNumDenTruePS[3] = {
Expand All @@ -165,16 +168,11 @@
template <typename C, typename T>
void analyzeEventAndTrackSelection(const C& collision, const T& tracks);

Filter trackFilter = (nabs(aod::track::eta) < cfgTrkEtaCut) &&
(aod::track::pt > cfgTrkLowPtCut);
Filter trackFilter = (nabs(aod::track::eta) < cfgTrkEtaCut) && (aod::track::pt > cfgTrkLowPtCut);

using CollisionTableMCTrue = aod::McCollisions;
using CollisionTableMC =
soa::SmallGroups<soa::Join<aod::McCollisionLabels, aod::Collisions,
aod::EvSels, aod::PVMults>>;
using TrackTableMC =
soa::Filtered<soa::Join<aod::Tracks, aod::TracksExtra, aod::TracksDCA,
aod::McTrackLabels, aod::TrackSelection>>;
using CollisionTableMC = soa::SmallGroups<soa::Join<aod::McCollisionLabels, aod::Collisions, aod::EvSels, aod::PVMults>>;
using TrackTableMC = soa::Filtered<soa::Join<aod::Tracks, aod::TracksExtra, aod::TracksDCA, aod::McTrackLabels, aod::TrackSelection>>;
using ParticleTableMC = aod::McParticles;
Preslice<TrackTableMC> perCollision = aod::track::collisionId;
void processMC(CollisionTableMCTrue::iterator const& mcCollision,
Expand All @@ -183,59 +181,54 @@
BCsRun3 const&);
PROCESS_SWITCH(ueCharged, processMC, "process MC", false);

using CollisionTableMCData =
soa::Join<aod::Collisions, aod::McCollisionLabels, aod::EvSels,
aod::PVMults>;
using TrackTableMCData =
soa::Filtered<soa::Join<aod::Tracks, aod::McTrackLabels, aod::TracksExtra,
aod::TracksDCA, aod::TrackSelection>>;
using CollisionTableMCData = soa::Join<aod::Collisions, aod::McCollisionLabels, aod::EvSels, aod::PVMults>;
using TrackTableMCData = soa::Filtered<soa::Join<aod::Tracks, aod::McTrackLabels, aod::TracksExtra, aod::TracksDCA, aod::TrackSelection>>;
void processDataMC(CollisionTableMCData::iterator const& collision,
TrackTableMCData const& tracks,
ParticleTableMC const& particles,
aod::McCollisions const& mcCollisions);
PROCESS_SWITCH(ueCharged, processDataMC, "process data MC", false);

using CollisionTableData =
soa::Join<aod::Collisions, aod::EvSels, aod::PVMults>;
using TrackTableData =
soa::Filtered<soa::Join<aod::Tracks, aod::TracksExtra, aod::TracksDCA,
aod::TrackSelection>>;
using CollisionTableData = soa::Join<aod::Collisions, aod::EvSels, aod::PVMults>;
using TrackTableData = soa::Filtered<soa::Join<aod::Tracks, aod::TracksExtra, aod::TracksDCA, aod::TrackSelection>>;
void processData(CollisionTableData::iterator const& collision,
TrackTableData const& tracks, aod::FT0s const&,
BCsRun3 const&);
PROCESS_SWITCH(ueCharged, processData, "process data", false);

// add new method
};

WorkflowSpec defineDataProcessing(ConfigContext const& cfgc)
{
WorkflowSpec workflow{};
workflow.push_back(adaptAnalysisTask<ueCharged>(cfgc));
return workflow;
}

// implementation
float ueCharged::deltaPhi(float phia, float phib,
float rangeMin = -o2::constants::math::PI / 2.0,

Check failure on line 211 in PWGMM/UE/Tasks/uecharged.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[pi-multiple-fraction] 

Use multiples/fractions of PI defined in o2::constants::math.
float rangeMax = 3.0 * o2::constants::math::PI /
2.0)
{
float dphi = -999;

if (phia < 0) {
phia += 2 * o2::constants::math::PI;

Check failure on line 218 in PWGMM/UE/Tasks/uecharged.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[pi-multiple-fraction] 

Use multiples/fractions of PI defined in o2::constants::math.

Check failure on line 218 in PWGMM/UE/Tasks/uecharged.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[two-pi-add-subtract] 

Use RecoDecay::constrainAngle to restrict angle to a given range.
} else if (phia > 2 * o2::constants::math::PI) {

Check failure on line 219 in PWGMM/UE/Tasks/uecharged.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[pi-multiple-fraction] 

Use multiples/fractions of PI defined in o2::constants::math.
phia -= 2 * o2::constants::math::PI;

Check failure on line 220 in PWGMM/UE/Tasks/uecharged.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[pi-multiple-fraction] 

Use multiples/fractions of PI defined in o2::constants::math.

Check failure on line 220 in PWGMM/UE/Tasks/uecharged.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[two-pi-add-subtract] 

Use RecoDecay::constrainAngle to restrict angle to a given range.
}
if (phib < 0) {
phib += 2 * o2::constants::math::PI;

Check failure on line 223 in PWGMM/UE/Tasks/uecharged.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[two-pi-add-subtract] 

Use RecoDecay::constrainAngle to restrict angle to a given range.
} else if (phib > 2 * o2::constants::math::PI) {
phib -= 2 * o2::constants::math::PI;

Check failure on line 225 in PWGMM/UE/Tasks/uecharged.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[two-pi-add-subtract] 

Use RecoDecay::constrainAngle to restrict angle to a given range.
}
dphi = phib - phia;
if (dphi < rangeMin) {
dphi += 2 * o2::constants::math::PI;

Check failure on line 229 in PWGMM/UE/Tasks/uecharged.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[two-pi-add-subtract] 

Use RecoDecay::constrainAngle to restrict angle to a given range.
} else if (dphi > rangeMax) {
dphi -= 2 * o2::constants::math::PI;

Check failure on line 231 in PWGMM/UE/Tasks/uecharged.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[two-pi-add-subtract] 

Use RecoDecay::constrainAngle to restrict angle to a given range.
}

return dphi;
Expand Down Expand Up @@ -289,7 +282,6 @@
{{ptAxis}, {121, -3.025, 3.025, "#it{DCA}_{xy} (cm)"}});
ue.add("hPtDCAMat", "Material; DCA_xy; Nch", HistType::kTH2D,
{{ptAxis}, {121, -3.025, 3.025, "#it{DCA}_{xy} (cm)"}});

ue.add("hmultTrue", "mult true", HistType::kTH1F,
{{200, -0.5, 199.5, " "}});
ue.add("hmultTrueGen", "mult true all Gen", HistType::kTH1F,
Expand All @@ -299,8 +291,8 @@
HistType::kTH1D, {ptAxist});

for (int i = 0; i < 3; ++i) {
ue.add(hPtVsPtLeadingTrue[i].data(), " ", HistType::kTH2D,
{{ptAxist}, {ptAxis}});
ue.add(hPtVsPtLeadingTrue[i].data(), " ", HistType::kTH2D, {{ptAxist}, {ptAxis}});
ue.add(hPtVsPtLeadingTruePS[i].data(), " ", HistType::kTH2D, {{ptAxist}, {ptAxis}});
ue.add(pNumDenTrueAll[i].data(), "", HistType::kTProfile, {ptAxist});
ue.add(pSumPtTrueAll[i].data(), "", HistType::kTProfile, {ptAxist});
ue.add(pNumDenTrue[i].data(), "", HistType::kTProfile, {ptAxist});
Expand All @@ -324,14 +316,21 @@
ue.add("phiEta", ";#eta;#varphi", HistType::kTH2F,
{{50, -2.5, 2.5}, {200, 0., 2 * o2::constants::math::PI, " "}});
ue.add("hvtxZ", "vtxZ", HistType::kTH1F, {{40, -20.0, 20.0, " "}});

ue.add("hCounter", "Counter; sel; Nev", HistType::kTH1D, {{7, 0, 7, " "}});
ue.add("hPtLeadingRecPS", "rec pTleading after physics selection",
HistType::kTH1D, {ptAxist});
ue.add("hPtLeadingMeasured", "measured pTleading after physics selection",
HistType::kTH1D, {ptAxist});
ue.add("hPtLeadingVsTracks", "", HistType::kTProfile, {{ptAxist}});

auto h = ue.get<TH1>(HIST("hCounter"));
h->GetXaxis()->SetBinLabel(1, "Events read");
h->GetXaxis()->SetBinLabel(2, "INEL");
h->GetXaxis()->SetBinLabel(3, "Sel8");
h->GetXaxis()->SetBinLabel(4, "NoSameBunchPileup");
h->GetXaxis()->SetBinLabel(5, "IsGoodZvtxFT0vsPV");
h->GetXaxis()->SetBinLabel(6, "posZ passed");

for (int i = 0; i < 3; ++i) {
ue.add(pNumDenMeasuredPS[i].data(),
"Number Density; ; #LT #it{N}_{trk} #GT", HistType::kTProfile,
Expand Down Expand Up @@ -534,7 +533,16 @@
ue.fill(HIST("hPtInPrimGen"), particle.pt());
}
ue.fill(HIST("hmultTrueGen"), multTrue);
if (std::abs(mcCollision.posZ()) > 10.f && multTrueINEL <= 0) {

if (cfgINELCut == 1 && !o2::pwglf::isINELgt0mc(particles, pdg)) {
return;
}

if (cfgINELCut == 2 && !o2::pwglf::isINELgt1mc(particles, pdg)) {
return;
}

if (std::abs(mcCollision.posZ()) > 10.f) {
return;
}

Expand Down Expand Up @@ -648,6 +656,14 @@

ue.fill(HIST("hCounter"), 0);

if (cfgINELCut == 1 && !collision.isInelGt0()) {
return;
}

if (cfgINELCut == 2 && !collision.isInelGt1()) {
return;
}

ue.fill(HIST("hCounter"), 1);

if (sel8 && !collision.sel8()) {
Expand All @@ -668,6 +684,7 @@
}

ue.fill(HIST("hCounter"), 3);

if (goodzvertex &&
!collision.selection_bit(o2::aod::evsel::kIsGoodZvtxFT0vsPV)) {
return;
Expand Down Expand Up @@ -860,6 +877,14 @@
const P& particles)
{

if (cfgINELCut == 1 && !o2::pwglf::isINELgt0mc(particles, pdg)) {
return;
}

if (cfgINELCut == 2 && !o2::pwglf::isINELgt1mc(particles, pdg)) {
return;
}

ue.fill(HIST("hStat"), collision.size());
auto vtxZ = collision.posZ();

Expand Down Expand Up @@ -960,6 +985,14 @@

ue.fill(HIST("hCounter"), 0);

if (cfgINELCut == 1 && !collision.isInelGt0()) {
return;
}

if (cfgINELCut == 2 && !collision.isInelGt1()) {
return;
}

ue.fill(HIST("hCounter"), 1);

if (sel8 && !collision.sel8()) {
Expand All @@ -978,18 +1011,20 @@
!collision.selection_bit(o2::aod::evsel::kNoSameBunchPileup)) {
return;
}

ue.fill(HIST("hCounter"), 3);

if (goodzvertex &&
!collision.selection_bit(o2::aod::evsel::kIsGoodZvtxFT0vsPV)) {
return;
}

ue.fill(HIST("hCounter"), 4);

// only PS
if ((std::abs(collision.posZ()) >= 10.f)) {
return;
}

ue.fill(HIST("hCounter"), 5);

ue.fill(HIST(pNumDenTruePS[0]), flPtTrue, ueTrue[0]);
Expand Down Expand Up @@ -1020,7 +1055,24 @@
continue;
}
ue.fill(HIST("hPtInPrim"), particle.pt());

// remove the autocorrelation
if (flIndexTrue == particle.globalIndex()) {
continue;
}
double dPhi = deltaPhi(particle.phi(), flPhiTrue);

// definition of the topological regions
if (std::abs(dPhi) < o2::constants::math::PI / 3.0) { // near side
ue.fill(HIST(hPtVsPtLeadingTruePS[0]), flPtTrue, particle.pt());
} else if (std::abs(dPhi - o2::constants::math::PI) <
o2::constants::math::PI / 3.0) { // away side
ue.fill(HIST(hPtVsPtLeadingTruePS[1]), flPtTrue, particle.pt());
} else { // transverse side
ue.fill(HIST(hPtVsPtLeadingTruePS[2]), flPtTrue, particle.pt());
}
}

ue.fill(HIST("hmultTrue"), multTrue);

// loop over selected tracks
Expand Down Expand Up @@ -1229,6 +1281,14 @@
const T& tracks)
{

if (cfgINELCut == 1 && !collision.isInelGt0()) {
return;
}

if (cfgINELCut == 2 && !collision.isInelGt1()) {
return;
}

// z-vertex from FT0 vs PV analysis

const auto& foundBC = collision.template foundBC_as<BCsRun3>();
Expand Down
Loading