Store impact parameter for q-pythia.

[u/mrichter/AliRoot.git] / PYTHIA6 / AliGenPythia.h

diff --git a/PYTHIA6/AliGenPythia.h b/PYTHIA6/AliGenPythia.h
index 8dde2b1d3f2a284a9814437c3e3e00d752718004..ba751f8977340211ff1cfb68a8debec812861ae6 100644 (file)
--- a/PYTHIA6/AliGenPythia.h
+++ b/PYTHIA6/AliGenPythia.h
@@ -25,6 +25,7 @@ class AliGenPythiaEventHeader;
 class AliGenEventHeader;
 class AliStack;
 class AliRunLoader;
+class TObjArray; 
 
 class AliGenPythia : public AliGenMC
 {
@@ -97,6 +98,7 @@ class AliGenPythia : public AliGenMC
     virtual void  SetPi0InEMCAL(Bool_t b)        {fCheckEMCAL   = b; fPi0InCalo        = b;}
     virtual void  SetPi0InPHOS(Bool_t b)         {fCheckPHOS    = b; fPi0InCalo        = b;}
     virtual void  SetPhotonInEMCAL(Bool_t b)     {fCheckEMCAL   = b; fPhotonInCalo     = b;}
+    virtual void  SetElectronInEMCAL(Bool_t b)   {fEleInEMCAL   = b;}
     virtual void  SetPhotonInPHOS(Bool_t b)      {fCheckPHOS    = b; fPhotonInCalo     = b;}
 
     // Trigger on a minimum multiplicity
@@ -105,6 +107,7 @@ class AliGenPythia : public AliGenMC
     virtual void  SetPhotonInPHOSeta(Bool_t b)   {fCheckPHOSeta = b; fPhotonInCalo     = b;}
     virtual void  SetFragPhotonOrPi0MinPt(Float_t pt)      {fFragPhotonOrPi0MinPt = pt;}
     virtual void  SetPhotonMinPt(Float_t pt)     {fPhotonMinPt = pt;}
+    virtual void  SetElectronMinPt(Float_t pt)     {fElectronMinPt = pt;}
     // Trigger and rotate event 
     void RotatePhi(Int_t iphcand, Bool_t& okdd);
     // Trigger on a single particle
@@ -133,8 +136,13 @@ class AliGenPythia : public AliGenMC
     //
     // Quenching
     //
-    // Set quenching mode 0 = no, 1 = AM, 2 = IL
+    // Set quenching mode 0 = no, 1 = AM, 2 = IL,  3 = NA, 4 = ACS
     virtual void SetQuench(Int_t flag = 0) {fQuench = flag;}
+    // Set transport coefficient.
+    void SetQhat(Float_t qhat) {fQhat = qhat;}
+    //Set initial medium length.
+    void SetLength(Float_t length) {fLength = length;}
+
     virtual void SetHadronisation(Int_t flag = 1) {fHadronisation = flag;}
     virtual void SetReadFromFile(const Text_t *filname) {fFileName = filname;  fReadFromFile = 1;}    
 
@@ -153,7 +161,8 @@ class AliGenPythia : public AliGenMC
                             Float_t thresh = 0., Float_t etseed = 4.,
                             Float_t minet = 10., Float_t r = 1.);
     
-    void LoadEvent(AliStack* stack, Int_t flag = 0, Int_t reHadr = 0);
+  void LoadEvent(AliStack* stack, Int_t flag = 0, Int_t reHadr = 0);
+  void LoadEvent(TObjArray* stack, Int_t flag = 0, Int_t reHadr = 0);
     // Getters
     virtual Process_t    GetProcess() const {return fProcess;}
     virtual StrucFunc_t  GetStrucFunc() const {return fStrucFunc;}
@@ -212,6 +221,9 @@ class AliGenPythia : public AliGenMC
     Int_t       fNpartons;          //Number of partons before hadronisation
     Int_t       fReadFromFile;      //read partons from file
     Int_t       fQuench;            //Flag for quenching
+    Float_t     fQhat;              //Transport coefficient (GeV^2/fm)
+    Float_t     fLength;            //Medium length (fm)
+    Float_t     fImpact;            //Impact parameter for quenching simulation (q-pythia) 
     Float_t     fPtKick;            //Transverse momentum kick
     Bool_t      fFullEvent;         //!Write Full event if true
     AliDecayer  *fDecayer;          //!Pointer to the decayer instance
@@ -266,11 +278,13 @@ class AliGenPythia : public AliGenMC
     Bool_t fFragPhotonInCalo; // Option to ask for Fragmentation Photon in calorimeters acceptance
     Bool_t fPi0InCalo;        // Option to ask for Pi0 in calorimeters acceptance
     Bool_t fPhotonInCalo;     // Option to ask for Decay Photon in calorimeter acceptance
+    Bool_t fEleInEMCAL;       // Option to ask for Electron in EMCAL acceptance
     Bool_t fCheckEMCAL;       // Option to ask for FragPhoton or Pi0 in calorimeters EMCAL acceptance
     Bool_t fCheckPHOS;        // Option to ask for FragPhoton or Pi0 in calorimeters PHOS acceptance
     Bool_t fCheckPHOSeta;     // Option to ask for PHOS eta acceptance
     Float_t fFragPhotonOrPi0MinPt; // Minimum momentum of Fragmentation Photon or Pi0
     Float_t fPhotonMinPt;          // Minimum momentum of Photon 
+    Float_t fElectronMinPt;        // Minimum momentum of Electron 
     //Calorimeters eta-phi acceptance 
     Float_t fPHOSMinPhi;           // Minimum phi PHOS
     Float_t fPHOSMaxPhi;           // Maximum phi PHOS
@@ -283,7 +297,7 @@ class AliGenPythia : public AliGenMC
     AliGenPythia(const AliGenPythia &Pythia);
     AliGenPythia & operator=(const AliGenPythia & rhs);
 
-    ClassDef(AliGenPythia,8) // AliGenerator interface to Pythia
+    ClassDef(AliGenPythia,9) // AliGenerator interface to Pythia
 };
 #endif