From 492e52b77638acf828ea4c1a733f06db599901ac Mon Sep 17 00:00:00 2001
From: lmassacr <laure.marie.massacrier@cern.ch>
Date: Fri, 9 Aug 2024 12:13:52 +0200
Subject: [PATCH] adding the possibility to cut on the MFT/MCH chi^2

---
 Modules/MUON/Common/src/TracksTask.cxx | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/Modules/MUON/Common/src/TracksTask.cxx b/Modules/MUON/Common/src/TracksTask.cxx
index 626f52e844..d4788c3d84 100644
--- a/Modules/MUON/Common/src/TracksTask.cxx
+++ b/Modules/MUON/Common/src/TracksTask.cxx
@@ -103,6 +103,7 @@ void TracksTask::createTrackHistos(const Activity& activity)
   double cutChi2Max = getConfigurationParameter<double>(mCustomParameters, "cutChi2Max", 1000, activity);
   double nSigmaPDCA = getConfigurationParameter<double>(mCustomParameters, "nSigmaPDCA", 6, activity);
   double matchScoreMaxMFT = getConfigurationParameter<double>(mCustomParameters, "matchScoreMaxMFT", 1000, activity);
+  double matchChi2MaxMFT = getConfigurationParameter<double>(mCustomParameters, "matchChi2MaxMFT", 1000, activity);
   double diMuonTimeCut = getConfigurationParameter<double>(mCustomParameters, "diMuonTimeCut", 100, activity) / 1000;
 
   int etaBins = getConfigurationParameter<int>(mCustomParameters, "etaBins", 200, activity);
@@ -165,6 +166,12 @@ void TracksTask::createTrackHistos(const Activity& activity)
         return false;
       return true;
     },
+    // MFT-MCH chi2 score
+    [matchChi2MaxMFT](const MuonTrack& t) {
+      if (t.hasMFT() && t.hasMCH() && t.getMatchInfoFwd().getMFTMCHMatchingChi2() > matchChi2MaxMFT)
+        return false;
+      return true;
+    },
     // MCH chi2 cut
     [cutChi2Min, cutChi2Max](const MuonTrack& t) { return ((t.getChi2OverNDFMCH() >= cutChi2Min) && (t.getChi2OverNDFMCH() <= cutChi2Max)); }
   };