WriterAscii.h

// -*- C++ -*-
//
// This file is part of HepMC
// Copyright (C) 2014-2023 The HepMC collaboration (see AUTHORS for details)
//
#ifndef HEPMC3_WRITERASCII_H
#define HEPMC3_WRITERASCII_H
#include "HepMC3/Writer.h"
#include "HepMC3/GenEvent.h"
#include "HepMC3/GenRunInfo.h"
#include <string>
#include <fstream>
 
namespace HepMC3 {
 
class WriterAscii : public Writer {
public:

    WriterAscii(const std::string& filename,
                std::shared_ptr<GenRunInfo> run = std::shared_ptr<GenRunInfo>());

    WriterAscii(std::ostream& stream,
                std::shared_ptr<GenRunInfo> run = std::shared_ptr<GenRunInfo>());
    WriterAscii(std::shared_ptr<std::ostream> s_stream,
                std::shared_ptr<GenRunInfo> run = std::shared_ptr<GenRunInfo>());

    ~WriterAscii();

    void write_event(const GenEvent& evt)  override;

    void write_run_info();

    bool failed()  override;

    void close()  override;

    void set_precision(const int& prec );
    int precision() const;
private:


    void allocate_buffer();

    void set_buffer_size(const size_t& size );

    static std::string escape(const std::string& s);

    void flush();

    void forced_flush();

 


    void write_string( const std::string &str );

    void write_vertex(const ConstGenVertexPtr& v);

    void write_particle(const ConstGenParticlePtr& p, int second_field);

 
private:
 
    std::ofstream m_file; 
    std::shared_ptr<std::ostream> m_shared_stream  = nullptr;
    std::ostream* m_stream = nullptr; 
 
    int m_precision = 16; 
    char* m_buffer = nullptr;  
    char* m_cursor = nullptr;  
    unsigned long m_buffer_size = 262144; 
    std::string m_float_printf_specifier; 
    std::string m_particle_printf_specifier; 
    std::string m_vertex_short_printf_specifier; 
    std::string m_vertex_long_printf_specifier; 
};
 
 
} // namespace HepMC3
 
#endif