#ifndef StringBuilder_h
#define StringBuilder_h
#include <wtf/Vector.h>
#include <wtf/text/WTFString.h>
namespace WTF {
class StringBuilder {
public:
StringBuilder()
: m_length(0)
{
}
void append(const UChar*, unsigned);
void append(const char*, unsigned);
void append(const String& string)
{
if (!m_length && !m_buffer) {
m_string = string;
m_length = string.length();
return;
}
append(string.characters(), string.length());
}
void append(const char* characters)
{
if (characters)
append(characters, strlen(characters));
}
void append(UChar c)
{
if (m_buffer && m_length < m_buffer->length() && m_string.isNull())
m_bufferCharacters[m_length++] = c;
else
append(&c, 1);
}
void append(char c)
{
if (m_buffer && m_length < m_buffer->length() && m_string.isNull())
m_bufferCharacters[m_length++] = (unsigned char)c;
else
append(&c, 1);
}
String toString()
{
if (m_string.isNull()) {
shrinkToFit();
reifyString();
}
return m_string;
}
String toStringPreserveCapacity()
{
if (m_string.isNull())
reifyString();
return m_string;
}
unsigned length() const
{
return m_length;
}
bool isEmpty() const { return !length(); }
void reserveCapacity(unsigned newCapacity);
void resize(unsigned newSize);
void shrinkToFit();
UChar operator[](unsigned i) const
{
ASSERT(i < m_length);
if (!m_string.isNull())
return m_string[i];
ASSERT(m_buffer);
return m_buffer->characters()[i];
}
void clear()
{
m_length = 0;
m_string = String();
m_buffer = 0;
}
private:
void allocateBuffer(const UChar* currentCharacters, unsigned requiredLength);
UChar* appendUninitialized(unsigned length);
void reifyString();
unsigned m_length;
String m_string;
RefPtr<StringImpl> m_buffer;
UChar* m_bufferCharacters;
};
}
using WTF::StringBuilder;
#endif // StringBuilder_h