Update master to aliroot

[u/mrichter/AliRoot.git] / ANALYSIS / macros / train / AddMCGenPythia.C

AliGenerator* AddMCGenPythia(Float_t e_cms = 2760., Double_t ptHardMin = 0., Double_t ptHardMax = 1., Int_t tune = 2,Int_t cr=1,Float_t ptWeight=0) 
{
  //Add Pythia generator: pt-hard bin or min bias

  gSystem->Load("liblhapdf");
 
  AliGenerator *genP = NULL;
  genP = CreatePythia6Gen(e_cms, ptHardMin, ptHardMax, tune,cr,ptWeight);
  
  return genP;
}

AliGenerator* CreatePythia6Gen(Float_t e_cms, Int_t ptHardMin, Int_t ptHardMax, Int_t tune, Int_t cr,Float_t ptWeight) {
    
  gSystem->Load("libpythia6.4.25");
  gSystem->Load("libEGPythia6");
  gSystem->Load("libAliPythia6");

  AliGenPythia* genP = new AliGenPythia(1);

  //   vertex position and smearing 
  genP->SetVertexSmear(kPerEvent);

  // structure function
  // use kCTEQ5l for Perugia tunes 
  // except for tunes: Perugia * (325, MRSTLO*), Perugia 6 (326, CTEQ6L),  
  // Perugia 11 M (355, MRST LO**), Perugia 11 C (356, CTEQ6L1) 
  genP->SetStrucFunc(kCTEQ5L); 

  //   charm, beauty, charm_unforced, beauty_unforced, jpsi, jpsi_chi, mb
  if(ptHardMin>0.) {
   
    genP->SetProcess(kPyJets);
    genP->SetPtHard((float)ptHardMin,(float)ptHardMax);
    if(ptWeight>0) genP->SetWeightPower(ptWeight);
  } else
    genP->SetProcess(kPyMb); // Minimum Bias  

  //   Centre of mass energy 
  genP->SetEnergyCMS(e_cms); // in GeV
    
  genP->UseNewMultipleInteractionsScenario(); // for all Pythia versions >= 6.3

  if(tune == 0){ // tune Perugia0
    genP->SetTune(320);
    if(cr==0) genP->SetTune(324);
  }
  if(tune == 1){ // tune Perugia2010
    genP->SetTune(327);
    if(cr==0) genP->SetTune(324);
  } 
  if(tune == 2){ // tune Perugia2011 ('central' Perugia 2011)
    genP->SetTune(350);
    if(cr==0) genP->SetTune(354);
  }
  genP->Print();
  return genP;
}