Correct error handling in ExtractQA

[u/mrichter/AliRoot.git] / PYTHIA6 / AliDecayerPythia.h

diff --git a/PYTHIA6/AliDecayerPythia.h b/PYTHIA6/AliDecayerPythia.h
index 3673f12971faf82a633e243f85225d73090d4248..e048f049a37e3520bea573b210d161fa003be587 100644 (file)
--- a/PYTHIA6/AliDecayerPythia.h
+++ b/PYTHIA6/AliDecayerPythia.h
@@ -28,7 +28,7 @@ public AliDecayer
     virtual void    SetForceDecay(Int_t decay)
       {SetForceDecay((Decay_t) decay);}
     virtual void    ForceDecay();
-    
+    virtual void    HeavyFlavourOff() {fHeavyFlavour = kFALSE;}
     virtual Float_t GetPartialBranchingRatio(Int_t ipart);
     virtual Float_t GetLifetime(Int_t kf);
     virtual void    WriteDecayTable();
@@ -42,9 +42,10 @@ public AliDecayer
     void     ForceParticleDecay(Int_t particle, Int_t product, Int_t mult);
     void     ForceParticleDecay(Int_t particle, Int_t* products, Int_t* mult, Int_t npart);
     void     ForceHadronicD(Int_t optUser4Bodies=1);
-    void     ForceOmega();    
+    void     ForceOmega();
+    void     SwitchOffHeavyFlavour();
     Float_t  GetBraPart(Int_t kf);
-    void  Copy(TObject &decayer) const;
+    void     Copy(TObject &decayer) const;
 
     AliDecayerPythia &operator=(const AliDecayerPythia &decayer) 
     {decayer.Copy(*this);return(*this);}
@@ -52,11 +53,12 @@ public AliDecayer
     
  private:
     AliPythia*  fPythia;          //! Pointer to AliPythia
-    Decay_t     fDecay;           // Forced decay mode
+    Decay_t     fDecay;           //  Forced decay mode
     Float_t     fBraPart[501];    //! Branching ratios
+    Bool_t      fHeavyFlavour;    //! Flag for heavy flavors
     static Bool_t fgInit;         //! initialization flag 
     
-    ClassDef(AliDecayerPythia,2)  // AliDecayer implementation using Pythia  
+    ClassDef(AliDecayerPythia, 3) // AliDecayer implementation using Pythia  
 };
 #endif