coverity fix

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

diff --git a/EVGEN/AliGenBox.h b/EVGEN/AliGenBox.h
index 238fa8313225ada3d28f90df48af61ea053b42b3..708ba3a514812217fff05dc8567e3ce7a69c77fb 100644 (file)
--- a/EVGEN/AliGenBox.h
+++ b/EVGEN/AliGenBox.h
@@ -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
@@ -21,12 +21,16 @@ class AliGenBox : public AliGenerator
   virtual ~AliGenBox() {}
   virtual void Generate();
   virtual void Init();
+  virtual void SetEtaRange(Float_t etamin, Float_t etamax)
+      {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