From 8bc17c24f01e7c4c8b45a28c85ef09a603c45940 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: Thu, 18 Dec 2025 21:12:16 +0100
Subject: [PATCH] Update femtoDreamUtils.h

---
 PWGCF/FemtoDream/Core/femtoDreamUtils.h | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/PWGCF/FemtoDream/Core/femtoDreamUtils.h b/PWGCF/FemtoDream/Core/femtoDreamUtils.h
index 53cd9e6a36b..beff8c509ac 100644
--- a/PWGCF/FemtoDream/Core/femtoDreamUtils.h
+++ b/PWGCF/FemtoDream/Core/femtoDreamUtils.h
@@ -54,9 +54,18 @@ inline float getMass(int pdgCode)
     case o2::constants::physics::Pdg::kPhi:
       mass = o2::constants::physics::MassPhi;
       break;
+    case o2::constants::physics::Pdg::kD0:
+      mass = o2::constants::physics::MassD0;
+      break;
+    case o2::constants::physics::Pdg::kDStar:
+      mass = o2::constants::physics::MassDStar;
+      break;
     case o2::constants::physics::Pdg::kDPlus:
       mass = o2::constants::physics::MassDPlus;
       break;
+    case o2::constants::physics::Pdg::kDS:
+      mass = o2::constants::physics::MassDS;
+      break;
     case o2::constants::physics::Pdg::kLambdaCPlus:
       mass = o2::constants::physics::MassLambdaCPlus;
       break;