worldspawn/libs/stream/stringstream.h

149 lines
3.4 KiB
C++

/*
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(){
}
explicit 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