streams.hh

// -*- tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*-
// vi: set et ts=4 sw=2 sts=2:
 
#ifndef DUNE_GRID_IO_FILE_VTK_STREAMS_HH
#define DUNE_GRID_IO_FILE_VTK_STREAMS_HH
 
#include <ostream>
 
#include <dune/grid/io/file/vtk/b64enc.hh>
 
namespace Dune {
 
  class Base64Stream {
    std::ostream& s;
    b64chunk chunk;
    char obuf[4];
 
  public:
 
    Base64Stream(std::ostream& s_)
      : s(s_)
    {
      // reset chunk
      chunk.reset();
    }
 
 
    template <class X>
    void write(X & data)
    {
      char* p = reinterpret_cast<char*>(&data);
      for (size_t len = sizeof(X); len > 0; len--,p++)
      {
        chunk.put(*p);
        if (chunk.size == 3)
        {
          chunk.write(obuf);
          s.write(obuf,4);
        }
      }
    }
 
 
    void flush()
    {
      if (chunk.size > 0)
      {
        chunk.write(obuf);
        s.write(obuf,4);
      }
    }
 
 
    ~Base64Stream() {
      flush();
    }
  };
 
  class RawStream
  {
  public:
    inline RawStream (std::ostream& theStream)
      : s(theStream)
    {}
 
    template<class T>
    void write (T data)
    {
      char* p = reinterpret_cast<char*>(&data);
      s.write(p,sizeof(T));
    }
  private:
    std::ostream& s;
  };
 
} // namespace Dune
 
#endif // DUNE_GRID_IO_FILE_VTK_STREAMS_HH