GenParticleInteractingFilter.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2017 CERN for the benefit of the ATLAS collaboration
*/
 
// GenParticleInteractingFilter.cxx, (c) ATLAS Detector software
 
// class header include
#include "GenParticleInteractingFilter.h"
 
// HepMC includes
#include "AtlasHepMC/GenParticle.h"
#include "AtlasHepMC/Flow.h"
 
// Helper function
#include "TruthUtils/HepMCHelpers.h"
 
ISF::GenParticleInteractingFilter::GenParticleInteractingFilter( const std::string& t,
                                                                 const std::string& n,
                                                                 const IInterface* p )
  : base_class(t,n,p)
{
  declareProperty("AdditionalInteractingParticleTypes", m_additionalInteractingParticleTypes);
  declareProperty("AdditionalNonInteractingParticleTypes", m_additionalNonInteractingParticleTypes);
}
 
StatusCode ISF::GenParticleInteractingFilter::initialize()
{
  if(m_additionalInteractingParticleTypes.empty()) {
    ATH_MSG_DEBUG("No additional particle types will be classified as interacting.");
  }
  else {
    ATH_MSG_DEBUG("Will classify particles with the following additional PDG codes as interacting:");
    for(const auto& pdg_id : m_additionalInteractingParticleTypes) {
      ATH_MSG_DEBUG("   " << pdg_id);
    }
  }
  if(m_additionalNonInteractingParticleTypes.empty()) {
    ATH_MSG_DEBUG("No additional particle types will be classified as non-interacting.");
  }
  else {
    ATH_MSG_DEBUG("Will classify particles with the following additional PDG codes as non-interacting:");
    for(const auto& pdg_id : m_additionalNonInteractingParticleTypes) {
      ATH_MSG_DEBUG("   " << pdg_id);
    }
  }
  if (!m_additionalInteractingParticleTypes.empty() && !m_additionalNonInteractingParticleTypes.empty()) {
    //Insanity check
    bool clash(false);
    for (const auto& pdg_id : m_additionalInteractingParticleTypes) {
      if (find(m_additionalNonInteractingParticleTypes.begin(),
               m_additionalNonInteractingParticleTypes.end(),
               pdg_id) != m_additionalNonInteractingParticleTypes.end())
        {
          ATH_MSG_ERROR("Particle with PDG code " << pdg_id << " requested to be classified as both interacting and non-interacting.");
          clash=true;
        }
    }
    if (clash) {
      ATH_MSG_ERROR("Please check your configuration!");
      return StatusCode::FAILURE;
    }
  }
 
  return StatusCode::SUCCESS;
}
 
#ifdef HEPMC3
bool ISF::GenParticleInteractingFilter::pass(const HepMC::ConstGenParticlePtr& particle) const {
  const int pdg_id = particle->pdg_id();
#else
bool ISF::GenParticleInteractingFilter::pass(const HepMC::GenParticle& particle) const {
  const int pdg_id = particle.pdg_id();
#endif
  const bool isInteracting = find(m_additionalInteractingParticleTypes.begin(),
                                m_additionalInteractingParticleTypes.end(),
                                pdg_id) != m_additionalInteractingParticleTypes.end();
  const bool isNonInteracting = find(m_additionalNonInteractingParticleTypes.begin(),
                                     m_additionalNonInteractingParticleTypes.end(),
                                pdg_id) != m_additionalNonInteractingParticleTypes.end();
  return !( !MC::isInteracting( pdg_id ) || isNonInteracting) || isInteracting;
}