From 50a201e3e8a84d711ed4a6fc2fe8b15e77506125 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?BiaoZhang=20=28=E5=BC=A0=E5=BD=AA=29?=
 <52267892+zhangbiao-phy@users.noreply.github.com>
Date: Sat, 14 Dec 2024 01:30:02 +0100
Subject: [PATCH] Add pt and inv.mass cuts also in mixed event

---
 PWGHF/HFC/Tasks/taskCharmHadronsFemtoDream.cxx | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/PWGHF/HFC/Tasks/taskCharmHadronsFemtoDream.cxx b/PWGHF/HFC/Tasks/taskCharmHadronsFemtoDream.cxx
index f421bd35ff3..9faaa37360a 100644
--- a/PWGHF/HFC/Tasks/taskCharmHadronsFemtoDream.cxx
+++ b/PWGHF/HFC/Tasks/taskCharmHadronsFemtoDream.cxx
@@ -396,6 +396,14 @@ struct HfTaskCharmHadronsFemtoDream {
           invMass = p2.m(std::array{o2::constants::physics::MassPiPlus, o2::constants::physics::MassKPlus, o2::constants::physics::MassProton});
         }
 
+        if (invMass < charmHadMinInvMass || invMass > charmHadMaxInvMass) {
+          continue;
+        }
+
+        if (p2.pt() < charmHadMinPt || p2.pt() > charmHadMaxPt) {
+          continue;
+        }
+
         int charmHadMc = 0;
         int originType = 0;
         if constexpr (isMc) {