zfstream.h
上传用户:dangjiwu
上传日期:2013-07-19
资源大小:42019k
文件大小:3k
源码类别:

Symbian

开发平台:

Visual C++

  1. #ifndef _zfstream_h
  2. #define _zfstream_h
  3. #include <fstream.h>
  4. #include "zlib.h"
  5. class gzfilebuf : public streambuf {
  6. public:
  7.   gzfilebuf( );
  8.   virtual ~gzfilebuf();
  9.   gzfilebuf *open( const char *name, int io_mode );
  10.   gzfilebuf *attach( int file_descriptor, int io_mode );
  11.   gzfilebuf *close();
  12.   int setcompressionlevel( short comp_level );
  13.   int setcompressionstrategy( short comp_strategy );
  14.   inline int is_open() const { return (file !=NULL); }
  15.   virtual streampos seekoff( streamoff, ios::seek_dir, int );
  16.   virtual int sync();
  17. protected:
  18.   virtual int underflow();
  19.   virtual int overflow( int = EOF );
  20. private:
  21.   gzFile file;
  22.   short mode;
  23.   short own_file_descriptor;
  24.   int flushbuf();
  25.   int fillbuf();
  26. };
  27. class gzfilestream_common : virtual public ios {
  28.   friend class gzifstream;
  29.   friend class gzofstream;
  30.   friend gzofstream &setcompressionlevel( gzofstream &, int );
  31.   friend gzofstream &setcompressionstrategy( gzofstream &, int );
  32. public:
  33.   virtual ~gzfilestream_common();
  34.   void attach( int fd, int io_mode );
  35.   void open( const char *name, int io_mode );
  36.   void close();
  37. protected:
  38.   gzfilestream_common();
  39. private:
  40.   gzfilebuf *rdbuf();
  41.   gzfilebuf buffer;
  42. };
  43. class gzifstream : public gzfilestream_common, public istream {
  44. public:
  45.   gzifstream();
  46.   gzifstream( const char *name, int io_mode = ios::in );
  47.   gzifstream( int fd, int io_mode = ios::in );
  48.   virtual ~gzifstream();
  49. };
  50. class gzofstream : public gzfilestream_common, public ostream {
  51. public:
  52.   gzofstream();
  53.   gzofstream( const char *name, int io_mode = ios::out );
  54.   gzofstream( int fd, int io_mode = ios::out );
  55.   virtual ~gzofstream();
  56. };
  57. template<class T> class gzomanip {
  58.   friend gzofstream &operator<<(gzofstream &, const gzomanip<T> &);
  59. public:
  60.   gzomanip(gzofstream &(*f)(gzofstream &, T), T v) : func(f), val(v) { }
  61. private:
  62.   gzofstream &(*func)(gzofstream &, T);
  63.   T val;
  64. };
  65. template<class T> gzofstream &operator<<(gzofstream &s,
  66.  const gzomanip<T> &m) {
  67.   return (*m.func)(s, m.val);
  68.   
  69. }
  70. inline gzofstream &setcompressionlevel( gzofstream &s, int l ) {
  71.   (s.rdbuf())->setcompressionlevel(l);
  72.   return s;
  73. }
  74. inline gzofstream &setcompressionstrategy( gzofstream &s, int l ) {
  75.   (s.rdbuf())->setcompressionstrategy(l);
  76.   return s;
  77. }
  78. inline gzomanip<int> setcompressionlevel(int l)
  79. {
  80.   return gzomanip<int>(&setcompressionlevel,l);
  81. }
  82. inline gzomanip<int> setcompressionstrategy(int l)
  83. {
  84.   return gzomanip<int>(&setcompressionstrategy,l);
  85. }
  86. #endif