Skip to content

[PWGCF] Fix missing indices issue with derive data table, added cuts #9545

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
shouqiye merged 1 commit into from
Jan 27, 2025
Merged

[PWGCF] Fix missing indices issue with derive data table, added cuts #9545

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
54 changes: 44 additions & 10 deletions PWGCF/EbyEFluctuations/Tasks/netprotonCumulantsMc.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -88,22 +88,22 @@ DECLARE_SOA_TABLE(ProtRecCollEbyeTables, "AOD", "PROTRECCOLLEBYETABLE",
rec_ebyecolltable::AntiProtNoRec);
using ProtRecCollEbyeTable = ProtRecCollEbyeTables::iterator;

DECLARE_SOA_TABLE(ProtRecCollTables, "AOD", "PROTRECCOLLTABLE",
DECLARE_SOA_TABLE(RecCollTables, "AOD", "RECCOLLTABLE",
o2::soa::Index<>,
rec_ebyecolltable::CentralityRec);
using ProtRecCollTable = ProtRecCollTables::iterator;
using RecCollTable = RecCollTables::iterator;

namespace rec_ebyetracktable
{
DECLARE_SOA_INDEX_COLUMN(ProtRecCollTable, protRecCollTable);
DECLARE_SOA_INDEX_COLUMN(RecCollTable, recCollTable);
DECLARE_SOA_COLUMN(Pt, pt, float);
DECLARE_SOA_COLUMN(Eta, eta, float);
DECLARE_SOA_COLUMN(Charge, charge, int);
} // namespace rec_ebyetracktable

DECLARE_SOA_TABLE(ProtRecCompleteEbyeTables, "AOD", "PROTRECCOMPLETEEBYETABLE",
o2::soa::Index<>,
rec_ebyetracktable::ProtRecCollTableId,
rec_ebyetracktable::RecCollTableId,
rec_ebyetracktable::Pt,
rec_ebyetracktable::Eta,
rec_ebyetracktable::Charge);
Expand Down Expand Up @@ -143,14 +143,18 @@ struct NetprotonCumulantsMc {

HistogramRegistry histos{"Histos", {}, OutputObjHandlingPolicy::AnalysisObject};

// Filter command for rec (data/MC)***********
Filter collisionFilter = nabs(aod::collision::posZ) < cfgCutVertex && (aod::evsel::sel8 == true);
// Filter command for rec (data)***********
Filter collisionFilter = nabs(aod::collision::posZ) < cfgCutVertex;
Filter trackFilter = (nabs(aod::track::eta) < 0.8f) && (aod::track::pt > cfgCutPtLower) && (aod::track::pt < 5.0f) && ((requireGlobalTrackInFilter()) || (aod::track::isGlobalTrackSDD == (uint8_t) true)) && (aod::track::tpcChi2NCl < cfgCutTpcChi2NCl) && (aod::track::itsChi2NCl < cfgCutItsChi2NCl) && (aod::track::dcaZ < cfgCutDCAz) && (aod::track::dcaXY < cfgCutDCAxy);

// filtering collisions and tracks for real data***********
using AodCollisions = soa::Filtered<soa::Join<aod::Collisions, aod::EvSels, aod::CentFV0As, aod::CentFT0Ms, aod::CentFT0As, aod::CentFT0Cs, aod::CentFDDMs>>;
using AodTracks = soa::Filtered<soa::Join<aod::Tracks, aod::TrackSelection, aod::TracksExtra, aod::TracksDCA, aod::pidTPCFullPr, aod::pidTOFFullPr, aod::pidTPCFullKa, aod::pidTOFFullKa, aod::pidTPCFullPi, aod::pidTOFFullPi, aod::pidTPCFullEl, aod::pidTOFFullEl>>;

// // Filter command for rec (MC)***********
// Filter collisionMcFilter = nabs(aod::collision::posZ) < cfgCutVertex && (aod::evsel::sel8 == true);
// Filter trackMcFilter = (nabs(aod::track::eta) < 0.8f) && (aod::track::pt > cfgCutPtLower) && (aod::track::pt < 5.0f) && ((requireGlobalTrackInFilter()) || (aod::track::isGlobalTrackSDD == (uint8_t) true)) && (aod::track::tpcChi2NCl < cfgCutTpcChi2NCl) && (aod::track::itsChi2NCl < cfgCutItsChi2NCl) && (aod::track::dcaZ < cfgCutDCAz) && (aod::track::dcaXY < cfgCutDCAxy);

// filtering collisions and tracks for MC rec data***********
using MyMCRecCollisions = soa::Filtered<soa::Join<aod::Collisions, aod::EvSels, aod::CentFV0As, aod::CentFT0Ms, aod::CentFT0As, aod::CentFT0Cs, aod::CentFDDMs, aod::McCollisionLabels>>;
using MyMCRecCollision = MyMCRecCollisions::iterator;
Expand Down Expand Up @@ -192,6 +196,12 @@ struct NetprotonCumulantsMc {
histos.add("hrecEtaAll", "Reconstructed All particles;#eta", kTH1F, {{100, -2.01, 2.01}});
histos.add("hrecEtaProton", "Reconstructed Proton;#eta", kTH1F, {{100, -2.01, 2.01}});
histos.add("hrecEtaAntiproton", "Reconstructed Antiprotons;#eta", kTH1F, {{100, -2.01, 2.01}});
histos.add("hrecDcaXYAll", "Reconstructed All particles;DCA_{xy} (in cm)", kTH1F, {{400, -2.01, 2.01}});
histos.add("hrecDcaXYProton", "Reconstructed Proton;DCA_{xy} (in cm)", kTH1F, {{400, -2.01, 2.01}});
histos.add("hrecDcaXYAntiproton", "Reconstructed Antiprotons;DCA_{xy} (in cm)", kTH1F, {{400, -2.01, 2.01}});
histos.add("hrecDcaZAll", "Reconstructed All particles;DCA_{z} (in cm)", kTH1F, {{400, -2.01, 2.01}});
histos.add("hrecDcaZProton", "Reconstructed Proton;DCA_{z} (in cm)", kTH1F, {{400, -2.0, 2.0}});
histos.add("hrecDcaZAntiproton", "Reconstructed Antiprotons;DCA_{z} (in cm)", kTH1F, {{400, -2.0, 2.0}});
histos.add("hrecPtDistProtonVsCentrality", "Reconstructed proton number vs centrality in 2D", kTH2F, {ptAxis, centAxis});
histos.add("hrecPtDistAntiprotonVsCentrality", "Reconstructed antiproton number vs centrality in 2D", kTH2F, {ptAxis, centAxis});
histos.add("hrecNetProtonVsCentrality", "Reconstructed net-proton number vs centrality in 2D", kTH2F, {netprotonAxis, centAxis});
Expand All @@ -209,6 +219,9 @@ struct NetprotonCumulantsMc {
histos.add("hgenPtAll", "Generated All particles;#it{p}_{T} (GeV/#it{c})", kTH1F, {ptAxis});
histos.add("hgenPtProton", "Generated Protons;#it{p}_{T} (GeV/#it{c})", kTH1F, {ptAxis});
histos.add("hgenPtAntiproton", "Generated Antiprotons;#it{p}_{T} (GeV/#it{c})", kTH1F, {ptAxis});
histos.add("hrecPartPtAll", "Reconstructed All particles filled mcparticle pt;#it{p}_{T} (GeV/#it{c})", kTH1F, {ptAxis});
histos.add("hrecPartPtProton", "Reconstructed Protons filled mcparticle pt;#it{p}_{T} (GeV/#it{c})", kTH1F, {ptAxis});
histos.add("hrecPartPtAntiproton", "Reconstructed Antiprotons filled mcparticle pt;#it{p}_{T} (GeV/#it{c})", kTH1F, {ptAxis});
histos.add("hgenPhiAll", "Generated All particles;#phi", kTH1F, {{100, 0., 7.}});
histos.add("hgenPhiProton", "Generated Protons;#phi", kTH1F, {{100, 0., 7.}});
histos.add("hgenPhiAntiproton", "Generated Antiprotons;#phi", kTH1F, {{100, 0., 7.}});
Expand All @@ -228,6 +241,12 @@ struct NetprotonCumulantsMc {
}
}

// template <typename T>
// bool selectionTrack(const T& candidate)
// {

// }

template <typename T>
bool selectionPIDold(const T& candidate)
{
Expand Down Expand Up @@ -376,11 +395,14 @@ struct NetprotonCumulantsMc {
PROCESS_SWITCH(NetprotonCumulantsMc, processMCGen, "Process Generated", true);

Produces<aod::ProtRecCollEbyeTables> recEbyeCollisions; //! MC Rec table creation
Produces<aod::ProtRecCollTables> recCollisions; //! MC Rec table creation
Produces<aod::RecCollTables> recCollisions; //! MC Rec table creation
Produces<aod::ProtRecCompleteEbyeTables> recEbyeCompleteCollisions; //! MC Rec table creation with tracks

void processMCRec(MyMCRecCollision const& collision, MyMCTracks const& tracks, aod::McCollisions const&, aod::McParticles const&)
{
if (!collision.sel8()) {
return;
}
if (!collision.has_mcCollision()) {
return;
}
Expand All @@ -405,10 +427,16 @@ struct NetprotonCumulantsMc {
}

auto particle = track.mcParticle();
if ((particle.pt() < cfgCutPtLower) || (particle.pt() > 5.0f) || (std::abs(particle.eta()) > 0.8f)) {
continue;
}
if (particle.isPhysicalPrimary()) {
histos.fill(HIST("hrecPtAll"), particle.pt());
histos.fill(HIST("hrecPartPtAll"), particle.pt());
histos.fill(HIST("hrecPtAll"), track.pt());
histos.fill(HIST("hrecEtaAll"), particle.eta());
histos.fill(HIST("hrecPhiAll"), particle.phi());
histos.fill(HIST("hrecDcaXYAll"), track.dcaXY());
histos.fill(HIST("hrecDcaZAll"), track.dcaZ());

bool trackSelected = false;
if (cfgPIDchoice == 0)
Expand All @@ -419,21 +447,27 @@ struct NetprotonCumulantsMc {
if (trackSelected) {
recEbyeCompleteCollisions(recCollisions.lastIndex(), particle.pt(), particle.eta(), track.sign());
if (track.sign() > 0) {
histos.fill(HIST("hrecPtProton"), particle.pt()); //! hist for p rec
histos.fill(HIST("hrecPartPtProton"), particle.pt()); //! hist for p rec
histos.fill(HIST("hrecPtProton"), track.pt()); //! hist for p rec
histos.fill(HIST("hrecPtDistProtonVsCentrality"), particle.pt(), cent);
histos.fill(HIST("hrecEtaProton"), particle.eta());
histos.fill(HIST("hrecPhiProton"), particle.phi());
histos.fill(HIST("hrecDcaXYProton"), track.dcaXY());
histos.fill(HIST("hrecDcaZProton"), track.dcaZ());
if (particle.pt() < cfgCutPtUpper)
nProt = nProt + 1.0;
if (particle.pdgCode() == 2212) {
histos.fill(HIST("hrecTruePtProton"), particle.pt()); //! hist for p purity
}
}
if (track.sign() < 0) {
histos.fill(HIST("hrecPtAntiproton"), particle.pt()); //! hist for anti-p rec
histos.fill(HIST("hrecPartPtAntiproton"), particle.pt()); //! hist for anti-p rec
histos.fill(HIST("hrecPtAntiproton"), track.pt()); //! hist for anti-p rec
histos.fill(HIST("hrecPtDistAntiprotonVsCentrality"), particle.pt(), cent);
histos.fill(HIST("hrecEtaAntiproton"), particle.eta());
histos.fill(HIST("hrecPhiAntiproton"), particle.phi());
histos.fill(HIST("hrecDcaXYAntiproton"), track.dcaXY());
histos.fill(HIST("hrecDcaZAntiproton"), track.dcaZ());
if (particle.pt() < cfgCutPtUpper)
nAntiprot = nAntiprot + 1.0;
if (particle.pdgCode() == -2212) {
Expand Down
Loading