Coverity fix

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

diff --git a/PWG2/RESONANCES/AliRsnPairDef.h b/PWG2/RESONANCES/AliRsnPairDef.h
index eaf691c09b538445f83ece6e37c9e0757e865f36..c3ba2db2956f523a01a3a8c6e619073e709c9f1c 100644 (file)
--- a/PWG2/RESONANCES/AliRsnPairDef.h
+++ b/PWG2/RESONANCES/AliRsnPairDef.h
@@ -1,62 +1,53 @@
-//
-// Class AliRsnPairDef
-//
-// Defines a decay channel for a resonance,
-// resulting in a specified PDG code for the mother,
-// and the particle type for the daughters, defined
-// according to the internal PID format of the package
-//
-// author: A. Pulvirenti (alberto.pulvirenti@ct.infn.it)
-//
-
 #ifndef ALIRSNPAIRDEF_H
 #define ALIRSNPAIRDEF_H
 
-#include <TString.h>
-
-#include "AliPID.h"
-
-class AliRsnDaughter;
-
-class AliRsnPairDef : public TObject
-{
-  public:
-
-    AliRsnPairDef();
-    AliRsnPairDef(Char_t sign1, AliPID::EParticleType type1,
-                  Char_t sign2, AliPID::EParticleType type2, Int_t motherPDG = 0);
-    AliRsnPairDef(AliPID::EParticleType type1, Char_t sign1,
-                  AliPID::EParticleType type2, Char_t sign2, Int_t motherPDG = 0);
-    AliRsnPairDef(const AliRsnPairDef &copy);
-    const AliRsnPairDef& operator= (const AliRsnPairDef &copy);
-    virtual ~AliRsnPairDef() { }
-
-    // getters
-    Char_t                GetCharge(Int_t i) const {if (i>=0&&i<2) return fCharge[i]; else return 0;}
-    AliPID::EParticleType GetType(Int_t i) const {if (i>=0&&i<2) return fType[i]; else return AliPID::kUnknown;}
-    Double_t              GetMass(Int_t i) const {if (i>=0&&i<2) return fMass[i]; else return 0.0;}
-    Int_t                 GetMotherPDG() const {return fMotherPDG;}
-    TString               GetPairName();
-
-    // setters
-    Bool_t SetPairElement(Int_t i, Char_t charge, AliPID::EParticleType pid);
-    Bool_t SetPair(Char_t ch1, AliPID::EParticleType pid1, Char_t ch2, AliPID::EParticleType pid2);
-    void   SetMotherPDG(Int_t pdg) {fMotherPDG = pdg;}
-
-    // pair information methods
-    Bool_t IsLikeSign() {return (fCharge[0] == fCharge[1]);}
-    Bool_t HasEqualTypes() {return (fType[0] == fType[1]);}
-
-  private:
-
-    // pair parameters
-    Int_t                  fMotherPDG;  // PDG code of true mother (if known)
-    Double_t               fMass[2];    // mass of particles
-    Char_t                 fCharge[2];  // charge of particles
-    AliPID::EParticleType  fType[2];    // PID of particles
-
-    // ROOT dictionary
-    ClassDef(AliRsnPairDef, 1)
+/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
+ * See cxx source for full Copyright notice                               */
+ 
+////////////////////////////////////////////////////////////////////////////////
+//
+//  Resonance decay tree definition.
+//
+////////////////////////////////////////////////////////////////////////////////
+
+#include "AliRsnDaughter.h"
+#include "AliRsnDaughterDef.h"
+
+class AliRsnPairDef : public TObject {
+public:
+
+   AliRsnPairDef();
+   AliRsnPairDef(EPARTYPE type1, Char_t ch1, EPARTYPE type2, Char_t ch2, Int_t motherPDG = 0, Double_t motherMass = 0.0);
+   AliRsnPairDef(AliRsnDaughter::ESpecies type1, Char_t ch1, AliRsnDaughter::ESpecies type2, Char_t ch2, Int_t motherPDG = 0, Double_t motherMass = 0.0);
+   AliRsnPairDef(const AliRsnPairDef &copy);
+   const AliRsnPairDef& operator= (const AliRsnPairDef &copy);
+   virtual ~AliRsnPairDef() { }
+
+   virtual const char*     GetName()       const {return Form("%s_%s", fDef1.GetName(), fDef2.GetName());}
+   Int_t                    GetMotherPDG()  const {return fMotherPDG;}
+   Double_t                 GetMotherMass() const {return fMotherMass;}
+   AliRsnDaughterDef&       GetDef1()             {return fDef1;}
+   AliRsnDaughterDef&       GetDef2()             {return fDef2;}
+   AliRsnDaughterDef&       GetDef(Int_t i)       {if (i<1) return GetDef1(); else return GetDef2();}
+
+   void SetMotherPDG(Int_t pdg)                 {fMotherPDG = pdg;}
+   void SetMotherMass(Double_t mass)            {fMotherMass = mass;}
+   void SetDef1(AliRsnDaughterDef *def)         {if (def) fDef1 = (*def);}
+   void SetDef2(AliRsnDaughterDef *def)         {if (def) fDef2 = (*def);}
+   void SetDef(Int_t i, AliRsnDaughterDef *def) {if (!def) return; if (i<1) fDef1 = (*def); else fDef2 = (*def);}
+
+   Bool_t IsLikeSign()  const {return (fDef1.GetChargeC() == fDef2.GetChargeC());}
+   Bool_t HasEqualPID() const {return (fDef1.GetPID() == fDef2.GetPID());}
+
+private:
+
+   Double_t          fMotherMass;  // nominal mass of true mother
+   Int_t             fMotherPDG;   // PDG code of true mother (if known)
+   AliRsnDaughterDef fDef1;        // definitions for daughter #1 (see class)
+   AliRsnDaughterDef fDef2;        // definitions for daughter #2 (see class)
+
+   // ROOT dictionary
+   ClassDef(AliRsnPairDef, 1)
 };
 
 #endif