stringstream.h

Go to the documentation of this file.

/*
 Copyright (C) 2001-2006, William Joseph.
 All Rights Reserved.

 This file is part of GtkRadiant.

 GtkRadiant is free software; you can redistribute it and/or modify
 it under the terms of the GNU General Public License as published by
 the Free Software Foundation; either version 2 of the License, or
 (at your option) any later version.

 GtkRadiant is distributed in the hope that it will be useful,
 but WITHOUT ANY WARRANTY; without even the implied warranty of
 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 GNU General Public License for more details.

 You should have received a copy of the GNU General Public License
 along with GtkRadiant; if not, write to the Free Software
 Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
 */

#if !defined(INCLUDED_STREAM_STRINGSTREAM_H)
#define INCLUDED_STREAM_STRINGSTREAM_H

#include "itextstream.h"
#include "string/string.h"
#include <vector>

class StringBuffer
{
        std::vector<char> m_string;
    public:
        StringBuffer ()
        {
            m_string.push_back('\0');
        }
        explicit StringBuffer (std::size_t capacity)
        {
            m_string.reserve(capacity);
            m_string.push_back('\0');
        }
        explicit StringBuffer (const char* string) :
            m_string(string, string + string_length(string) + 1)
        {
        }

        typedef std::vector<char>::iterator iterator;
        typedef std::vector<char>::const_iterator const_iterator;

        iterator begin ()
        {
            return m_string.begin();
        }
        const_iterator begin () const
        {
            return m_string.begin();
        }
        iterator end ()
        {
            return m_string.end() - 1;
        }
        const_iterator end () const
        {
            return m_string.end() - 1;
        }

        void push_back (char c)
        {
            m_string.insert(end(), c);
        }
        void pop_back ()
        {
            m_string.erase(end() - 1);
        }
        void push_range (const char* first, const char* last)
        {
            m_string.insert(end(), first, last);
        }
        void push_string (const char* string)
        {
            push_range(string, string + string_length(string));
        }
        char* c_str ()
        {
            return &(*m_string.begin());
        }
        const char* c_str () const
        {
            return &(*m_string.begin());
        }

        char& back ()
        {
            return *(end() - 1);
        }
        const char& back () const
        {
            return *(end() - 1);
        }
        bool empty () const
        {
            return m_string.size() == 1;
        }
        void clear ()
        {
            m_string.clear();
            m_string.push_back('\0');
        }
};

class StringOutputStream: public TextOutputStream
{
        StringBuffer m_string;
    public:
        typedef StringBuffer::iterator iterator;
        typedef StringBuffer::const_iterator const_iterator;

        StringOutputStream ()
        {
        }
        StringOutputStream (std::size_t capacity) :
            m_string(capacity)
        {
        }
        ~StringOutputStream ()
        {
        }

        std::size_t write (const char* buffer, std::size_t length)
        {
            m_string.push_range(buffer, buffer + length);
            return length;
        }

        iterator begin ()
        {
            return m_string.begin();
        }
        const_iterator begin () const
        {
            return m_string.begin();
        }
        iterator end ()
        {
            return m_string.end();
        }
        const_iterator end () const
        {
            return m_string.end();
        }

        bool empty () const
        {
            return m_string.empty();
        }
        char* c_str ()
        {
            return m_string.c_str();
        }
        const char* c_str () const
        {
            return m_string.c_str();
        }
        void clear ()
        {
            m_string.clear();
        }
};

template<typename T>
inline StringOutputStream& operator<< (StringOutputStream& ostream, const T& t)
{
    return ostream_write(ostream, t);
}

#endif