From 406c3ba8b0c343b995fd437f37634040dcd71f22 Mon Sep 17 00:00:00 2001
From: ddobrigk <david.dobrigkeit.chinellato@cern.ch>
Date: Wed, 29 Sep 2021 09:11:31 -0300
Subject: [PATCH 1/2] Add DCA cut

---
 ALICE3/Tasks/alice3-qa-multiplicity.cxx | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/ALICE3/Tasks/alice3-qa-multiplicity.cxx b/ALICE3/Tasks/alice3-qa-multiplicity.cxx
index 981e2c8afed..0f23e175fd5 100644
--- a/ALICE3/Tasks/alice3-qa-multiplicity.cxx
+++ b/ALICE3/Tasks/alice3-qa-multiplicity.cxx
@@ -13,6 +13,9 @@
 // O2 includes
 #include "Framework/AnalysisTask.h"
 #include "Framework/runDataProcessing.h"
+#include "Common/Core/TrackSelection.h"
+#include "Common/DataModel/TrackSelectionTables.h"
+#include "Common/Core/trackUtilities.h"
 
 using namespace o2;
 using namespace o2::framework;
@@ -25,6 +28,7 @@ struct ALICE3MultTask {
   Configurable<float> MaxEta{"MaxEta", 0.8f, "Maximum eta in range"};
   Configurable<float> MaxMult{"MaxMult", 1000.f, "Maximum multiplicity in range"};
   Configurable<float> MinContrib{"MinContrib", 2.f, "Minimum number of contributors to the PV"};
+  Configurable<float> MaxDCA{"MaxDCA", 25.f, "Max DCAxy and DCAz for counted tracks"};
 
   void init(InitContext&)
   {
@@ -35,13 +39,16 @@ struct ALICE3MultTask {
   }
 
   int nevs = 0;
-  void process(const o2::aod::Collision& collision, const o2::aod::Tracks& tracks)
+  void process(const o2::aod::Collision& collision, const soa::Join<aod::Tracks, aod::TracksExtended>& tracks)
   {
     int nTracks = 0;
     for (const auto& track : tracks) {
       if (track.eta() < MinEta || track.eta() > MaxEta) {
         continue;
       }
+      if (abs(track.dcaXY()) > MaxDCA || abs(track.dcaZ()) > MaxDCA) {
+        continue;
+      }
       nTracks++;
     }
     LOG(info) << nevs++ << ") Event " << collision.globalIndex() << " has " << nTracks << " tracks";

From ab76973eeb05da324b7edf148a93a24459b856a2 Mon Sep 17 00:00:00 2001
From: ddobrigk <david.dobrigkeit.chinellato@cern.ch>
Date: Wed, 29 Sep 2021 10:12:27 -0300
Subject: [PATCH 2/2] remove unnecessary header

---
 ALICE3/Tasks/alice3-qa-multiplicity.cxx | 1 -
 1 file changed, 1 deletion(-)

diff --git a/ALICE3/Tasks/alice3-qa-multiplicity.cxx b/ALICE3/Tasks/alice3-qa-multiplicity.cxx
index 0f23e175fd5..d0869e257ac 100644
--- a/ALICE3/Tasks/alice3-qa-multiplicity.cxx
+++ b/ALICE3/Tasks/alice3-qa-multiplicity.cxx
@@ -15,7 +15,6 @@
 #include "Framework/runDataProcessing.h"
 #include "Common/Core/TrackSelection.h"
 #include "Common/DataModel/TrackSelectionTables.h"
-#include "Common/Core/trackUtilities.h"
 
 using namespace o2;
 using namespace o2::framework;