X-Git-Url: https://git.saurik.com/wxWidgets.git/blobdiff_plain/f6bcfd974ef26faf6f91a62cac09827e09463fd1..54946cc95b8493fb092b6a508bf22a1cd6fa4693:/include/wx/buffer.h?ds=sidebyside diff --git a/include/wx/buffer.h b/include/wx/buffer.h index 6556f39cd6..dac2adf1bd 100644 --- a/include/wx/buffer.h +++ b/include/wx/buffer.h @@ -71,7 +71,8 @@ public: wxASSERT_MSG( wcs, wxT("NULL string in wxWCharBuffer") ); if (wcs) { -#if (defined(__BORLANDC__) && (__BORLANDC__ > 0x530)) +#if ( defined(__BORLANDC__) && (__BORLANDC__ > 0x530) ) \ + || ( defined(__MWERKS__) && defined(__WXMSW__) ) size_t siz = (std::wcslen(wcs)+1)*sizeof(wchar_t); #else size_t siz = (::wcslen(wcs)+1)*sizeof(wchar_t); @@ -125,6 +126,119 @@ private: #define wxWX2WCbuf wxWCharBuffer #endif // Unicode/ANSI + + +// ---------------------------------------------------------------------------- +// A class for holding growable data buffers (not necessarily strings) +// ---------------------------------------------------------------------------- + +class wxMemoryBuffer +{ +public: + enum { BLOCK_SIZE = 1024 }; + wxMemoryBuffer(size_t size=wxMemoryBuffer::BLOCK_SIZE) + { + wxASSERT(size > 0); + m_data = malloc(size); + wxASSERT(m_data != NULL); + m_size = size; + m_len = 0; + } + + ~wxMemoryBuffer() { free(m_data); } + + // Accessors + void* GetData() const { return m_data; } + size_t GetBufSize() const { return m_size; } + size_t GetDataLen() const { return m_len; } + + void SetBufSize(size_t size) { ResizeIfNeeded(size); } + void SetDataLen(size_t len) + { + wxASSERT(len <= m_size); + m_len = len; + } + + // Ensure the buffer is big enough and return a pointer to it + void* GetWriteBuf(size_t sizeNeeded) + { + ResizeIfNeeded(sizeNeeded); + return m_data; + } + // Update the length after the write + void UngetWriteBuf(size_t sizeUsed) { SetDataLen(sizeUsed); } + + // Like the above, but appends to the buffer + void* GetAppendBuf(size_t sizeNeeded) + { + ResizeIfNeeded(m_len + sizeNeeded); + return (char*)m_data + m_len; + } + void UngetAppendBuf(size_t sizeUsed) { SetDataLen(m_len + sizeUsed); } + + // Other ways to append to the buffer + void AppendByte(char data) { + ResizeIfNeeded(m_len + 1); + *(((char*)m_data)+m_len) = data; + m_len += 1; + } + void AppendData(void* data, size_t len) + { + memcpy(GetAppendBuf(len), data, len); + UngetAppendBuf(len); + } + + operator const char *() const { return (const char*)m_data; } + + + // Copy and assignment + wxMemoryBuffer(const wxMemoryBuffer& src) + { + m_data = src.m_data; + m_size = src.m_size; + m_len = src.m_len; + + // no reference count yet... + ((wxMemoryBuffer*)&src)->m_data = NULL; + ((wxMemoryBuffer*)&src)->m_size = 0; + ((wxMemoryBuffer*)&src)->m_len = 0; + } + + wxMemoryBuffer& operator=(const wxMemoryBuffer& src) + { + m_data = src.m_data; + m_size = src.m_size; + m_len = src.m_len; + + // no reference count yet... + ((wxMemoryBuffer*)&src)->m_data = NULL; + ((wxMemoryBuffer*)&src)->m_size = 0; + ((wxMemoryBuffer*)&src)->m_len = 0; + + return *this; + } + + +protected: + + void ResizeIfNeeded(size_t newSize) + { + if (newSize > m_size) + { + m_data = realloc(m_data, newSize + wxMemoryBuffer::BLOCK_SIZE); + wxASSERT(m_data != NULL); + m_size = newSize + wxMemoryBuffer::BLOCK_SIZE; + } + } + +private: + void* m_data; + size_t m_size; + size_t m_len; +}; + + + // ---------------------------------------------------------------------------- // template class for any kind of data // ----------------------------------------------------------------------------