Skip to content

[PWGCF] IdentifiedBfFilter Moved N Sigma Plots #10650

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
victor-gonzalez merged 6 commits into from
Mar 26, 2025
Merged

[PWGCF] IdentifiedBfFilter Moved N Sigma Plots #10650

Changes from all commits
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
2fdf4ad
Moved where MC N sigma histograms were filled to better measure the s…
bghanley1995 Mar 20, 2025
fe4a56d
Moved true N sigma plots earlier in the PID process. Added histograms…
bghanley1995 Mar 26, 2025
6e7db7b
Please consider the following formatting changes
alibuild Mar 26, 2025
aefaef5
Merge pull request #27 from alibuild/alibot-cleanup-10650
bghanley1995 Mar 26, 2025
40ceab8
Please consider the following formatting changes
alibuild Mar 26, 2025
45c1021
Merge pull request #28 from alibuild/alibot-cleanup-10650
bghanley1995 Mar 26, 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
94 changes: 69 additions & 25 deletions PWGCF/TwoParticleCorrelations/TableProducer/identifiedBfFilter.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,7 @@
/// \file identifiedBfFilter.cxx
/// \brief Filters collisions and tracks according to selection criteria
/// \author bghanley1995@gmail.com

#include "PWGCF/TwoParticleCorrelations/TableProducer/identifiedBfFilter.h"

#include <cmath>
Expand Down Expand Up @@ -156,8 +157,8 @@ TH2F* fhCompatibleCollisionsZVtxRms = nullptr;
TH2S* fhTruePIDMismatch = nullptr;
TH1S* fhTruePIDCorrect = nullptr;

TH2F* fhTrueNSigmaTPC[kIdBfNoOfSpecies] = {nullptr};
TH2F* fhTrueNSigmaTOF[kIdBfNoOfSpecies] = {nullptr};
std::vector<std::vector<TH2F*>> fhTrueNSigmaTPC = {o2::analysis::identifiedbffilter::kIdBfNoOfSpecies, {o2::analysis::identifiedbffilter::kIdBfNoOfSpecies, nullptr}};
std::vector<std::vector<TH2F*>> fhTrueNSigmaTOF = {o2::analysis::identifiedbffilter::kIdBfNoOfSpecies, {o2::analysis::identifiedbffilter::kIdBfNoOfSpecies, nullptr}};

TH1F* fhTrueCentMultB = nullptr;
TH1F* fhTrueCentMultA = nullptr;
Expand Down Expand Up @@ -1025,16 +1026,18 @@ struct IdentifiedBfFilterTracks {
79, -39.5, 39.5);
}

for (int sp = 0; sp < kIdBfNoOfSpecies; ++sp) {
fhTrueNSigmaTPC[sp] = new TH2F(TString::Format("fhTrueNSigmaTPC_%s", speciesName[sp]).Data(),
TString::Format("N #sigma from TPC vs P for generated %s;N #sigma;p (GeV/c)", speciesTitle[sp]).Data(),
48, -6, 6,
ptbins, ptlow, ptup);

fhTrueNSigmaTOF[sp] = new TH2F(TString::Format("fhTrueNSigmaTOF_%s", speciesName[sp]).Data(),
TString::Format("N #sigma from TOF vs P for generated %s;N #sigma;p (GeV/c)", speciesTitle[sp]).Data(),
48, -6, 6,
ptbins, ptlow, ptup);
for (int sp1 = 0; sp1 < kIdBfNoOfSpecies; ++sp1) {
for (int sp2 = 0; sp2 < kIdBfNoOfSpecies; ++sp2) {
fhTrueNSigmaTPC[sp1][sp2] = new TH2F(TString::Format("fhTrueNSigmaTPC%s_%s", speciesName[sp1], speciesName[sp2]).Data(),
TString::Format("N #sigma %s from TPC vs P for generated %s;N #sigma;p (GeV/c)", speciesTitle[sp1], speciesTitle[sp2]).Data(),
48, -6, 6,
ptbins, ptlow, ptup);

fhTrueNSigmaTOF[sp1][sp2] = new TH2F(TString::Format("fhTrueNSigmaTOF%s_%s", speciesName[sp1], speciesName[sp2]).Data(),
TString::Format("N #sigma %s from TOF vs P for generated %s;N #sigma;p (GeV/c)", speciesTitle[sp1], speciesTitle[sp2]).Data(),
48, -6, 6,
ptbins, ptlow, ptup);
}
}

/* add the hstograms to the output list */
Expand Down Expand Up @@ -1073,9 +1076,11 @@ struct IdentifiedBfFilterTracks {
fOutputList->Add(fhTrueNPosNegA[sp]);
fOutputList->Add(fhTrueDeltaNA[sp]);
}
for (int sp = 0; sp < kIdBfNoOfSpecies; ++sp) {
fOutputList->Add(fhTrueNSigmaTPC[sp]);
fOutputList->Add(fhTrueNSigmaTOF[sp]);
for (int sp1 = 0; sp1 < kIdBfNoOfSpecies; ++sp1) {
for (int sp2 = 0; sp2 < kIdBfNoOfSpecies; ++sp2) {
fOutputList->Add(fhTrueNSigmaTPC[sp1][sp2]);
fOutputList->Add(fhTrueNSigmaTOF[sp1][sp2]);
}
}
}
/* initialize access to the CCDB */
Expand All @@ -1090,7 +1095,9 @@ struct IdentifiedBfFilterTracks {
template <typename CollisionObjects, typename TrackObject>
int8_t selectTrackAmbiguousCheck(CollisionObjects const& collisions, TrackObject const& track);
template <typename ParticleObject>
inline void identifyPIDMismatch(ParticleObject const& particle, MatchRecoGenSpecies const& trkId, float tpcNSigma[kIdBfNoOfSpecies], float tofNSigma[kIdBfNoOfSpecies], float tpcInnerParam);
inline void identifyPIDMismatch(ParticleObject const& particle, MatchRecoGenSpecies const& trkId);
template <typename ParticleObject>
inline void identifyRealNSigma(ParticleObject const& particle, std::vector<float> tpcNSigma, std::vector<float> tofNSigma, float tpcInnerParam);
template <typename ParticleObject>
inline MatchRecoGenSpecies identifyParticle(ParticleObject const& particle);
template <typename TrackObject>
Expand Down Expand Up @@ -1338,7 +1345,7 @@ inline MatchRecoGenSpecies IdentifiedBfFilterTracks::identifyParticle(ParticleOb
}

template <typename ParticleObject>
inline void IdentifiedBfFilterTracks::identifyPIDMismatch(ParticleObject const& particle, MatchRecoGenSpecies const& trkId, float tpcNSigma[kIdBfNoOfSpecies], float tofNSigma[kIdBfNoOfSpecies], float tpcInnerParam)
inline void IdentifiedBfFilterTracks::identifyPIDMismatch(ParticleObject const& particle, MatchRecoGenSpecies const& trkId)
{
MatchRecoGenSpecies realPID = kWrongSpecies;
int pdgcode = std::fabs(particle.pdgCode());
Expand Down Expand Up @@ -1369,11 +1376,44 @@ inline void IdentifiedBfFilterTracks::identifyPIDMismatch(ParticleObject const&
} else {
fhTruePIDMismatch->Fill(realPID, trkId);
}
}
}

fhTrueNSigmaTPC[realPID]->Fill(tpcNSigma[realPID], tpcInnerParam);
fhTrueNSigmaTOF[realPID]->Fill(tofNSigma[realPID], tpcInnerParam);
template <typename ParticleObject>
inline void IdentifiedBfFilterTracks::identifyRealNSigma(ParticleObject const& particle, std::vector<float> tpcNSigma, std::vector<float> tofNSigma, float tpcInnerParam)
{

// fill histo based on realTPC and tpcNSigma value for center align
MatchRecoGenSpecies realPID = kWrongSpecies;
int pdgcode = std::fabs(particle.pdgCode());
switch (pdgcode) {
case pdgcodeEl:
Copy link
Collaborator

@vkucera vkucera Mar 26, 2025

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Suggested change
case pdgcodeEl:
case kElectron:

These should be fixed.

Copy link
Collaborator

@victor-gonzalez victor-gonzalez Mar 26, 2025

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@bghanley1995, @vkucera is right!
You don't need to redefine

constexpr int pdgcodeEl = 11;
constexpr int pdgcodePi = 211;
constexpr int pdgcodeKa = 321;
constexpr int pdgcodePr = 2212;

what is already defined in TPDGCode.h
Please, address this in the next iteration

realPID = kIdBfElectron;
break;
case pdgcodePi:
realPID = kIdBfPion;
break;
case pdgcodeKa:
realPID = kIdBfKaon;
break;
case pdgcodePr:
realPID = kIdBfProton;
break;
default:
if (traceOutOfSpeciesParticles) {
LOGF(info, "Wrong particle passed selection cuts. PDG code: %d", pdgcode);
}
realPID = kWrongSpecies;
break;
}
if (!(realPID < 0)) {
fhTrueNSigmaTPC[kIdBfElectron][realPID]->Fill(tpcNSigma[kIdBfElectron], tpcInnerParam);
fhTrueNSigmaTOF[kIdBfElectron][realPID]->Fill(tofNSigma[kIdBfElectron], tpcInnerParam);
fhTrueNSigmaTPC[kIdBfPion][realPID]->Fill(tpcNSigma[kIdBfPion], tpcInnerParam);
fhTrueNSigmaTOF[kIdBfPion][realPID]->Fill(tofNSigma[kIdBfPion], tpcInnerParam);
fhTrueNSigmaTPC[kIdBfKaon][realPID]->Fill(tpcNSigma[kIdBfKaon], tpcInnerParam);
fhTrueNSigmaTOF[kIdBfKaon][realPID]->Fill(tofNSigma[kIdBfKaon], tpcInnerParam);
fhTrueNSigmaTPC[kIdBfProton][realPID]->Fill(tpcNSigma[kIdBfProton], tpcInnerParam);
fhTrueNSigmaTOF[kIdBfProton][realPID]->Fill(tofNSigma[kIdBfProton], tpcInnerParam);
}
}

Expand Down Expand Up @@ -1409,8 +1449,8 @@ void fillNSigmaHistos(TrackObject const& track)

fhNSigmaTOF[kIdBfElectron]->Fill(actualTOFNSigma[kIdBfElectron], track.tpcInnerParam());
fhNSigmaTOF[kIdBfPion]->Fill(actualTOFNSigma[kIdBfPion], track.tpcInnerParam());
fhNSigmaTOF[kIdBfKaon]->Fill(track.tofNSigmaKa(), track.tpcInnerParam());
fhNSigmaTOF[kIdBfProton]->Fill(track.tofNSigmaPr(), track.tpcInnerParam());
fhNSigmaTOF[kIdBfKaon]->Fill(actualTOFNSigma[kIdBfKaon], track.tpcInnerParam());
fhNSigmaTOF[kIdBfProton]->Fill(actualTOFNSigma[kIdBfProton], track.tpcInnerParam());

fhNSigmaCombo[kIdBfElectron]->Fill(sqrtf(actualTOFNSigma[kIdBfElectron] * actualTOFNSigma[kIdBfElectron] + actualTPCNSigma[kIdBfElectron] * actualTPCNSigma[kIdBfElectron]), track.tpcInnerParam());
fhNSigmaCombo[kIdBfPion]->Fill(sqrtf(actualTOFNSigma[kIdBfPion] * actualTOFNSigma[kIdBfPion] + actualTPCNSigma[kIdBfPion] * actualTPCNSigma[kIdBfPion]), track.tpcInnerParam());
Expand All @@ -1429,14 +1469,14 @@ inline MatchRecoGenSpecies IdentifiedBfFilterTracks::identifyTrack(TrackObject c

fillNSigmaHistos(track);

float actualTPCNSigma[kIdBfNoOfSpecies];
std::vector<float> actualTPCNSigma(kIdBfNoOfSpecies, 0.);

actualTPCNSigma[kIdBfElectron] = track.tpcNSigmaEl();
actualTPCNSigma[kIdBfPion] = track.tpcNSigmaPi();
actualTPCNSigma[kIdBfKaon] = track.tpcNSigmaKa();
actualTPCNSigma[kIdBfProton] = track.tpcNSigmaPr();

float actualTOFNSigma[kIdBfNoOfSpecies];
std::vector<float> actualTOFNSigma(kIdBfNoOfSpecies, 0.);

actualTOFNSigma[kIdBfElectron] = track.tofNSigmaEl();
actualTOFNSigma[kIdBfPion] = track.tofNSigmaPi();
Expand All @@ -1445,6 +1485,10 @@ inline MatchRecoGenSpecies IdentifiedBfFilterTracks::identifyTrack(TrackObject c

float nsigmas[kIdBfNoOfSpecies];

if constexpr (framework::has_type_v<aod::mctracklabel::McParticleId, typename TrackObject::all_columns>) {
identifyRealNSigma(track.template mcParticle_as<aod::McParticles>(), actualTPCNSigma, actualTOFNSigma, track.tpcInnerParam());
}

if (loadfromccdb) {
for (int iSp = 0; iSp < kIdBfNoOfSpecies; iSp++) {
actualTPCNSigma[iSp] = actualTPCNSigma[iSp] - fhNSigmaCorrection[iSp]->GetBinContent(fhNSigmaCorrection[iSp]->FindBin(track.tpcInnerParam()));
Expand Down Expand Up @@ -1519,7 +1563,7 @@ inline MatchRecoGenSpecies IdentifiedBfFilterTracks::identifyTrack(TrackObject c
fhNSigmaTPCIdTrks[spMinNSigma]->Fill(actualTPCNSigma[spMinNSigma], track.tpcInnerParam());

if constexpr (framework::has_type_v<aod::mctracklabel::McParticleId, typename TrackObject::all_columns>) {
identifyPIDMismatch(track.template mcParticle_as<aod::McParticles>(), spMinNSigma, actualTPCNSigma, actualTOFNSigma, track.tpcInnerParam());
identifyPIDMismatch(track.template mcParticle_as<aod::McParticles>(), spMinNSigma);
}
return spMinNSigma;
}
Expand Down
Loading