hepmcoutput.cc

Go to the documentation of this file.

/*
 *
 *    Copyright (c) 2020-2022
 *      SMASH Team
 *
 *    GNU General Public License (GPLv3 or later)
 *
 */
 
#include "smash/hepmcoutput.h"
 
#include "HepMC3/Print.h"
#include "HepMC3/WriterAscii.h"
 
#ifdef SMASH_USE_HEPMC_ROOTIO
#include "HepMC3/WriterRootTree.h"
#endif
 
namespace smash {
// clang-format off
// clang-format on
HepMcOutput::HepMcOutput(const std::filesystem::path &path, std::string name,
                         const bool full_event, std::string HepMC3_output_type)
    : HepMcInterface(name, full_event),
      filename_(path / (name + "." + HepMC3_output_type)) {
  filename_unfinished_ = filename_;
  filename_unfinished_ += +".unfinished";
#ifdef SMASH_USE_HEPMC_ROOTIO
  if (HepMC3_output_type == "asciiv3") {
#endif
    output_file_ = std::make_unique<HepMC3::WriterAscii>(
        filename_unfinished_.string(), event_.run_info());
    output_type_ = asciiv3;
#ifdef SMASH_USE_HEPMC_ROOTIO
  } else {
    output_file_ = std::make_unique<HepMC3::WriterRootTree>(
        filename_unfinished_.string(), event_.run_info());
    output_type_ = treeroot;
  }
#endif
}
HepMcOutput::~HepMcOutput() {
  logg[LOutput].debug() << "Renaming file " << filename_unfinished_ << " to "
                        << filename_ << std::endl;
  output_file_->close();
  std::filesystem::rename(filename_unfinished_, filename_);
}
 
void HepMcOutput::at_eventend(const Particles &particles,
                              const int32_t event_number,
                              const EventInfo &event) {
  HepMcInterface::at_eventend(particles, event_number, event);
  logg[LOutput].debug() << "Writing event " << event_number << " with "
                        << event_.particles().size() << " particles and "
                        << event_.vertices().size() << " vertices to output "
                        << std::endl;
  output_file_->write_event(event_);
}
 
}  // namespace smash