http://savannah.cern.ch/bugs/?103960

[u/mrichter/AliRoot.git] / EVGEN / AliGenAfterBurnerFlow.cxx

diff --git a/EVGEN/AliGenAfterBurnerFlow.cxx b/EVGEN/AliGenAfterBurnerFlow.cxx
index 8e816c86663e865522010c4538535694db8451fa..4ce8a5784399c514b6248a919c14e4f4e34f8ae0 100644 (file)
--- a/EVGEN/AliGenAfterBurnerFlow.cxx
+++ b/EVGEN/AliGenAfterBurnerFlow.cxx
@@ -36,10 +36,13 @@
 #include "AliGenAfterBurnerFlow.h"
 #include "AliGenCocktailAfterBurner.h"
 #include "AliMC.h"
+#include "AliRun.h"
 #include "AliCollisionGeometry.h"
 #include "AliGenCocktailEntry.h"
 
 
+using std::cout;
+using std::endl;
 ClassImp(AliGenAfterBurnerFlow)
 
 
@@ -212,7 +215,7 @@ void AliGenAfterBurnerFlow::SetFlowParameters(Int_t pdg, Int_t order, Int_t type
   // private function
   // 
 
-  if(TMath::Abs(pdg)>fgkPDG){
+  if(TMath::Abs(pdg)>=fgkPDG){
     Error("AliAfterBurnerFlow","Overflow");
     return;
   }
@@ -343,7 +346,7 @@ Float_t AliGenAfterBurnerFlow::GetCoefficient(Int_t pdg, Int_t n, Float_t Pt, Fl
 
 ////////////////////////////////////////////////////////////////////////////////////////////////////
 
-Float_t AliGenAfterBurnerFlow::GetNpNorm(Int_t npart)
+Float_t AliGenAfterBurnerFlow::GetNpNorm(Int_t npart) const
 {
   //
   // Calculate npart norm.
@@ -367,9 +370,9 @@ Float_t AliGenAfterBurnerFlow::GetNpNorm(Int_t npart)
 
 ////////////////////////////////////////////////////////////////////////////////////////////////////
 
-Bool_t AliGenAfterBurnerFlow::IsPrimary(Int_t pdg)
+Bool_t AliGenAfterBurnerFlow::IsPrimary(Int_t pdg) const
 {
-  if(pdg>fgkPDG) return kFALSE;
+  if(pdg>=fgkPDG) return kFALSE;
   return fIsPrim[pdg];
 }
 
@@ -377,6 +380,7 @@ Bool_t AliGenAfterBurnerFlow::IsPrimary(Int_t pdg)
 
 Double_t CalcAngle(Double_t phi, Double_t phi0, Double_t phiRP, Double_t v2, Double_t v1=0.)
 {
+    // Calculate relative angle
   Double_t phi1 = phi-(phi+2*v1*TMath::Sin(phi-phiRP)+v2*TMath::Sin(2*(phi-phiRP))-phi0)/
     (1.+2*v1*TMath::Cos(phi-phiRP)+ 2*v2*TMath::Cos(2*(phi-phiRP)));
   if(TMath::Abs(phi/phi1-1.)<0.00001) return phi1;
@@ -387,6 +391,7 @@ Double_t CalcAngle(Double_t phi, Double_t phi0, Double_t phiRP, Double_t v2, Dou
 
 void AliGenAfterBurnerFlow::InitPrimaries()
 {
+  // Init the primary particle list
   for(Int_t i=0; i<fgkPDG; i++) fIsPrim[i]=kFALSE;
 
   //mesons
@@ -483,7 +488,6 @@ void AliGenAfterBurnerFlow::Generate()
   // M.G. Poghosyan 
   // PWG2 meeting on 06.05.2008 and 03.06.2008
 
-  Info("Generate()","");
 
   if (0) 
     for(Int_t ii=0; ii<fCounter;ii++)
@@ -567,7 +571,7 @@ void AliGenAfterBurnerFlow::Generate()
       //exclude incoming protons in PYTHIA     
       if(particle->GetPdgCode()==21) continue;
 
-      if(TMath::Abs(pdg)>fgkPDG) continue; 
+      if(TMath::Abs(pdg)>=fgkPDG) continue; 
       // is particle primary?
       if(!fIsPrim[TMath::Abs(pdg)]) continue; 
 
@@ -609,6 +613,7 @@ void AliGenAfterBurnerFlow::Generate()
 
 void AliGenAfterBurnerFlow::Rotate(Int_t i, Double_t phi, Bool_t IsPrim)
 {
+  // Rotation
   TParticle*  particle = fStack->Particle(i);
   
   TLorentzVector momentum;