Change SetDir to SetDirection
[u/mrichter/AliRoot.git] / ZDC / AliGenZDC.h
index a135240366065b2aeb76a8adc2ae79945f41dccd..a7825decb574a7495350f496922d61a0d74c61bc 100644 (file)
@@ -30,7 +30,7 @@ public:
   // Parameters that could be set for generation
   virtual void SetParticle(Int_t ipart) {fIpart=ipart;};
   virtual void SetMomentum(Float_t ptot) {fPMin=ptot; fPMax=ptot;};
-  virtual void SetDir(Float_t zpsrp, Float_t cosx, Float_t cosy, Float_t cosz)
+  virtual void SetDirection(Float_t zpsrp, Float_t cosx, Float_t cosy, Float_t cosz)
                 {fPseudoRapidity=zpsrp; fCosx=cosx; fCosy=cosy; fCosz=cosz;};
   virtual void SetFermi(Int_t Fflag) {fFermiflag=Fflag;};
   virtual void SetDiv(Float_t bmdiv, Float_t bmcra, Int_t iflcr) 
@@ -39,9 +39,10 @@ public:
   // Getters 
   Double_t GetFermi2p(Int_t key) {return fProbintp[key];}
   Double_t GetFermi2n(Int_t key) {return fProbintn[key];}
-  Float_t  GetMomentum(Int_t key) {return fP[key];};
+  Float_t  GetInMomentum(Int_t key) {return fPInit[key];};
   Float_t  GetBoostMomentum(Int_t key) {return fBoostP[key];};
   Float_t  GetDivMomentum(Int_t key) {return fDivP[key];};
+  Float_t  GetTrackMomentum(Int_t key) {return fPTrack[key];};
   
   // Fermi smearing, beam divergence and crossing angle               
   virtual void FermiTwoGaussian(Double_t A, Float_t Z, Double_t* pp, 
@@ -66,9 +67,11 @@ protected:
   Double_t fProbintp[201];      // for protons
   Double_t fProbintn[201];      // for neutrons
   Double_t fPp[201];            // for protons
-  Double_t fP[3];               // momentum
-  Float_t fBoostP[3];           // boost vector
-  Float_t fDivP[3];             // divergence
+  Double_t fP[3];               // temporary momentum
+  Float_t  fPInit[3];           // initial momentum 
+  Float_t  fBoostP[3];          // boosted momentum
+  Float_t  fDivP[3];            // divergence
+  Float_t  fPTrack[3];          // track momentum
   
    ClassDef(AliGenZDC,1)  // Generator for AliZDC class
 };