small update

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

diff --git a/EVGEN/AliGenBox.h b/EVGEN/AliGenBox.h
index 73f0600e7ad1fba8c60827ce673de21525feff11..6996b9f83c8b179c7fb5fdc73a2d045edb9d441e 100644 (file)
--- a/EVGEN/AliGenBox.h
+++ b/EVGEN/AliGenBox.h
@@ -19,14 +19,19 @@ class AliGenBox : public AliGenerator
   AliGenBox();
   AliGenBox(Int_t npart);
   virtual ~AliGenBox() {}
+  virtual void GenerateN(Int_t ntimes);
   virtual void Generate();
   virtual void Init();
+  virtual void SetEtaRange(Float_t etamin, Float_t etamax)
+      {SetBit(kEtaRange);fEtaMin = etamin; fEtaMax = etamax;}
   virtual void SetPart(Int_t part) {fIpart=part;}
+  virtual void SetParticleType(Int_t part) {SetPart(part);}
 protected:
 
   Int_t fIpart; // Particle type
-
-  ClassDef(AliGenBox,1) // Square box random generator
+  Float_t fEtaMin;  // Minimum eta 
+  Float_t fEtaMax;  // Maximum eta
+  ClassDef(AliGenBox,2) // Square box random generator
 };
 
 #endif