From 386fc45822242f0a35b85c4178aa6805d7335932 Mon Sep 17 00:00:00 2001
From: Fabrizio Grosa <fabrizio.grosa@cern.ch>
Date: Wed, 12 Jun 2024 13:34:25 +0200
Subject: [PATCH 1/2] DPG/AOTTrack: check proper process functions

---
 DPG/Tasks/AOTTrack/tagAndProbeDmesons.cxx | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/DPG/Tasks/AOTTrack/tagAndProbeDmesons.cxx b/DPG/Tasks/AOTTrack/tagAndProbeDmesons.cxx
index c9bf20d2231..2d149d24072 100644
--- a/DPG/Tasks/AOTTrack/tagAndProbeDmesons.cxx
+++ b/DPG/Tasks/AOTTrack/tagAndProbeDmesons.cxx
@@ -272,6 +272,10 @@ struct TagTwoProngDisplacedVertices {
 
   void init(InitContext&)
   {
+    if ((doprocessPiPiFromDplus && doprocessPiPiFromDplusMc) || (doprocessKaKaFromDsOrDplus && doprocessKaKaFromDsOrDplusMc) || (doprocessKaPiFromDstar && doprocessKaPiFromDstarMc) || (doprocessKaKaFromDzero && doprocessKaKaFromDzeroMc)) {
+      LOGP(fatal, "The process functions for the same channel with and without MC truth cannot be enabled at the same time! Please check your configuration");
+    }
+
     std::string ccdbUrl = "http://alice-ccdb.cern.ch";
     ccdb->setURL(ccdbUrl.data());
     ccdb->setCaching(true);
@@ -297,20 +301,20 @@ struct TagTwoProngDisplacedVertices {
     const AxisSpec axisMassKaKa{200, constants::physics::MassPhi - 0.05f, constants::physics::MassPhi + 0.05f};
     const AxisSpec axisMassKaPi{400, constants::physics::MassD0 - 0.2f, constants::physics::MassD0 + 0.2f};
 
-    if (doprocessPiPiFromDplus) {
+    if (doprocessPiPiFromDplus || doprocessPiPiFromDplusMc) {
       registry.add<TH2>("hMassPiPiVsPt", ";#it{p}_{T}(#pi#pi) (GeV/#it{c}); #it{M}(#pi#pi) (GeV/#it{c}^{2})", HistType::kTH2D, {axisPt, axisMassPiPi});
     }
-    if (doprocessKaKaFromDsOrDplus) {
+    if (doprocessKaKaFromDsOrDplus || doprocessKaKaFromDsOrDplusMc) {
       registry.add<TH2>("hMassKaKaVsPt", ";#it{p}_{T}(KK) (GeV/#it{c}); #it{M}(KK) (GeV/#it{c}^{2})", HistType::kTH2D, {axisPt, axisMassKaKa});
     }
-    if (doprocessKaPiFromDstar) {
+    if (doprocessKaPiFromDstar || doprocessKaPiFromDstarMc) {
       if (!studyDzeroReflections) {
         registry.add<TH2>("hMassKaPiVsPt", ";#it{p}_{T}(K#pi) (GeV/#it{c}); #it{M}(K#pi) (GeV/#it{c}^{2})", HistType::kTH2D, {axisPt, axisMassKaPi});
       } else {
         registry.add<THnSparse>("hMassKaPiVsPt", ";#it{p}_{T}(K#pi) (GeV/#it{c}); #it{M}(K#pi) (GeV/#it{c}^{2}); #it{M}(#piK) (GeV/#it{c}^{2}); ReflFag", HistType::kTHnSparseF, {axisPtDzeroRefl, axisMassKaPi, axisMassKaPi, axisReflFlag});
       }
     }
-    if (doprocessKaKaFromDzero) {
+    if (doprocessKaKaFromDzero || doprocessKaKaFromDzeroMc) {
       registry.add<TH2>("hMassDzeroKaKaVsPt", ";#it{p}_{T}(K#pi) (GeV/#it{c}); #it{M}(K#pi) (GeV/#it{c}^{2})", HistType::kTH2D, {axisPt, axisMassKaPi});
     }
 

From c7d225d9c2171d52d244b3254bdfcd8b05a9849d Mon Sep 17 00:00:00 2001
From: Fabrizio Grosa <fabrizio.grosa@cern.ch>
Date: Wed, 12 Jun 2024 13:43:36 +0200
Subject: [PATCH 2/2] Remove undefined process function

---
 DPG/Tasks/AOTTrack/tagAndProbeDmesons.cxx | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/DPG/Tasks/AOTTrack/tagAndProbeDmesons.cxx b/DPG/Tasks/AOTTrack/tagAndProbeDmesons.cxx
index 2d149d24072..4481e366e8d 100644
--- a/DPG/Tasks/AOTTrack/tagAndProbeDmesons.cxx
+++ b/DPG/Tasks/AOTTrack/tagAndProbeDmesons.cxx
@@ -272,7 +272,7 @@ struct TagTwoProngDisplacedVertices {
 
   void init(InitContext&)
   {
-    if ((doprocessPiPiFromDplus && doprocessPiPiFromDplusMc) || (doprocessKaKaFromDsOrDplus && doprocessKaKaFromDsOrDplusMc) || (doprocessKaPiFromDstar && doprocessKaPiFromDstarMc) || (doprocessKaKaFromDzero && doprocessKaKaFromDzeroMc)) {
+    if ((doprocessPiPiFromDplus && doprocessPiPiFromDplusMc) || (doprocessKaKaFromDsOrDplus && doprocessKaKaFromDsOrDplusMc) || (doprocessKaPiFromDstar && doprocessKaPiFromDstarMc)) {
       LOGP(fatal, "The process functions for the same channel with and without MC truth cannot be enabled at the same time! Please check your configuration");
     }
 
@@ -314,7 +314,7 @@ struct TagTwoProngDisplacedVertices {
         registry.add<THnSparse>("hMassKaPiVsPt", ";#it{p}_{T}(K#pi) (GeV/#it{c}); #it{M}(K#pi) (GeV/#it{c}^{2}); #it{M}(#piK) (GeV/#it{c}^{2}); ReflFag", HistType::kTHnSparseF, {axisPtDzeroRefl, axisMassKaPi, axisMassKaPi, axisReflFlag});
       }
     }
-    if (doprocessKaKaFromDzero || doprocessKaKaFromDzeroMc) {
+    if (doprocessKaKaFromDzero) {
       registry.add<TH2>("hMassDzeroKaKaVsPt", ";#it{p}_{T}(K#pi) (GeV/#it{c}); #it{M}(K#pi) (GeV/#it{c}^{2})", HistType::kTH2D, {axisPt, axisMassKaPi});
     }