hepmcoutput.h

Go to the documentation of this file.

 
/*
 *
 *    Copyright (c) 2014-2020
 *      SMASH Team
 *
 *    GNU General Public License (GPLv3 or later)
 *
 */
 
#ifndef SRC_INCLUDE_SMASH_HEPMCOUTPUT_H_
#define SRC_INCLUDE_SMASH_HEPMCOUTPUT_H_
 
#include <memory>
#include <string>
 
#include "forwarddeclarations.h"
#include "outputinterface.h"
#include "outputparameters.h"
 
#include <boost/filesystem.hpp>
 
#include "HepMC3/GenEvent.h"
#include "HepMC3/GenParticle.h"
#include "HepMC3/GenVertex.h"
#include "HepMC3/WriterAscii.h"
 
namespace smash {
 
class HepMcOutput : public OutputInterface {
 public:
  HepMcOutput(const bf::path &path, std::string name,
              const OutputParameters &out_par, const int total_N,
              const int proj_N);
 
  ~HepMcOutput();
 
  void at_eventstart(const Particles &particles, const int event_number,
                     const EventInfo &) override;
 
  void at_eventend(const Particles &particles, const int32_t event_number,
                   const EventInfo &event) override;
 
 private:
  static const int status_code_for_beam_particles;
  static const int status_code_for_final_particles;
 
  const bf::path filename_;
  bf::path filename_unfinished_;
 
  const int total_N_;
  const int proj_N_;
 
  int construct_nuclear_pdg_code(int na, int nz) const;
 
  std::unique_ptr<HepMC3::WriterAscii> output_file_;
 
  std::unique_ptr<HepMC3::GenEvent> current_event_;
 
  HepMC3::GenVertexPtr vertex_;
};
 
}  // namespace smash
 
#endif  // SRC_INCLUDE_SMASH_HEPMCOUTPUT_H_