hepmcinterface.h

Go to the documentation of this file.

/*
 *
 *    Copyright (c) 2021 Christian Holm Christensen
 *    Copyright (c) 2021-2024
 *      SMASH Team
 *
 *    GNU General Public License (GPLv3 or later)
 *
 */
 
#ifndef SRC_INCLUDE_SMASH_HEPMCINTERFACE_H_
#define SRC_INCLUDE_SMASH_HEPMCINTERFACE_H_
 
#include <map>
#include <memory>
#include <string>
#include <utility>
#include <valarray>
 
#include "HepMC3/GenCrossSection.h"
#include "HepMC3/GenEvent.h"
#include "HepMC3/GenHeavyIon.h"
#include "HepMC3/GenParticle.h"
#include "HepMC3/GenVertex.h"
 
#include "action.h"
#include "forwarddeclarations.h"
#include "outputinterface.h"
#include "outputparameters.h"
namespace smash {
 
class HepMcInterface : public OutputInterface {
 public:
  using AZ = std::pair<int, int>;
  HepMcInterface(const std::string& name, const bool full_event);
  void at_eventstart(const Particles& particles, const EventLabel& event_label,
                     const EventInfo& event) override;
  void at_interaction(const Action& action, const double density) override;
  void at_eventend(const Particles& particles, const EventLabel& event_label,
                   const EventInfo& event) override;
 
 protected:
  enum Status {
    beam = 4,  // Beam particle
    fnal = 1,  // final state `final` is a reserved word
    dcy = 2,   // Decay
    off = 100
  };
  using IdMap = std::map<int, HepMC3::GenParticlePtr>;
  using CollCounter = std::valarray<int>;
  void clear();
  int get_status(const ProcessType& t) const;
  HepMC3::GenParticlePtr make_gen(int pid, int status,
                                  const smash::FourVector& mom,
                                  double mass = -1);
  HepMC3::GenParticlePtr make_register(const ParticleData& p,
                                       int status = Status::fnal);
  HepMC3::GenParticlePtr find_or_make(const ParticleData& p,
                                      int status = Status::fnal,
                                      bool force_new = false);
  int ion_pdg(const AZ& az) const;
  HepMC3::GenEvent event_;
  HepMC3::GenHeavyIonPtr ion_;
  HepMC3::GenCrossSectionPtr xs_;
  HepMC3::GenVertexPtr ip_;
  IdMap map_;
  CollCounter coll_;
  int ncoll_;
  int ncoll_hard_;
  bool full_event_;
};
}  // namespace smash
 
#endif  // SRC_INCLUDE_SMASH_HEPMCINTERFACE_H_