From 898bb7f872e2969feec2f6455c2f40223bbab737 Mon Sep 17 00:00:00 2001
From: morsch <morsch@f7af4fe6-9843-0410-8265-dc069ae4e863>
Date: Tue, 30 Jun 2009 11:22:09 +0000
Subject: [PATCH] Possibility to set eta-range (needed for CORRFW)

---
 EVGEN/AliGenBox.cxx | 11 ++++++++++-
 EVGEN/AliGenBox.h   |  7 +++++--
 2 files changed, 15 insertions(+), 3 deletions(-)

diff --git a/EVGEN/AliGenBox.cxx b/EVGEN/AliGenBox.cxx
index 9c0440b3894..4c3d7975c58 100644
--- a/EVGEN/AliGenBox.cxx
+++ b/EVGEN/AliGenBox.cxx
@@ -79,7 +79,15 @@ void AliGenBox::Generate()
 
     for(i=0;i<fNpart;i++) {
 	Rndm(random,3);
-	theta=fThetaMin+random[0]*(fThetaMax-fThetaMin);
+	
+	if (TestBit(kThetaRange)) {
+	    theta = fThetaMin+random[0]*(fThetaMax-fThetaMin);
+	} else {
+	    Float_t eta = fEtaMin+random[0]*(fEtaMax-fEtaMin);
+	    theta = 2. * TMath::ATan(TMath::Exp(-eta));
+	}
+	    
+	    
 	if(TestBit(kMomentumRange)) {
 	    pmom=fPMin+random[1]*(fPMax-fPMin);
 	    pt=pmom*TMath::Sin(theta);
@@ -88,6 +96,7 @@ void AliGenBox::Generate()
 	    pt=fPtMin+random[1]*(fPtMax-fPtMin);
 	    pmom=pt/TMath::Sin(theta);
 	}
+
 	phi=fPhiMin+random[2]*(fPhiMax-fPhiMin);
 	p[0] = pt*TMath::Cos(phi);
 	p[1] = pt*TMath::Sin(phi);
diff --git a/EVGEN/AliGenBox.h b/EVGEN/AliGenBox.h
index 73f0600e7ad..73d2db9efee 100644
--- a/EVGEN/AliGenBox.h
+++ b/EVGEN/AliGenBox.h
@@ -21,12 +21,15 @@ class AliGenBox : public AliGenerator
   virtual ~AliGenBox() {}
   virtual void Generate();
   virtual void Init();
+  virtual void SetEtaRange(Float_t etamin, Float_t 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
-- 
2.43.0