SIOWriter.h

Go to the documentation of this file.

#ifndef SIO_SIOWRITER_H
#define SIO_SIOWRITER_H 1

// -- std headers
#include <string>
#include <vector>

// -- lcio headers
#include "IO/LCWriter.h"
#include "EVENT/LCEvent.h"
#include "EVENT/LCRunHeader.h"
#include "SIO/LCIORecords.h"
#include "SIO/LCIORandomAccessMgr.h"

// -- sio headers
#include "sio/definitions.h"
#include "sio/buffer.h"
#include <sio/compression/zlib.h>

namespace SIO {

  class SIOWriter : public IO::LCWriter {
  public:

    SIOWriter() = default ;
    SIOWriter(const SIOWriter&) = delete ;
    SIOWriter& operator=(const SIOWriter&) = delete ;
    ~SIOWriter() = default ;

    virtual void open(const std::string & filename)  ;

    virtual void open(const std::string & filename, int writeMode)  ;

    virtual void setCompressionLevel(int level) ;


    virtual void writeRunHeader(const EVENT::LCRunHeader * hdr)  ;

    virtual void writeEvent(const EVENT::LCEvent * evt)  ;

    virtual void close()  ;

    virtual void flush()  ;

  private:
    void getSIOFileName(const std::string& filename, std::string& sioFilename)  ;

  private:
    sio::ofstream                        _stream {} ;
    sio::buffer                          _rawBuffer {2*sio::mbyte} ;
    sio::buffer                          _compBuffer {sio::mbyte} ;
    sio::zlib_compression                _compressor {} ;
    SIOHandlerMgr                        _eventHandlerMgr {} ;
    LCIORandomAccessMgr                  _raMgr {} ;
  }; // class

} // namespace.

#endif /* ifndef SIO_SIOWRITER_H */