small update
[u/mrichter/AliRoot.git] / EVGEN / AliGenBox.h
index 238fa83..6996b9f 100644 (file)
@@ -5,11 +5,11 @@
 
 /* $Id$ */
 
-/*
- Generator for particles in a preset
- kinematic range (flat distribution)
- Comments and suggestions: andreas.morsch@cern.ch
-*/
+
+// Generator for particles in a preset
+// kinematic range (flat distribution)
+// Comments and suggestions: andreas.morsch@cern.ch
+
 
 #include "AliGenerator.h"
 class AliGenBox : public AliGenerator
@@ -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