StringWriter.cpp

Go to the documentation of this file.

#include <string.h>     // memcpy
#include <dcl/StringWriter.h>
#include <dcl/size_t.h>
 
#if __DCL_DEBUG
#undef __THIS_FILE__
static const char_t __THIS_FILE__[] = __T("dcl/StringWriter.cpp");
#endif
 
__DCL_BEGIN_NAMESPACE
 
IMPLEMENT_CLASSINFO(StringWriter, Writer)
 
String StringWriter::toString() const
{
    String r;
    if (__buf != NULL) {
        // STRING::assign 에서 __buf의 참조카운터가 증가한다.
        r.assign(__buf);
    }
    return r;
}
 
StringWriter::StringWriter(size_t _capacity)
{
    __DCL_ASSERT_PARAM(_capacity > 0);
 
    __buf = NULL;
    __capacity = _capacity;
}
 
StringWriter::~StringWriter()
{
    if (__buf) {
        try {
            close();
        }
        catch (Exception* cause) {
            __DCL_TRACE1(__T("%ls\n"), cause->toString().data());
            cause->destroy();
        }
    }
}
 
void StringWriter::close()
            __DCL_THROWS1(IOException*)
{
    if (__buf != NULL) {
        __buf->release();
        __buf = NULL;
    }
}
 
Writer& StringWriter::write(const wchar_t* _buf, size_t _n)
            __DCL_THROWS1(IOException*)
{
    __DCL_ASSERT_PARAM(_buf != NULL);
 
    if (_n) {
        if (__buf == NULL)
            __buf = CharBuffer::create_e(__MAX(_n, __capacity));
        else if (__buf->__refCount > 1) {
            CharBuffer* buf = CharBuffer::create_e(__MAX(_n, __capacity));
            memcpy(buf->data(), __buf->data(), (__buf->__dataLength + 1) * sizeof(wchar_t));
            buf->__dataLength = __buf->__dataLength;
            __DCL_ASSERT(buf->data()[buf->__dataLength] == L'\0');
            __buf->release();
            __buf = buf;
        }
        else {
            __DCL_ASSERT(__buf->__refCount == 1);
        }
 
        CharBuffer::write(__buf, _buf, _n);
    }
    return *this;
}
 
int StringWriter::vprintf(const wchar_t* _format, va_list _arglist)
        __DCL_THROWS1(IOException*)
{
    __DCL_ASSERT_PARAM(_format != NULL);
 
    if (__buf == NULL)
        __buf = CharBuffer::create_e(__capacity);
    else if (__buf->__refCount > 1) {
        CharBuffer* buf = CharBuffer::create_e(__capacity);
        memcpy(buf->data(), __buf->data(), (__buf->__dataLength + 1) * sizeof(wchar_t));
        buf->__dataLength = __buf->__dataLength;
        __DCL_ASSERT(buf->data()[buf->__dataLength] == L'\0');
        __buf->release();
        __buf = buf;
    }
    else {
        __DCL_ASSERT(__buf->__refCount == 1);
    }
 
    return CharBuffer::vformat(__buf, _format, _arglist);
}
 
size_t StringWriter::writeTo(Writer& _writer) const
            __DCL_THROWS1(IOException*)
{
    if (__buf != NULL && __buf->__dataLength) {
        _writer.write(__buf->data(), __buf->__dataLength);
        return __buf->__dataLength;
    }
    return 0;
}
 
size_t StringWriter::flushTo(Writer& _writer)
            __DCL_THROWS1(IOException*)
{
    size_t n = writeTo(_writer);
    if (n > 0) {
        reset();
    }
    return n;
}
 
void StringWriter::reset()
{
    if (__buf != NULL) {
        if (__buf->__refCount == 1) {
            __buf->__dataLength = 0;
            __buf->data()[__buf->__dataLength] =  __T('\0');
        }
        else {
            __DCL_ASSERT(__buf->__refCount > 1);
            __buf->release();
            __buf = NULL;
        }
    }
}
 
__DCL_END_NAMESPACE