- correct minor issues with forcred gamma decays, and speed up
[u/mrichter/AliRoot.git] / EVGEN / AliGenHIJINGpara.h
index effa0fa..89c5030 100644 (file)
@@ -5,8 +5,14 @@
 
 /* $Id$ */
 
+// Parameterisation of pi and K, eta and pt distributions
+// used for the ALICE TDRs.
+// Author: andreas.morsch@cern.ch
+
+class TF1;
+
 #include "AliGenerator.h"
-#include "TF1.h"
+#include "AliDecayer.h"
 
 class AliGenHIJINGpara : public AliGenerator
 {
@@ -14,20 +20,35 @@ class AliGenHIJINGpara : public AliGenerator
 
   AliGenHIJINGpara();
   AliGenHIJINGpara(Int_t npart);
-  AliGenHIJINGpara(const AliGenHIJINGpara &HIJINGpara);
-     
   virtual ~AliGenHIJINGpara();
+  virtual void SetCutVertexZ(Float_t cut=999999.) {fCutVertexZ = cut;}
   virtual void Generate();
   virtual void Init();
-  AliGenHIJINGpara & operator=(const AliGenHIJINGpara & rhs);
+  virtual void SetPtRange(Float_t ptmin = 0., Float_t ptmax=15.);
+  virtual void SetPi0Decays(Bool_t flag = kFALSE) {fPi0Decays = flag;}
+  virtual void SetWeighting(Int_t flag = 0) {fAnalog = flag;}
+  virtual void Draw( const char *);
+  
  protected:
+  Int_t   fNt;          // CurrentTrack;
+  Int_t   fNpartProd;   // Total number of particles produced  
+  Bool_t  fPi0Decays;   // Flag for pi0 decays
+  Float_t fPtWgtPi;     // pt weight for pions
+  Float_t fPtWgtKa;     // pt weight for kaons
+  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
+  AliDecayer* fDecayer; // ! Pointer to pythia object for decays
+
+  void DecayPi0(Float_t* orig, Float_t * p, Float_t time);
+
+ private:
+  AliGenHIJINGpara(const AliGenHIJINGpara &HIJINGpara);
+  AliGenHIJINGpara& operator = (const AliGenHIJINGpara &para) ;
 
-  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
 
-  ClassDef(AliGenHIJINGpara,1) // Hijing parametrisation generator
+  ClassDef(AliGenHIJINGpara,3) // Hijing parametrisation generator
 };
 #endif