fix const subtr tracks. Note: array has no gaps

[u/mrichter/AliRoot.git] / PWGJE / EMCALJetTasks / macros / AddTaskJetEmbeddingFromGen.C

diff --git a/PWGJE/EMCALJetTasks/macros/AddTaskJetEmbeddingFromGen.C b/PWGJE/EMCALJetTasks/macros/AddTaskJetEmbeddingFromGen.C
index acbd7a863da419d7922c1f8f8b6b4d5063021cc6..c389814ba54025c7c2688be8dceb1be964a77623 100644 (file)
--- a/PWGJE/EMCALJetTasks/macros/AddTaskJetEmbeddingFromGen.C
+++ b/PWGJE/EMCALJetTasks/macros/AddTaskJetEmbeddingFromGen.C
@@ -7,50 +7,54 @@ AliJetEmbeddingFromGenTask* AddTaskJetEmbeddingFromGen(
   Double_t        ecms           = 2760.,
   const char     *tracksName     = "GenParticles",
   const char     *taskName       = "JetEmbeddingFromGenTask",
-  const Double_t  minPt          = 10,
-  const Double_t  maxPt          = 10,
+  const Double_t  minPt          = 0.15,
+  const Double_t  maxPt          = 1000,
   const Double_t  minEta         = -0.9,
   const Double_t  maxEta         = 0.9,
   const Double_t  minPhi         = 0,
   const Double_t  maxPhi         = TMath::Pi() * 2,
   const Bool_t    copyArray      = kTRUE,
   const Bool_t    drawQA         = kTRUE,
-  const char     *partonInfoName = "PartonInfo"
+  const char     *pythiaInfoName = "PythiaInfo",
+  Float_t         ptWeight       = 0,
+  Int_t           kTune          = 2,
+  Int_t           kColorReco     = 1,
+  Float_t         kQuench        = 4.4e6,
+  Int_t           kAnglePyquen   = 2   
 )
 {
   AliGenerator *genGen = NULL;
   if(genType==0) { //PYTHIA Perugia 2011
     gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/train/AddMCGenPythia.C");
-    genGen = AddMCGenPythia(ecms, ptHardMin, ptHardMax, 2);
+    genGen = AddMCGenPythia(ecms, ptHardMin, ptHardMax, kTune,kColorReco,ptWeight);
   }
   else if(genType==1 || genType==2) { //QPYTHIA and PYQUEN
     gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/train/AddMCGenQuench.C");
-    genGen = AddMCGenQuench(ecms, ptHardMin, ptHardMax, genType);
+    genGen = AddMCGenQuench(ecms, ptHardMin, ptHardMax, genType,kQuench,kAnglePyquen,ptWeight);
   }
   if(!genGen)   {
     ::Error("AddTaskJetEmbeddingFromGenTask", "Generator does not exist");
     return NULL;
   }
 
-  AliJetEmbeddingFromGenTask *task = AddTaskJetEmbeddingFromGen(genGen,tracksName,taskName,minPt,maxPt,minEta,maxEta,minPhi,maxPhi,copyArray,drawQA,partonInfoName);
+  AliJetEmbeddingFromGenTask *task = AddTaskJetEmbeddingFromGen(genGen,tracksName,taskName,minPt,maxPt,minEta,maxEta,minPhi,maxPhi,copyArray,drawQA,pythiaInfoName);
 
   return task;
-
 }
 
 AliJetEmbeddingFromGenTask* AddTaskJetEmbeddingFromGen(
   AliGenerator   *genGen,
   const char     *tracksName     = "GenParticles",
   const char     *taskName       = "JetEmbeddingFromGenTask",
-  const Double_t  minPt          = 10,
-  const Double_t  maxPt          = 10,
+  const Double_t  minPt          = 0.15,
+  const Double_t  maxPt          = 1000,
   const Double_t  minEta         = -0.9,
   const Double_t  maxEta         = 0.9,
   const Double_t  minPhi         = 0,
   const Double_t  maxPhi         = TMath::Pi() * 2,
   const Bool_t    copyArray      = kTRUE,
   const Bool_t    drawQA         = kTRUE,
-  const char     *partonInfoName = ""
+  const char     *pythiaInfoName = ""
 )
 {  
   // Get the pointer to the existing analysis manager via the static access method.
@@ -81,7 +85,7 @@ AliJetEmbeddingFromGenTask* AddTaskJetEmbeddingFromGen(
     jetEmb->SetGen(genGen);
   }
   jetEmb->SetTracksName(tracksName);
-  jetEmb->SetPartonInfoName(partonInfoName);
+  jetEmb->SetPythiaInfoName(pythiaInfoName);
   jetEmb->SetEtaRange(minEta, maxEta);
   jetEmb->SetPhiRange(minPhi, maxPhi);
   jetEmb->SetPtRange(minPt, maxPt);