StrIOStream.h

Go to the documentation of this file.

#ifndef STRIOSTREAM_H_
#define STRIOSTREAM_H_

#include "IOStream.h"
#include "utils/Str.h"

class StrIOStream : public IOStream
{
public:

    StrIOStream( Str& str )
        : str_( str ),
          readPos_( 0 )
    {}

    virtual ~StrIOStream()
    {}

    virtual InStream& read( char* buffer, size_t num )
    {
        if( readPos_ >= str_.length() )
        {
            num = 0;
        }
        if( str_.length() - readPos_ < num )
        {
            num = str_.length() - readPos_;
        }
        strncpy( buffer, str_.cStr() + readPos_, num );
        //buffer[num] = 0;  // Can cause buffer overflow!
        readPos_ += num;
        setBytesRead( num );
        return *this;
    }

    const char* getName()
    {
        return "String";
    }

    void clear()
    {
        rewind();
        str_ = Str::EMPTY_STR;
    }

    void rewind()
    {
        readPos_ = 0;
    }

    virtual bool isReadable()
    {
        return !eof();
    }

    virtual bool eof()
    {
        return str_.length() <= readPos_;
    }

    virtual OutStream& write( const char* buffer, size_t num )
    {
        lastWritten_ = str_.size();
        str_ += Str( buffer, num );
        lastWritten_ = str_.length() - lastWritten_;
        return *this;
    }

    bool isWritable()
    {
        return true;
    }

protected:
    Str& str_;
    size_t readPos_;

private:
    // Note that the copy constructor below is private and not given a body.
    // Any attempt to call it will return a compiler error.
    StrIOStream( const StrIOStream& old ); // disallow copy constructor

};

#endif /*STRIOSTREAM_H_*/