small update

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

diff --git a/EVGEN/AliGenHIJINGpara.h b/EVGEN/AliGenHIJINGpara.h
index 5350809f3ce5824f2d66a8df52d29fa24e4a5504..89c50302cab46956ea4165d920d5cfb6fb1e8dcd 100644 (file)
--- a/EVGEN/AliGenHIJINGpara.h
+++ b/EVGEN/AliGenHIJINGpara.h
@@ -5,29 +5,50 @@
 
 /* $Id$ */
 
-#include "AliGenerator.h"
+// 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 "AliDecayer.h"
+
 class AliGenHIJINGpara : public AliGenerator
 {
  public:
 
   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