JBufferedIO.hh

Go to the documentation of this file.

#ifndef __JIO__JBUFFEREDIO__
#define __JIO__JBUFFEREDIO__
 
#include <algorithm>
#include <cstring>
 
#include "JLang/JSinglePointer.hh"
#include "JIO/JSerialisable.hh"
 
 
/**
 * \author mdejong
 */
 
namespace JIO {}
namespace JPP { using namespace JIO; }
 
namespace JIO {
 
 
  /**
   * Buffered binary input.
   * This class implements the JReader interface.
   */
  class JBufferedReader : 
    public JReader 
  {
  public:
    /**
     * Constructor.
     * Note that this object owns the reader pointed to.
     *
     * \param  __in    pointer to reader
     * \param  __size  size of internal buffer
     */
    JBufferedReader(JReader* __in, const int __size = 1048576) :
      in(__in)
    {
      size   = std::max(__size, 1024);
      buffer = new char[size];
 
      pos = 0;
      ls  = 0;
      eof = true;
    }
 
 
    /**
     * Destructor.
     */
    ~JBufferedReader()
    {
      delete [] buffer;
    }
 
 
    /**
     * Status of reader.
     *
     * \return         status of this reader
     */
    virtual bool getStatus() const 
    { 
      return in->getStatus() || !eof;
    }
 
 
    /**
     * Clear status of reader.
     */
    virtual void clear()
    {
      in->clear();
    }
 
 
    /**
     * Read byte array.
     *
     * \param  zbuf    pointer to byte array
     * \param  n       number of bytes
     * \return         number of bytes
     */
    virtual int read(char* zbuf, int n)
    { 
      for (int i = 0; i != n; ) {
 
        int m = n - i;
 
        if (m > ls - pos) {
          
          memmove(buffer, buffer + pos, ls - pos);
          
          ls  -= pos;
          pos  = 0;
          ls  += in->read(buffer + ls, size - ls); 
 
          if (m > ls) {
 
            if (ls == 0) {
 
              eof = true;
 
              return n - i;
            }
 
            m = ls;
          }
        }
 
        memcpy(zbuf + i, buffer + pos, m);
        
        i   += m;
        pos += m;
      }
 
      eof = false;
 
      return n;
    }
 
 
  protected:
    JLANG::JSinglePointer<JReader> in;
 
    char* buffer;  //!< internal buffer
    int   size;    //!< size of internal buffer
    int   pos;     //!< pointer to begin of available data
    int   ls;      //!< pointer to end   of available data
    bool  eof;     //!< end of file
  };
 
 
  /**
   * Buffered binary output.
   * This class implements the JWriter interface.
   */
  class JBufferedWriter : 
    public JWriter 
  {
  public:
    /**
     * Constructor.
     * Note that this object owns the writer pointed to.
     *
     * \param  __out   pointer to writer
     * \param  __size  size of internal buffer
     */
    JBufferedWriter(JWriter* __out, const int __size = 1048576) :
      out(__out)
    {
      size   = std::max(__size, 1024);
      buffer = new char[size];
 
      pos = 0;
    }
 
 
    /**
     * Destructor.
     */
    ~JBufferedWriter()
    {
      flush();
 
      delete [] buffer;
    }
 
 
    /**
     * Status of writer.
     *
     * \return         status of this writer
     */
    virtual bool getStatus() const 
    { 
      return (bool) *out;
    }
 
 
    /**
     * Write byte array.
     *
     * \param  zbuf    pointer to byte array
     * \param  n       number of bytes
     * \return         number of bytes
     */
    virtual int write(const char* zbuf, int n)
    { 
      for (int i = 0; i != n; ) {
 
        int m = n - i;
 
        if (m > size - pos) {
 
          flush();
        
          if (m > size - pos) {
            m = size - pos;
          }
        }
 
        memcpy(buffer + pos, zbuf + i, m);
        
        i   += m;
        pos += m;
      }
 
      return n;
    }
 
    
    /**
     * Write internal data to output.
     */
    void flush()
    {
      pos -= out->write(buffer, pos); 
    }
 
 
  protected:
    JLANG::JSinglePointer<JWriter> out;
 
    char* buffer;  //!< internal buffer
    int   size;    //!< size of internal buffer
    int   pos;     //!< pointer to end of buffered data
  };
}
 
#endif