Coverity fixes

[u/mrichter/AliRoot.git] / PWG2 / RESONANCES / AliRsnMother.h

diff --git a/PWG2/RESONANCES/AliRsnMother.h b/PWG2/RESONANCES/AliRsnMother.h
index cdfe3a46a9a021f41641bf5dfa0c00350c545725..049c4359426dedd340f4db80fce34dcb2ad780b5 100644 (file)
--- a/PWG2/RESONANCES/AliRsnMother.h
+++ b/PWG2/RESONANCES/AliRsnMother.h
@@ -1,61 +1,64 @@
-//
-// Class AliRsnMother
-//
-// Implementation of a pair of tracks, for several purposes
-// - computing the total 4-momentum & inv. mass for output histos filling
-// - evaluating cut checks on the pair of particles
-// - evaluating any kind of kinematic value over their sum
-//
-// authors: Martin Vala (martin.vala@cern.ch)
-//          Alberto Pulvirenti (alberto.pulvirenti@ct.infn.it)
-//
-
 #ifndef ALIRSNMOTHER_H
 #define ALIRSNMOTHER_H
 
-#include <TLorentzVector.h>
+/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
+ * See cxx source for full Copyright notice                               */
+ 
+////////////////////////////////////////////////////////////////////////////////
+//
+//  Interface to candidate resonance decaying into 2 bodies.
+//
+////////////////////////////////////////////////////////////////////////////////
 
-#include "AliRsnEvent.h"
+#include <TMath.h>
 #include "AliRsnDaughter.h"
 
-class AliRsnPairDef;
-
-class AliRsnMother : public TObject
-{
-  public:
+class AliRsnEvent;
 
-    AliRsnMother();
-    AliRsnMother(const AliRsnMother &obj);
-    AliRsnMother& operator=(const AliRsnMother &obj);
-    virtual ~AliRsnMother();
-    
-    TLorentzVector&   Sum()   {return fSum;}
-    TLorentzVector&   SumMC() {return fSumMC;}
-    Double_t          AngleTo(AliRsnDaughter track, Bool_t mc = kFALSE) const {return fSum.Angle(track.P(mc).Vect());}
-    Double_t          CosThetaStar(Bool_t first = kTRUE, Bool_t useMC = kFALSE);
+class AliRsnMother : public TObject {
+public:
 
-    AliRsnDaughter*   GetDaughter   (const Int_t &index) const {if (index==0||index==1) return fDaughter[index]; return 0x0;}
-    AliRsnDaughter&   GetDaughterRef(const Int_t &index) const {if (index==1) return (*fDaughter[1]); return (*fDaughter[0]);}
+   AliRsnMother() : fRefEvent(0), fSum(), fSumMC() {fDaughter[0] = fDaughter[1] = 0;}
+   AliRsnMother(const AliRsnMother &obj);
+   AliRsnMother& operator=(const AliRsnMother &obj);
+   virtual ~AliRsnMother();
 
-    Bool_t            IsLabelEqual() const {return abs(fDaughter[0]->GetLabel()) == abs(fDaughter[1]->GetLabel());}
-    Bool_t            IsIndexEqual() const {return (fDaughter[0]->GetID() == fDaughter[1]->GetID());}
-    Int_t             CommonMother(Int_t &m0, Int_t &m1) const;
-    Int_t             CommonMother() const {Int_t d0, d1; return CommonMother(d0,d1);}
+   // setters (4-vectors cannot be set)
+   void  SetDaughter(Int_t i, AliRsnDaughter *d) {fDaughter[CkID(i)] = d;}
+   void  SetRefEvent(AliRsnEvent *event)         {fRefEvent = event;}
+   
+   // getters
+   AliRsnEvent*      GetRefEvent()               {return fRefEvent;}
+   AliRsnDaughter*   GetDaughter(const Int_t &i) {return fDaughter[CkID(i)];}
+   TLorentzVector&   Sum()                       {return fSum;}
+   TLorentzVector&   SumMC()                     {return fSumMC;}
+   
+   // checks
+   Bool_t    IsLabelEqual()  const {return TMath::Abs(fDaughter[0]->GetLabel()) == TMath::Abs(fDaughter[1]->GetLabel());}
+   Bool_t    IsIndexEqual()  const {return (fDaughter[0]->GetID() == fDaughter[1]->GetID());}
+   Bool_t    IsOwnerEqual()  const {return (fDaughter[0]->GetOwnerEvent() == fDaughter[1]->GetOwnerEvent());}
+   Int_t     CommonMother()  const {Int_t d0, d1; return CommonMother(d0, d1);}
+   Int_t     CommonMother(Int_t &m0, Int_t &m1) const;
+   
+   // useful computations/operations
+   void      ComputeSum(Double_t mass1, Double_t mass2);
+   Double_t  AngleTo(AliRsnDaughter *track, Bool_t mc = kFALSE);
+   Double_t  AngleToLeading(Bool_t &success);
+   Double_t  CosThetaStar(Bool_t first = kTRUE, Bool_t useMC = kFALSE);
+   void      ResetPair();
+   void      PrintInfo(const Option_t *option = "ALL") const;
+   Bool_t    CheckPair(Bool_t checkMC = kFALSE) const;
 
-    void              SetDaughters(AliRsnDaughter * const daughter1, Double_t mass1, AliRsnDaughter * const daughter2, Double_t mass2);
-    void              ResetPair();
-    void              PrintInfo(const Option_t *option = "ALL") const;
-    Bool_t            CheckPair() const;
-    Bool_t            MatchesDef(AliRsnPairDef *pairDef);
+private:
 
-  private:
+   Int_t CkID(Int_t i) {if (i < 1) return 0; else return 1;}
 
-    Bool_t           fUseMC;            // choose if momenta are taken from ESD/AOD or MC
-    AliRsnDaughter  *fDaughter[2];      // elements of the pair
-    TLorentzVector   fSum;              // sum computed from the two daughters
-    TLorentzVector   fSumMC;            // sum computed from the two daughters
+   AliRsnDaughter  *fDaughter[2];      // elements of the pair
+   AliRsnEvent     *fRefEvent;         // reference event
+   TLorentzVector   fSum;              // sum computed from the two daughters
+   TLorentzVector   fSumMC;            // sum computed from the two daughters
 
-    ClassDef(AliRsnMother,1)
+   ClassDef(AliRsnMother, 1)
 };
 
 #endif