X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;f=EVGEN%2FAliGenGrayParticles.h;h=f7f350552f1bbbf55529566b5f4ddb94911caf8f;hb=417b3b549cf058d2cb3c34499db9bc2922152a2e;hp=108dae67f31e62d22cf08c8b0d8d66489c32113d;hpb=ca33591a510f802e71e66ea3a23de4129f8f5398;p=u%2Fmrichter%2FAliRoot.git

diff --git a/EVGEN/AliGenGrayParticles.h b/EVGEN/AliGenGrayParticles.h
index 108dae67f31..f7f350552f1 100644
--- a/EVGEN/AliGenGrayParticles.h
+++ b/EVGEN/AliGenGrayParticles.h
@@ -6,6 +6,7 @@
 /* $Id$ */
 
 #include "AliGenerator.h"
+class AliGrayParticleModel;
 
 class AliGenGrayParticles : public AliGenerator
 {
@@ -21,7 +22,13 @@ public:
     virtual void SetCharge(Int_t c = 1) {fCharge = c;}
     virtual void SetTemperature(Double_t t = 0.05) {fTemperature = t;}
     virtual void SetBetaSource(Double_t b = 0.05) {fBetaSource = b;}
-    
+    //
+    virtual void SetGrayParticleModel(AliGrayParticleModel* model) 
+	{fGrayParticleModel = model;}
+    virtual Bool_t NeedsCollisionGeometry() {return kTRUE;}
+    virtual void   SetCollisionGeometry(AliCollisionGeometry* geom)
+	{fCollisionGeometry = geom;}
+	    
  protected:
     void     GenerateSlow(Int_t charge, Double_t T, Double_t beta, Float_t* q);
     Double_t Maxwell(Double_t m, Double_t p, Double_t t);
@@ -36,6 +43,8 @@ public:
     Int_t    fCharge;      // Slow nucleon charge
     Float_t  fTemperature; // Source Temperature
     Float_t  fBetaSource;  // Source beta
+    //
+    AliGrayParticleModel* fGrayParticleModel; // The gray particle model
   ClassDef(AliGenGrayParticles,1) // Gray Particle Generator
 };
 #endif