From fc69bdae04e3a0fbecb40bdd7618c81d7818a58a Mon Sep 17 00:00:00 2001
From: Chiara Zampolli <Chiara.Zampolli@cern.ch>
Date: Wed, 9 Oct 2024 16:50:04 +0200
Subject: [PATCH] adding cut on DCA and cosPA

---
 Detectors/GLOQC/include/GLOQC/ITSTPCMatchingQCParams.h |  2 ++
 Detectors/GLOQC/include/GLOQC/MatchITSTPCQC.h          | 10 ++++++++--
 Detectors/GLOQC/src/MatchITSTPCQC.cxx                  |  9 ++++++++-
 .../qc/src/ITSTPCMatchingQCSpec.cxx                    |  2 ++
 4 files changed, 20 insertions(+), 3 deletions(-)

diff --git a/Detectors/GLOQC/include/GLOQC/ITSTPCMatchingQCParams.h b/Detectors/GLOQC/include/GLOQC/ITSTPCMatchingQCParams.h
index 62c248209001a..6bd88a9be64c0 100644
--- a/Detectors/GLOQC/include/GLOQC/ITSTPCMatchingQCParams.h
+++ b/Detectors/GLOQC/include/GLOQC/ITSTPCMatchingQCParams.h
@@ -46,6 +46,8 @@ struct ITSTPCMatchingQCParams : public o2::conf::ConfigurableParamHelper<ITSTPCM
   float minTPCOccPbPb = 0.f;
   float maxTPCOccPbPb = 8.e6;
   int nBinsTPCOccPbPb = 8;
+  float maxK0DCA = 0.01;
+  float minK0CosPA = 0.995;
 
   O2ParamDef(ITSTPCMatchingQCParams, "ITSTPCMatchingQC");
 };
diff --git a/Detectors/GLOQC/include/GLOQC/MatchITSTPCQC.h b/Detectors/GLOQC/include/GLOQC/MatchITSTPCQC.h
index c3f7c47c8b981..8734cc4cd6ce4 100644
--- a/Detectors/GLOQC/include/GLOQC/MatchITSTPCQC.h
+++ b/Detectors/GLOQC/include/GLOQC/MatchITSTPCQC.h
@@ -249,8 +249,6 @@ class MatchITSTPCQC
   void setBz(float bz) { mBz = bz; }
   void setDoK0QC(bool v) { mDoK0QC = v; }
   bool getDoK0QC() const { return mDoK0QC; }
-  void setK0Scaling(float v) { mK0Scaling = v; }
-  float getK0Scaling() const { return mK0Scaling; }
 
   // ITS track
   void setMinPtITSCut(float v) { mPtITSCut = v; };
@@ -285,6 +283,12 @@ class MatchITSTPCQC
   void setMinTPCOccPbPb(float v) { mMinTPCOccPbPb = v; }
   void setMaxTPCOccPbPb(float v) { mMaxTPCOccPbPb = v; }
   void setNBinsTPCOccPbPb(int v) { mNBinsTPCOccPbPb = v; }
+  void setK0Scaling(float v) { mK0Scaling = v; }
+  float getK0Scaling() const { return mK0Scaling; }
+  void setK0MaxDCA(float v) { mK0MaxDCA = v; }
+  float getK0MaxDCA() { return mK0MaxDCA; }
+  void setK0MinCosPA(float v) { mK0MinCosPA = v; }
+  float getK0MinCosPA() const { return mK0MinCosPA; }
 
   void printParams()
   {
@@ -455,6 +459,8 @@ class MatchITSTPCQC
   float mMinTPCOccPbPb = 0.f;  // min TPC occupancy for K0s plot for PbPb collisions
   float mMaxTPCOccPbPb = 8.e6; // max TPC occupancy for K0s plot for PbPb collisions
   int mNBinsTPCOccPbPb = 8;    // number of bins in TPC occupancy for K0s plot for PbPb collisions
+  float mK0MaxDCA = 0.01;      // max DCA to select the K0
+  float mK0MinCosPA = 0.995;   // min cosPA to select the K0
 
   ClassDefNV(MatchITSTPCQC, 3);
 };
diff --git a/Detectors/GLOQC/src/MatchITSTPCQC.cxx b/Detectors/GLOQC/src/MatchITSTPCQC.cxx
index 6cbe886109762..15b85b692650c 100644
--- a/Detectors/GLOQC/src/MatchITSTPCQC.cxx
+++ b/Detectors/GLOQC/src/MatchITSTPCQC.cxx
@@ -1079,7 +1079,14 @@ bool MatchITSTPCQC::processV0(int iv, o2::globaltracking::RecoContainer& recoDat
   if (mMaxEtaK0 < std::abs(v0sel.getEta())) {
     return false;
   }
-  if (mCutK0Mass > 0 && std::abs(std::sqrt(v0sel.calcMass2AsK0()) - 0.497) > mCutK0Mass) {
+
+  if (mCutK0Mass > 0 && std::abs(std::sqrt(v0sel.calcMass2AsK0()) - 0.497) > mCutK0Mass || v0sel.getDCA() > mK0MaxDCA || v0sel.getCosPA() < mK0MinCosPA) {
+    if (v0sel.getDCA() > mK0MaxDCA && v0sel.getCosPA() < mK0MinCosPA) {
+      LOG(debug) << "v0sel.getDCA() = " << v0sel.getDCA() << " max is " << mK0MaxDCA << " returning ... ";
+    }
+    if (v0sel.getCosPA() > mK0MinCosPA) {
+      LOG(debug) << "v0sel.getCosPA() = " << v0sel.getCosPA() << " min is " << mK0MinCosPA << " returning ... ";
+    }
     return false;
   }
   // get the corresponding PV
diff --git a/Detectors/GlobalTrackingWorkflow/qc/src/ITSTPCMatchingQCSpec.cxx b/Detectors/GlobalTrackingWorkflow/qc/src/ITSTPCMatchingQCSpec.cxx
index 91cbdc9f7b7ce..2f9ca99d29a79 100644
--- a/Detectors/GlobalTrackingWorkflow/qc/src/ITSTPCMatchingQCSpec.cxx
+++ b/Detectors/GlobalTrackingWorkflow/qc/src/ITSTPCMatchingQCSpec.cxx
@@ -58,6 +58,8 @@ void ITSTPCMatchingQCDevice::init(InitContext& /*ic*/)
   mMatchITSTPCQC->setMinTPCOccPbPb(params.minTPCOccPbPb);
   mMatchITSTPCQC->setMaxTPCOccPbPb(params.maxTPCOccPbPb);
   mMatchITSTPCQC->setNBinsTPCOccPbPb(params.nBinsTPCOccPbPb);
+  mMatchITSTPCQC->setK0MaxDCA(params.maxK0DCA);
+  mMatchITSTPCQC->setK0MinCosPA(params.minK0CosPA);
   o2::base::GRPGeomHelper::instance().setRequest(mCCDBRequest);
   if (mUseMC) {
     mMatchITSTPCQC->setUseMC(mUseMC);