1 /////////////////////////////////////////////////////////////////////////////
3 // Purpose: stream classes
4 // Author: Guilhem Lavaux, Guillermo Rodriguez Garcia, Vadim Zeitlin
8 // Copyright: (c) Guilhem Lavaux
9 // Licence: wxWindows licence
10 /////////////////////////////////////////////////////////////////////////////
12 #ifndef _WX_WXSTREAM_H__
13 #define _WX_WXSTREAM_H__
20 #include "wx/object.h"
21 #include "wx/string.h"
22 #include "wx/filefn.h" // for wxFileOffset, wxInvalidOffset and wxSeekMode
24 class WXDLLIMPEXP_FWD_BASE wxStreamBase
;
25 class WXDLLIMPEXP_FWD_BASE wxInputStream
;
26 class WXDLLIMPEXP_FWD_BASE wxOutputStream
;
28 typedef wxInputStream
& (*__wxInputManip
)(wxInputStream
&);
29 typedef wxOutputStream
& (*__wxOutputManip
)(wxOutputStream
&);
31 WXDLLIMPEXP_BASE wxOutputStream
& wxEndL(wxOutputStream
& o_stream
);
33 // ----------------------------------------------------------------------------
35 // ----------------------------------------------------------------------------
39 wxSTREAM_NO_ERROR
= 0, // stream is in good state
40 wxSTREAM_EOF
, // EOF reached in Read() or similar
41 wxSTREAM_WRITE_ERROR
, // generic write error
42 wxSTREAM_READ_ERROR
// generic read error
47 // ============================================================================
48 // base stream classes: wxInputStream and wxOutputStream
49 // ============================================================================
51 // ---------------------------------------------------------------------------
52 // wxStreamBase: common (but non virtual!) base for all stream classes
53 // ---------------------------------------------------------------------------
55 class WXDLLIMPEXP_BASE wxStreamBase
: public wxObject
59 virtual ~wxStreamBase();
62 wxStreamError
GetLastError() const { return m_lasterror
; }
63 virtual bool IsOk() const { return GetLastError() == wxSTREAM_NO_ERROR
; }
64 bool operator!() const { return !IsOk(); }
66 // reset the stream state
67 void Reset(wxStreamError error
= wxSTREAM_NO_ERROR
) { m_lasterror
= error
; }
69 // this doesn't make sense for all streams, always test its return value
70 virtual size_t GetSize() const;
71 virtual wxFileOffset
GetLength() const { return wxInvalidOffset
; }
73 // returns true if the streams supports seeking to arbitrary offsets
74 virtual bool IsSeekable() const { return false; }
77 virtual wxFileOffset
OnSysSeek(wxFileOffset seek
, wxSeekMode mode
);
78 virtual wxFileOffset
OnSysTell() const;
81 wxStreamError m_lasterror
;
83 friend class wxStreamBuffer
;
85 DECLARE_ABSTRACT_CLASS(wxStreamBase
)
86 wxDECLARE_NO_COPY_CLASS(wxStreamBase
);
89 // ----------------------------------------------------------------------------
90 // wxInputStream: base class for the input streams
91 // ----------------------------------------------------------------------------
93 class WXDLLIMPEXP_BASE wxInputStream
: public wxStreamBase
96 // ctor and dtor, nothing exciting
98 virtual ~wxInputStream();
104 // return a character from the stream without removing it, i.e. it will
105 // still be returned by the next call to GetC()
107 // blocks until something appears in the stream if necessary, if nothing
108 // ever does (i.e. EOF) LastRead() will return 0 (and the return value is
109 // undefined), otherwise 1
112 // return one byte from the stream, blocking until it appears if
115 // on success returns a value between 0 - 255, or wxEOF on EOF or error.
118 // read at most the given number of bytes from the stream
120 // there are 2 possible situations here: either there is nothing at all in
121 // the stream right now in which case Read() blocks until something appears
122 // (use CanRead() to avoid this) or there is already some data available in
123 // the stream and then Read() doesn't block but returns just the data it
124 // can read without waiting for more
126 // in any case, if there are not enough bytes in the stream right now,
127 // LastRead() value will be less than size but greater than 0. If it is 0,
128 // it means that EOF has been reached.
129 virtual wxInputStream
& Read(void *buffer
, size_t size
);
131 // Read exactly the given number of bytes, unlike Read(), which may read
132 // less than the requested amount of data without returning an error, this
133 // method either reads all the data or returns false.
134 bool ReadAll(void *buffer
, size_t size
);
136 // copy the entire contents of this stream into streamOut, stopping only
137 // when EOF is reached or an error occurs
138 wxInputStream
& Read(wxOutputStream
& streamOut
);
144 // returns the number of bytes read by the last call to Read(), GetC() or
147 // this should be used to discover whether that call succeeded in reading
148 // all the requested data or not
149 virtual size_t LastRead() const { return wxStreamBase::m_lastcount
; }
151 // returns true if some data is available in the stream right now, so that
152 // calling Read() wouldn't block
153 virtual bool CanRead() const;
155 // is the stream at EOF?
157 // note that this cannot be really implemented for all streams and
158 // CanRead() is more reliable than Eof()
159 virtual bool Eof() const;
165 // put back the specified number of bytes into the stream, they will be
166 // fetched by the next call to the read functions
168 // returns the number of bytes really stuffed back
169 size_t Ungetch(const void *buffer
, size_t size
);
171 // put back the specified character in the stream
173 // returns true if ok, false on error
174 bool Ungetch(char c
);
177 // position functions
178 // ------------------
180 // move the stream pointer to the given position (if the stream supports
183 // returns wxInvalidOffset on error
184 virtual wxFileOffset
SeekI(wxFileOffset pos
, wxSeekMode mode
= wxFromStart
);
186 // return the current position of the stream pointer or wxInvalidOffset
187 virtual wxFileOffset
TellI() const;
190 // stream-like operators
191 // ---------------------
193 wxInputStream
& operator>>(wxOutputStream
& out
) { return Read(out
); }
194 wxInputStream
& operator>>(__wxInputManip func
) { return func(*this); }
197 // do read up to size bytes of data into the provided buffer
199 // this method should return 0 if EOF has been reached or an error occurred
200 // (m_lasterror should be set accordingly as well) or the number of bytes
202 virtual size_t OnSysRead(void *buffer
, size_t size
) = 0;
204 // write-back buffer support
205 // -------------------------
207 // return the pointer to a buffer big enough to hold sizeNeeded bytes
208 char *AllocSpaceWBack(size_t sizeNeeded
);
210 // read up to size data from the write back buffer, return the number of
212 size_t GetWBack(void *buf
, size_t size
);
214 // write back buffer or NULL if none
217 // the size of the buffer
220 // the current position in the buffer
223 friend class wxStreamBuffer
;
225 DECLARE_ABSTRACT_CLASS(wxInputStream
)
226 wxDECLARE_NO_COPY_CLASS(wxInputStream
);
229 // ----------------------------------------------------------------------------
230 // wxOutputStream: base for the output streams
231 // ----------------------------------------------------------------------------
233 class WXDLLIMPEXP_BASE wxOutputStream
: public wxStreamBase
237 virtual ~wxOutputStream();
240 virtual wxOutputStream
& Write(const void *buffer
, size_t size
);
242 // This is ReadAll() equivalent for Write(): it either writes exactly the
243 // given number of bytes or returns false, unlike Write() which can write
244 // less data than requested but still return without error.
245 bool WriteAll(const void *buffer
, size_t size
);
247 wxOutputStream
& Write(wxInputStream
& stream_in
);
249 virtual wxFileOffset
SeekO(wxFileOffset pos
, wxSeekMode mode
= wxFromStart
);
250 virtual wxFileOffset
TellO() const;
252 virtual size_t LastWrite() const { return wxStreamBase::m_lastcount
; }
255 virtual bool Close() { return true; }
257 wxOutputStream
& operator<<(wxInputStream
& out
) { return Write(out
); }
258 wxOutputStream
& operator<<( __wxOutputManip func
) { return func(*this); }
261 // to be implemented in the derived classes (it should have been pure
263 virtual size_t OnSysWrite(const void *buffer
, size_t bufsize
);
265 friend class wxStreamBuffer
;
267 DECLARE_ABSTRACT_CLASS(wxOutputStream
)
268 wxDECLARE_NO_COPY_CLASS(wxOutputStream
);
271 // ============================================================================
272 // helper stream classes
273 // ============================================================================
275 // ---------------------------------------------------------------------------
276 // A stream for measuring streamed output
277 // ---------------------------------------------------------------------------
279 class WXDLLIMPEXP_BASE wxCountingOutputStream
: public wxOutputStream
282 wxCountingOutputStream();
284 virtual wxFileOffset
GetLength() const;
285 bool Ok() const { return IsOk(); }
286 virtual bool IsOk() const { return true; }
289 virtual size_t OnSysWrite(const void *buffer
, size_t size
);
290 virtual wxFileOffset
OnSysSeek(wxFileOffset pos
, wxSeekMode mode
);
291 virtual wxFileOffset
OnSysTell() const;
296 DECLARE_DYNAMIC_CLASS(wxCountingOutputStream
)
297 wxDECLARE_NO_COPY_CLASS(wxCountingOutputStream
);
300 // ---------------------------------------------------------------------------
302 // ---------------------------------------------------------------------------
304 class WXDLLIMPEXP_BASE wxFilterInputStream
: public wxInputStream
307 wxFilterInputStream();
308 wxFilterInputStream(wxInputStream
& stream
);
309 wxFilterInputStream(wxInputStream
*stream
);
310 virtual ~wxFilterInputStream();
312 char Peek() { return m_parent_i_stream
->Peek(); }
314 wxFileOffset
GetLength() const { return m_parent_i_stream
->GetLength(); }
316 wxInputStream
*GetFilterInputStream() const { return m_parent_i_stream
; }
319 wxInputStream
*m_parent_i_stream
;
322 DECLARE_ABSTRACT_CLASS(wxFilterInputStream
)
323 wxDECLARE_NO_COPY_CLASS(wxFilterInputStream
);
326 class WXDLLIMPEXP_BASE wxFilterOutputStream
: public wxOutputStream
329 wxFilterOutputStream();
330 wxFilterOutputStream(wxOutputStream
& stream
);
331 wxFilterOutputStream(wxOutputStream
*stream
);
332 virtual ~wxFilterOutputStream();
334 wxFileOffset
GetLength() const { return m_parent_o_stream
->GetLength(); }
336 wxOutputStream
*GetFilterOutputStream() const { return m_parent_o_stream
; }
341 wxOutputStream
*m_parent_o_stream
;
344 DECLARE_ABSTRACT_CLASS(wxFilterOutputStream
)
345 wxDECLARE_NO_COPY_CLASS(wxFilterOutputStream
);
348 enum wxStreamProtocolType
350 wxSTREAM_PROTOCOL
, // wxFileSystem protocol (should be only one)
351 wxSTREAM_MIMETYPE
, // MIME types the stream handles
352 wxSTREAM_ENCODING
, // The HTTP Content-Encodings the stream handles
353 wxSTREAM_FILEEXT
// File extensions the stream handles
356 void WXDLLIMPEXP_BASE
wxUseFilterClasses();
358 class WXDLLIMPEXP_BASE wxFilterClassFactoryBase
: public wxObject
361 virtual ~wxFilterClassFactoryBase() { }
363 wxString
GetProtocol() const { return wxString(*GetProtocols()); }
364 wxString
PopExtension(const wxString
& location
) const;
366 virtual const wxChar
* const *GetProtocols(wxStreamProtocolType type
367 = wxSTREAM_PROTOCOL
) const = 0;
369 bool CanHandle(const wxString
& protocol
,
370 wxStreamProtocolType type
371 = wxSTREAM_PROTOCOL
) const;
374 wxString::size_type
FindExtension(const wxString
& location
) const;
376 DECLARE_ABSTRACT_CLASS(wxFilterClassFactoryBase
)
379 class WXDLLIMPEXP_BASE wxFilterClassFactory
: public wxFilterClassFactoryBase
382 virtual ~wxFilterClassFactory() { }
384 virtual wxFilterInputStream
*NewStream(wxInputStream
& stream
) const = 0;
385 virtual wxFilterOutputStream
*NewStream(wxOutputStream
& stream
) const = 0;
386 virtual wxFilterInputStream
*NewStream(wxInputStream
*stream
) const = 0;
387 virtual wxFilterOutputStream
*NewStream(wxOutputStream
*stream
) const = 0;
389 static const wxFilterClassFactory
*Find(const wxString
& protocol
,
390 wxStreamProtocolType type
391 = wxSTREAM_PROTOCOL
);
393 static const wxFilterClassFactory
*GetFirst();
394 const wxFilterClassFactory
*GetNext() const { return m_next
; }
396 void PushFront() { Remove(); m_next
= sm_first
; sm_first
= this; }
400 wxFilterClassFactory() : m_next(this) { }
402 wxFilterClassFactory
& operator=(const wxFilterClassFactory
&)
406 static wxFilterClassFactory
*sm_first
;
407 wxFilterClassFactory
*m_next
;
409 DECLARE_ABSTRACT_CLASS(wxFilterClassFactory
)
412 // ============================================================================
414 // ============================================================================
416 // ---------------------------------------------------------------------------
417 // Stream buffer: this class can be derived from and passed to
418 // wxBufferedStreams to implement custom buffering
419 // ---------------------------------------------------------------------------
421 class WXDLLIMPEXP_BASE wxStreamBuffer
431 wxStreamBuffer(wxStreamBase
& stream
, BufMode mode
)
433 InitWithStream(stream
, mode
);
436 wxStreamBuffer(size_t bufsize
, wxInputStream
& stream
)
438 InitWithStream(stream
, read
);
439 SetBufferIO(bufsize
);
442 wxStreamBuffer(size_t bufsize
, wxOutputStream
& stream
)
444 InitWithStream(stream
, write
);
445 SetBufferIO(bufsize
);
448 wxStreamBuffer(const wxStreamBuffer
& buf
);
449 virtual ~wxStreamBuffer();
452 virtual size_t Read(void *buffer
, size_t size
);
453 size_t Read(wxStreamBuffer
*buf
);
454 virtual size_t Write(const void *buffer
, size_t size
);
455 size_t Write(wxStreamBuffer
*buf
);
458 virtual char GetChar();
459 virtual void PutChar(char c
);
460 virtual wxFileOffset
Tell() const;
461 virtual wxFileOffset
Seek(wxFileOffset pos
, wxSeekMode mode
);
467 // NB: the buffer must always be allocated with malloc() if takeOwn is
468 // true as it will be deallocated by free()
469 void SetBufferIO(void *start
, void *end
, bool takeOwnership
= false);
470 void SetBufferIO(void *start
, size_t len
, bool takeOwnership
= false);
471 void SetBufferIO(size_t bufsize
);
472 void *GetBufferStart() const { return m_buffer_start
; }
473 void *GetBufferEnd() const { return m_buffer_end
; }
474 void *GetBufferPos() const { return m_buffer_pos
; }
475 size_t GetBufferSize() const { return m_buffer_end
- m_buffer_start
; }
476 size_t GetIntPosition() const { return m_buffer_pos
- m_buffer_start
; }
477 void SetIntPosition(size_t pos
) { m_buffer_pos
= m_buffer_start
+ pos
; }
478 size_t GetLastAccess() const { return m_buffer_end
- m_buffer_start
; }
479 size_t GetBytesLeft() const { return m_buffer_end
- m_buffer_pos
; }
481 void Fixed(bool fixed
) { m_fixed
= fixed
; }
482 void Flushable(bool f
) { m_flushable
= f
; }
486 size_t GetDataLeft();
489 wxStreamBase
*GetStream() const { return m_stream
; }
490 bool HasBuffer() const { return m_buffer_start
!= m_buffer_end
; }
492 bool IsFixed() const { return m_fixed
; }
493 bool IsFlushable() const { return m_flushable
; }
495 // only for input/output buffers respectively, returns NULL otherwise
496 wxInputStream
*GetInputStream() const;
497 wxOutputStream
*GetOutputStream() const;
499 #if WXWIN_COMPATIBILITY_2_6
500 // deprecated, for compatibility only
501 wxDEPRECATED( wxStreamBase
*Stream() );
502 #endif // WXWIN_COMPATIBILITY_2_6
504 // this constructs a dummy wxStreamBuffer, used by (and exists for)
505 // wxMemoryStreams only, don't use!
506 wxStreamBuffer(BufMode mode
);
509 void GetFromBuffer(void *buffer
, size_t size
);
510 void PutToBuffer(const void *buffer
, size_t size
);
512 // set the last error to the specified value if we didn't have it before
513 void SetError(wxStreamError err
);
515 // common part of several ctors
518 // common part of ctors taking wxStreamBase parameter
519 void InitWithStream(wxStreamBase
& stream
, BufMode mode
);
521 // init buffer variables to be empty
524 // free the buffer (always safe to call)
527 // the buffer itself: the pointers to its start and end and the current
528 // position in the buffer
529 char *m_buffer_start
,
533 // the stream we're associated with
534 wxStreamBase
*m_stream
;
540 bool m_destroybuf
, // deallocate buffer?
545 wxDECLARE_NO_ASSIGN_CLASS(wxStreamBuffer
);
548 // ---------------------------------------------------------------------------
549 // wxBufferedInputStream
550 // ---------------------------------------------------------------------------
552 class WXDLLIMPEXP_BASE wxBufferedInputStream
: public wxFilterInputStream
555 // create a buffered stream on top of the specified low-level stream
557 // if a non NULL buffer is given to the stream, it will be deleted by it,
558 // otherwise a default 1KB buffer will be used
559 wxBufferedInputStream(wxInputStream
& stream
,
560 wxStreamBuffer
*buffer
= NULL
);
562 // ctor allowing to specify the buffer size, it's just a more convenient
563 // alternative to creating wxStreamBuffer, calling its SetBufferIO(bufsize)
564 // and using the ctor above
565 wxBufferedInputStream(wxInputStream
& stream
, size_t bufsize
);
568 virtual ~wxBufferedInputStream();
571 wxInputStream
& Read(void *buffer
, size_t size
);
573 // Position functions
574 wxFileOffset
SeekI(wxFileOffset pos
, wxSeekMode mode
= wxFromStart
);
575 wxFileOffset
TellI() const;
576 bool IsSeekable() const { return m_parent_i_stream
->IsSeekable(); }
578 // the buffer given to the stream will be deleted by it
579 void SetInputStreamBuffer(wxStreamBuffer
*buffer
);
580 wxStreamBuffer
*GetInputStreamBuffer() const { return m_i_streambuf
; }
582 #if WXWIN_COMPATIBILITY_2_6
583 // deprecated, for compatibility only
584 wxDEPRECATED( wxStreamBuffer
*InputStreamBuffer() const );
585 #endif // WXWIN_COMPATIBILITY_2_6
588 virtual size_t OnSysRead(void *buffer
, size_t bufsize
);
589 virtual wxFileOffset
OnSysSeek(wxFileOffset seek
, wxSeekMode mode
);
590 virtual wxFileOffset
OnSysTell() const;
592 wxStreamBuffer
*m_i_streambuf
;
594 wxDECLARE_NO_COPY_CLASS(wxBufferedInputStream
);
597 // ----------------------------------------------------------------------------
598 // wxBufferedOutputStream
599 // ----------------------------------------------------------------------------
601 class WXDLLIMPEXP_BASE wxBufferedOutputStream
: public wxFilterOutputStream
604 // create a buffered stream on top of the specified low-level stream
606 // if a non NULL buffer is given to the stream, it will be deleted by it,
607 // otherwise a default 1KB buffer will be used
608 wxBufferedOutputStream(wxOutputStream
& stream
,
609 wxStreamBuffer
*buffer
= NULL
);
611 // ctor allowing to specify the buffer size, it's just a more convenient
612 // alternative to creating wxStreamBuffer, calling its SetBufferIO(bufsize)
613 // and using the ctor above
614 wxBufferedOutputStream(wxOutputStream
& stream
, size_t bufsize
);
616 virtual ~wxBufferedOutputStream();
618 wxOutputStream
& Write(const void *buffer
, size_t size
);
620 // Position functions
621 wxFileOffset
SeekO(wxFileOffset pos
, wxSeekMode mode
= wxFromStart
);
622 wxFileOffset
TellO() const;
623 bool IsSeekable() const { return m_parent_o_stream
->IsSeekable(); }
628 wxFileOffset
GetLength() const;
630 // the buffer given to the stream will be deleted by it
631 void SetOutputStreamBuffer(wxStreamBuffer
*buffer
);
632 wxStreamBuffer
*GetOutputStreamBuffer() const { return m_o_streambuf
; }
634 #if WXWIN_COMPATIBILITY_2_6
635 // deprecated, for compatibility only
636 wxDEPRECATED( wxStreamBuffer
*OutputStreamBuffer() const );
637 #endif // WXWIN_COMPATIBILITY_2_6
640 virtual size_t OnSysWrite(const void *buffer
, size_t bufsize
);
641 virtual wxFileOffset
OnSysSeek(wxFileOffset seek
, wxSeekMode mode
);
642 virtual wxFileOffset
OnSysTell() const;
644 wxStreamBuffer
*m_o_streambuf
;
646 wxDECLARE_NO_COPY_CLASS(wxBufferedOutputStream
);
649 #if WXWIN_COMPATIBILITY_2_6
650 inline wxStreamBase
*wxStreamBuffer::Stream() { return m_stream
; }
651 inline wxStreamBuffer
*wxBufferedInputStream::InputStreamBuffer() const { return m_i_streambuf
; }
652 inline wxStreamBuffer
*wxBufferedOutputStream::OutputStreamBuffer() const { return m_o_streambuf
; }
653 #endif // WXWIN_COMPATIBILITY_2_6
655 // ---------------------------------------------------------------------------
656 // wxWrapperInputStream: forwards all IO to another stream.
657 // ---------------------------------------------------------------------------
659 class WXDLLIMPEXP_BASE wxWrapperInputStream
: public wxFilterInputStream
662 // Constructor fully initializing the stream. The overload taking pointer
663 // takes ownership of the parent stream, the one taking reference does not.
665 // Notice that this class also has a default ctor but it's protected as the
666 // derived class is supposed to take care of calling InitParentStream() if
668 wxWrapperInputStream(wxInputStream
& stream
);
669 wxWrapperInputStream(wxInputStream
* stream
);
671 // Override the base class methods to forward to the wrapped stream.
672 virtual wxFileOffset
GetLength() const;
673 virtual bool IsSeekable() const;
676 virtual size_t OnSysRead(void *buffer
, size_t size
);
677 virtual wxFileOffset
OnSysSeek(wxFileOffset pos
, wxSeekMode mode
);
678 virtual wxFileOffset
OnSysTell() const;
680 // Ensure that our own last error is the same as that of the real stream.
682 // This method is const because the error must be updated even from const
683 // methods (in other words, it really should have been mutable in the first
685 void SynchronizeLastError() const
687 const_cast<wxWrapperInputStream
*>(this)->
688 Reset(m_parent_i_stream
->GetLastError());
691 // Default constructor, use InitParentStream() later.
692 wxWrapperInputStream();
694 // Set up the wrapped stream for an object initialized using the default
695 // constructor. The ownership logic is the same as above.
696 void InitParentStream(wxInputStream
& stream
);
697 void InitParentStream(wxInputStream
* stream
);
699 wxDECLARE_NO_COPY_CLASS(wxWrapperInputStream
);
703 #endif // wxUSE_STREAMS
705 #endif // _WX_WXSTREAM_H__