Bug fix (Chiara)

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

diff --git a/EVGEN/AliGenHIJINGpara.h b/EVGEN/AliGenHIJINGpara.h
index c476e2970acb30f75ba934689e7713669b913020..89c50302cab46956ea4165d920d5cfb6fb1e8dcd 100644 (file)
--- a/EVGEN/AliGenHIJINGpara.h
+++ b/EVGEN/AliGenHIJINGpara.h
@@ -9,9 +9,10 @@
 // used for the ALICE TDRs.
 // Author: andreas.morsch@cern.ch
 
-#include "AliGenerator.h"
 class TF1;
-class AliDecayerPythia;
+
+#include "AliGenerator.h"
+#include "AliDecayer.h"
 
 class AliGenHIJINGpara : public AliGenerator
 {
@@ -19,26 +20,34 @@ 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;}
- private:
-  void DecayPi0(Float_t* orig, Float_t * p);
+  virtual void SetWeighting(Int_t flag = 0) {fAnalog = flag;}
+  virtual void Draw( const char *);
+  
  protected:
-  Int_t   fNt;         // CurrentTrack;
-  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
+  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) ;
+
+
   ClassDef(AliGenHIJINGpara,3) // Hijing parametrisation generator
 };
 #endif