Setter for calling Generate n-times

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

diff --git a/EVGEN/AliOmegaDalitz.cxx b/EVGEN/AliOmegaDalitz.cxx
index c278a8b82074d1ea218046f3e570fa3c8a33764d..dcf7944c445d3a6a6628bdf3dea6bb71931e4431 100644 (file)
--- a/EVGEN/AliOmegaDalitz.cxx
+++ b/EVGEN/AliOmegaDalitz.cxx
@@ -49,7 +49,8 @@ ClassImp(AliOmegaDalitz)
 AliOmegaDalitz::AliOmegaDalitz():
        AliDecayer(),
         fEPMass(0),
-       fMPMass(0)
+       fMPMass(0),
+       fInit(0)
 {
     // Constructor
 }
@@ -122,7 +123,7 @@ void AliOmegaDalitz::Init()
 }
 
 
-void AliOmegaDalitz::Decay(Int_t idlepton, const TLorentzVector* pparent)
+void AliOmegaDalitz::Decay(Int_t idlepton, TLorentzVector* pparent)
 {
 //-----------------------------------------------------------------------------
 //
@@ -130,6 +131,11 @@ void AliOmegaDalitz::Decay(Int_t idlepton, const TLorentzVector* pparent)
 //
 //-----------------------------------------------------------------------------
 
+    if (!fInit) {
+       Init();
+       fInit=1;
+    }
+    
   Double_t pmass, lmass, omass, lpmass;
   Double_t e1, p1, e3, p3;
   Double_t betaSquare, lambda;
@@ -249,7 +255,7 @@ Int_t AliOmegaDalitz::ImportParticles(TClonesArray *particles)
 
 void AliOmegaDalitz::
 Rot(Double_t pin[3], Double_t pout[3], Double_t costheta, Double_t sintheta,
-    Double_t cosphi, Double_t sinphi)
+    Double_t cosphi, Double_t sinphi) const
 {
 // Perform rotation
   pout[0] = pin[0]*costheta*cosphi-pin[1]*sinphi+pin[2]*sintheta*cosphi;
@@ -315,7 +321,8 @@ void AliOmegaDalitz::Copy(TObject&) const
 AliOmegaDalitz::AliOmegaDalitz(const AliOmegaDalitz &dalitz)
   : AliDecayer(),
     fEPMass(0),
-    fMPMass(0)
+    fMPMass(0),
+    fInit(0)
 {
   // Copy constructor
   dalitz.Copy(*this);