IO_GenEvent.h

//--------------------------------------------------------------------------
#ifndef HEPMC_IO_GENEVENT_H
#define HEPMC_IO_GENEVENT_H

// garren@fnal.gov, July 2007
// with input from Gavin Salam, salam@lpthe.jussieu.fr
//
// event input/output in ascii format for machine reading
// This class persists all information found in a GenEvent

#include <fstream>
#include <string>
#include <map>
#include <vector>
#include "HepMC/IO_BaseClass.h"
#include "HepMC/IO_Exception.h"
#include "HepMC/Units.h"

namespace HepMC {

class GenEvent;
class GenVertex;
class GenParticle;
class HeavyIon;
class PdfInfo;


class IO_GenEvent : public IO_BaseClass {
public:
    IO_GenEvent( const std::string& filename="IO_GenEvent.dat", 
              std::ios::openmode mode=std::ios::out );
    IO_GenEvent( std::istream & );
    IO_GenEvent( std::ostream & );
    virtual       ~IO_GenEvent();

    void          write_event( const GenEvent* evt );
    bool          fill_next_event( GenEvent* evt );
    void          write_comment( const std::string comment );

    int           rdstate() const;  
    void          clear();  

    void          print( std::ostream& ostr = std::cout ) const;

    void use_input_units( Units::MomentumUnit, Units::LengthUnit );

    void precision( int );
        
    int           error_type()    const;
    const std::string & error_message() const;

private: // use of copy constructor is not allowed
    IO_GenEvent( const IO_GenEvent& ) : IO_BaseClass() {}

private: // data members
    std::ios::openmode  m_mode;
    std::fstream        m_file;
    std::ostream *      m_ostr;
    std::istream *      m_istr;
    std::ios *          m_iostr;
    bool                m_have_file;
    IO_Exception::ErrorType m_error_type;
    std::string         m_error_message;

};

// Inlines  //

inline int  IO_GenEvent::rdstate() const { 
    int state;
    if( m_istr ) {
        state =  (int)m_istr->rdstate();
    } else {
        state =  (int)m_ostr->rdstate();
    }
    return state; 
}

inline void IO_GenEvent::clear() { 
    if( m_istr ) {
        m_istr->clear();
    } else {
        m_ostr->clear();
    }
}

inline int IO_GenEvent::error_type() const {
    return m_error_type;
}

inline const std::string & IO_GenEvent::error_message() const {
    return m_error_message;
}

} // HepMC

#endif  // HEPMC_IO_GENEVENT_H
//--------------------------------------------------------------------------