Adding pA as option for AMPT AddTask macro

diff --git a/ANALYSIS/macros/train/AddMCGenAmpt.C b/ANALYSIS/macros/train/AddMCGenAmpt.C
index 1dce0b0590607a9c72c92b9922e5b68167b09d4a..3467b7a7d9b182087e21e6a867d2f59241d2af25 100644 (file)
--- a/ANALYSIS/macros/train/AddMCGenAmpt.C
+++ b/ANALYSIS/macros/train/AddMCGenAmpt.C
@@ -4,7 +4,8 @@ AliGenerator *AddMCGenAmpt(
                           Double_t bmax        = 20.0,    // maximum impact parameter
                           Double_t ptHardMin   = 3.0,     // minimum pt hard (was 3.0 in previous AMPT productions)
                           Bool_t stringMelting = kTRUE,   // string melting option 
-                          Bool_t useART        = kTRUE   // use hadronic rescattering phase (ART)
+                          Bool_t useART        = kTRUE,   // use hadronic rescattering phase (ART)
+                          Bool_t pAcollisions  = kFALSE   // pA instead of AA collisions 
                           )
 {
   // User defined generator
@@ -31,6 +32,9 @@ AliGenerator *AddMCGenAmpt(
   
   if(!useART)
     NTmax = 3;
+
+  if(pAcollisions)
+    NTmax = 1500; // this was used in earlier productions for p-Pb
   //=========================================================================
 
 
@@ -44,7 +48,10 @@ AliGenerator *AddMCGenAmpt(
   genAMPT->SetEnergyCMS(Energy);
   genAMPT->SetReferenceFrame("CMS");
   genAMPT->SetProjectile("A", 208, 82);
-  genAMPT->SetTarget    ("A", 208, 82);
+  if(pAcollisions)
+        genAMPT->SetTarget("P", 1, 1);
+  else
+    genAMPT->SetTarget("A", 208, 82);
   genAMPT->SetPtHardMin (ptHardMin);
   genAMPT->SetImpactParameterRange(bmin,bmax);
   //=========================================================================