Pythia Event Weight modifications

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

diff --git a/ANALYSIS/macros/train/AddMCGenQuench.C b/ANALYSIS/macros/train/AddMCGenQuench.C
index 20ed913d5fc3773d4c78914941240dcba6fbefa3..f3bbcf62ac2dba6653dbb2468e9c98d0e8ade120 100644 (file)
--- a/ANALYSIS/macros/train/AddMCGenQuench.C
+++ b/ANALYSIS/macros/train/AddMCGenQuench.C
@@ -23,16 +23,16 @@
 //If you want k in fm then you have to divide by the squared of hbarc (hbar c=0.197 GeV fm)
 // This gives k=4.4e^6 fm, which is the quench value we set in SetQhat
 
-AliGenerator* AddMCGenQuench(Float_t e_cms = 2760., Double_t ptHardMin = 0., Double_t ptHardMax = 0., Int_t fpythia = 1, Double_t quench=4.4e6, Int_t ianglepyquen = 2) 
+AliGenerator* AddMCGenQuench(Float_t e_cms = 2760., Double_t ptHardMin = 0., Double_t ptHardMax = 0., Int_t fpythia = 1, Double_t quench=4.4e6, Int_t ianglepyquen = 2,Int_t ptWeight=0) 
 {
   //Add Pythia generator: pt-hard bin or min bias
 
   gSystem->Load("liblhapdf.so");
  
-  return CreatePythia6Gen(e_cms, ptHardMin, ptHardMax, fpythia, quench, ianglepyquen);
+  return CreatePythia6Gen(e_cms, ptHardMin, ptHardMax, fpythia, quench, ianglepyquen,ptWeight);
 }
 
-AliGenerator* CreatePythia6Gen(Float_t e_cms, Int_t ptHardMin, Int_t ptHardMax, Int_t fpythia, Double_t quench = 4.4e6, Int_t ianglepyquen=2) {
+AliGenerator* CreatePythia6Gen(Float_t e_cms, Int_t ptHardMin, Int_t ptHardMax, Int_t fpythia, Double_t quench = 4.4e6, Int_t ianglepyquen=2,Int_t ptWeight=0) {
     
   gSystem->Load("libqpythia.so");
   gSystem->Load("libEGPythia6.so");
@@ -47,7 +47,8 @@ AliGenerator* CreatePythia6Gen(Float_t e_cms, Int_t ptHardMin, Int_t ptHardMax,
   if (ptHardMin>0.) {
     genP->SetProcess(kPyJets);
     genP->SetPtHard((float)ptHardMin,(float)ptHardMax);
-  } else
+    genP->SetWeightPower(ptWeight);
+   } else
     genP->SetProcess(kPyMb); // Minimum Bias  
 
   //   Centre of mass energy