Bug corrected.

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

diff --git a/EVGEN/AliGenGrayParticles.h b/EVGEN/AliGenGrayParticles.h
index 108dae67f31e62d22cf08c8b0d8d66489c32113d..f7f350552f1bbbf55529566b5f4ddb94911caf8f 100644 (file)
--- 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