From 481b70cc8593c498f812c48400ed0e042a5d2c36 Mon Sep 17 00:00:00 2001
From: bianchil <Livio.Bianchi@cern.ch>
Date: Wed, 11 Nov 2020 17:40:51 +0100
Subject: [PATCH] Added strangeness generator for pbpb 2018 cascades (mc-to-mc
 embedding enabled)

---
 .../PWGLF/Hijing_Str_pbpb2018.C               | 51 +++++++++++++++++++
 1 file changed, 51 insertions(+)
 create mode 100644 MC/CustomGenerators/PWGLF/Hijing_Str_pbpb2018.C

diff --git a/MC/CustomGenerators/PWGLF/Hijing_Str_pbpb2018.C b/MC/CustomGenerators/PWGLF/Hijing_Str_pbpb2018.C
new file mode 100644
index 00000000..5e4f73fc
--- /dev/null
+++ b/MC/CustomGenerators/PWGLF/Hijing_Str_pbpb2018.C
@@ -0,0 +1,51 @@
+AliGenerator *GeneratorCustom(TString opt = "")
+{
+
+  AliGenCocktail *ctl  = (AliGenCocktail*) GeneratorCocktail("Hijing_Str_pbpb2018");
+
+  //check if this is an embedding simulation. Otherwise, simulate hijing event before injection
+  TString simulation = gSystem->Getenv("CONFIG_SIMULATION");
+  bool isEmbedding=kFALSE;
+  if(!simulation.Contains("Embed")){
+    AliGenerator *hij = GeneratorHijing();
+    ctl->AddGenerator(hij, "Hijing", 1.);
+  }
+  else isEmbedding=kTRUE;
+
+  //if it is embedding, check impact parameter. If impossible, Fatal
+  double bEvalFormula=-1;
+  if(isEmbedding){
+    if(bminConfig>bmaxConfig)Fatal("Hijing_Str_pbpb2018","max impact parameter smaller than min impact parameter");
+    if(bminConfig<0.)Fatal("Hijing_Str_pbpb2018","min impact parameter <0");
+    if(bmaxConfig>20)Fatal("Hijing_Str_pbpb2018","unphysical impact parameter");
+    bEvalFormula=(bmaxConfig+bminConfig)/2.;
+  }
+  
+  //decide ho many particles to inject, depending on the impact parameter
+  int ninj[4] = {30, 10, 3, 1};
+  int iinj = 0;
+  if(isEmbedding){
+    if     ( bEvalFormula>0 && bEvalFormula<6 ) iinj = 0;
+    else if( bEvalFormula>5 && bEvalFormula<9 ) iinj = 1;
+    else if( bEvalFormula>5.5&& bEvalFormula<11.5 ) iinj = 2;
+    else if( bEvalFormula>11 && bEvalFormula<15) iinj = 3;
+  }
+  else{
+    TString optList[4] = {"a", "b", "c", "d"};
+    for (int iopt = 0; iopt < 4; iopt++)
+      if (opt.EqualTo(optList[iopt]))
+        iinj = iopt;
+  }
+
+  //define and add signal generators
+  AliGenerator   *ixim = GeneratorInjector(ninj[iinj], 3312, 0.5, 12., -0.6, 0.6);
+  AliGenerator   *ixip = GeneratorInjector(ninj[iinj],-3312, 0.5, 12., -0.6, 0.6);
+  AliGenerator   *iomm = GeneratorInjector(ninj[iinj], 3334, 0.5, 10., -0.6, 0.6);
+  AliGenerator   *iomp = GeneratorInjector(ninj[iinj],-3334, 0.5, 10., -0.6, 0.6);
+  ctl->AddGenerator(ixim, "Injector (Xi-)", 1.);
+  ctl->AddGenerator(ixip, "Injector (Xi+)", 1.);
+  ctl->AddGenerator(iomm, "Injector (Omega-)", 1.);
+  ctl->AddGenerator(iomp, "Injector (Omega+)", 1.);
+
+  return ctl;
+}