GenEventValidityChecker.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
// class header include
#include "GenEventValidityChecker.h"
 
// HepMC includes
#include "AtlasHepMC/GenEvent.h"
 
// CLHEP includes
#include "CLHEP/Geometry/Point3D.h"
 
namespace Simulation
{
  GenEventValidityChecker::GenEventValidityChecker( const std::string& t,
                                                    const std::string& n,
                                                    const IInterface* p )
    : base_class(t,n,p)
  {
  }
 
  StatusCode GenEventValidityChecker::initialize()
  {
    ATH_MSG_VERBOSE("Initializing ...");
    ATH_MSG_VERBOSE("Initialize successful");
    return StatusCode::SUCCESS;
  }
 
  StatusCode GenEventValidityChecker::finalize()
  {
    ATH_MSG_VERBOSE("Finalizing ...");
    ATH_MSG_VERBOSE("Finalize successful");
    return StatusCode::SUCCESS;
  }
 
  StatusCode GenEventValidityChecker::manipulate(HepMC::GenEvent& ge, const EventContext&) const
  {
    bool allOK = true;
 
    // loop over the vertices in the GenEvent
#ifdef HEPMC3
    auto vtxIt  = ge.vertices().begin();
    auto vtxEnd = ge.vertices().end();
#else
    auto vtxIt  = ge.vertices_begin();
    auto vtxEnd = ge.vertices_end();
#endif
    for( ; vtxIt != vtxEnd; ++vtxIt) {
      // for quick access:
      const HepMC::FourVector &curPos = (*vtxIt)->position();
 
      // check if all position values are in range
      allOK &= std::isfinite( curPos.x() );
      allOK &= std::isfinite( curPos.y() );
      allOK &= std::isfinite( curPos.z() );
      // in case m_checkTime==false --> always return true here:
      allOK &= std::isfinite( curPos.t() ) || !m_checkTime;
    }
 
    if (allOK) {
      ATH_MSG_DEBUG("All vertices in the given GenEvent are valid.");
      return StatusCode::SUCCESS;
    }
 
    ATH_MSG_ERROR("At least one vertex in the given GenEvent has an invalid position value (NaN or inf).");
    return StatusCode::FAILURE;
  }
 
}