Skip to content

PWGLF: Add task to study track propagation #4004

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 2 commits into from
Nov 27, 2023
Merged

PWGLF: Add task to study track propagation #4004

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
ed638d4
Add task to study track propagation
creetz16 Nov 27, 2023
5187a12
Formatting
creetz16 Nov 27, 2023
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
5 changes: 5 additions & 0 deletions PWGLF/Tasks/QC/CMakeLists.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -39,6 +39,11 @@ o2physics_add_dpl_workflow(kfperformancestudy
PUBLIC_LINK_LIBRARIES O2::DetectorsBase O2Physics::AnalysisCore
COMPONENT_NAME Analysis)

o2physics_add_dpl_workflow(lfpropstudy
SOURCES lfpropStudy.cxx
PUBLIC_LINK_LIBRARIES O2Physics::AnalysisCore
COMPONENT_NAME Analysis)

o2physics_add_dpl_workflow(tpc-dedx-postcalibration
SOURCES tpc_dEdx_postcalibration.cxx
PUBLIC_LINK_LIBRARIES O2Physics::AnalysisCore
Expand Down
264 changes: 264 additions & 0 deletions PWGLF/Tasks/QC/lfpropStudy.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,264 @@
// Copyright 2019-2020 CERN and copyright holders of ALICE O2.
// See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
// All rights not expressly granted are reserved.
//
// This software is distributed under the terms of the GNU General Public
// License v3 (GPL Version 3), copied verbatim in the file "COPYING".
//
// In applying this license CERN does not waive the privileges and immunities
// granted to it by virtue of its status as an Intergovernmental Organization
// or submit itself to any jurisdiction.
//
/// \file lfpropStudy.cxx
/// \since 27-11-2023
/// \author Carolina Reetz <c.reetz@cern.ch>
/// \brief QA task to study properties of propagated tracks

#include "Framework/runDataProcessing.h"
#include "Framework/AnalysisTask.h"
#include "Framework/AnalysisDataModel.h"
#include "ReconstructionDataFormats/Track.h"
#include "ReconstructionDataFormats/DCA.h"
#include "Common/Core/RecoDecay.h"
#include "Common/Core/trackUtilities.h"
#include "Common/Core/TrackSelection.h"
#include "Common/Core/TrackSelectionDefaults.h"
#include "Common/DataModel/TrackSelectionTables.h"
#include "Common/DataModel/EventSelection.h"
#include "Common/DataModel/Centrality.h"
#include "Common/DataModel/Multiplicity.h"
#include "Common/DataModel/PIDResponse.h"
#include "TPDGCode.h"

using namespace o2;
using namespace o2::framework;
using namespace o2::framework::expressions;

using TracksLabeled = soa::Join<aod::StoredTracks, aod::StoredTracksCov, aod::TracksDCA, aod::TracksDCACov, aod::TracksExtra, aod::McTrackLabels>;
using Tracks = soa::Join<aod::StoredTracks, aod::StoredTracksCov, aod::TracksDCA, aod::TracksDCACov, aod::TracksExtra>;

struct lfpropStudy {

ConfigurableAxis axisDCAxy{"axisDCAxy", {1000, -10.f, 10.f}, "DCA_{xy} (cm)"};
ConfigurableAxis axisDCAz{"axisDCAz", {1000, -10.f, 10.f}, "DCA_{z} (cm)"};
ConfigurableAxis axisMom{"axisMom", {1000, 0.0f, 10.f}, "p (GeV/c)"};

Configurable<bool> d_noITS{"d_noITS", true, "flag to select tracks without ITS"};
Configurable<float> d_pTMin{"d_pTMin", 0.3, "minimum track momentum"};
Configurable<float> d_TPCrowsMin{"d_TPCrowsMin", 70, "minimum number of TPC crossed rows"};
Configurable<float> d_TPCrowsOverFindMin{"d_TPCrowsOverFindMin", 0.8, "minimum for ratio of TPC crossed rows over findable"};

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

Filter collisionFilter = (aod::evsel::sel8 == true);

void init(InitContext const&)
{
histos.add("hEventCounter", "hEventCounter", kTH1F, {{1, 0.0f, 1.0f}});

// momentum
histos.add("hPxEl", "hPxEl", kTH1F, {axisMom});
histos.add("hPyEl", "hPyEl", kTH1F, {axisMom});
histos.add("hPzEl", "hPzEl", kTH1F, {axisMom});
histos.add("hPtEl", "hPtEl", kTH1F, {axisMom});
histos.add("hPxPi", "hPxPi", kTH1F, {axisMom});
histos.add("hPyPi", "hPyPi", kTH1F, {axisMom});
histos.add("hPzPi", "hPzPi", kTH1F, {axisMom});
histos.add("hPtPi", "hPtPi", kTH1F, {axisMom});
histos.add("hPxKa", "hPxKa", kTH1F, {axisMom});
histos.add("hPyKa", "hPyKa", kTH1F, {axisMom});
histos.add("hPzKa", "hPzKa", kTH1F, {axisMom});
histos.add("hPtKa", "hPtKa", kTH1F, {axisMom});
histos.add("hPxPr", "hPxPr", kTH1F, {axisMom});
histos.add("hPyPr", "hPyPr", kTH1F, {axisMom});
histos.add("hPzPr", "hPzPr", kTH1F, {axisMom});
histos.add("hPtPr", "hPtPr", kTH1F, {axisMom});
histos.add("hPxDe", "hPxDe", kTH1F, {axisMom});
histos.add("hPyDe", "hPyDe", kTH1F, {axisMom});
histos.add("hPzDe", "hPzDe", kTH1F, {axisMom});
histos.add("hPtDe", "hPtDe", kTH1F, {axisMom});
histos.add("hPxTr", "hPxTr", kTH1F, {axisMom});
histos.add("hPyTr", "hPyTr", kTH1F, {axisMom});
histos.add("hPzTr", "hPzTr", kTH1F, {axisMom});
histos.add("hPtTr", "hPtTr", kTH1F, {axisMom});
histos.add("hPxHe", "hPxHe", kTH1F, {axisMom});
histos.add("hPyHe", "hPyHe", kTH1F, {axisMom});
histos.add("hPzHe", "hPzHe", kTH1F, {axisMom});
histos.add("hPtHe", "hPtHe", kTH1F, {axisMom});
histos.add("hPxAl", "hPxAl", kTH1F, {axisMom});
histos.add("hPyAl", "hPyAl", kTH1F, {axisMom});
histos.add("hPzAl", "hPzAl", kTH1F, {axisMom});
histos.add("hPtAl", "hPtAl", kTH1F, {axisMom});

// DCA
histos.add("hDCAxyEl", "hDCAxyEl", kTH1F, {axisDCAxy});
histos.add("hDCAzEl", "hDCAzEl", kTH1F, {axisDCAz});
histos.add("hDCAxyPi", "hDCAxyPi", kTH1F, {axisDCAxy});
histos.add("hDCAzPi", "hDCAzPi", kTH1F, {axisDCAz});
histos.add("hDCAxyKa", "hDCAxyKa", kTH1F, {axisDCAxy});
histos.add("hDCAzKa", "hDCAzKa", kTH1F, {axisDCAz});
histos.add("hDCAxyPr", "hDCAxyPr", kTH1F, {axisDCAxy});
histos.add("hDCAzPr", "hDCAzPr", kTH1F, {axisDCAz});
histos.add("hDCAxyDe", "hDCAxyDe", kTH1F, {axisDCAxy});
histos.add("hDCAzDe", "hDCAzDe", kTH1F, {axisDCAz});
histos.add("hDCAxyTr", "hDCAxyTr", kTH1F, {axisDCAxy});
histos.add("hDCAzTr", "hDCAzTr", kTH1F, {axisDCAz});
histos.add("hDCAxyHe", "hDCAxyHe", kTH1F, {axisDCAxy});
histos.add("hDCAzHe", "hDCAzHe", kTH1F, {axisDCAz});
histos.add("hDCAxyAl", "hDCAxyAl", kTH1F, {axisDCAxy});
histos.add("hDCAzAl", "hDCAzAl", kTH1F, {axisDCAz});
}

void processData(soa::Filtered<soa::Join<aod::Collisions, aod::EvSels>>::iterator const& collision, Tracks const& Tracks)
{
histos.fill(HIST("hEventCounter"), 0.5);
for (auto& track : Tracks) {

// track selection
if (d_noITS && track.hasITS())
continue; // optional: only look at tracks which start outside the ITS
if (track.trackType() == aod::track::TrackIU)
continue; // only look at tracks which were propagated
if (track.tpcNClsCrossedRows() < d_TPCrowsMin || track.tpcCrossedRowsOverFindableCls() < d_TPCrowsOverFindMin)
continue;

if (track.pidForTracking() == o2::track::PID::Electron) {
histos.fill(HIST("hPxEl"), track.px());
histos.fill(HIST("hPyEl"), track.py());
histos.fill(HIST("hPzEl"), track.pz());
histos.fill(HIST("hPtEl"), sqrt(track.px() * track.px() + track.py() * track.py()));
histos.fill(HIST("hDCAxyEl"), track.dcaXY());
histos.fill(HIST("hDCAzEl"), track.dcaZ());
} else if (track.pidForTracking() == o2::track::PID::Pion) {
histos.fill(HIST("hPxPi"), track.px());
histos.fill(HIST("hPyPi"), track.py());
histos.fill(HIST("hPzPi"), track.pz());
histos.fill(HIST("hPtPi"), sqrt(track.px() * track.px() + track.py() * track.py()));
histos.fill(HIST("hDCAxyPi"), track.dcaXY());
histos.fill(HIST("hDCAzPi"), track.dcaZ());
} else if (track.pidForTracking() == o2::track::PID::Kaon) {
histos.fill(HIST("hPxKa"), track.px());
histos.fill(HIST("hPyKa"), track.py());
histos.fill(HIST("hPzKa"), track.pz());
histos.fill(HIST("hPtKa"), sqrt(track.px() * track.px() + track.py() * track.py()));
histos.fill(HIST("hDCAxyKa"), track.dcaXY());
histos.fill(HIST("hDCAzKa"), track.dcaZ());
} else if (track.pidForTracking() == o2::track::PID::Proton) {
histos.fill(HIST("hPxPr"), track.px());
histos.fill(HIST("hPyPr"), track.py());
histos.fill(HIST("hPzPr"), track.pz());
histos.fill(HIST("hPtPr"), sqrt(track.px() * track.px() + track.py() * track.py()));
histos.fill(HIST("hDCAxyPr"), track.dcaXY());
histos.fill(HIST("hDCAzPr"), track.dcaZ());
} else if (track.pidForTracking() == o2::track::PID::Deuteron) {
histos.fill(HIST("hPxDe"), track.px());
histos.fill(HIST("hPyDe"), track.py());
histos.fill(HIST("hPzDe"), track.pz());
histos.fill(HIST("hPtDe"), sqrt(track.px() * track.px() + track.py() * track.py()));
histos.fill(HIST("hDCAxyDe"), track.dcaXY());
histos.fill(HIST("hDCAzDe"), track.dcaZ());
} else if (track.pidForTracking() == o2::track::PID::Triton) {
histos.fill(HIST("hPxTr"), track.px());
histos.fill(HIST("hPyTr"), track.py());
histos.fill(HIST("hPzTr"), track.pz());
histos.fill(HIST("hPtTr"), sqrt(track.px() * track.px() + track.py() * track.py()));
histos.fill(HIST("hDCAxyTr"), track.dcaXY());
histos.fill(HIST("hDCAzTr"), track.dcaZ());
} else if (track.pidForTracking() == o2::track::PID::Helium3) {
histos.fill(HIST("hPxHe"), track.px());
histos.fill(HIST("hPyHe"), track.py());
histos.fill(HIST("hPzHe"), track.pz());
histos.fill(HIST("hPtHe"), sqrt(track.px() * track.px() + track.py() * track.py()));
histos.fill(HIST("hDCAxyHe"), track.dcaXY());
histos.fill(HIST("hDCAzHe"), track.dcaZ());
} else if (track.pidForTracking() == o2::track::PID::Alpha) {
histos.fill(HIST("hPxAl"), track.px());
histos.fill(HIST("hPyAl"), track.py());
histos.fill(HIST("hPzAl"), track.pz());
histos.fill(HIST("hPtAl"), sqrt(track.px() * track.px() + track.py() * track.py()));
histos.fill(HIST("hDCAxyAl"), track.dcaXY());
histos.fill(HIST("hDCAzAl"), track.dcaZ());
}
}
}
PROCESS_SWITCH(lfpropStudy, processData, "process data", true);

void processMC(soa::Filtered<soa::Join<aod::Collisions, aod::EvSels>>::iterator const& collision, TracksLabeled const& Tracks, aod::McParticles const& particlesMC)
{
histos.fill(HIST("hEventCounter"), 0.5);
for (auto& track : Tracks) {

if (!track.has_mcParticle() || track.mcParticleId() <= -1 || track.mcParticleId() > particlesMC.size())
continue;
if (d_noITS && track.hasITS())
continue; // optional: only look at tracks which start outside the ITS
if (track.trackType() == aod::track::TrackIU)
continue; // only look at tracks which were propagated

if (track.mcParticle().pdgCode() == kElectron) {
histos.fill(HIST("hPxEl"), track.px());
histos.fill(HIST("hPyEl"), track.py());
histos.fill(HIST("hPzEl"), track.pz());
histos.fill(HIST("hPtEl"), sqrt(track.px() * track.px() + track.py() * track.py()));
histos.fill(HIST("hDCAxyEl"), track.dcaXY());
histos.fill(HIST("hDCAzEl"), track.dcaZ());
} else if (track.mcParticle().pdgCode() == kPiPlus) {
histos.fill(HIST("hPxPi"), track.px());
histos.fill(HIST("hPyPi"), track.py());
histos.fill(HIST("hPzPi"), track.pz());
histos.fill(HIST("hPtPi"), sqrt(track.px() * track.px() + track.py() * track.py()));
histos.fill(HIST("hDCAxyPi"), track.dcaXY());
histos.fill(HIST("hDCAzPi"), track.dcaZ());
} else if (track.mcParticle().pdgCode() == kKPlus) {
histos.fill(HIST("hPxKa"), track.px());
histos.fill(HIST("hPyKa"), track.py());
histos.fill(HIST("hPzKa"), track.pz());
histos.fill(HIST("hPtKa"), sqrt(track.px() * track.px() + track.py() * track.py()));
histos.fill(HIST("hDCAxyKa"), track.dcaXY());
histos.fill(HIST("hDCAzKa"), track.dcaZ());
} else if (track.mcParticle().pdgCode() == kProton) {
histos.fill(HIST("hPxPr"), track.px());
histos.fill(HIST("hPyPr"), track.py());
histos.fill(HIST("hPzPr"), track.pz());
histos.fill(HIST("hPtPr"), sqrt(track.px() * track.px() + track.py() * track.py()));
histos.fill(HIST("hDCAxyPr"), track.dcaXY());
histos.fill(HIST("hDCAzPr"), track.dcaZ());
} else if (track.mcParticle().pdgCode() == 1000010020) {
histos.fill(HIST("hPxDe"), track.px());
histos.fill(HIST("hPyDe"), track.py());
histos.fill(HIST("hPzDe"), track.pz());
histos.fill(HIST("hPtDe"), sqrt(track.px() * track.px() + track.py() * track.py()));
histos.fill(HIST("hDCAxyDe"), track.dcaXY());
histos.fill(HIST("hDCAzDe"), track.dcaZ());
} else if (track.mcParticle().pdgCode() == 1000010030) {
histos.fill(HIST("hPxTr"), track.px());
histos.fill(HIST("hPyTr"), track.py());
histos.fill(HIST("hPzTr"), track.pz());
histos.fill(HIST("hPtTr"), sqrt(track.px() * track.px() + track.py() * track.py()));
histos.fill(HIST("hDCAxyTr"), track.dcaXY());
histos.fill(HIST("hDCAzTr"), track.dcaZ());
} else if (track.mcParticle().pdgCode() == 1000020030) {
histos.fill(HIST("hPxHe"), track.px());
histos.fill(HIST("hPyHe"), track.py());
histos.fill(HIST("hPzHe"), track.pz());
histos.fill(HIST("hPtHe"), sqrt(track.px() * track.px() + track.py() * track.py()));
histos.fill(HIST("hDCAxyHe"), track.dcaXY());
histos.fill(HIST("hDCAzHe"), track.dcaZ());
} else if (track.mcParticle().pdgCode() == 1000020040) {
histos.fill(HIST("hPxAl"), track.px());
histos.fill(HIST("hPyAl"), track.py());
histos.fill(HIST("hPzAl"), track.pz());
histos.fill(HIST("hPtAl"), sqrt(track.px() * track.px() + track.py() * track.py()));
histos.fill(HIST("hDCAxyAl"), track.dcaXY());
histos.fill(HIST("hDCAzAl"), track.dcaZ());
}
}
}
PROCESS_SWITCH(lfpropStudy, processMC, "process MC", false);
};

WorkflowSpec defineDataProcessing(ConfigContext const& cfgc)
{
return WorkflowSpec{
adaptAnalysisTask<lfpropStudy>(cfgc)};
}