[PWGLF] QA of ITS-TPC matching efficiency for V0 daughters (AliceO2Gr…

…oup#9813)
samrangy · Feb 7, 2025 · 873dd2c · 873dd2c
1 parent 6cb56b5
commit 873dd2c
Show file tree

Hide file tree

Showing 2 changed files with 337 additions and 0 deletions.
diff --git a/PWGLF/Tasks/QC/CMakeLists.txt b/PWGLF/Tasks/QC/CMakeLists.txt
@@ -129,3 +129,8 @@ o2physics_add_dpl_workflow(str-derived-genqa
     SOURCES strderivedGenQA.cxx
     PUBLIC_LINK_LIBRARIES O2Physics::AnalysisCore
     COMPONENT_NAME Analysis)
+
+o2physics_add_dpl_workflow(its-tpc-matching-vzeros
+    SOURCES lfITSTPCMatchingSecondaryTracksQA.cxx
+    PUBLIC_LINK_LIBRARIES O2Physics::AnalysisCore
+    COMPONENT_NAME Analysis)
diff --git a/PWGLF/Tasks/QC/lfITSTPCMatchingSecondaryTracksQA.cxx b/PWGLF/Tasks/QC/lfITSTPCMatchingSecondaryTracksQA.cxx
@@ -0,0 +1,332 @@
+// 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 lfITSTPCMatchingSecondaryTracksQA.cxx
+///
+/// \brief task for QA of ITS-TPC matching efficiency of secondary tracks from V0s
+/// \author Alberto Caliva ([email protected]), Francesca Ercolessi ([email protected]), Nicolò Jacazio ([email protected])
+/// \since May 22, 2024
+
+#include <TLorentzVector.h>
+#include <TMath.h>
+#include <TObjArray.h>
+#include <TPDGCode.h>
+#include <TVector2.h>
+#include <TVector3.h>
+#include <cmath>
+#include <vector>
+#include <string>
+#include "Common/Core/RecoDecay.h"
+#include "Common/Core/trackUtilities.h"
+#include "Common/DataModel/EventSelection.h"
+#include "Common/DataModel/Multiplicity.h"
+#include "Common/DataModel/PIDResponse.h"
+#include "Common/DataModel/TrackSelectionTables.h"
+#include "Framework/ASoAHelpers.h"
+#include "Framework/AnalysisDataModel.h"
+#include "Framework/AnalysisTask.h"
+#include "Framework/runDataProcessing.h"
+#include "PWGLF/DataModel/LFStrangenessTables.h"
+#include "ReconstructionDataFormats/Track.h"
+
+using namespace std;
+using namespace o2;
+using namespace o2::soa;
+using namespace o2::aod;
+using namespace o2::framework;
+using namespace o2::framework::expressions;
+using namespace o2::constants::physics;
+using namespace o2::constants::math;
+using std::array;
+
+using SelCollisions = soa::Join<aod::Collisions, aod::EvSels, aod::CentFT0Ms>;
+using SimCollisions = soa::Join<aod::Collisions, aod::EvSels, aod::CentFT0Ms, aod::McCollisionLabels>;
+using StrHadronDaughterTracks = soa::Join<aod::Tracks, aod::TracksIU, aod::TracksExtra, aod::TracksCovIU, aod::TracksDCA, aod::pidTPCFullPi, aod::pidTPCFullKa, aod::pidTPCFullPr, aod::pidTOFFullPi, aod::pidTOFFullKa, aod::pidTOFFullPr>;
+using MCTracks = soa::Join<StrHadronDaughterTracks, aod::McTrackLabels>;
+
+struct LfITSTPCMatchingSecondaryTracksQA {
+
+  HistogramRegistry registryData{"registryData", {}, OutputObjHandlingPolicy::AnalysisObject, true, true};
+  HistogramRegistry registryMC{"registryMC", {}, OutputObjHandlingPolicy::AnalysisObject, true, true};
+
+  // Global Parameters
+  Configurable<double> zVtx{"zVtx", 10.0, "Maximum zVertex"};
+
+  // Track Parameters
+  Configurable<float> minITSnCls{"minITSnCls", 3.0f, "min number of ITS clusters"};
+  Configurable<float> minNCrossedRowsTPC{"minNCrossedRowsTPC", 80.0f, "min number of TPC crossed rows"};
+  Configurable<float> minNCrossedRowsOverFindable{"minNCrossedRowsOverFindable", 0.8f, "min number of TPC crossed rows/findable"};
+  Configurable<float> maxChi2TPC{"maxChi2TPC", 4.0f, "max chi2 per cluster TPC"};
+  Configurable<float> maxChi2ITS{"maxChi2ITS", 36.0f, "max chi2 per cluster ITS"};
+  Configurable<float> etaMin{"etaMin", -0.8f, "eta min"};
+  Configurable<float> etaMax{"etaMax", +0.8f, "eta max"};
+  Configurable<float> nsigmaTPCmin{"nsigmaTPCmin", -3.0f, "Minimum nsigma TPC"};
+  Configurable<float> nsigmaTPCmax{"nsigmaTPCmax", +3.0f, "Maximum nsigma TPC"};
+  Configurable<float> nsigmaTOFmin{"nsigmaTOFmin", -3.0f, "Minimum nsigma TOF"};
+  Configurable<float> nsigmaTOFmax{"nsigmaTOFmax", +3.0f, "Maximum nsigma TOF"};
+  Configurable<bool> requireTOF{"requireTOF", false, "require TOF hit"};
+  Configurable<bool> requireItsHits{"requireItsHits", false, "require ITS hits"};
+  Configurable<std::vector<float>> requiredHit{"requiredHit", {0, 0, 0, 1, 1, 1, 1}, "required ITS Hits"};
+
+  // V0 Parameters
+  Configurable<float> minimumV0Radius{"minimumV0Radius", 0.5f, "Minimum V0 Radius"};
+  Configurable<float> maximumV0Radius{"maximumV0Radius", 40.0f, "Maximum V0 Radius"};
+  Configurable<float> dcanegtoPVmin{"dcanegtoPVmin", 0.1f, "Minimum DCA Neg To PV"};
+  Configurable<float> dcapostoPVmin{"dcapostoPVmin", 0.1f, "Minimum DCA Pos To PV"};
+  Configurable<float> v0cospaMin{"v0cospaMin", 0.99f, "Minimum V0 CosPA"};
+  Configurable<float> dcaV0DaughtersMax{"dcaV0DaughtersMax", 0.5f, "Maximum DCA Daughters"};
+  Configurable<float> mK0Min{"mK0Min", 0.48f, "K0 mass lower cut"};
+  Configurable<float> mK0Max{"mK0Max", 0.52f, "K0 mass upper cut"};
+
+  void init(InitContext const&)
+  {
+    // Event Counters
+    if (doprocessData) {
+      registryData.add("number_of_events_data", "number of events in data", HistType::kTH1D, {{20, 0, 20, "Event Cuts"}});
+      registryData.add("trkPionTpc", "trkPionTpc", HistType::kTH1D, {{1000, 0, 10, "#it{p}_{T} (GeV/#it{c})"}});
+      registryData.add("trkPionTpcIts", "trkPionTpcIts", HistType::kTH1D, {{1000, 0, 10, "#it{p}_{T} (GeV/#it{c})"}});
+      registryData.add("secPionTpc", "secPionTpc", HistType::kTH1D, {{1000, 0, 10, "#it{p}_{T} (GeV/#it{c})"}});
+      registryData.add("secPionTpcIts", "secPionTpcIts", HistType::kTH1D, {{1000, 0, 10, "#it{p}_{T} (GeV/#it{c})"}});
+    }
+
+    if (doprocessMC) {
+      registryData.add("number_of_events_mc", "number of events in mc", HistType::kTH1D, {{20, 0, 20, "Event Cuts"}});
+      registryData.add("trkPionTpcMc", "trkPionTpcMc", HistType::kTH1D, {{1000, 0, 10, "#it{p}_{T} (GeV/#it{c})"}});
+      registryData.add("trkPionTpcItsMc", "trkPionTpcItsMc", HistType::kTH1D, {{1000, 0, 10, "#it{p}_{T} (GeV/#it{c})"}});
+      registryData.add("secPionTpcMc", "secPionTpcMc", HistType::kTH1D, {{1000, 0, 10, "#it{p}_{T} (GeV/#it{c})"}});
+      registryData.add("secPionTpcItsMc", "secPionTpcItsMc", HistType::kTH1D, {{1000, 0, 10, "#it{p}_{T} (GeV/#it{c})"}});
+    }
+  }
+
+  bool hasHitOnITSlayer(uint8_t itsClsmap, int layer)
+  {
+    unsigned char testBit = 1 << layer;
+    return (itsClsmap & testBit);
+  }
+
+  template <typename TpcTrack>
+  bool passedTrackSelectionTpc(const TpcTrack& track)
+  {
+    if (!track.hasTPC())
+      return false;
+    if (track.tpcNClsCrossedRows() < minNCrossedRowsTPC)
+      return false;
+    if ((track.tpcNClsCrossedRows() / track.tpcNClsFindable()) < minNCrossedRowsOverFindable)
+      return false;
+    if (track.tpcChi2NCl() > maxChi2TPC)
+      return false;
+    if (track.eta() < etaMin || track.eta() > etaMax)
+      return false;
+    return true;
+  }
+
+  // K0s Selections
+  template <typename K0short>
+  bool passedK0ShortSelection(const K0short& v0)
+  {
+    if (v0.v0cosPA() < v0cospaMin)
+      return false;
+    if (v0.v0radius() < minimumV0Radius || v0.v0radius() > maximumV0Radius)
+      return false;
+    if (v0.dcaV0daughters() > dcaV0DaughtersMax)
+      return false;
+    if (std::fabs(v0.dcapostopv()) < dcapostoPVmin)
+      return false;
+    if (std::fabs(v0.dcanegtopv()) < dcanegtoPVmin)
+      return false;
+    if (v0.mK0Short() < mK0Min || v0.mK0Short() > mK0Max)
+      return false;
+
+    return true;
+  }
+
+  template <typename pionTrack>
+  bool passedPionSelection(const pionTrack& track)
+  {
+    // TPC Selection
+    if (track.tpcNSigmaPi() < nsigmaTPCmin || track.tpcNSigmaPi() > nsigmaTPCmax)
+      return false;
+
+    // TOF Selection
+    if (requireTOF) {
+      if (track.tofNSigmaPi() < nsigmaTOFmin || track.tofNSigmaPi() > nsigmaTOFmax)
+        return false;
+    }
+    return true;
+  }
+
+  template <typename ItsTrack>
+  bool passedTrackSelectionIts(const ItsTrack& track)
+  {
+    if (!track.hasITS())
+      return false;
+    if (track.itsNCls() < minITSnCls)
+      return false;
+    if (track.itsChi2NCl() > maxChi2ITS)
+      return false;
+
+    auto requiredItsHit = static_cast<std::vector<float>>(requiredHit);
+
+    if (requireItsHits) {
+      for (int i = 0; i < 7; i++) {
+        if (requiredItsHit[i] > 0 && !hasHitOnITSlayer(track.itsClusterMap(), i)) {
+          return false;
+        }
+      }
+    }
+
+    return true;
+  }
+
+  void processData(SelCollisions::iterator const& collision, aod::V0Datas const& fullV0s, StrHadronDaughterTracks const& tracks)
+  {
+    registryData.fill(HIST("number_of_events_data"), 0.5);
+
+    // Event Selection
+    if (!collision.sel8() || std::fabs(collision.posZ()) > zVtx)
+      return;
+    registryData.fill(HIST("number_of_events_data"), 1.5);
+
+    for (const auto& track : tracks) {
+
+      if (!passedTrackSelectionTpc(track))
+        continue;
+      if (!passedPionSelection(track))
+        continue;
+
+      registryData.fill(HIST("trkPionTpc"), track.pt());
+
+      if (!passedTrackSelectionIts(track))
+        continue;
+
+      registryData.fill(HIST("trkPionTpcIts"), track.pt());
+    }
+
+    for (const auto& v0 : fullV0s) {
+
+      const auto& posTrack = v0.posTrack_as<StrHadronDaughterTracks>();
+      const auto& negTrack = v0.negTrack_as<StrHadronDaughterTracks>();
+      if (!passedK0ShortSelection(v0))
+        continue;
+
+      if (!passedTrackSelectionTpc(posTrack))
+        continue;
+      if (!passedTrackSelectionTpc(negTrack))
+        continue;
+      if (!passedPionSelection(posTrack))
+        continue;
+      if (!passedPionSelection(negTrack))
+        continue;
+
+      registryData.fill(HIST("secPionTpc"), posTrack.pt());
+      registryData.fill(HIST("secPionTpc"), negTrack.pt());
+
+      if (!passedTrackSelectionIts(posTrack))
+        continue;
+      registryData.fill(HIST("secPionTpcIts"), posTrack.pt());
+
+      if (!passedTrackSelectionIts(negTrack))
+        continue;
+      registryData.fill(HIST("secPionTpcIts"), negTrack.pt());
+    }
+  }
+  PROCESS_SWITCH(LfITSTPCMatchingSecondaryTracksQA, processData, "Process data", true);
+
+  Preslice<aod::V0Datas> perCollisionV0 = o2::aod::v0data::collisionId;
+  Preslice<MCTracks> perCollisionTrk = o2::aod::track::collisionId;
+
+  void processMC(SimCollisions const& collisions, MCTracks const& mcTracks, aod::V0Datas const& fullV0s, const aod::McParticles&)
+  {
+    for (const auto& collision : collisions) {
+      registryMC.fill(HIST("number_of_events_mc"), 0.5);
+
+      if (!collision.sel8() || std::fabs(collision.posZ()) > zVtx)
+        continue;
+      registryMC.fill(HIST("number_of_events_mc"), 1.5);
+
+      auto v0sPerColl = fullV0s.sliceBy(perCollisionV0, collision.globalIndex());
+      auto tracksPerColl = mcTracks.sliceBy(perCollisionTrk, collision.globalIndex());
+
+      for (const auto& track : tracksPerColl) {
+        if (!passedTrackSelectionTpc(track))
+          continue;
+        if (!passedPionSelection(track))
+          continue;
+        if (!track.has_mcParticle())
+          continue;
+        const auto particle = track.mcParticle();
+        if (std::fabs(particle.pdgCode()) != 211)
+          continue;
+
+        registryMC.fill(HIST("trkPionTpcMc"), track.pt());
+        if (!passedTrackSelectionIts(track))
+          continue;
+
+        registryMC.fill(HIST("trkPionTpcItsMc"), track.pt());
+      }
+
+      for (const auto& v0 : v0sPerColl) {
+
+        const auto& posTrack = v0.posTrack_as<MCTracks>();
+        const auto& negTrack = v0.negTrack_as<MCTracks>();
+        if (!passedK0ShortSelection(v0))
+          continue;
+        if (!passedTrackSelectionTpc(posTrack))
+          continue;
+        if (!passedTrackSelectionTpc(negTrack))
+          continue;
+        if (!passedPionSelection(posTrack))
+          continue;
+        if (!passedPionSelection(negTrack))
+          continue;
+        if (!posTrack.has_mcParticle())
+          continue;
+        if (!negTrack.has_mcParticle())
+          continue;
+
+        auto posParticle = posTrack.mcParticle_as<aod::McParticles>();
+        auto negParticle = negTrack.mcParticle_as<aod::McParticles>();
+        if (!posParticle.has_mothers())
+          continue;
+        if (!negParticle.has_mothers())
+          continue;
+
+        int pdgParent(0);
+        for (const auto& particleMotherOfNeg : negParticle.mothers_as<aod::McParticles>()) {
+          for (const auto& particleMotherOfPos : posParticle.mothers_as<aod::McParticles>()) {
+            if (particleMotherOfNeg == particleMotherOfPos) {
+              pdgParent = particleMotherOfNeg.pdgCode();
+            }
+          }
+        }
+        if (pdgParent != 310)
+          continue;
+
+        registryMC.fill(HIST("secPionTpcMc"), posTrack.pt());
+        registryMC.fill(HIST("secPionTpcMc"), negTrack.pt());
+
+        if (!passedTrackSelectionIts(posTrack))
+          continue;
+        registryMC.fill(HIST("secPionTpcItsMc"), posTrack.pt());
+
+        if (!passedTrackSelectionIts(negTrack))
+          continue;
+        registryMC.fill(HIST("secPionTpcItsMc"), negTrack.pt());
+      }
+    }
+  }
+  PROCESS_SWITCH(LfITSTPCMatchingSecondaryTracksQA, processMC, "Process MC", false);
+};
+
+WorkflowSpec defineDataProcessing(ConfigContext const& cfgc)
+{
+  return WorkflowSpec{adaptAnalysisTask<LfITSTPCMatchingSecondaryTracksQA>(cfgc)};
+}