IO_EPOS.h

Go to the documentation of this file.

//--------------------------------------------------------------------------
#ifndef HEPMC_IO_EPOS_H
#define HEPMC_IO_EPOS_H
//THIS IS A BRUTAL COPY OF EPOS g_Wrapper from HEPMC
//We need it because the EPOS generator needs a largeer version of EPOS g to store the event
#include <map>
#include <vector>
#include "HepMC/IO_BaseClass.h"
#include "GeneratorInterface/ReggeGribovPartonMCInterface/interface/EPOS_Wrapper.h"
 
namespace EPOS {
  /*
    namespace HepMC{
      class GenEvent;
      class GenVertex;
      class GenParticle;
    }  
*/
 
  class IO_EPOS : public HepMC::IO_BaseClass {
  public:
    IO_EPOS();
    ~IO_EPOS() override;
    bool fill_next_event(HepMC::GenEvent*) override;
    void write_event(const HepMC::GenEvent*) override;
    void print(std::ostream& ostr = std::cout) const override;
 
    // see comments below for these switches.
    bool trust_both_mothers_and_daughters() const;
    bool trust_mothers_before_daughters() const;
    bool print_inconsistency_errors() const;
    bool trust_beam_particles() const;
    void set_trust_mothers_before_daughters(bool b = true);
    void set_trust_both_mothers_and_daughters(bool b = false);
    void set_print_inconsistency_errors(bool b = true);
    void set_trust_beam_particles(bool b = true);
    void set_skip_nuclear_fragments(bool b = true);
 
  protected:  // for internal use only
    HepMC::GenParticle* build_particle(int index);
    void build_production_vertex(int i, std::vector<HepMC::GenParticle*>& hepevt_particle, HepMC::GenEvent* evt);
    void build_end_vertex(int i, std::vector<HepMC::GenParticle*>& hepevt_particle, HepMC::GenEvent* evt);
    int find_in_map(const std::map<HepMC::GenParticle*, int>& m, HepMC::GenParticle* p) const;
 
  private:  // use of copy constructor is not allowed
    IO_EPOS(const IO_EPOS&) : HepMC::IO_BaseClass() {}
 
  private:  // data members
    bool m_trust_mothers_before_daughters;
    bool m_trust_both_mothers_and_daughters;
    bool m_print_inconsistency_errors;
    bool m_trust_beam_particles;
    bool m_skip_nucl_frag;
  };
 
  // INLINES access methods //
  inline bool IO_EPOS::trust_both_mothers_and_daughters() const { return m_trust_both_mothers_and_daughters; }
 
  inline bool IO_EPOS::trust_mothers_before_daughters() const { return m_trust_mothers_before_daughters; }
 
  inline bool IO_EPOS::print_inconsistency_errors() const { return m_print_inconsistency_errors; }
 
  inline void IO_EPOS::set_trust_both_mothers_and_daughters(bool b) { m_trust_both_mothers_and_daughters = b; }
 
  inline void IO_EPOS::set_trust_mothers_before_daughters(bool b) { m_trust_mothers_before_daughters = b; }
 
  inline void IO_EPOS::set_print_inconsistency_errors(bool b) { m_print_inconsistency_errors = b; }
 
  inline bool IO_EPOS::trust_beam_particles() const { return m_trust_beam_particles; }
 
  inline void IO_EPOS::set_trust_beam_particles(bool b) { m_trust_beam_particles = b; }
 
  inline void IO_EPOS::set_skip_nuclear_fragments(bool b) { m_skip_nucl_frag = b; }
 
}  // namespace EPOS
 
#endif  // HEPMC_IO_EPOS_H
//--------------------------------------------------------------------------