Extract starlight from the cocktail for Psi2sToMuPi decays (#28)

alisw · May 19, 2017 · 063af58 · 063af58
1 parent 774c38c
commit 063af58
Showing 1 changed file with 14 additions and 2 deletions.
diff --git a/MC/CustomGenerators/PWGUD/Starlight_tuned_bslope.C b/MC/CustomGenerators/PWGUD/Starlight_tuned_bslope.C
@@ -1,12 +1,24 @@
 AliGenerator* GeneratorCustom() {
-  AliGenStarLight *genStarLight = (AliGenStarLight *) GeneratorStarlight();
+  AliGenerator* gen = GeneratorStarlight();
+
   if (processConfig.Contains("Incoh")) {
     Float_t bslope = 3.75;
     if      (systemConfig.EqualTo("p-Pb")) bslope = 6.4;   // Pb towards muon arm
     else if (systemConfig.EqualTo("Pb-p")) bslope = 3.75;  // p  towards muon arm
     printf("GeneratorCustom: b-slope = %.2f\n",bslope);
+
+    AliGenStarLight *genStarLight = 0;
+    if (processConfig.Contains("Psi2sToElPi") || processConfig.Contains("Psi2sToMuPi") || processConfig.Contains("RhoPrime")) {
+      // printf("Extract starlight pointer from the cocktail\n");
+      AliGenCocktail* genCocktail = (AliGenCocktail *) gen;
+      genStarLight = (AliGenStarLight*) genCocktail->FirstGenerator()->Generator();
+    } else {
+      // printf("Take Starlight pointer directly\n");
+      genStarLight = (AliGenStarLight *) gen;
+    }
     genStarLight->SetParameter("BSLOPE_DEFINITION = 1");
     genStarLight->SetParameter(Form("BSLOPE_VALUE      = %.2f",bslope));
   }
-  return genStarLight;
+
+  return gen;
 }