Decay pi0 if requested.
[u/mrichter/AliRoot.git] / EVGEN / AliGenHIJINGpara.h
index 3a1fda4..4d34f38 100644 (file)
@@ -11,6 +11,7 @@
 
 #include "AliGenerator.h"
 class TF1;
+class AliDecayerPythia;
 
 class AliGenHIJINGpara : public AliGenerator
 {
@@ -25,15 +26,18 @@ class AliGenHIJINGpara : public AliGenerator
   virtual void Generate();
   virtual void Init();
   AliGenHIJINGpara & operator=(const AliGenHIJINGpara & rhs);
-  virtual void SetPtRange(Float_t ptmin=0, Float_t ptmax=15);
-
+  virtual void SetPtRange(Float_t ptmin = 0., Float_t ptmax=15.);
+  virtual void SetPi0Decays(Bool_t flag = kFALSE) {fPi0Decays = flag;}
+ private:
+  void DecayPi0(Float_t* orig, Float_t * p);
  protected:
   Float_t fCutVertexZ; // Vertex truncation
+  Bool_t  fPi0Decays;  // Flag for pi0 decays
   TF1* fPtpi;          // Parametrised pt distribution for pi
   TF1* fPtka;          // Parametrised pt distribution for ka
   TF1* fETApic;        // Parametrised eta distribution for pi
   TF1* fETAkac;        // Parametrised eta distribution fro ka
-
+  AliDecayerPythia  *fDecayer; // ! Pointer to pythia object for decays
   ClassDef(AliGenHIJINGpara,2) // Hijing parametrisation generator
 };
 #endif