Fixes for bug #50317: Correct AliGenPythia for B-jet studies (Jennifer)
[u/mrichter/AliRoot.git] / PYTHIA6 / AliGenPythia.h
index 0dfa1e5..6096f56 100644 (file)
@@ -98,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
@@ -106,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
@@ -275,11 +277,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
@@ -292,7 +296,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