Skip to content

Please consider the following formatting changes to #9620 #30

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
ilikmeta merged 1 commit into from
Jan 30, 2025
+132 −112
Merged

Please consider the following formatting changes to #9620 #30

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
244 changes: 132 additions & 112 deletions PWGCF/Flow/Tasks/FlowGFWPbPb.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -85,7 +85,6 @@
O2_DEFINE_CONFIGURABLE(cfgTrackSelRun3ITSMatch, bool, false, "System check: Run3ITSMatch")
O2_DEFINE_CONFIGURABLE(cfgTrackSel, bool, false, "System check: track selection")


ConfigurableAxis axisVertex{"axisVertex", {20, -10, 10}, "vertex axis for histograms"};
ConfigurableAxis axisPhi{"axisPhi", {60, 0.0, constants::math::TwoPI}, "phi axis for histograms"};
ConfigurableAxis axisPhiMod{"axisPhiMod", {100, 0, constants::math::PI / 9}, "fmod(#varphi,#pi/9)"};
Expand Down Expand Up @@ -129,9 +128,30 @@
enum ExtraProfile {

// here are TProfiles for vn-ft0 correlations that are not implemented in GFW
kc22, kc24, kc26, kc28, kc22etagap, kc32, kc32etagap, kc34,
kc22Nch, kc24Nch, kc26Nch, kc28Nch, kc22Nchetagap, kc32Nch, kc32Nchetagap, kc34Nch,
kc22Nch05, kc24Nch05, kc26Nch05, kc28Nch05, kc22Nch05etagap, kc32Nch05, kc32Nch05etagap, kc34Nch05,
kc22,
kc24,
kc26,
kc28,
kc22etagap,
kc32,
kc32etagap,
kc34,
kc22Nch,
kc24Nch,
kc26Nch,
kc28Nch,
kc22Nchetagap,
kc32Nch,
kc32Nchetagap,
kc34Nch,
kc22Nch05,
kc24Nch05,
kc26Nch05,
kc28Nch05,
kc22Nch05etagap,
kc32Nch05,
kc32Nch05etagap,
kc34Nch05,

// Count the total number of enum
kCount_ExtraProfile
Expand All @@ -152,17 +172,17 @@
kNOOFEVENTSTEPS
};

// Additional Event selection cuts - Copy from flowGenericFramework.cxx
TrackSelection myTrackSel;
TF1* fPhiCutLow = nullptr;
TF1* fPhiCutHigh = nullptr;
TF1* fMultPVCutLow = nullptr;
TF1* fMultPVCutHigh = nullptr;
TF1* fMultCutLow = nullptr;
TF1* fMultCutHigh = nullptr;
TF1* fMultMultPVCut = nullptr;
TF1* fT0AV0AMean = nullptr;
TF1* fT0AV0ASigma = nullptr;
// Additional Event selection cuts - Copy from flowGenericFramework.cxx
TrackSelection myTrackSel;
TF1* fPhiCutLow = nullptr;
TF1* fPhiCutHigh = nullptr;
TF1* fMultPVCutLow = nullptr;
TF1* fMultPVCutHigh = nullptr;
TF1* fMultCutLow = nullptr;
TF1* fMultCutHigh = nullptr;
TF1* fMultMultPVCut = nullptr;
TF1* fT0AV0AMean = nullptr;
TF1* fT0AV0ASigma = nullptr;

void init(InitContext const&) // Initialization
{
Expand Down Expand Up @@ -209,7 +229,7 @@
registry.add("multV0A_multT0A_Aft", "after cut;mulplicity T0A;mulplicity V0A", {HistType::kTH2D, {axisT0A, axisT0A}});
registry.add("multT0C_centT0C_Aft", "after cut;Centrality T0C;mulplicity T0C", {HistType::kTH2D, {axisCentForQA, axisT0C}});

//FT0 plots
// FT0 plots
registry.add("FT0CAmp", ";FT0C amplitude;Events", kTH1F, {axisFT0CAmp});
registry.add("FT0AAmp", ";FT0A amplitude;Events", kTH1F, {axisFT0AAmp});
registry.add("FT0MAmp", ";FT0M amplitude;Events", kTH1F, {axisFT0MAmp});
Expand Down Expand Up @@ -359,13 +379,14 @@
fPhiCutHigh = new TF1("fPhiCutHigh", "0.1/x+pi/18.0+0.06", 0, 100);
}

if (cfgTrackSelRun3ITSMatch) {
myTrackSel = getGlobalTrackSelectionRun3ITSMatch(TrackSelection::GlobalTrackRun3ITSMatching::Run3ITSall7Layers, TrackSelection::GlobalTrackRun3DCAxyCut::Default);}
else {
myTrackSel = getGlobalTrackSelectionRun3ITSMatch(TrackSelection::GlobalTrackRun3ITSMatching::Run3ITSibAny, TrackSelection::GlobalTrackRun3DCAxyCut::Default);}
if (cfgTrackSelRun3ITSMatch) {
myTrackSel = getGlobalTrackSelectionRun3ITSMatch(TrackSelection::GlobalTrackRun3ITSMatching::Run3ITSall7Layers, TrackSelection::GlobalTrackRun3DCAxyCut::Default);
} else {
myTrackSel = getGlobalTrackSelectionRun3ITSMatch(TrackSelection::GlobalTrackRun3ITSMatching::Run3ITSibAny, TrackSelection::GlobalTrackRun3DCAxyCut::Default);
}

myTrackSel.SetMinNClustersTPC(cfgCutTPCclu);
myTrackSel.SetMinNClustersITS(cfgCutITSclu);
myTrackSel.SetMinNClustersTPC(cfgCutTPCclu);
myTrackSel.SetMinNClustersITS(cfgCutITSclu);

} // end of Initialization

Expand Down Expand Up @@ -413,7 +434,7 @@
fFC->FillProfile(corrconf.Head.c_str(), cent, val, dnx, rndm);
return;
}
for (Int_t i = 1; i <= fPtAxis->GetNbins(); i++) {

Check warning on line 437 in PWGCF/Flow/Tasks/FlowGFWPbPb.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[root-entity] 

Consider replacing ROOT entities with equivalents from standard C++ or from O2.
dnx = fGFW->Calculate(corrconf, i - 1, kTRUE).real();
if (dnx == 0)
continue;
Expand Down Expand Up @@ -515,7 +536,7 @@
float vtxz = -999;
if (collision.numContrib() > 1) {
vtxz = collision.posZ();
float zRes = sqrt(collision.covZZ());

Check warning on line 539 in PWGCF/Flow/Tasks/FlowGFWPbPb.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[std-prefix] 

Use std:: prefix for names from the std namespace.
if (zRes > 0.25 && collision.numContrib() < 20)
vtxz = -999;
}
Expand Down Expand Up @@ -563,7 +584,7 @@
{
double phimodn = track.phi();
if (field < 0) // for negative polarity field
phimodn = TMath::TwoPi() - phimodn;

Check warning on line 587 in PWGCF/Flow/Tasks/FlowGFWPbPb.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[external-pi] 

Consider using the PI constant (and its multiples and fractions) defined in o2::constants::math.
if (track.sign() < 0) // for negative charge
phimodn = TMath::TwoPi() - phimodn;
if (phimodn < 0)
Expand All @@ -578,23 +599,23 @@
return true;
}

template <typename TTrack>
bool trackSelected(TTrack track)
{
template <typename TTrack>
bool trackSelected(TTrack track)
{

if (cfgTrackSel) {
return myTrackSel.IsSelected(track);
} else {
return (track.tpcNClsFound() >= cfgCutTPCclu);
}
}
if (cfgTrackSel) {
return myTrackSel.IsSelected(track);
} else {
return (track.tpcNClsFound() >= cfgCutTPCclu);
}
}

// Apply process filters
Filter collisionFilter = nabs(aod::collision::posZ) < cfgCutVertex;
Filter trackFilter = (nabs(aod::track::eta) < cfgCutEta) && (aod::track::pt > cfgCutPtMin) && (aod::track::pt < cfgCutPtMax) && ((requireGlobalTrackInFilter()) || (aod::track::isGlobalTrackSDD == (uint8_t) true)) && (aod::track::tpcChi2NCl < cfgCutChi2prTPCcls) && (nabs(aod::track::dcaZ) < cfgCutDCAz) && (nabs(aod::track::dcaXY) < cfgCutDCAxy);

using Colls = soa::Filtered<soa::Join<aod::Collisions, aod::EvSels, aod::Mults, aod::CentFT0Cs>>; // collisions filter
using AodTracks = soa::Filtered<soa::Join<aod::Tracks, aod::TrackSelection, aod::TracksDCA, aod::TracksExtra>>; // tracks filter
using Colls = soa::Filtered<soa::Join<aod::Collisions, aod::EvSels, aod::Mults, aod::CentFT0Cs>>; // collisions filter
using AodTracks = soa::Filtered<soa::Join<aod::Tracks, aod::TrackSelection, aod::TracksDCA, aod::TracksExtra>>; // tracks filter

void processData(Colls::iterator const& collision, aod::BCsWithTimestamps const&, AodTracks const& tracks, aod::FT0s const&)
{
Expand Down Expand Up @@ -659,13 +680,13 @@
double ft0cAmp = 0;

if (collision.has_foundFT0()) {
auto ft0 = collision.foundFT0();
for (const auto& amplitude : ft0.amplitudeA()) {
ft0aAmp += amplitude;
}
for (const auto& amplitude : ft0.amplitudeC()) {
ft0cAmp += amplitude;
}
auto ft0 = collision.foundFT0();
for (const auto& amplitude : ft0.amplitudeA()) {
ft0aAmp += amplitude;
}
for (const auto& amplitude : ft0.amplitudeC()) {
ft0cAmp += amplitude;
}
}

registry.fill(HIST("FT0AAmp"), ft0aAmp);
Expand All @@ -688,7 +709,7 @@

for (const auto& track : tracks) {

if (!trackSelected(track))
if (!trackSelected(track))
continue;

if (cfgUseAdditionalTrackCut && !trackSelected(track, Magnetfield))
Expand All @@ -715,13 +736,13 @@
registry.fill(HIST("hDCAxy"), track.dcaXY(), track.pt());
}

globaltracks_nch++;
globaltracks_nch++;

if (withinPtRef)
fGFW->Fill(track.eta(), 1, track.phi(), wacc * weff, 1);
fGFW->Fill(track.eta(), 1, track.phi(), wacc * weff, 1);

if (FineBinning == true)
fGFW->Fill(track.eta(), 1, track.phi(), wacc * weff, 1);
fGFW->Fill(track.eta(), 1, track.phi(), wacc * weff, 1);

} // End of track loop

Expand All @@ -748,17 +769,16 @@
FillProfile(corrconfigs.at(7), HIST("c34Nch"), globaltracks_nch);

// 0-5% centrality Nch
if (centrality>=0 && centrality<=5){
FillProfile(corrconfigs.at(0), HIST("c22Nch05"), globaltracks_nch);
FillProfile(corrconfigs.at(1), HIST("c24Nch05"), globaltracks_nch);
FillProfile(corrconfigs.at(2), HIST("c26Nch05"), globaltracks_nch);
FillProfile(corrconfigs.at(3), HIST("c28Nch05"), globaltracks_nch);
FillProfile(corrconfigs.at(4), HIST("c22Nch05etagap"), globaltracks_nch);
FillProfile(corrconfigs.at(5), HIST("c32Nch05"), globaltracks_nch);
FillProfile(corrconfigs.at(6), HIST("c32Nch05etagap"), globaltracks_nch);
FillProfile(corrconfigs.at(7), HIST("c34Nch05"), globaltracks_nch);
}

if (centrality >= 0 && centrality <= 5) {
FillProfile(corrconfigs.at(0), HIST("c22Nch05"), globaltracks_nch);
FillProfile(corrconfigs.at(1), HIST("c24Nch05"), globaltracks_nch);
FillProfile(corrconfigs.at(2), HIST("c26Nch05"), globaltracks_nch);
FillProfile(corrconfigs.at(3), HIST("c28Nch05"), globaltracks_nch);
FillProfile(corrconfigs.at(4), HIST("c22Nch05etagap"), globaltracks_nch);
FillProfile(corrconfigs.at(5), HIST("c32Nch05"), globaltracks_nch);
FillProfile(corrconfigs.at(6), HIST("c32Nch05etagap"), globaltracks_nch);
FillProfile(corrconfigs.at(7), HIST("c34Nch05"), globaltracks_nch);
}

// Filling Bootstrap Samples
int sampleIndex = static_cast<int>(cfgNbootstrap * l_Random);
Expand All @@ -780,80 +800,80 @@
FillProfile(corrconfigs.at(6), bootstrapArray[sampleIndex][kc32Nchetagap], globaltracks_nch);
FillProfile(corrconfigs.at(7), bootstrapArray[sampleIndex][kc34Nch], globaltracks_nch);

if (centrality>=0 && centrality<=5){
FillProfile(corrconfigs.at(0), bootstrapArray[sampleIndex][kc22Nch05], globaltracks_nch);
FillProfile(corrconfigs.at(1), bootstrapArray[sampleIndex][kc24Nch05], globaltracks_nch);
FillProfile(corrconfigs.at(2), bootstrapArray[sampleIndex][kc26Nch05], globaltracks_nch);
FillProfile(corrconfigs.at(3), bootstrapArray[sampleIndex][kc28Nch05], globaltracks_nch);
FillProfile(corrconfigs.at(4), bootstrapArray[sampleIndex][kc22Nch05etagap], globaltracks_nch);
FillProfile(corrconfigs.at(5), bootstrapArray[sampleIndex][kc32Nch05], globaltracks_nch);
FillProfile(corrconfigs.at(6), bootstrapArray[sampleIndex][kc32Nch05etagap], globaltracks_nch);
FillProfile(corrconfigs.at(7), bootstrapArray[sampleIndex][kc34Nch05], globaltracks_nch);
}
if (centrality >= 0 && centrality <= 5) {
FillProfile(corrconfigs.at(0), bootstrapArray[sampleIndex][kc22Nch05], globaltracks_nch);
FillProfile(corrconfigs.at(1), bootstrapArray[sampleIndex][kc24Nch05], globaltracks_nch);
FillProfile(corrconfigs.at(2), bootstrapArray[sampleIndex][kc26Nch05], globaltracks_nch);
FillProfile(corrconfigs.at(3), bootstrapArray[sampleIndex][kc28Nch05], globaltracks_nch);
FillProfile(corrconfigs.at(4), bootstrapArray[sampleIndex][kc22Nch05etagap], globaltracks_nch);
FillProfile(corrconfigs.at(5), bootstrapArray[sampleIndex][kc32Nch05], globaltracks_nch);
FillProfile(corrconfigs.at(6), bootstrapArray[sampleIndex][kc32Nch05etagap], globaltracks_nch);
FillProfile(corrconfigs.at(7), bootstrapArray[sampleIndex][kc34Nch05], globaltracks_nch);
}

// Filling Flow Container
for (uint l_ind = 0; l_ind < corrconfigs.size(); l_ind++) {
FillFC(corrconfigs.at(l_ind), centrality, l_Random);
}

} // End of process
PROCESS_SWITCH(flowGFWPbPb, processData, "Process analysis for Run 3 data", false);

// Filter the Reco tracks
Filter mytrackFilter = (nabs(aod::track::eta) < cfgCutEta) && (aod::track::pt > cfgCutPtMin) && (aod::track::pt < cfgCutPtMax) && (nabs(aod::track::dcaXY) < cfgCutDCAxy);
using MyTracks = soa::Filtered<soa::Join<aod::Tracks, aod::TracksExtra, aod::TrackSelection, aod::TracksDCA, aod::McTrackLabels>>;
using MyCollisions = soa::Join<aod::Collisions, aod::CentFT0Cs>;

void processReco(MyCollisions::iterator const& collision, MyTracks const& tracks, aod::McParticles const&)
{
registry.fill(HIST("eventCounter"), 0.5);
const auto centrality = collision.centFT0C();
registry.fill(HIST("hCenMCRec"), centrality);
for (const auto& track : tracks) {
if (track.tpcNClsCrossedRows() < 70)
continue;

if (track.has_mcParticle()) {
registry.fill(HIST("hPtMCRec"), track.pt());
if (centrality >0 && centrality<=5){
registry.fill(HIST("hPtMCRec05"), track.pt());
}
if (centrality >=50 && centrality<=60){
registry.fill(HIST("hPtMCRec5060"), track.pt());
}
PROCESS_SWITCH(flowGFWPbPb, processData, "Process analysis for Run 3 data", false);

// Filter the Reco tracks
Filter mytrackFilter = (nabs(aod::track::eta) < cfgCutEta) && (aod::track::pt > cfgCutPtMin) && (aod::track::pt < cfgCutPtMax) && (nabs(aod::track::dcaXY) < cfgCutDCAxy);
using MyTracks = soa::Filtered<soa::Join<aod::Tracks, aod::TracksExtra, aod::TrackSelection, aod::TracksDCA, aod::McTrackLabels>>;
using MyCollisions = soa::Join<aod::Collisions, aod::CentFT0Cs>;

void processReco(MyCollisions::iterator const& collision, MyTracks const& tracks, aod::McParticles const&)
{
registry.fill(HIST("eventCounter"), 0.5);
const auto centrality = collision.centFT0C();
registry.fill(HIST("hCenMCRec"), centrality);
for (const auto& track : tracks) {
if (track.tpcNClsCrossedRows() < 70)
continue;

if (track.has_mcParticle()) {
registry.fill(HIST("hPtMCRec"), track.pt());
if (centrality > 0 && centrality <= 5) {
registry.fill(HIST("hPtMCRec05"), track.pt());
}
if (centrality >= 50 && centrality <= 60) {
registry.fill(HIST("hPtMCRec5060"), track.pt());
}
}
}
PROCESS_SWITCH(flowGFWPbPb, processReco, "process reconstructed information", false);

// Filter for MCParticle
Filter particleFilter = (nabs(aod::mcparticle::eta) < cfgCutEta) && (aod::mcparticle::pt > cfgCutPtMin) && (aod::mcparticle::pt < cfgCutPtMax);
using MyMcParticles = soa::Filtered<aod::McParticles>;
using MyMcCollisionsFT0Cs = soa::Join<o2::aod::Collisions, o2::aod::CentFT0Cs>;

void processSim(aod::McCollision const&, soa::SmallGroups<soa::Join<o2::aod::Collisions, o2::aod::McCollisionLabels>> const& collisions, MyMcParticles const& mcParticles, MyMcCollisionsFT0Cs const& mcCollisionsFT0Cs )
{
if (collisions.size() > -1) {
registry.fill(HIST("mcEventCounter"), 0.5);
for (const auto& mcCollisionsFT0C : mcCollisionsFT0Cs) {
registry.fill(HIST("hCenMCGen"), mcCollisionsFT0C.centFT0C());
}
}
PROCESS_SWITCH(flowGFWPbPb, processReco, "process reconstructed information", false);

// Filter for MCParticle
Filter particleFilter = (nabs(aod::mcparticle::eta) < cfgCutEta) && (aod::mcparticle::pt > cfgCutPtMin) && (aod::mcparticle::pt < cfgCutPtMax);
using MyMcParticles = soa::Filtered<aod::McParticles>;
using MyMcCollisionsFT0Cs = soa::Join<o2::aod::Collisions, o2::aod::CentFT0Cs>;

void processSim(aod::McCollision const&, soa::SmallGroups<soa::Join<o2::aod::Collisions, o2::aod::McCollisionLabels>> const& collisions, MyMcParticles const& mcParticles, MyMcCollisionsFT0Cs const& mcCollisionsFT0Cs)
{
if (collisions.size() > -1) {
registry.fill(HIST("mcEventCounter"), 0.5);
for (const auto& mcCollisionsFT0C : mcCollisionsFT0Cs) {
registry.fill(HIST("hCenMCGen"), mcCollisionsFT0C.centFT0C());
}

for (const auto& mcCollisionsFT0C : mcCollisionsFT0Cs) {
const auto centrality = mcCollisionsFT0C.centFT0C();
for (const auto& mcParticle : mcParticles) {
registry.fill(HIST("hPtMCGen"), mcParticle.pt());
if (centrality >0 && centrality<=5){
registry.fill(HIST("hPtMCGen05"), mcParticle.pt());
}
if (centrality >=50 && centrality<=60){
registry.fill(HIST("hPtMCGen5060"), mcParticle.pt());
}
for (const auto& mcParticle : mcParticles) {
registry.fill(HIST("hPtMCGen"), mcParticle.pt());
if (centrality > 0 && centrality <= 5) {
registry.fill(HIST("hPtMCGen05"), mcParticle.pt());
}
if (centrality >= 50 && centrality <= 60) {
registry.fill(HIST("hPtMCGen5060"), mcParticle.pt());
}
}
}
}
PROCESS_SWITCH(flowGFWPbPb, processSim, "process pure simulation information", false);
}
PROCESS_SWITCH(flowGFWPbPb, processSim, "process pure simulation information", false);

}; // End of struct

Expand Down
Loading