WriterHepMC3.h

/*
 * WriterHepMC3.h
 *
 *  Created on: Feb 11, 2021
 *      Author: Pawel Sznajder (NCBJ) 
 */
 
#ifndef MODULES_WRITER_WRITERHEPMC3_H_
#define MODULES_WRITER_WRITERHEPMC3_H_
 
#include <ElementaryUtils/parameters/Parameters.h>
#include <memory>
#include <string>
#include <vector>
 
#include "../../beans/physics/Event.h"
#include "WriterHepMC3Type.h"
#include "WriterModule.h"
 
namespace HepMC3 {
class Writer;
} /* namespace HepMC3 */
 
namespace EPIC {
 
class WriterHepMC3: public WriterModule {
 
public:
    static const std::string PARAMETER_NAME_HEPMC3_WRITER_TYPE; 
 
    static const unsigned int classId; 
 
    WriterHepMC3(const std::string &className);
 
    WriterHepMC3(const WriterHepMC3 &other);
 
    virtual ~WriterHepMC3();
 
    virtual WriterHepMC3 *clone() const;
    virtual void configure(const ElemUtils::Parameters &parameters);
 
    virtual void open();
    virtual void saveGenerationInformation(
            const GenerationInformation& generationInformation);
    virtual void close();
    virtual void write(const Event &event);
    virtual void write(const std::vector<Event> &events);
 
    //********************************************************
    //*** SETTERS AND GETTERS ********************************
    //********************************************************
 
    WriterHepMC3Type::Type getWriterHepMC3Type() const;
 
    void setWriterHepMC3Type(WriterHepMC3Type::Type writerHepMc3Type);
 
private:
 
    int getParticleCode(ParticleCodeType::Type type) const;
 
    WriterHepMC3Type::Type m_writerHepMC3Type;      
    std::shared_ptr<HepMC3::Writer> m_writerHepMC3; 
};
 
} /* namespace EPIC */
 
#endif /* MODULES_WRITER_WRITERHEPMC3_H_ */