Possibility to introduce centrality dependence into the existing AliGenMUONCocktailpp.

Smbat Grigoryan <Smbat.Grigoryan@cern.ch>

diff --git a/EVGEN/AliGenCocktail.cxx b/EVGEN/AliGenCocktail.cxx
index 7524d942d825c6053e329057a1505018ed10b595..09acfc75528348c091f57305551a2c86ace6d585 100644 (file)
--- a/EVGEN/AliGenCocktail.cxx
+++ b/EVGEN/AliGenCocktail.cxx
@@ -211,6 +211,7 @@ AddGenerator(AliGenerator *Generator, const char* Name, Float_t RateExp, TFormul
            }
          //
          //      Number of signals is calculated from Collision Geometry
+         //      and entry with given centrality bin is selected
          //
          if (entry->Formula() != 0)
            {
@@ -221,14 +222,19 @@ AddGenerator(AliGenerator *Generator, const char* Name, Float_t RateExp, TFormul
              AliCollisionGeometry* coll = (collentry->Generator())->CollisionGeometry();
              Float_t b  = coll->ImpactParameter();
              Int_t nsig = Int_t(entry->Formula()->Eval(b));
-             if (nsig < 1) nsig = 1;
-             AliInfo(Form("Signal Events %13.3f %5d %5d\n", b, coll->HardScatters(), nsig));
-             ntimes = nsig;
+             Int_t bin = entry->Bin() - 100;
+             if (bin > 0) {
+               if (bin != nsig) continue;
+             } else {
+               if (nsig < 1) nsig = 1;
+               AliInfo(Form("Signal Events %13.3f %5d %5d\n", b, coll->HardScatters(), nsig));
+               ntimes = nsig;
+             }
            }
          
-         entry->Generator()->SetVertex(fVertex.At(0), fVertex.At(1), fVertex.At(2));
+         gen->SetVertex(fVertex.At(0), fVertex.At(1), fVertex.At(2));
          
-         for (Int_t i = 0; i < ntimes; i++) entry->Generator()->Generate();
+         for (Int_t i = 0; i < ntimes; i++) gen->Generate();
          entry->SetLast(partArray->GetEntriesFast());
          preventry = entry;
        }

diff --git a/EVGEN/AliGenCocktailEntry.h b/EVGEN/AliGenCocktailEntry.h
index 7c1f24d713727c8b05f41c0342a95b0b3668bb1e..17b3e0988784f6c9961fa588d11635584233d832 100644 (file)
--- a/EVGEN/AliGenCocktailEntry.h
+++ b/EVGEN/AliGenCocktailEntry.h
@@ -30,6 +30,7 @@ class AliGenCocktailEntry : public TNamed
     Int_t GetFirst() const {return fFirst;}
     Int_t GetLast () const {return fLast;}
     Float_t Rate()   const {return fRate;}
+    Int_t   Bin()    const {return Int_t(fRate);}
     void  PrintInfo() const;
     TFormula* Formula() const {return fFormula;}
     AliGenCocktailEntry & operator =(const AliGenCocktailEntry & rhs);