stream_writer.h

Go to the documentation of this file.

#ifndef ODVA_ETHERNETIP_SERIALIZATION_STREAM_WRITER_H
#define ODVA_ETHERNETIP_SERIALIZATION_STREAM_WRITER_H

#include <iostream>
#include <boost/shared_ptr.hpp>
#include <boost/asio.hpp>
#include "odva_ethernetip/serialization/writer.h"

using boost::shared_ptr;
using boost::asio::const_buffer;
using boost::asio::buffer_cast;
using boost::asio::buffer_size;
using std::ostream;

namespace eip {
namespace serialization {

class StreamWriter : public Writer
{
public:

  StreamWriter(shared_ptr<ostream> osp) : osp_(osp) { }

  template <typename T>
  void write(T& v)
  {
    osp_->write((char*)&v, sizeof(v));
  }

  void writeBytes(const void* p, size_t n)
  {
    osp_->write((char*)p, n);
  }

  void writeBuffer(const_buffer b)
  {
    osp_->write(buffer_cast<const char*>(b), buffer_size(b));
  }

  size_t getByteCount()
  {
    return osp_->tellp();
  }

private:
  shared_ptr<ostream> osp_;
};

} // namespace serialization
} // namespace eip
#endif  // ODVA_ETHERNETIP_SERIALIZATION_STREAM_WRITER_H