SimTestToolBase.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
#include "SimTestToolBase.h"
 
#include "AtlasHepMC/GenEvent.h"
#include "TruthUtils/MagicNumbers.h"
#include "GeneratorObjects/McEventCollection.h"
 
SimTestToolBase::SimTestToolBase(const std::string& type, 
                   const std::string& name,
                   const IInterface* parent) : 
  base_class(type, name, parent)
{
  declareProperty("HistPath",  m_path="/truth/");
  declareProperty("McEventKey", m_key="TruthEvent");
}
 
HepMC::ConstGenParticlePtr   SimTestToolBase::getPrimary() 
{
  const McEventCollection* mcCollection;
  if (evtStore()->retrieve(mcCollection,m_key).isSuccess()) {
    DataVector<HepMC::GenEvent>::const_iterator e;
    for (e=mcCollection->begin();e!=mcCollection->end(); ++e) {
      for (auto p : (**e)) {
    if (!HepMC::is_simulation_particle(p)) {
      return p;
    }
      }
    }
  }
  return 0;
}