Have to create Digits trees and branches by hand in the case of multiple event files.

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

diff --git a/EVGEN/AliGenPythia.h b/EVGEN/AliGenPythia.h
index 588b8ae19df078a29ef9b8506af25c92f0fafc4a..3e5ec381f653e199fbb6fd8709d9b9670fdcdc1d 100644 (file)
--- a/EVGEN/AliGenPythia.h
+++ b/EVGEN/AliGenPythia.h
@@ -7,10 +7,10 @@
 
 
 #include "AliGenerator.h"
-#include "AliDecayer.h"
 #include "GenTypeDefs.h"
 #include <TArrayI.h>    
 
+class AliDecayer;
 class AliPythia;
 class TParticle;
 
@@ -34,6 +34,8 @@ class AliGenPythia : public AliGenerator
     virtual void    SetEnergyCMS(Float_t energy=5500) {fEnergyCMS=energy;}
     // force decay type
     virtual void    SetForceDecay(Decay_t decay=semimuonic) {fForceDecay=decay;}
+    // treat protons as inside nuclei
+    virtual void    SetNuclei(Int_t a1, Int_t a2);
     // get cross section of process
     virtual Float_t GetXsection() {return fXsection;}
     // Check PDG code
@@ -50,10 +52,13 @@ class AliGenPythia : public AliGenerator
     TArrayI     fParentSelect;  // Parent particles to be selected 
     TArrayI     fChildSelect;   // Decay products to be selected
     Float_t     fXsection;      // Cross-section
-    AliPythia   *fPythia;       // Pythia 
+    AliPythia   *fPythia;       //! Pythia 
     Float_t     fPtHardMin;     // lower pT-hard cut 
     Float_t     fPtHardMax;     // higher pT-hard cut
-    AliDecayer  *fDecayer;
+    Int_t       fNucA1;         // mass number nucleus side 1
+    Int_t       fNucA2;         // mass number nucleus side 2
+    
+    AliDecayer  *fDecayer;      // pointer to the decayer instance
  private:
     // check if particle is selected as parent particle
     Bool_t ParentSelected(Int_t ip);