BytesOutputStream.cpp

Go to the documentation of this file.

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