/*
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>


/// \brief A wrapper around a STL vector of char.
/// Maintains a null-terminated array of char.
/// Provides a limited STL-style interface to push and pop characters at the end of the string.
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');
  }
};

/// \brief A TextOutputStream which writes to a StringBuffer.
/// Similar to std::stringstream.
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)
  {
  }
  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