X-Git-Url: https://git.saurik.com/wxWidgets.git/blobdiff_plain/67c8c225f469529d86afa1c4e817e96d97c22d18..fd85b0640ea48f66a27e5908930e9f004de698b6:/src/common/stream.cpp diff --git a/src/common/stream.cpp b/src/common/stream.cpp index 38499f5793..9763842c4a 100644 --- a/src/common/stream.cpp +++ b/src/common/stream.cpp @@ -17,8 +17,9 @@ // ---------------------------------------------------------------------------- // headers // ---------------------------------------------------------------------------- + #ifdef __GNUG__ -#pragma implementation "stream.h" + #pragma implementation "stream.h" #endif // For compilers that support precompilation, includes "wx.h". @@ -80,6 +81,8 @@ void wxStreamBuffer::Init() wxStreamBuffer::wxStreamBuffer(wxStreamBase& stream, BufMode mode) { + Init(); + m_stream = &stream; m_mode = mode; @@ -89,6 +92,8 @@ wxStreamBuffer::wxStreamBuffer(wxStreamBase& stream, BufMode mode) wxStreamBuffer::wxStreamBuffer(BufMode mode) { + Init(); + m_stream = new wxStreamBase; m_mode = mode; @@ -129,6 +134,16 @@ wxStreamBuffer::~wxStreamBuffer() delete m_stream; } +wxInputStream *wxStreamBuffer::GetInputStream() const +{ + return m_mode == write ? NULL : (wxInputStream *)m_stream; +} + +wxOutputStream *wxStreamBuffer::GetOutputStream() const +{ + return m_mode == read ? NULL : (wxOutputStream *)m_stream; +} + void wxStreamBuffer::SetBufferIO(void *buffer_start, void *buffer_end, bool takeOwnership) @@ -150,7 +165,7 @@ void wxStreamBuffer::SetBufferIO(void *start, m_buffer_size = len; // if we own it, we free it - m_destroybuf = !takeOwnership; + m_destroybuf = takeOwnership; ResetBuffer(); } @@ -185,9 +200,11 @@ void wxStreamBuffer::ResetBuffer() // fill the buffer with as much data as possible (only for read buffers) bool wxStreamBuffer::FillBuffer() { - wxCHECK_MSG( m_stream, FALSE, _T("should have a stream in wxStreamBuffer") ); + wxInputStream *inStream = GetInputStream(); - size_t count = m_stream->OnSysRead(m_buffer_start, m_buffer_size); + wxCHECK_MSG( inStream, FALSE, _T("should have a stream in wxStreamBuffer") ); + + size_t count = inStream->OnSysRead(m_buffer_start, m_buffer_size); if ( !count ) return FALSE; @@ -206,10 +223,12 @@ bool wxStreamBuffer::FlushBuffer() if ( m_buffer_pos == m_buffer_start ) return FALSE; - wxCHECK_MSG( m_stream, FALSE, _T("should have a stream in wxStreamBuffer") ); + wxOutputStream *outStream = GetOutputStream(); + + wxCHECK_MSG( outStream, FALSE, _T("should have a stream in wxStreamBuffer") ); size_t current = m_buffer_pos - m_buffer_start; - size_t count = m_stream->OnSysWrite(m_buffer_start, current); + size_t count = outStream->OnSysWrite(m_buffer_start, current); if ( count != current ) return FALSE; @@ -281,12 +300,14 @@ void wxStreamBuffer::PutToBuffer(const void *buffer, size_t size) void wxStreamBuffer::PutChar(char c) { - wxCHECK_RET( m_stream, _T("should have a stream in wxStreamBuffer") ); + wxOutputStream *outStream = GetOutputStream(); + + wxCHECK_RET( outStream, _T("should have a stream in wxStreamBuffer") ); // if we don't have buffer at all, just forward this call to the stream, if ( !HasBuffer() ) { - m_stream->OnSysWrite(&c, 1); + outStream->OnSysWrite(&c, 1); } else { @@ -324,12 +345,14 @@ char wxStreamBuffer::Peek() char wxStreamBuffer::GetChar() { - wxCHECK_MSG( m_stream, 0, _T("should have a stream in wxStreamBuffer") ); + wxInputStream *inStream = GetInputStream(); + + wxCHECK_MSG( inStream, 0, _T("should have a stream in wxStreamBuffer") ); char c; if ( !HasBuffer() ) { - m_stream->OnSysRead(&c, 1); + inStream->OnSysRead(&c, 1); } else { @@ -350,16 +373,16 @@ char wxStreamBuffer::GetChar() size_t wxStreamBuffer::Read(void *buffer, size_t size) { - wxCHECK_MSG( m_stream, 0, _T("should have a stream in wxStreamBuffer") ); + wxInputStream *inStream = GetInputStream(); - wxCHECK_MSG( m_mode != write, 0, _T("can't read from this buffer") ); + wxCHECK_MSG( inStream, 0, _T("should have a stream in wxStreamBuffer") ); // lasterror is reset before all new IO calls m_stream->m_lasterror = wxStream_NOERROR; if ( !HasBuffer() ) { - m_stream->m_lastcount = m_stream->OnSysRead(buffer, size); + m_stream->m_lastcount = inStream->OnSysRead(buffer, size); } else // we have a buffer, use it { @@ -421,8 +444,9 @@ size_t wxStreamBuffer::Read(wxStreamBuffer *dbuf) size_t wxStreamBuffer::Write(const void *buffer, size_t size) { - wxCHECK_MSG( m_stream, 0, _T("should have a stream in wxStreamBuffer") ); - wxCHECK_MSG( m_mode != read, 0, _T("can't write to this buffer") ); + wxOutputStream *outStream = GetOutputStream(); + + wxCHECK_MSG( outStream, 0, _T("should have a stream in wxStreamBuffer") ); // lasterror is reset before all new IO calls m_stream->m_lasterror = wxStream_NOERROR; @@ -430,7 +454,7 @@ size_t wxStreamBuffer::Write(const void *buffer, size_t size) if ( !HasBuffer() && m_fixed ) { // no buffer, just forward the call to the stream - m_stream->m_lastcount = m_stream->OnSysWrite(buffer, size); + m_stream->m_lastcount = outStream->OnSysWrite(buffer, size); } else // we [may] have a buffer, use it { @@ -607,16 +631,6 @@ wxStreamBase::~wxStreamBase() { } -size_t wxStreamBase::OnSysRead(void *WXUNUSED(buffer), size_t WXUNUSED(size)) -{ - return 0; -} - -size_t wxStreamBase::OnSysWrite(const void *WXUNUSED(buffer), size_t WXUNUSED(bufsize)) -{ - return 0; -} - off_t wxStreamBase::OnSysSeek(off_t WXUNUSED(seek), wxSeekMode WXUNUSED(mode)) { return wxInvalidOffset; @@ -643,13 +657,24 @@ wxInputStream::~wxInputStream() free(m_wback); } +size_t wxInputStream::OnSysRead(void * WXUNUSED(buffer), + size_t WXUNUSED(bufsize)) +{ + return 0; +} + bool wxInputStream::Eof() const { wxInputStream *self = wxConstCast(this, wxInputStream); char c; self->Read(&c, 1); - if ( GetLastError() == wxSTREAM_EOF ) + + // some streams can know that they're at EOF before actually trying to + // read beyond the end of stream (e.g. files) while others have no way of + // knowing it, so to provide the same behaviour in all cases we only + // return TRUE from here if the character really couldn't be read + if ( !self->LastRead() && GetLastError() == wxSTREAM_EOF ) { return TRUE; } @@ -843,6 +868,12 @@ wxOutputStream::~wxOutputStream() { } +size_t wxOutputStream::OnSysWrite(const void * WXUNUSED(buffer), + size_t WXUNUSED(bufsize)) +{ + return 0; +} + void wxOutputStream::PutC(char c) { Write(&c, 1); @@ -945,6 +976,7 @@ off_t wxCountingOutputStream::OnSysTell() const wxFilterInputStream::wxFilterInputStream() { + m_parent_i_stream = NULL; } wxFilterInputStream::wxFilterInputStream(wxInputStream& stream) @@ -962,6 +994,7 @@ wxFilterInputStream::~wxFilterInputStream() wxFilterOutputStream::wxFilterOutputStream() { + m_parent_o_stream = NULL; } wxFilterOutputStream::wxFilterOutputStream(wxOutputStream& stream) @@ -977,17 +1010,27 @@ wxFilterOutputStream::~wxFilterOutputStream() // wxBufferedInputStream // ---------------------------------------------------------------------------- -wxBufferedInputStream::wxBufferedInputStream(wxInputStream& s) +wxBufferedInputStream::wxBufferedInputStream(wxInputStream& s, + wxStreamBuffer *buffer) : wxFilterInputStream(s) { - m_i_streambuf = new wxStreamBuffer(*this, wxStreamBuffer::read); + if ( buffer ) + { + // use the buffer provided by the user + m_i_streambuf = buffer; + } + else // create a default buffer + { + m_i_streambuf = new wxStreamBuffer(*this, wxStreamBuffer::read); - m_i_streambuf->SetBufferIO(1024); + m_i_streambuf->SetBufferIO(1024); + } } wxBufferedInputStream::~wxBufferedInputStream() { - m_parent_i_stream->SeekI(-m_i_streambuf->GetBytesLeft(), wxFromCurrent); + m_parent_i_stream->SeekI(-(off_t)m_i_streambuf->GetBytesLeft(), + wxFromCurrent); delete m_i_streambuf; } @@ -1003,7 +1046,7 @@ wxInputStream& wxBufferedInputStream::Read(void *buf, size_t size) retsize = GetWBack(buf, size); m_lastcount = retsize; - if (retsize == size) + if ( retsize == size ) { m_lasterror = wxStream_NOERROR; return *this; @@ -1041,15 +1084,32 @@ off_t wxBufferedInputStream::OnSysTell() const return m_parent_i_stream->TellI(); } +void wxBufferedInputStream::SetInputStreamBuffer(wxStreamBuffer *buffer) +{ + wxCHECK_RET( buffer, _T("wxBufferedInputStream needs buffer") ); + + delete m_i_streambuf; + m_i_streambuf = buffer; +} + // ---------------------------------------------------------------------------- // wxBufferedOutputStream // ---------------------------------------------------------------------------- -wxBufferedOutputStream::wxBufferedOutputStream(wxOutputStream& s) +wxBufferedOutputStream::wxBufferedOutputStream(wxOutputStream& s, + wxStreamBuffer *buffer) : wxFilterOutputStream(s) { - m_o_streambuf = new wxStreamBuffer(*this, wxStreamBuffer::write); - m_o_streambuf->SetBufferIO(1024); + if ( buffer ) + { + m_o_streambuf = buffer; + } + else // create a default one + { + m_o_streambuf = new wxStreamBuffer(*this, wxStreamBuffer::write); + + m_o_streambuf->SetBufferIO(1024); + } } wxBufferedOutputStream::~wxBufferedOutputStream() @@ -1102,6 +1162,14 @@ size_t wxBufferedOutputStream::GetSize() const return m_parent_o_stream->GetSize() + m_o_streambuf->GetIntPosition(); } +void wxBufferedOutputStream::SetOutputStreamBuffer(wxStreamBuffer *buffer) +{ + wxCHECK_RET( buffer, _T("wxBufferedOutputStream needs buffer") ); + + delete m_o_streambuf; + m_o_streambuf = buffer; +} + // ---------------------------------------------------------------------------- // Some IOManip function // ----------------------------------------------------------------------------