include/wx/zstream.h

   1 /////////////////////////////////////////////////////////////////////////////
   2 // Name:        wx/zstream.h
   3 // Purpose:     Memory stream classes
   4 // Author:      Guilhem Lavaux
   5 // Modified by: Mike Wetherell
   6 // Created:     11/07/98
   7 // RCS-ID:      $Id$
   8 // Copyright:   (c) Guilhem Lavaux
   9 // Licence:     wxWindows licence
  10 /////////////////////////////////////////////////////////////////////////////
  11 #ifndef _WX_WXZSTREAM_H__
  12 #define _WX_WXZSTREAM_H__
  13
  14 #include "wx/defs.h"
  15
  16 #if wxUSE_ZLIB && wxUSE_STREAMS
  17
  18 #include "wx/stream.h"
  19
  20 // Compression level
  21 enum wxZlibCompressionLevels {
  22     wxZ_DEFAULT_COMPRESSION = -1,
  23     wxZ_NO_COMPRESSION = 0,
  24     wxZ_BEST_SPEED = 1,
  25     wxZ_BEST_COMPRESSION = 9
  26 };
  27
  28 // Flags
  29 enum wxZLibFlags {
  30     wxZLIB_NO_HEADER = 0,    // raw deflate stream, no header or checksum
  31     wxZLIB_ZLIB = 1,         // zlib header and checksum
  32     wxZLIB_GZIP = 2,         // gzip header and checksum, requires zlib 1.2.1+
  33     wxZLIB_AUTO = 3          // autodetect header zlib or gzip
  34 };
  35
  36 class WXDLLIMPEXP_BASE wxZlibInputStream: public wxFilterInputStream {
  37  public:
  38   wxZlibInputStream(wxInputStream& stream, int flags = wxZLIB_AUTO);
  39   wxZlibInputStream(wxInputStream *stream, int flags = wxZLIB_AUTO);
  40   virtual ~wxZlibInputStream();
  41
  42   char Peek() { return wxInputStream::Peek(); }
  43   wxFileOffset GetLength() const { return wxInputStream::GetLength(); }
  44
  45   static bool CanHandleGZip();
  46
  47   bool SetDictionary(const char *data, const size_t datalen);
  48   bool SetDictionary(const wxMemoryBuffer &buf);
  49
  50  protected:
  51   size_t OnSysRead(void *buffer, size_t size);
  52   wxFileOffset OnSysTell() const { return m_pos; }
  53
  54  private:
  55   void Init(int flags);
  56
  57  protected:
  58   size_t m_z_size;
  59   unsigned char *m_z_buffer;
  60   struct z_stream_s *m_inflate;
  61   wxFileOffset m_pos;
  62
  63   wxDECLARE_NO_COPY_CLASS(wxZlibInputStream);
  64 };
  65
  66 class WXDLLIMPEXP_BASE wxZlibOutputStream: public wxFilterOutputStream {
  67  public:
  68   wxZlibOutputStream(wxOutputStream& stream, int level = -1, int flags = wxZLIB_ZLIB);
  69   wxZlibOutputStream(wxOutputStream *stream, int level = -1, int flags = wxZLIB_ZLIB);
  70   virtual ~wxZlibOutputStream() { Close(); }
  71
  72   void Sync() { DoFlush(false); }
  73   bool Close();
  74   wxFileOffset GetLength() const { return m_pos; }
  75
  76   static bool CanHandleGZip();
  77
  78   bool SetDictionary(const char *data, const size_t datalen);
  79   bool SetDictionary(const wxMemoryBuffer &buf);
  80
  81  protected:
  82   size_t OnSysWrite(const void *buffer, size_t size);
  83   wxFileOffset OnSysTell() const { return m_pos; }
  84
  85   virtual void DoFlush(bool final);
  86
  87  private:
  88   void Init(int level, int flags);
  89
  90  protected:
  91   size_t m_z_size;
  92   unsigned char *m_z_buffer;
  93   struct z_stream_s *m_deflate;
  94   wxFileOffset m_pos;
  95
  96   wxDECLARE_NO_COPY_CLASS(wxZlibOutputStream);
  97 };
  98
  99 class WXDLLIMPEXP_BASE wxZlibClassFactory: public wxFilterClassFactory
 100 {
 101 public:
 102     wxZlibClassFactory();
 103
 104     wxFilterInputStream *NewStream(wxInputStream& stream) const
 105         { return new wxZlibInputStream(stream); }
 106     wxFilterOutputStream *NewStream(wxOutputStream& stream) const
 107         { return new wxZlibOutputStream(stream, -1); }
 108     wxFilterInputStream *NewStream(wxInputStream *stream) const
 109         { return new wxZlibInputStream(stream); }
 110     wxFilterOutputStream *NewStream(wxOutputStream *stream) const
 111         { return new wxZlibOutputStream(stream, -1); }
 112
 113     const wxChar * const *GetProtocols(wxStreamProtocolType type
 114                                        = wxSTREAM_PROTOCOL) const;
 115
 116 private:
 117     DECLARE_DYNAMIC_CLASS(wxZlibClassFactory)
 118 };
 119
 120 class WXDLLIMPEXP_BASE wxGzipClassFactory: public wxFilterClassFactory
 121 {
 122 public:
 123     wxGzipClassFactory();
 124
 125     wxFilterInputStream *NewStream(wxInputStream& stream) const
 126         { return new wxZlibInputStream(stream); }
 127     wxFilterOutputStream *NewStream(wxOutputStream& stream) const
 128         { return new wxZlibOutputStream(stream, -1); }
 129     wxFilterInputStream *NewStream(wxInputStream *stream) const
 130         { return new wxZlibInputStream(stream); }
 131     wxFilterOutputStream *NewStream(wxOutputStream *stream) const
 132         { return new wxZlibOutputStream(stream, -1); }
 133
 134     const wxChar * const *GetProtocols(wxStreamProtocolType type
 135                                        = wxSTREAM_PROTOCOL) const;
 136
 137 private:
 138     DECLARE_DYNAMIC_CLASS(wxGzipClassFactory)
 139 };
 140
 141 #endif
 142   // wxUSE_ZLIB && wxUSE_STREAMS
 143
 144 #endif
 145    // _WX_WXZSTREAM_H__
 146