Skip to content

[spectraTOF] add more DCA histograms #7754

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
njacazio merged 11 commits into from
Sep 27, 2024
Merged

[spectraTOF] add more DCA histograms #7754

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
11 commits
Select commit Hold shift + click to select a range
e697403
Update on spectraTOF task.
mtorresc15 Sep 20, 2024
7696367
Update on spectraTOF task.
mtorresc15 Sep 20, 2024
d38fc16
Update on spectraTOF task.
mtorresc15 Sep 20, 2024
a0d0c3e
Update on spectraTOF.cxx task.
mtorresc15 Sep 20, 2024
ac7995c
Update on spectraTOF.cxx task.
mtorresc15 Sep 20, 2024
68155fd
Update
njacazio Sep 21, 2024
ab3a615
FIx
njacazio Sep 22, 2024
b04727c
Merge pull request #4 from njacazio/m-master
mtorresc15 Sep 23, 2024
09eeb14
Update spectraTOF.h
njacazio Sep 23, 2024
58890f8
Merge branch 'AliceO2Group:master' into master
mtorresc15 Sep 23, 2024
7449eee
Update spectraTOF.cxx
njacazio Sep 27, 2024
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
8 changes: 4 additions & 4 deletions PWGLF/DataModel/spectraTOF.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -398,11 +398,11 @@ DECLARE_SOA_DYNAMIC_COLUMN(TRDSignal, trdSignal, //! Dummy
[](float /*v*/) -> float { return 0.f; });
DECLARE_SOA_DYNAMIC_COLUMN(P, p, [](float signedpt, float eta) -> float { return std::abs(signedpt) * cosh(eta); });
DECLARE_SOA_DYNAMIC_COLUMN(TrackType, trackType, [](float /*v*/) -> uint8_t { return o2::aod::track::TrackTypeEnum::Track; });
DECLARE_SOA_COLUMN(IsGlobalTrack, isGlobalTrack, bool); // if a track passed the isGlobalTrack requirement
DECLARE_SOA_COLUMN(IsGlobalTrackWoDCA, isGlobalTrackWoDCA, bool); // if a track passed the isGlobalTrackWoDCA requirement
DECLARE_SOA_DYNAMIC_COLUMN(Flags, flags, [](float /*v*/) -> uint32_t { return 0; }); // Dummy
DECLARE_SOA_COLUMN(IsGlobalTrack, isGlobalTrack, bool); // if a track passed the isGlobalTrack requirement
DECLARE_SOA_COLUMN(IsGlobalTrackWoDCA, isGlobalTrackWoDCA, bool); // if a track passed the isGlobalTrackWoDCA requirement
DECLARE_SOA_DYNAMIC_COLUMN(Flags, flags, [](float /*v*/) -> uint32_t { return 0; }); // Dummy
DECLARE_SOA_DYNAMIC_COLUMN(TRDPattern, trdPattern, [](float /*v*/) -> uint8_t { return 0; }); // Dummy
DECLARE_SOA_DYNAMIC_COLUMN(Rapidity, rapidity, //! Track rapidity, computed under the mass assumption given as input
DECLARE_SOA_DYNAMIC_COLUMN(Rapidity, rapidity, //! Track rapidity, computed under the mass assumption given as input
[](float signedPt, float eta, float mass) -> float {
const auto pt = std::abs(signedPt);
const auto p = std::abs(signedPt) * cosh(eta);
Expand Down
106 changes: 91 additions & 15 deletions PWGLF/Tasks/Nuspex/spectraTOF.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -36,7 +36,7 @@
#include "Framework/O2DatabasePDGPlugin.h"
#include "PWGLF/Utils/inelGt.h"
#include "PWGLF/DataModel/mcCentrality.h"

#include "Common/Core/RecoDecay.h"
#include "TPDGCode.h"

using namespace o2;
Expand All @@ -52,6 +52,16 @@ std::array<std::shared_ptr<TH3>, NpCharge> hDcaXYZMat;
std::array<std::shared_ptr<TH2>, NpCharge> hDcaXYWrongCollisionPrm;
std::array<std::shared_ptr<TH2>, NpCharge> hDcaXYWrongCollisionStr;
std::array<std::shared_ptr<TH2>, NpCharge> hDcaXYWrongCollisionMat;
std::array<std::shared_ptr<TH2>, NpCharge> hDcaXYMC; // DCA xy in the MC
std::array<std::shared_ptr<TH2>, NpCharge> hDcaZMC; // DCA z in the MC
std::array<std::shared_ptr<TH2>, NpCharge> hDcaXYMCD0; // DCA xy in the MC for particles from D0
std::array<std::shared_ptr<TH2>, NpCharge> hDcaZMCD0; // DCA z in the MC for particles from D0
std::array<std::shared_ptr<TH2>, NpCharge> hDcaXYMCCharm; // DCA xy in the MC for particles from charm
std::array<std::shared_ptr<TH2>, NpCharge> hdcaZMCCharm; // DCA z in the MC for particles from charm
std::array<std::shared_ptr<TH2>, NpCharge> hDcaXYMCBeauty; // DCA xy in the MC for particles from beauty
std::array<std::shared_ptr<TH2>, NpCharge> hDcaZMCBeauty; // DCA z in the MC for particles from beauty
std::array<std::shared_ptr<TH2>, NpCharge> hDcaXYMCNotHF; // DCA xy in the MC for particles from not a HF
std::array<std::shared_ptr<TH2>, NpCharge> hDcaZMCNotHF; // DCA z in the MC for particles from not a HF

// Spectra task
struct tofSpectra {
Expand Down Expand Up @@ -83,6 +93,7 @@ struct tofSpectra {
Configurable<bool> enableTPCTOFHistograms{"enableTPCTOFHistograms", true, "Enables TPC TOF histograms"};
Configurable<bool> enableDCAxyzHistograms{"enableDCAxyzHistograms", false, "Enables DCAxyz correlation histograms"};
Configurable<bool> enableDCAxyphiHistograms{"enableDCAxyphiHistograms", false, "Enables DCAxyphi correlation histograms"};
Configurable<bool> enableDCAvsmotherHistograms{"enableDCAvsmotherHistograms", false, "Enables DCA vs mother histograms"};

struct : ConfigurableGroup {
ConfigurableAxis binsPt{"binsPt", {VARIABLE_WIDTH, 0.0, 0.1, 0.12, 0.14, 0.16, 0.18, 0.2, 0.25, 0.3, 0.35, 0.4, 0.45, 0.5, 0.55, 0.6, 0.65, 0.7, 0.75, 0.8, 0.85, 0.9, 0.95, 1.0, 1.1, 1.2, 1.3, 1.4, 1.5, 1.6, 1.7, 1.8, 1.9, 2.0, 2.2, 2.4, 2.6, 2.8, 3.0, 3.2, 3.4, 3.6, 3.8, 4.0, 4.2, 4.4, 4.6, 4.8, 5.0}, "Binning of the pT axis"};
Expand Down Expand Up @@ -611,17 +622,19 @@ struct tofSpectra {
histos.add(hpt_den_prm_mcgoodev[i].data(), pTCharge[i], kTH2D, {ptAxis, multAxis});
histos.add(hpt_den_prm_mcbadev[i].data(), pTCharge[i], kTH2D, {ptAxis, multAxis});

const std::string cpName = Form("/%s/%s", (i < Np) ? "pos" : "neg", pN[i % Np]);
if (enableDCAxyzHistograms) {
hDcaXYZPrm[i] = histos.add<TH3>(Form("dcaprm/%s/%s", (i < Np) ? "pos" : "neg", pN[i % Np]), pTCharge[i], kTH3D, {ptAxis, dcaXyAxis, dcaZAxis});
hDcaXYZStr[i] = histos.add<TH3>(Form("dcastr/%s/%s", (i < Np) ? "pos" : "neg", pN[i % Np]), pTCharge[i], kTH3D, {ptAxis, dcaXyAxis, dcaZAxis});
hDcaXYZMat[i] = histos.add<TH3>(Form("dcamat/%s/%s", (i < Np) ? "pos" : "neg", pN[i % Np]), pTCharge[i], kTH3D, {ptAxis, dcaXyAxis, dcaZAxis});
hDcaXYZPrm[i] = histos.add<TH3>("dcaprm" + cpName, pTCharge[i], kTH3D, {ptAxis, dcaXyAxis, dcaZAxis});
hDcaXYZStr[i] = histos.add<TH3>("dcastr" + cpName, pTCharge[i], kTH3D, {ptAxis, dcaXyAxis, dcaZAxis});
hDcaXYZMat[i] = histos.add<TH3>("dcamat" + cpName, pTCharge[i], kTH3D, {ptAxis, dcaXyAxis, dcaZAxis});
if (enableDcaGoodEvents) {
histos.add(hdcaxyprmgoodevs[i].data(), pTCharge[i], kTH3D, {ptAxis, dcaXyAxis, dcaZAxis});
}
} else {
hDcaXYWrongCollisionPrm[i] = histos.add<TH2>(Form("dcaxywrongcollprm/%s/%s", (i < Np) ? "pos" : "neg", pN[i % Np]), pTCharge[i], kTH2D, {ptAxis, dcaXyAxis});
hDcaXYWrongCollisionStr[i] = histos.add<TH2>(Form("dcaxywrongcollstr/%s/%s", (i < Np) ? "pos" : "neg", pN[i % Np]), pTCharge[i], kTH2D, {ptAxis, dcaXyAxis});
hDcaXYWrongCollisionMat[i] = histos.add<TH2>(Form("dcaxywrongcollmat/%s/%s", (i < Np) ? "pos" : "neg", pN[i % Np]), pTCharge[i], kTH2D, {ptAxis, dcaXyAxis});
hDcaXYWrongCollisionPrm[i] = histos.add<TH2>("dcaxywrongcollprm" + cpName, pTCharge[i], kTH2D, {ptAxis, dcaXyAxis});
hDcaXYWrongCollisionStr[i] = histos.add<TH2>("dcaxywrongcollstr" + cpName, pTCharge[i], kTH2D, {ptAxis, dcaXyAxis});
hDcaXYWrongCollisionMat[i] = histos.add<TH2>("dcaxywrongcollmat" + cpName, pTCharge[i], kTH2D, {ptAxis, dcaXyAxis});

histos.add(hdcaxyprm[i].data(), pTCharge[i], kTH2D, {ptAxis, dcaXyAxis});
histos.add(hdcazprm[i].data(), pTCharge[i], kTH2D, {ptAxis, dcaZAxis});
histos.add(hdcaxystr[i].data(), pTCharge[i], kTH2D, {ptAxis, dcaXyAxis});
Expand All @@ -632,6 +645,18 @@ struct tofSpectra {
histos.add(hdcaxyprmgoodevs[i].data(), pTCharge[i], kTH2D, {ptAxis, dcaXyAxis});
histos.add(hdcazprmgoodevs[i].data(), pTCharge[i], kTH2D, {ptAxis, dcaZAxis});
}
if (enableDCAvsmotherHistograms) {
hDcaXYMC[i] = histos.add<TH2>("dcaxymc" + cpName, pTCharge[i], kTH2D, {ptAxis, dcaXyAxis});
hDcaZMC[i] = histos.add<TH2>("dcazmc" + cpName, pTCharge[i], kTH2D, {ptAxis, dcaZAxis});
hDcaXYMCNotHF[i] = histos.add<TH2>("dcaxynothf" + cpName, pTCharge[i], kTH2D, {ptAxis, dcaXyAxis});
hDcaZMCNotHF[i] = histos.add<TH2>("dcaznothf" + cpName, pTCharge[i], kTH2D, {ptAxis, dcaZAxis});
hDcaXYMCD0[i] = histos.add<TH2>("dcaxyD0" + cpName, pTCharge[i], kTH2D, {ptAxis, dcaXyAxis});
hDcaZMCD0[i] = histos.add<TH2>("dcazD0" + cpName, pTCharge[i], kTH2D, {ptAxis, dcaZAxis});
hDcaXYMCCharm[i] = histos.add<TH2>("dcaxycharm" + cpName, pTCharge[i], kTH2D, {ptAxis, dcaXyAxis});
hdcaZMCCharm[i] = histos.add<TH2>("dcazcharm" + cpName, pTCharge[i], kTH2D, {ptAxis, dcaZAxis});
hDcaXYMCBeauty[i] = histos.add<TH2>("dcaxybeauty" + cpName, pTCharge[i], kTH2D, {ptAxis, dcaXyAxis});
hDcaZMCBeauty[i] = histos.add<TH2>("dcazbeauty" + cpName, pTCharge[i], kTH2D, {ptAxis, dcaZAxis});
}
}

// Mismatched info
Expand Down Expand Up @@ -1314,7 +1339,7 @@ struct tofSpectra {
histos.fill(HIST("nsigmatof/test_occupancy/neg/pr"), track.pt(), nsigmaTOFPr, multiplicity, occupancy);

} // track
} // process function
} // process function
PROCESS_SWITCH(tofSpectra, processOccupancy, "check for occupancy plots", false);

void processStandard(CollisionCandidates::iterator const& collision,
Expand Down Expand Up @@ -1525,7 +1550,10 @@ struct tofSpectra {

using RecoMCCollisions = soa::Join<aod::Collisions, aod::McCollisionLabels, aod::EvSels, aod::CentFT0As, aod::CentFT0Cs, aod::TPCMults, aod::PVMults, aod::MultZeqs, aod::CentFT0Ms>; // RD
template <std::size_t i, typename TrackType, typename ParticleType>
void fillTrackHistograms_MC(TrackType const& track, ParticleType const& mcParticle, RecoMCCollisions::iterator const& collision)
void fillTrackHistograms_MC(TrackType const& track,
ParticleType::iterator const& mcParticle,
RecoMCCollisions::iterator const& collision,
ParticleType const& mcParticles)
{
if (!isParticleEnabled<i>()) { // Check if the particle is enabled
return;
Expand All @@ -1549,7 +1577,12 @@ struct tofSpectra {
if (std::abs(mcParticle.y()) > trkselOptions.cfgCutY) {
return;
}
if (!mcParticle.isPhysicalPrimary()) {

if (enableDCAvsmotherHistograms) {
hDcaXYMC[i]->Fill(track.pt(), track.dcaXY());
hDcaZMC[i]->Fill(track.pt(), track.dcaZ());
}
if (!mcParticle.isPhysicalPrimary()) { // Secondaries (weak decays and material)
if (mcParticle.getProcess() == 4) {
if (enableDCAxyzHistograms) {
hDcaXYZStr[i]->Fill(track.pt(), track.dcaXY(), track.dcaZ());
Expand All @@ -1565,18 +1598,61 @@ struct tofSpectra {
histos.fill(HIST(hdcazmat[i]), track.pt(), track.dcaZ());
}
}
} else {
} else { // Primaries
if (enableDCAxyzHistograms) {
hDcaXYZPrm[i]->Fill(track.pt(), track.dcaXY(), track.dcaZ());
if (enableDcaGoodEvents.value && collision.has_mcCollision()) {
histos.fill(HIST(hdcaxyprmgoodevs[i]), track.pt(), track.dcaXY(), track.dcaZ());
}
} else {
// DCAxy for all primaries
histos.fill(HIST(hdcaxyprm[i]), track.pt(), track.dcaXY());
histos.fill(HIST(hdcazprm[i]), track.pt(), track.dcaZ());
if (enableDcaGoodEvents.value && collision.has_mcCollision()) {
histos.fill(HIST(hdcaxyprmgoodevs[i]), track.pt(), track.dcaXY());
histos.fill(HIST(hdcazprmgoodevs[i]), track.pt(), track.dcaZ());
}
if (enableDcaGoodEvents.value && collision.has_mcCollision()) {
histos.fill(HIST(hdcaxyprmgoodevs[i]), track.pt(), track.dcaXY());
histos.fill(HIST(hdcazprmgoodevs[i]), track.pt(), track.dcaZ());
}

if (enableDCAvsmotherHistograms) {
bool IsD0Mother = false;
bool IsCharmMother = false;
bool IsBeautyMother = false;

if (mcParticle.has_mothers()) {
const int charmOrigin = RecoDecay::getCharmHadronOrigin(mcParticles, mcParticle, false);
for (const auto& mother : mcParticle.template mothers_as<aod::McParticles>()) {
const int motherPdgCode = mother.pdgCode();
if (motherPdgCode == 421) {
IsD0Mother = true;
}
if (charmOrigin == RecoDecay::OriginType::NonPrompt) {
if ((motherPdgCode) / 1000 == 5 || (motherPdgCode) / 100 == 5) {
IsBeautyMother = true;
}
}
if (charmOrigin == RecoDecay::OriginType::Prompt) {
if ((motherPdgCode) / 1000 == 4 || (motherPdgCode) / 100 == 4) {
IsCharmMother = true;
}
}
}
}
if (IsD0Mother) {
hDcaXYMCD0[i]->Fill(track.pt(), track.dcaXY());
hDcaZMCD0[i]->Fill(track.pt(), track.dcaZ());
}
if (IsCharmMother) {
hDcaXYMCCharm[i]->Fill(track.pt(), track.dcaXY());
hdcaZMCCharm[i]->Fill(track.pt(), track.dcaZ());
}
if (IsBeautyMother) {
hDcaXYMCBeauty[i]->Fill(track.pt(), track.dcaXY());
hDcaZMCBeauty[i]->Fill(track.pt(), track.dcaZ());
}
if (!IsCharmMother && !IsBeautyMother) {
hDcaXYMCNotHF[i]->Fill(track.pt(), track.dcaXY());
hDcaZMCNotHF[i]->Fill(track.pt(), track.dcaZ());
}
}
}
Expand Down Expand Up @@ -1817,7 +1893,7 @@ struct tofSpectra {
const auto& mcParticle = track.mcParticle();

static_for<0, 17>([&](auto i) {
fillTrackHistograms_MC<i>(track, mcParticle, track.collision_as<RecoMCCollisions>());
fillTrackHistograms_MC<i>(track, mcParticle, track.collision_as<RecoMCCollisions>(), mcParticles);
});
}
if (includeCentralityMC) {
Expand Down