PdgIdFilter.h

Go to the documentation of this file.

 
/*
  Copyright (C) 2002-2017 CERN for the benefit of the ATLAS collaboration
*/
 
// PdgIdFilter.h 
// Header file for class PdgIdFilter
// Author: S.Binet<binet@cern.ch>
#ifndef ANALYSISUTILS_PDGIDFILTER_H 
#define ANALYSISUTILS_PDGIDFILTER_H 
 
// STL includes
#include <list>
 
// Gaudi includes
#include "AthenaKernel/getMessageSvc.h"
#include "GaudiKernel/MsgStream.h"
 
// AnalysisUtils includes
#include "AnalysisUtils/IFilter.h"
#include "AnalysisUtils/ParticleCandidateList.h"
 
// Forward declaration
 
template <typename T>
class PdgIdFilter : virtual public IFilter<T>
{ 
 
  // Public methods: 
 public: 
 
  PdgIdFilter();
 
  PdgIdFilter( const PdgIdFilter<T>& rhs );
 
  // Constructor with parameters: 
 
  virtual ~PdgIdFilter(); 
 
  PdgIdFilter<T>& operator=( const PdgIdFilter<T>& rhs ); 
 
  // Const methods: 
 
  virtual bool isAccepted( const T * element ) const;
 
  const ParticleCandidateList * pdgIds() const;
 
  bool matchSign() const;
 
  // Non-const methods: 
  void setFilter( const IFilterCuts * filter );
 
  void setPdgId ( const int& pdgId );
  void setPdgIds( const std::list<int>& pdgIds );
  void setPdgIds( const ParticleCandidateList& pdgIds );
 
  void setMatchSign( const bool matchSign );
 
  // Protected data: 
 protected: 
  
  ParticleCandidateList m_pdgIds;
 
  bool m_matchSign;
 
}; 
 
//MsgStream& operator<<( MsgStream & msg, const PdgIdFilter &obj );
 
 
template<typename T>
inline PdgIdFilter<T>::PdgIdFilter() :
  IFilterCuts(),
  IFilter<T>(),
  m_pdgIds(),
  m_matchSign( true )
{}
 
template<typename T>
inline PdgIdFilter<T>::PdgIdFilter( const PdgIdFilter<T>& rhs ) :
  IFilterCuts(rhs),
  IFilter<T>(rhs),
  m_pdgIds( rhs.m_pdgIds ),
  m_matchSign( rhs.m_matchSign )
{}
 
template<typename T>
PdgIdFilter<T>::~PdgIdFilter(){}
 
template<typename T>
inline
PdgIdFilter<T>& PdgIdFilter<T>::operator=(const PdgIdFilter<T>& rhs)
{
  if ( this != &rhs ) {
    IFilterCuts::operator=(rhs);
    IFilter<T>::operator=(rhs);
    m_pdgIds    = rhs.m_pdgIds;
    m_matchSign = rhs.m_matchSign;
  }
  return *this;
}
 
// Const methods: 
 
template<typename T>
bool PdgIdFilter<T>::isAccepted( const T * element ) const
{
  if ( false == m_pdgIds.empty() ) {
    const int pdgId = element->pdgId();
    //std::cout << "Received PDG: " << pdgId << std::endl;
    //m_pdgIds.dropList();
    if ( m_pdgIds.hasInList( pdgId, m_matchSign ) ) return true;
    else return false;
  }
 
  return true;
}
 
template<typename T>
inline const ParticleCandidateList * PdgIdFilter<T>::pdgIds() const
{
  return &m_pdgIds;
}
 
template<typename T>
inline bool PdgIdFilter<T>::matchSign() const
{
  return m_matchSign;
}
 
template<typename T>
void PdgIdFilter<T>::setFilter( const IFilterCuts * filter )
{
  if ( filter ) {
    const PdgIdFilter<T> * pdgFilter = 
      dynamic_cast<const PdgIdFilter<T> *>(filter );
    
    if ( pdgFilter ) {
      operator=(*pdgFilter);
    } else {
      MsgStream log( Athena::getMessageSvc(), "PdgIdFilter" );
      log << MSG::ERROR
      << "Can't dynamic_cast " << typeid(filter).name() 
      << " to a PdgIdFilter"
      << endmsg;
    }
  } //> filter is not a NULL pointer
}
 
template<typename T>
inline void PdgIdFilter<T>::setPdgId( const int& pdgId )
{
  m_pdgIds.add( pdgId );
}
 
template<typename T>
void PdgIdFilter<T>::setPdgIds( const std::list<int>& pdgIds )
{
  for ( auto itr = pdgIds.begin();
    itr != pdgIds.end();
    ++itr ) {
    m_pdgIds.add( *itr );
  }
}
 
template<typename T>
inline void PdgIdFilter<T>::setPdgIds( const ParticleCandidateList& pdgIds )
{
  setPdgIds( pdgIds.list() );
}
 
template<typename T>
inline void PdgIdFilter<T>::setMatchSign( const bool matchSign )
{
  m_matchSign = matchSign;
}
 
#endif //> ANALYSISUTILS_PDGIDFILTER_H