TruthEtIsolations.h

Go to the documentation of this file.

 
/*
  Copyright (C) 2002-2021 CERN for the benefit of the ATLAS collaboration
*/
 
// TruthEtIsolations.h 
// Header file for class TruthEtIsolations
// Author  : Sebastien Binet : April 2007
#ifndef MCPARTICLEEVENT_TRUTHETISOLATIONS_H 
#define MCPARTICLEEVENT_TRUTHETISOLATIONS_H 

 
// STL includes
#include <vector>
#include <iosfwd>
#include <string>
#include <stdexcept>
#include <unordered_map>
 
// Gaudi includes
 
#include "AthLinks/ElementLink.h"
 
// GeneratorObjects includes
#include "GeneratorObjects/McEventCollection.h"
 
// McParticleEvent includes
#include "McParticleEvent/TruthParticleParamDefs.h"
 
// Forward declaration
class McEventCollection;
 
class TruthEtIsolations
{ 

  // Public typedefs: 
 public: 

  typedef ElementLink<McEventCollection> GenEventLink_t;

  typedef McAod::EtIsolations EtIsol_t;

  typedef std::unordered_map<int,EtIsol_t> EtIsolMap_t;

  typedef EtIsolMap_t::const_iterator const_iterator;

  // Public methods: 
 public: 

  TruthEtIsolations();
  
  TruthEtIsolations( const TruthEtIsolations& rhs );
  
  TruthEtIsolations( const GenEventLink_t& genEvtLink );

  TruthEtIsolations( const std::string& genEvtName,
             const std::size_t  genEvtIdx );

  TruthEtIsolations& operator=( const TruthEtIsolations& rhs );

  ~TruthEtIsolations();

  // Const methods: 

  const std::string& genEventName() const;

  std::size_t genEventIdx() const;

  std::size_t size() const;

  bool hasEtIsol( const HepMC::ConstGenParticlePtr& genParticle ) const;

  bool hasEtIsol( const int genParticleBarcode ) const;

  const EtIsol_t* etIsolations(const HepMC::ConstGenParticlePtr& genParticle ) const;

  const EtIsol_t* etIsolations( const int genParticleBarcode ) const;

  double etIsol( const HepMC::ConstGenParticlePtr& genParticle,
         const TruthParticleParameters::ConeSize coneIdx ) const;

  double etIsol( const int genParticleBarcode,
         const TruthParticleParameters::ConeSize coneIdx ) const;

  const_iterator begin() const;

  const_iterator end() const;

  // Non-const methods: 
 
  // Set functions

  void setEtIsol( const HepMC::ConstGenParticlePtr& genParticle,
          const TruthParticleParameters::ConeSize coneIdx,
                  const double etIsol );

  void setEtIsol( const HepMC::ConstGenParticlePtr& genParticle,
          const std::vector<double>& etIsols );
  
  void setEtIsol( const HepMC::ConstGenParticlePtr& genParticle,
          const EtIsol_t& etIsols );
  
  // Protected data: 
 protected:
  friend class TruthEtIsolationsCnv_p1;

  GenEventLink_t m_genEvtLink;

  EtIsolMap_t m_etIsolations;

  static const EtIsol_t m_dfltEtIsols;
}; 

// I/O functions
std::ostream& 
operator<<( std::ostream& out, const TruthEtIsolations& etIsols );

// Inline methods: 
 
inline std::size_t TruthEtIsolations::size() const
{ return m_etIsolations.size(); }
 
inline bool 
TruthEtIsolations::hasEtIsol( const int bc ) const
{
  return m_etIsolations.end() != m_etIsolations.find( bc );
}
 
inline const TruthEtIsolations::EtIsol_t* 
TruthEtIsolations::etIsolations( const int bc ) const
{
  const EtIsolMap_t::const_iterator i = m_etIsolations.find( bc );
  return ( m_etIsolations.end() == i )
    ? 0
    : &i->second;
}
  
inline
double 
TruthEtIsolations::etIsol( const int bc,
               const TruthParticleParameters::ConeSize idx ) const
{
  return (*etIsolations(bc))[idx];
}
 
inline TruthEtIsolations::const_iterator TruthEtIsolations::begin() const
{ return m_etIsolations.begin(); }
 
inline TruthEtIsolations::const_iterator TruthEtIsolations::end() const
{ return m_etIsolations.end(); }
 
 
#endif //> MCPARTICLEEVENT_TRUTHETISOLATIONS_H