include/wx/stream.h

   1 /////////////////////////////////////////////////////////////////////////////
   2 // Name:        wx/stream.h
   3 // Purpose:     stream classes
   4 // Author:      Guilhem Lavaux, Guillermo Rodriguez Garcia, Vadim Zeitlin
   5 // Modified by:
   6 // Created:     11/07/98
   7 // RCS-ID:      $Id$
   8 // Copyright:   (c) Guilhem Lavaux
   9 // Licence:     wxWindows licence
  10 /////////////////////////////////////////////////////////////////////////////
  11
  12 #ifndef _WX_WXSTREAM_H__
  13 #define _WX_WXSTREAM_H__
  14
  15 #include "wx/defs.h"
  16
  17 #if wxUSE_STREAMS
  18
  19 #include <stdio.h>
  20 #include "wx/object.h"
  21 #include "wx/string.h"
  22 #include "wx/filefn.h"  // for wxFileOffset, wxInvalidOffset and wxSeekMode
  23
  24 class WXDLLIMPEXP_BASE wxStreamBase;
  25 class WXDLLIMPEXP_BASE wxInputStream;
  26 class WXDLLIMPEXP_BASE wxOutputStream;
  27
  28 typedef wxInputStream& (*__wxInputManip)(wxInputStream&);
  29 typedef wxOutputStream& (*__wxOutputManip)(wxOutputStream&);
  30
  31 WXDLLIMPEXP_BASE wxOutputStream& wxEndL(wxOutputStream& o_stream);
  32
  33 // ----------------------------------------------------------------------------
  34 // constants
  35 // ----------------------------------------------------------------------------
  36
  37 enum wxStreamError
  38 {
  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
  43 };
  44
  45 // ============================================================================
  46 // base stream classes: wxInputStream and wxOutputStream
  47 // ============================================================================
  48
  49 // ---------------------------------------------------------------------------
  50 // wxStreamBase: common (but non virtual!) base for all stream classes
  51 // ---------------------------------------------------------------------------
  52
  53 class WXDLLIMPEXP_BASE wxStreamBase
  54 {
  55 public:
  56     wxStreamBase();
  57     virtual ~wxStreamBase();
  58
  59     // error testing
  60     wxStreamError GetLastError() const { return m_lasterror; }
  61     virtual bool IsOk() const { return GetLastError() == wxSTREAM_NO_ERROR; }
  62     bool operator!() const { return !IsOk(); }
  63
  64     // reset the stream state
  65     void Reset() { m_lasterror = wxSTREAM_NO_ERROR; }
  66
  67     // this doesn't make sense for all streams, always test its return value
  68     virtual size_t GetSize() const;
  69     virtual wxFileOffset GetLength() const { return wxInvalidOffset; }
  70
  71     // returns true if the streams supports seeking to arbitrary offsets
  72     virtual bool IsSeekable() const { return false; }
  73
  74 protected:
  75     virtual wxFileOffset OnSysSeek(wxFileOffset seek, wxSeekMode mode);
  76     virtual wxFileOffset OnSysTell() const;
  77
  78     size_t m_lastcount;
  79     wxStreamError m_lasterror;
  80
  81     friend class wxStreamBuffer;
  82
  83     DECLARE_NO_COPY_CLASS(wxStreamBase)
  84 };
  85
  86 // ----------------------------------------------------------------------------
  87 // wxInputStream: base class for the input streams
  88 // ----------------------------------------------------------------------------
  89
  90 class WXDLLIMPEXP_BASE wxInputStream : public wxStreamBase
  91 {
  92 public:
  93     // ctor and dtor, nothing exciting
  94     wxInputStream();
  95     virtual ~wxInputStream();
  96
  97
  98     // IO functions
  99     // ------------
 100
 101     // return a character from the stream without removing it, i.e. it will
 102     // still be returned by the next call to GetC()
 103     //
 104     // blocks until something appears in the stream if necessary, if nothing
 105     // ever does (i.e. EOF) LastRead() will return 0 (and the return value is
 106     // undefined), otherwise 1
 107     virtual char Peek();
 108
 109     // return one character from the stream, blocking until it appears if
 110     // necessary
 111     //
 112     // if EOF, return value is undefined and LastRead() will return 0 and not 1
 113     char GetC();
 114
 115     // read at most the given number of bytes from the stream
 116     //
 117     // there are 2 possible situations here: either there is nothing at all in
 118     // the stream right now in which case Read() blocks until something appears
 119     // (use CanRead() to avoid this) or there is already some data available in
 120     // the stream and then Read() doesn't block but returns just the data it
 121     // can read without waiting for more
 122     //
 123     // in any case, if there are not enough bytes in the stream right now,
 124     // LastRead() value will be less than size but greater than 0. If it is 0,
 125     // it means that EOF has been reached.
 126     virtual wxInputStream& Read(void *buffer, size_t size);
 127
 128     // copy the entire contents of this stream into streamOut, stopping only
 129     // when EOF is reached or an error occurs
 130     wxInputStream& Read(wxOutputStream& streamOut);
 131
 132
 133     // status functions
 134     // ----------------
 135
 136     // returns the number of bytes read by the last call to Read(), GetC() or
 137     // Peek()
 138     //
 139     // this should be used to discover whether that call succeeded in reading
 140     // all the requested data or not
 141     virtual size_t LastRead() const { return wxStreamBase::m_lastcount; }
 142
 143     // returns true if some data is available in the stream right now, so that
 144     // calling Read() wouldn't block
 145     virtual bool CanRead() const;
 146
 147     // is the stream at EOF?
 148     //
 149     // note that this cannot be really implemented for all streams and
 150     // CanRead() is more reliable than Eof()
 151     virtual bool Eof() const;
 152
 153
 154     // write back buffer
 155     // -----------------
 156
 157     // put back the specified number of bytes into the stream, they will be
 158     // fetched by the next call to the read functions
 159     //
 160     // returns the number of bytes really stuffed back
 161     size_t Ungetch(const void *buffer, size_t size);
 162
 163     // put back the specified character in the stream
 164     //
 165     // returns true if ok, false on error
 166     bool Ungetch(char c);
 167
 168
 169     // position functions
 170     // ------------------
 171
 172     // move the stream pointer to the given position (if the stream supports
 173     // it)
 174     //
 175     // returns wxInvalidOffset on error
 176     virtual wxFileOffset SeekI(wxFileOffset pos, wxSeekMode mode = wxFromStart);
 177
 178     // return the current position of the stream pointer or wxInvalidOffset
 179     virtual wxFileOffset TellI() const;
 180
 181
 182     // stream-like operators
 183     // ---------------------
 184
 185     wxInputStream& operator>>(wxOutputStream& out) { return Read(out); }
 186     wxInputStream& operator>>(__wxInputManip func) { return func(*this); }
 187
 188 protected:
 189     // do read up to size bytes of data into the provided buffer
 190     //
 191     // this method should return 0 if EOF has been reached or an error occurred
 192     // (m_lasterror should be set accordingly as well) or the number of bytes
 193     // read
 194     virtual size_t OnSysRead(void *buffer, size_t size) = 0;
 195
 196     // write-back buffer support
 197     // -------------------------
 198
 199     // return the pointer to a buffer big enough to hold sizeNeeded bytes
 200     char *AllocSpaceWBack(size_t sizeNeeded);
 201
 202     // read up to size data from the write back buffer, return the number of
 203     // bytes read
 204     size_t GetWBack(void *buf, size_t size);
 205
 206     // write back buffer or NULL if none
 207     char *m_wback;
 208
 209     // the size of the buffer
 210     size_t m_wbacksize;
 211
 212     // the current position in the buffer
 213     size_t m_wbackcur;
 214
 215     friend class wxStreamBuffer;
 216
 217     DECLARE_NO_COPY_CLASS(wxInputStream)
 218 };
 219
 220 // ----------------------------------------------------------------------------
 221 // wxOutputStream: base for the output streams
 222 // ----------------------------------------------------------------------------
 223
 224 class WXDLLIMPEXP_BASE wxOutputStream : public wxStreamBase
 225 {
 226 public:
 227     wxOutputStream();
 228     virtual ~wxOutputStream();
 229
 230     void PutC(char c);
 231     virtual wxOutputStream& Write(const void *buffer, size_t size);
 232     wxOutputStream& Write(wxInputStream& stream_in);
 233
 234     virtual wxFileOffset SeekO(wxFileOffset pos, wxSeekMode mode = wxFromStart);
 235     virtual wxFileOffset TellO() const;
 236
 237     virtual size_t LastWrite() const { return wxStreamBase::m_lastcount; }
 238
 239     virtual void Sync();
 240     virtual bool Close() { return true; }
 241
 242     wxOutputStream& operator<<(wxInputStream& out) { return Write(out); }
 243     wxOutputStream& operator<<( __wxOutputManip func) { return func(*this); }
 244
 245 protected:
 246     // to be implemented in the derived classes (it should have been pure
 247     // virtual)
 248     virtual size_t OnSysWrite(const void *buffer, size_t bufsize);
 249
 250     friend class wxStreamBuffer;
 251
 252     DECLARE_NO_COPY_CLASS(wxOutputStream)
 253 };
 254
 255 // ============================================================================
 256 // helper stream classes
 257 // ============================================================================
 258
 259 // ---------------------------------------------------------------------------
 260 // A stream for measuring streamed output
 261 // ---------------------------------------------------------------------------
 262
 263 class WXDLLIMPEXP_BASE wxCountingOutputStream : public wxOutputStream
 264 {
 265 public:
 266     wxCountingOutputStream();
 267
 268     wxFileOffset GetLength() const;
 269     bool Ok() const { return IsOk(); }
 270     bool IsOk() const { return true; }
 271
 272 protected:
 273     virtual size_t OnSysWrite(const void *buffer, size_t size);
 274     virtual wxFileOffset OnSysSeek(wxFileOffset pos, wxSeekMode mode);
 275     virtual wxFileOffset OnSysTell() const;
 276
 277     size_t m_currentPos;
 278
 279     DECLARE_NO_COPY_CLASS(wxCountingOutputStream)
 280 };
 281
 282 // ---------------------------------------------------------------------------
 283 // "Filter" streams
 284 // ---------------------------------------------------------------------------
 285
 286 class WXDLLIMPEXP_BASE wxFilterInputStream : public wxInputStream
 287 {
 288 public:
 289     wxFilterInputStream();
 290     wxFilterInputStream(wxInputStream& stream);
 291     wxFilterInputStream(wxInputStream *stream);
 292     virtual ~wxFilterInputStream();
 293
 294     char Peek() { return m_parent_i_stream->Peek(); }
 295
 296     wxFileOffset GetLength() const { return m_parent_i_stream->GetLength(); }
 297
 298     wxInputStream *GetFilterInputStream() const { return m_parent_i_stream; }
 299
 300 protected:
 301     wxInputStream *m_parent_i_stream;
 302     bool m_owns;
 303
 304     DECLARE_NO_COPY_CLASS(wxFilterInputStream)
 305 };
 306
 307 class WXDLLIMPEXP_BASE wxFilterOutputStream : public wxOutputStream
 308 {
 309 public:
 310     wxFilterOutputStream();
 311     wxFilterOutputStream(wxOutputStream& stream);
 312     wxFilterOutputStream(wxOutputStream *stream);
 313     virtual ~wxFilterOutputStream();
 314
 315     wxFileOffset GetLength() const { return m_parent_o_stream->GetLength(); }
 316
 317     wxOutputStream *GetFilterOutputStream() const { return m_parent_o_stream; }
 318
 319     bool Close();
 320
 321 protected:
 322     wxOutputStream *m_parent_o_stream;
 323     bool m_owns;
 324
 325     DECLARE_NO_COPY_CLASS(wxFilterOutputStream)
 326 };
 327
 328 enum wxStreamProtocolType
 329 {
 330     wxSTREAM_PROTOCOL,      // wxFileSystem protocol (should be only one)
 331     wxSTREAM_MIMETYPE,      // Mime types the stream handles
 332     wxSTREAM_ENCODING,      // The http Content-Encodings the stream handles
 333     wxSTREAM_FILEEXTENSION  // File extensions the stream handles
 334 };
 335
 336 void WXDLLIMPEXP_BASE wxUseFilterClasses();
 337
 338 class WXDLLIMPEXP_BASE wxFilterClassFactoryBase : public wxObject
 339 {
 340 public:
 341     virtual ~wxFilterClassFactoryBase() { }
 342
 343     wxString GetProtocol() const { return wxString(*GetProtocols()); }
 344     wxString PopExtension(const wxString& location) const;
 345
 346     virtual const wxChar * const *GetProtocols(wxStreamProtocolType type
 347                                                = wxSTREAM_PROTOCOL) const = 0;
 348
 349     bool CanHandle(const wxChar *protocol,
 350                    wxStreamProtocolType type
 351                    = wxSTREAM_PROTOCOL) const;
 352
 353 protected:
 354     wxString::size_type FindExtension(const wxChar *location) const;
 355
 356     DECLARE_ABSTRACT_CLASS(wxFilterClassFactoryBase)
 357 };
 358
 359 class WXDLLIMPEXP_BASE wxFilterClassFactory : public wxFilterClassFactoryBase
 360 {
 361 public:
 362     virtual ~wxFilterClassFactory() { }
 363
 364     virtual wxFilterInputStream  *NewStream(wxInputStream& stream)  const = 0;
 365     virtual wxFilterOutputStream *NewStream(wxOutputStream& stream) const = 0;
 366     virtual wxFilterInputStream  *NewStream(wxInputStream *stream)  const = 0;
 367     virtual wxFilterOutputStream *NewStream(wxOutputStream *stream) const = 0;
 368
 369     static const wxFilterClassFactory *Find(const wxChar *protocol,
 370                                             wxStreamProtocolType type
 371                                             = wxSTREAM_PROTOCOL);
 372
 373     static const wxFilterClassFactory *GetFirst();
 374     const wxFilterClassFactory *GetNext() const { return m_next; }
 375
 376     void PushFront() { Remove(); m_next = sm_first; sm_first = this; }
 377     void Remove();
 378
 379 protected:
 380     wxFilterClassFactory() : m_next(this) { }
 381
 382     wxFilterClassFactory& operator=(const wxFilterClassFactory&)
 383         { return *this; }
 384
 385 private:
 386     static wxFilterClassFactory *sm_first;
 387     wxFilterClassFactory *m_next;
 388
 389     DECLARE_ABSTRACT_CLASS(wxFilterClassFactory)
 390 };
 391
 392 // ============================================================================
 393 // buffered streams
 394 // ============================================================================
 395
 396 // ---------------------------------------------------------------------------
 397 // Stream buffer: this class can be derived from and passed to
 398 // wxBufferedStreams to implement custom buffering
 399 // ---------------------------------------------------------------------------
 400
 401 class WXDLLIMPEXP_BASE wxStreamBuffer
 402 {
 403 public:
 404     enum BufMode
 405     {
 406         read,
 407         write,
 408         read_write
 409     };
 410
 411     wxStreamBuffer(wxStreamBase& stream, BufMode mode);
 412     wxStreamBuffer(const wxStreamBuffer& buf);
 413     virtual ~wxStreamBuffer();
 414
 415     // Filtered IO
 416     virtual size_t Read(void *buffer, size_t size);
 417     size_t Read(wxStreamBuffer *buf);
 418     virtual size_t Write(const void *buffer, size_t size);
 419     size_t Write(wxStreamBuffer *buf);
 420
 421     virtual char Peek();
 422     virtual char GetChar();
 423     virtual void PutChar(char c);
 424     virtual wxFileOffset Tell() const;
 425     virtual wxFileOffset Seek(wxFileOffset pos, wxSeekMode mode);
 426
 427     // Buffer control
 428     void ResetBuffer();
 429
 430     // NB: the buffer must always be allocated with malloc() if takeOwn is
 431     //     true as it will be deallocated by free()
 432     void SetBufferIO(void *start, void *end, bool takeOwnership = false);
 433     void SetBufferIO(void *start, size_t len, bool takeOwnership = false);
 434     void SetBufferIO(size_t bufsize);
 435     void *GetBufferStart() const { return m_buffer_start; }
 436     void *GetBufferEnd() const { return m_buffer_end; }
 437     void *GetBufferPos() const { return m_buffer_pos; }
 438     size_t GetBufferSize() const { return m_buffer_size; }
 439     size_t GetIntPosition() const { return m_buffer_pos - m_buffer_start; }
 440     void SetIntPosition(size_t pos) { m_buffer_pos = m_buffer_start + pos; }
 441     size_t GetLastAccess() const { return m_buffer_end - m_buffer_start; }
 442     size_t GetBytesLeft() const { return m_buffer_end - m_buffer_pos; }
 443
 444     void Fixed(bool fixed) { m_fixed = fixed; }
 445     void Flushable(bool f) { m_flushable = f; }
 446
 447     bool FlushBuffer();
 448     bool FillBuffer();
 449     size_t GetDataLeft();
 450
 451     // misc accessors
 452     wxStreamBase *GetStream() const { return m_stream; }
 453     bool HasBuffer() const { return m_buffer_size != 0; }
 454
 455     bool IsFixed() const { return m_fixed; }
 456     bool IsFlushable() const { return m_flushable; }
 457
 458     // only for input/output buffers respectively, returns NULL otherwise
 459     wxInputStream *GetInputStream() const;
 460     wxOutputStream *GetOutputStream() const;
 461
 462 #if WXWIN_COMPATIBILITY_2_6
 463     // deprecated, for compatibility only
 464     wxDEPRECATED( wxStreamBase *Stream() );
 465 #endif // WXWIN_COMPATIBILITY_2_6
 466
 467     // this constructs a dummy wxStreamBuffer, used by (and exists for)
 468     // wxMemoryStreams only, don't use!
 469     wxStreamBuffer(BufMode mode);
 470
 471 protected:
 472     void GetFromBuffer(void *buffer, size_t size);
 473     void PutToBuffer(const void *buffer, size_t size);
 474
 475     // set the last error to the specified value if we didn't have it before
 476     void SetError(wxStreamError err);
 477
 478     // common part of several ctors
 479     void Init();
 480
 481     // init buffer variables to be empty
 482     void InitBuffer();
 483
 484     // free the buffer (always safe to call)
 485     void FreeBuffer();
 486
 487     // the buffer itself: the pointers to its start and end and the current
 488     // position in the buffer
 489     char *m_buffer_start,
 490          *m_buffer_end,
 491          *m_buffer_pos;
 492
 493     // the buffer size
 494     // FIXME: isn't it the same as m_buffer_end - m_buffer_start? (VZ)
 495     size_t m_buffer_size;
 496
 497     // the stream we're associated with
 498     wxStreamBase *m_stream;
 499
 500     // its mode
 501     BufMode m_mode;
 502
 503     // flags
 504     bool m_destroybuf,      // deallocate buffer?
 505          m_fixed,
 506          m_flushable;
 507
 508 private:
 509 // Cannot use
 510 //  DECLARE_NO_COPY_CLASS(wxStreamBuffer)
 511 // because copy constructor is explicitly declared above;
 512 // but no copy assignment operator is defined, so declare
 513 // it private to prevent the compiler from defining it:
 514     wxStreamBuffer& operator=(const wxStreamBuffer&);
 515 };
 516
 517 // ---------------------------------------------------------------------------
 518 // wxBufferedInputStream
 519 // ---------------------------------------------------------------------------
 520
 521 class WXDLLIMPEXP_BASE wxBufferedInputStream : public wxFilterInputStream
 522 {
 523 public:
 524     // if a non NULL buffer is given to the stream, it will be deleted by it
 525     wxBufferedInputStream(wxInputStream& stream,
 526                           wxStreamBuffer *buffer = NULL);
 527     virtual ~wxBufferedInputStream();
 528
 529     char Peek();
 530     wxInputStream& Read(void *buffer, size_t size);
 531
 532     // Position functions
 533     wxFileOffset SeekI(wxFileOffset pos, wxSeekMode mode = wxFromStart);
 534     wxFileOffset TellI() const;
 535     bool IsSeekable() const { return m_parent_i_stream->IsSeekable(); }
 536
 537     // the buffer given to the stream will be deleted by it
 538     void SetInputStreamBuffer(wxStreamBuffer *buffer);
 539     wxStreamBuffer *GetInputStreamBuffer() const { return m_i_streambuf; }
 540
 541 #if WXWIN_COMPATIBILITY_2_6
 542     // deprecated, for compatibility only
 543     wxDEPRECATED( wxStreamBuffer *InputStreamBuffer() const );
 544 #endif // WXWIN_COMPATIBILITY_2_6
 545
 546 protected:
 547     virtual size_t OnSysRead(void *buffer, size_t bufsize);
 548     virtual wxFileOffset OnSysSeek(wxFileOffset seek, wxSeekMode mode);
 549     virtual wxFileOffset OnSysTell() const;
 550
 551     wxStreamBuffer *m_i_streambuf;
 552
 553     DECLARE_NO_COPY_CLASS(wxBufferedInputStream)
 554 };
 555
 556 // ----------------------------------------------------------------------------
 557 // wxBufferedOutputStream
 558 // ----------------------------------------------------------------------------
 559
 560 class WXDLLIMPEXP_BASE wxBufferedOutputStream : public wxFilterOutputStream
 561 {
 562 public:
 563     // if a non NULL buffer is given to the stream, it will be deleted by it
 564     wxBufferedOutputStream(wxOutputStream& stream,
 565                            wxStreamBuffer *buffer = NULL);
 566     virtual ~wxBufferedOutputStream();
 567
 568     wxOutputStream& Write(const void *buffer, size_t size);
 569
 570     // Position functions
 571     wxFileOffset SeekO(wxFileOffset pos, wxSeekMode mode = wxFromStart);
 572     wxFileOffset TellO() const;
 573     bool IsSeekable() const { return m_parent_o_stream->IsSeekable(); }
 574
 575     void Sync();
 576     bool Close();
 577
 578     wxFileOffset GetLength() const;
 579
 580     // the buffer given to the stream will be deleted by it
 581     void SetOutputStreamBuffer(wxStreamBuffer *buffer);
 582     wxStreamBuffer *GetOutputStreamBuffer() const { return m_o_streambuf; }
 583
 584 #if WXWIN_COMPATIBILITY_2_6
 585     // deprecated, for compatibility only
 586     wxDEPRECATED( wxStreamBuffer *OutputStreamBuffer() const );
 587 #endif // WXWIN_COMPATIBILITY_2_6
 588
 589 protected:
 590     virtual size_t OnSysWrite(const void *buffer, size_t bufsize);
 591     virtual wxFileOffset OnSysSeek(wxFileOffset seek, wxSeekMode mode);
 592     virtual wxFileOffset OnSysTell() const;
 593
 594     wxStreamBuffer *m_o_streambuf;
 595
 596     DECLARE_NO_COPY_CLASS(wxBufferedOutputStream)
 597 };
 598
 599 #if WXWIN_COMPATIBILITY_2_6
 600     inline wxStreamBase *wxStreamBuffer::Stream() { return m_stream; }
 601     inline wxStreamBuffer *wxBufferedInputStream::InputStreamBuffer() const { return m_i_streambuf; }
 602     inline wxStreamBuffer *wxBufferedOutputStream::OutputStreamBuffer() const { return m_o_streambuf; }
 603 #endif // WXWIN_COMPATIBILITY_2_6
 604
 605 #endif // wxUSE_STREAMS
 606
 607 #endif // _WX_WXSTREAM_H__