Restored call to CreateDigitizer (F.Prino)

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

#ifndef ALIRSNDAUGHTERDEF_H
#define ALIRSNDAUGHTERDEF_H

/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */
 
////////////////////////////////////////////////////////////////////////////////
//
//  Single daughter definitions.
//
////////////////////////////////////////////////////////////////////////////////

#include "AliRsnDaughter.h"

class AliRsnDaughterDef : public TObject {
public:

   AliRsnDaughterDef();
   AliRsnDaughterDef(EPARTYPE type, Char_t charge = 0);
   AliRsnDaughterDef(AliRsnDaughter::ESpecies type, Char_t charge = 0);
   AliRsnDaughterDef(AliRsnDaughter::ERefType refType, Char_t charge = 0);
   AliRsnDaughterDef(const AliRsnDaughterDef &copy);
   const AliRsnDaughterDef& operator= (const AliRsnDaughterDef &copy);
   virtual ~AliRsnDaughterDef() { }

   // getters
   Double_t                 GetMass()    const {return fMass;}
   Char_t                   GetChargeC() const {return fCharge;}
   Short_t                  GetChargeS() const {if (fCharge == '+') return 1; else if (fCharge == '-') return -1; else return 0;}
   AliRsnDaughter::ESpecies GetPID()     const {return fPID;}
   AliRsnDaughter::ERefType GetRefType() const {return fRefType;}
   virtual const char*      GetName()    const {return Form("%s%c", AliRsnDaughter::SpeciesName(fPID), fCharge);}

   // setters
   void SetPID(AliRsnDaughter::ESpecies pid = AliRsnDaughter::kUnknown)     {fPID = pid; fRefType = AliRsnDaughter::RefType(pid);}
   void SetCharge(Char_t charge = 0)                                        {fCharge = charge;}
   void SetRefType(AliRsnDaughter::ERefType type = AliRsnDaughter::kNoType) {fRefType = type;}
   
   // checker
   Bool_t MatchesPID(AliRsnDaughter *daughter);
   Bool_t MatchesCharge(AliRsnDaughter *daughter);
   Bool_t MatchesRefType(AliRsnDaughter *daughter);
   Bool_t MatchesDaughter(AliRsnDaughter *daughter, Bool_t truePID = kFALSE);
   
   // external checker
   Bool_t MatchesPDG(Int_t pdgCode)      {return (AliRsnDaughter::SpeciesPDG(fPID) == pdgCode);}
   Bool_t MatchesCharge(Short_t charge)  {return (GetChargeS() == charge);}

private:

   AliRsnDaughter::ESpecies  fPID;     // PID of particles
   Double_t                  fMass;    // mass of particles (subordinate to fPID)
   Char_t                    fCharge;  // charge of particles
   AliRsnDaughter::ERefType  fRefType; // object reference type (track/V0/cascade)

   // ROOT dictionary
   ClassDef(AliRsnDaughterDef, 1)
};

//__________________________________________________________________________________________________
inline Bool_t AliRsnDaughterDef::MatchesPID(AliRsnDaughter *daughter)
{
//
// Checks if the daughter true PID (taken from MC) matches
// that expected by this object.
// Works only if an MC is present and the fPID data member
// is set to something different from 'kUnknown'.
// If above conditions are not satisfied, it returns always kTRUE.
//

   if (fPID == AliRsnDaughter::kUnknown || daughter->GetRefMC() == 0x0) 
      return kTRUE;
   else
      return (AliRsnDaughter::SpeciesPDG(fPID) == daughter->GetPDG());
}
 
//__________________________________________________________________________________________________
inline Bool_t AliRsnDaughterDef::MatchesCharge(AliRsnDaughter *daughter)
{
//
// Checks that the daughter charge matches that expected by this object.
// Works only if the fCharge data member is set to '+', '-' or '0', 
// otherwise it accepts everything.
//

   switch (fCharge) {
      case '+': return daughter->IsPos();
      case '-': return daughter->IsNeg();
      case '0': return daughter->IsNeutral();
      default : return kTRUE;
   }
}
   
//__________________________________________________________________________________________________
inline Bool_t AliRsnDaughterDef::MatchesRefType(AliRsnDaughter *daughter)
{
//
// Checks that the daughter object type matches that expected by this object.
// Works only if the fRefType data member is different from AliRsnDaughter::kNoType,
// otherwise it accepts everything.
//

   switch (fRefType) {
      case AliRsnDaughter::kTrack  : return daughter->IsTrack();
      case AliRsnDaughter::kV0     : return daughter->IsV0();
      case AliRsnDaughter::kCascade: return daughter->IsCascade();
      default                      : return kTRUE;
   }
}

#endif