WriteHepMC.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#include "TruthIO/WriteHepMC.h"
#ifdef HEPMC3
#include "HepMC3/WriterAscii.h"
#endif
 
WriteHepMC::WriteHepMC(const std::string& name, ISvcLocator* pSvcLocator)
  : GenBase(name, pSvcLocator)
{
  declareProperty("OutputFile", m_outfile="events.hepmc");
  declareProperty("Precision", m_precision=8);
  declareProperty("Format", m_format="hepmc2");
}
 
 
StatusCode WriteHepMC::initialize() {
  CHECK(GenBase::initialize());
#ifdef HEPMC3
  if (m_format == "hepmc2") {
    auto writer = new HepMC3::WriterAsciiHepMC2(m_outfile);
    writer->set_precision(m_precision);
    m_hepmcio.reset(writer);
  }
  if (m_format == "hepmc3") {
    auto writer = new HepMC3::WriterAscii(m_outfile);
    writer->set_precision(m_precision);
    m_hepmcio.reset(writer);
  }
#else
  m_hepmcio.reset( new HepMC::IO_GenEvent(m_outfile) );
  m_hepmcio->precision(m_precision);
#endif
  return StatusCode::SUCCESS;
}
 
 
StatusCode WriteHepMC::execute() {
  // Just write out the first (i.e. signal) event in the collection
#ifdef HEPMC3
  m_hepmcio->write_event(*(event_const()));
#else
  m_hepmcio->write_event(event_const());
#endif
  return StatusCode::SUCCESS;
}