X-Git-Url: https://git.saurik.com/wxWidgets.git/blobdiff_plain/32fc4afbb253f55d4b35b7ee7a2ca9a7eea49fd9..48a1108eb87970373e1b1d5debeeaedd8d1916ea:/src/common/mstream.cpp?ds=sidebyside diff --git a/src/common/mstream.cpp b/src/common/mstream.cpp index 3a1264e3fb..f778677728 100644 --- a/src/common/mstream.cpp +++ b/src/common/mstream.cpp @@ -1,133 +1,174 @@ ///////////////////////////////////////////////////////////////////////////// -// Name: mstream.cpp +// Name: src/common/mstream.cpp // Purpose: "Memory stream" classes // Author: Guilhem Lavaux -// Modified by: +// Modified by: VZ (23.11.00): general code review // Created: 04/01/98 // RCS-ID: $Id$ // Copyright: (c) Guilhem Lavaux -// Licence: wxWindows license +// Licence: wxWindows licence ///////////////////////////////////////////////////////////////////////////// -#ifdef __GNUG__ -#pragma implementation "mstream.h" +// ============================================================================ +// declarations +// ============================================================================ + +// ---------------------------------------------------------------------------- +// headers +// ---------------------------------------------------------------------------- + +// For compilers that support precompilation, includes "wx.h". +#include "wx/wxprec.h" + +#ifdef __BORLANDC__ + #pragma hdrstop #endif +#if wxUSE_STREAMS + +#include "wx/mstream.h" + +#ifndef WX_PRECOMP + #include "wx/stream.h" +#endif //WX_PRECOMP + #include -#include -#include -wxMemoryStreamBase::wxMemoryStreamBase(char *data, size_t length, int iolimit) +// ============================================================================ +// implementation +// ============================================================================ + +// ---------------------------------------------------------------------------- +// wxMemoryInputStream +// ---------------------------------------------------------------------------- + +wxMemoryInputStream::wxMemoryInputStream(const void *data, size_t len) { - m_buffer = data; - m_iolimit = iolimit; - m_persistent = FALSE; - m_length = length; - m_position_i = m_position_o = 0; + m_i_streambuf = new wxStreamBuffer(wxStreamBuffer::read); + m_i_streambuf->SetBufferIO((void *)data, len); // const_cast + m_i_streambuf->SetIntPosition(0); // seek to start pos + m_i_streambuf->Fixed(true); + + m_length = len; } -wxMemoryStreamBase::~wxMemoryStreamBase() +wxMemoryInputStream::wxMemoryInputStream(const wxMemoryOutputStream& stream) { - free(m_buffer); + const wxFileOffset lenFile = stream.GetLength(); + if ( lenFile == wxInvalidOffset ) + { + m_i_streambuf = NULL; + m_lasterror = wxSTREAM_EOF; + return; + } + + const size_t len = wx_truncate_cast(size_t, lenFile); + wxASSERT_MSG( len == lenFile + size_t(0), _T("huge files not supported") ); + + m_i_streambuf = new wxStreamBuffer(wxStreamBuffer::read); + m_i_streambuf->SetBufferIO(len); // create buffer + stream.CopyTo(m_i_streambuf->GetBufferStart(), len); + m_i_streambuf->SetIntPosition(0); // seek to start pos + m_i_streambuf->Fixed(true); + m_length = len; } -wxInputStream& wxMemoryStreamBase::Read(void *buffer, size_t size) +wxMemoryInputStream::~wxMemoryInputStream() { - if (m_iolimit == 2) { - m_eof = TRUE; - return *this; - } - if (m_position_i+size > m_length) - size = m_length-m_position_i; - - memcpy((void *)((unsigned long)buffer+m_position_i), m_buffer, size); - m_position_i += size; - m_lastread = size; - - return *this; + delete m_i_streambuf; } -size_t wxMemoryStreamBase::SeekI(int pos, wxWhenceType whence) +char wxMemoryInputStream::Peek() { - if (m_iolimit == 2) - return 0; + char *buf = (char *)m_i_streambuf->GetBufferStart(); + size_t pos = m_i_streambuf->GetIntPosition(); + if ( pos == m_length ) + { + m_lasterror = wxSTREAM_READ_ERROR; - switch (whence) { - case wxBeginPosition: - if ((size_t)pos > m_length) - return m_position_i; - return (m_position_i = pos); + return 0; + } + + return buf[pos]; +} - case wxCurrentPosition: - if ((size_t)(m_position_i+pos) > m_length) - return m_position_i; +size_t wxMemoryInputStream::OnSysRead(void *buffer, size_t nbytes) +{ + size_t pos = m_i_streambuf->GetIntPosition(); + if ( pos == m_length ) + { + m_lasterror = wxSTREAM_EOF; - return (m_position_i += pos); + return 0; + } - case wxEndPosition: - if ((size_t)(m_length-pos) > m_length) - return m_position_i; + m_i_streambuf->Read(buffer, nbytes); + m_lasterror = wxSTREAM_NO_ERROR; - return (m_position_i = m_length-pos); - } + return m_i_streambuf->GetIntPosition() - pos; +} - return m_position_i; +wxFileOffset wxMemoryInputStream::OnSysSeek(wxFileOffset pos, wxSeekMode mode) +{ + return m_i_streambuf->Seek(pos, mode); } -wxOutputStream& wxMemoryStreamBase::Write(const void *buffer, size_t size) +wxFileOffset wxMemoryInputStream::OnSysTell() const { - if (m_iolimit == 1) { - m_bad = TRUE; - return *this; - } - - if (m_position_o+size > m_length) - if (!ChangeBufferSize(m_position_o+size)) { - m_bad = TRUE; - return *this; - } + return m_i_streambuf->Tell(); +} - memcpy(m_buffer+m_position_o, buffer, size); - m_position_o += size; - m_lastwrite = size; +// ---------------------------------------------------------------------------- +// wxMemoryOutputStream +// ---------------------------------------------------------------------------- - return *this; +wxMemoryOutputStream::wxMemoryOutputStream(void *data, size_t len) +{ + m_o_streambuf = new wxStreamBuffer(wxStreamBuffer::write); + if ( data ) + m_o_streambuf->SetBufferIO(data, len); + m_o_streambuf->Fixed(false); + m_o_streambuf->Flushable(false); } -size_t wxMemoryStreamBase::SeekO(int pos, wxWhenceType whence) +wxMemoryOutputStream::~wxMemoryOutputStream() { - if (m_iolimit == 2) - return 0; - - switch (whence) { - case wxBeginPosition: - if ((size_t)pos > m_length) - return m_position_o; - return (m_position_o = pos); + delete m_o_streambuf; +} - case wxCurrentPosition: - if ((size_t)(m_position_o+pos) > m_length) - return m_position_o; +size_t wxMemoryOutputStream::OnSysWrite(const void *buffer, size_t nbytes) +{ + size_t oldpos = m_o_streambuf->GetIntPosition(); + m_o_streambuf->Write(buffer, nbytes); + size_t newpos = m_o_streambuf->GetIntPosition(); - return (m_position_o += pos); + // FIXME can someone please explain what this does? (VZ) + if ( !newpos ) + newpos = m_o_streambuf->GetBufferSize(); - case wxEndPosition: - if ((size_t)(m_length-pos) > m_length) - return m_position_o; + return newpos - oldpos; +} - return (m_position_o = m_length-pos); - } +wxFileOffset wxMemoryOutputStream::OnSysSeek(wxFileOffset pos, wxSeekMode mode) +{ + return m_o_streambuf->Seek(pos, mode); +} - return m_position_o; +wxFileOffset wxMemoryOutputStream::OnSysTell() const +{ + return m_o_streambuf->Tell(); } -bool wxMemoryStreamBase::ChangeBufferSize(size_t new_size) +size_t wxMemoryOutputStream::CopyTo(void *buffer, size_t len) const { - m_length = new_size; - if (!m_buffer) - m_buffer = (char *)malloc(m_length); - else - m_buffer = (char *)realloc(m_buffer, m_length); + wxCHECK_MSG( buffer, 0, _T("must have buffer to CopyTo") ); + + if ( len > GetSize() ) + len = GetSize(); - return (m_buffer != NULL); + memcpy(buffer, m_o_streambuf->GetBufferStart(), len); + + return len; } + +#endif // wxUSE_STREAMS