From d31224d513ec714f002fd1060aec8d6eea5452aa Mon Sep 17 00:00:00 2001
From: Cevat Batuhan Tolon <85768122+ctolon@users.noreply.github.com>
Date: Mon, 13 Dec 2021 00:37:20 +0300
Subject: [PATCH] added psi2S signals in the MCSignalLibrary (#414)

---
 PWGDQ/Core/MCSignalLibrary.h | 39 ++++++++++++++++++++++++++++++++++--
 1 file changed, 37 insertions(+), 2 deletions(-)

diff --git a/PWGDQ/Core/MCSignalLibrary.h b/PWGDQ/Core/MCSignalLibrary.h
index b9e46c474c86b..f1eb29959a3cd 100644
--- a/PWGDQ/Core/MCSignalLibrary.h
+++ b/PWGDQ/Core/MCSignalLibrary.h
@@ -68,7 +68,7 @@ MCSignal* o2::aod::dqmcsignals::GetMCSignal(const char* name)
   }
   if (!nameStr.compare("eFromPromptJpsi")) {
     MCProng prong(3, {11, 443, 503}, {true, true, true}, {false, false, true}, {0, 0, 0}, {0, 0, 0}, {false, false, false});
-    signal = new MCSignal(name, "Electrons from beauty jpsi decays", {prong}, {-1});
+    signal = new MCSignal(name, "Electrons from prompt jpsi decays", {prong}, {-1});
     return signal;
   }
   if (!nameStr.compare("Jpsi")) {
@@ -86,6 +86,31 @@ MCSignal* o2::aod::dqmcsignals::GetMCSignal(const char* name)
     signal = new MCSignal(name, "Prompt jpsi (not from beauty)", {prong}, {-1});
     return signal;
   }
+  if (!nameStr.compare("eFromNonpromptPsi2S")) {
+    MCProng prong(3, {11, 100443, 503}, {true, true, true}, {false, false, false}, {0, 0, 0}, {0, 0, 0}, {false, false, false});
+    signal = new MCSignal(name, "Electrons from beauty psi2s decays", {prong}, {-1});
+    return signal;
+  }
+  if (!nameStr.compare("eFromPromptPsi2S")) {
+    MCProng prong(3, {11, 100443, 503}, {true, true, true}, {false, false, true}, {0, 0, 0}, {0, 0, 0}, {false, false, false});
+    signal = new MCSignal(name, "Electrons from prompt psi2s decays", {prong}, {-1});
+    return signal;
+  }
+  if (!nameStr.compare("Psi2S")) {
+    MCProng prong(1, {100443}, {true}, {false}, {0}, {0}, {false});
+    signal = new MCSignal(name, "Inclusive psi2s", {prong}, {-1});
+    return signal;
+  }
+  if (!nameStr.compare("nonPromptPsi2S")) {
+    MCProng prong(2, {100443, 503}, {true, true}, {false, false}, {0, 0}, {0, 0}, {false, false});
+    signal = new MCSignal(name, "Non-prompt psi2s", {prong}, {-1});
+    return signal;
+  }
+  if (!nameStr.compare("promptPsi2S")) {
+    MCProng prong(2, {100443, 503}, {true, true}, {false, true}, {0, 0}, {0, 0}, {false, false});
+    signal = new MCSignal(name, "Prompt psi2s (not from beauty)", {prong}, {-1});
+    return signal;
+  }
   if (!nameStr.compare("anyBeautyHadron")) {
     MCProng prong(1, {503}, {true}, {false}, {0}, {0}, {false});
     signal = new MCSignal(name, "All beauty hadrons", {prong}, {-1});
@@ -224,6 +249,11 @@ MCSignal* o2::aod::dqmcsignals::GetMCSignal(const char* name)
     signal = new MCSignal(name, "Electrons from psi2s decays", {prong}, {-1});
     return signal;
   }
+  if (!nameStr.compare("muFromPsi2S")) {
+    MCProng prong(2, {13, 100443}, {true, true}, {false, false}, {0, 0}, {0, 0}, {false, false});
+    signal = new MCSignal(name, "muons from psi2s decays", {prong}, {-1});
+    return signal;
+  }
   if (!nameStr.compare("eFromLMeeLFQ")) {
     MCProng prong(2, {11, 900}, {true, true}, {false, false}, {0, 0}, {0, 0}, {false, false});
     signal = new MCSignal(name, "Electrons from LF meson + quarkonia decays", {prong}, {-1}); //pi0,eta,eta',rho,omega,phi,jpsi,psi2s mesons
@@ -319,7 +349,12 @@ MCSignal* o2::aod::dqmcsignals::GetMCSignal(const char* name)
   }
   if (!nameStr.compare("eeFromPsi2S")) {
     MCProng prong(2, {11, 100443}, {true, true}, {false, false}, {0, 0}, {0, 0}, {false, false});
-    signal = new MCSignal(name, "ee pairs from psi2S decays", {prong, prong}, {1, 1}); // signal at pair level
+    signal = new MCSignal(name, "ee pairs from psi2s decays", {prong, prong}, {1, 1}); // signal at pair level
+    return signal;
+  }
+  if (!nameStr.compare("mumuFromPsi2S")) {
+    MCProng prong(2, {13, 100443}, {true, true}, {false, false}, {0, 0}, {0, 0}, {false, false});
+    signal = new MCSignal(name, "mumu pairs from psi2s decays", {prong, prong}, {1, 1}); // signal at pair level
     return signal;
   }
   if (!nameStr.compare("eeFromLMeeLFQ")) {