ParticleBase.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2018 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef PARTICLEEVENT_PARTICLEBASE_H
#define PARTICLEEVENT_PARTICLEBASE_H
 
#include "EventKernel/IParticle.h"
 
#include "VxVertex/VxContainer.h"
#include "VxVertex/VxCandidate.h"
#include "AthLinks/ElementLink.h"
#include "AthContainers/DataVector.h"
#include "NavFourMom/IParticleContainer.h"
 
class ParticleBase : public virtual IParticle
{
 public:
  
  ParticleBase();
  
  ParticleBase( const ParticleBase& rhs );
 
  ParticleBase& operator=( const ParticleBase& rhs );
 
  virtual ~ParticleBase();
 
  // standard IParticle methods 
  virtual ParticleDataType::DataType dataType()    const;
 
  virtual const Trk::RecVertex* origin() const ;
 
  virtual const ElementLink<VxContainer>& originLink () const { return m_origin; }
 
  virtual bool hasCharge()         const;
 
  virtual ChargeType charge()      const;
 
  virtual bool hasPdgId()          const;
 
  virtual int pdgId()     const;
  
  // Set functions (for IParticle)
  void set_dataType(ParticleDataType::DataType x ) { m_dataType = x;}
  void set_charge(ChargeType x ) { m_charge = x; m_hasCharge = true;}
  void set_pdgId(int x ) { m_pdgId = x; m_hasPdgId = true;}
  void reset_charge() { m_hasCharge = false; }
  void reset_pdgId() { m_hasPdgId = false; }
  void set_origin(const VxContainer* theContainer, int index);
  void set_origin(const VxContainer* theContainer, const Trk::VxCandidate * vertex );
  void set_origin(const ElementLink<VxContainer>& origin);
 
 private:
 
  ElementLink< VxContainer > m_origin;
  ChargeType m_charge;
  bool m_hasCharge;
  int m_pdgId;
  bool m_hasPdgId;
  ParticleDataType::DataType m_dataType;  
 
};
 
inline void ParticleBase::set_origin(const VxContainer* const theContainer, int index)
{
  m_origin.toIndexedElement( *theContainer, index) ;
}
inline void ParticleBase::set_origin(const VxContainer* theContainer, const Trk::VxCandidate * vertex )
{
  m_origin.toContainedElement(*theContainer, vertex);
}
inline void ParticleBase::set_origin(const ElementLink<VxContainer>& origin)
{
  m_origin = origin;
}
inline   ParticleDataType::DataType ParticleBase::dataType() const 
{
  return m_dataType;
}
inline bool ParticleBase::hasCharge()  const { return m_hasCharge;}
inline ChargeType ParticleBase::charge() const { return m_charge ;}
inline bool ParticleBase::hasPdgId() const {return m_hasPdgId ;}
inline int ParticleBase::pdgId() const {return m_pdgId;}
 
DATAVECTOR_VIRTBASES1 (ParticleBase, IParticle);
 
#endif // PARTICLEEVENT_PARTICLEBASE_H