| 1 |  | 
 
 
 
 
 | 2 | #ifndef zfstream_h | 
 
 
 
 
 | 3 | #define zfstream_h | 
 
 
 
 
 | 4 |  | 
 
 
 
 
 | 5 | #include <fstream.h> | 
 
 
 
 
 | 6 | #include "zlib.h" | 
 
 
 
 
 | 7 |  | 
 
 
 
 
 | 8 | class gzfilebuf : public streambuf { | 
 
 
 
 
 | 9 |  | 
 
 
 
 
 | 10 | public: | 
 
 
 
 
 | 11 |  | 
 
 
 
 
 | 12 | gzfilebuf( ); | 
 
 
 
 
 | 13 | virtual ~gzfilebuf(); | 
 
 
 
 
 | 14 |  | 
 
 
 
 
 | 15 | gzfilebuf *open( const char *name, int io_mode ); | 
 
 
 
 
 | 16 | gzfilebuf *attach( int file_descriptor, int io_mode ); | 
 
 
 
 
 | 17 | gzfilebuf *close(); | 
 
 
 
 
 | 18 |  | 
 
 
 
 
 | 19 | int setcompressionlevel( int comp_level ); | 
 
 
 
 
 | 20 | int setcompressionstrategy( int comp_strategy ); | 
 
 
 
 
 | 21 |  | 
 
 
 
 
 | 22 | inline int is_open() const { return (file !=NULL); } | 
 
 
 
 
 | 23 |  | 
 
 
 
 
 | 24 | virtual streampos seekoff( streamoff, ios::seek_dir, int ); | 
 
 
 
 
 | 25 |  | 
 
 
 
 
 | 26 | virtual int sync(); | 
 
 
 
 
 | 27 |  | 
 
 
 
 
 | 28 | protected: | 
 
 
 
 
 | 29 |  | 
 
 
 
 
 | 30 | virtual int underflow(); | 
 
 
 
 
 | 31 | virtual int overflow( int = EOF ); | 
 
 
 
 
 | 32 |  | 
 
 
 
 
 | 33 | private: | 
 
 
 
 
 | 34 |  | 
 
 
 
 
 | 35 | gzFile file; | 
 
 
 
 
 | 36 | short mode; | 
 
 
 
 
 | 37 | short own_file_descriptor; | 
 
 
 
 
 | 38 |  | 
 
 
 
 
 | 39 | int flushbuf(); | 
 
 
 
 
 | 40 | int fillbuf(); | 
 
 
 
 
 | 41 |  | 
 
 
 
 
 | 42 | }; | 
 
 
 
 
 | 43 |  | 
 
 
 
 
 | 44 | class gzfilestream_common : virtual public ios { | 
 
 
 
 
 | 45 |  | 
 
 
 
 
 | 46 | friend class gzifstream; | 
 
 
 
 
 | 47 | friend class gzofstream; | 
 
 
 
 
 | 48 | friend gzofstream &setcompressionlevel( gzofstream &, int ); | 
 
 
 
 
 | 49 | friend gzofstream &setcompressionstrategy( gzofstream &, int ); | 
 
 
 
 
 | 50 |  | 
 
 
 
 
 | 51 | public: | 
 
 
 
 
 | 52 | virtual ~gzfilestream_common(); | 
 
 
 
 
 | 53 |  | 
 
 
 
 
 | 54 | void attach( int fd, int io_mode ); | 
 
 
 
 
 | 55 | void open( const char *name, int io_mode ); | 
 
 
 
 
 | 56 | void close(); | 
 
 
 
 
 | 57 |  | 
 
 
 
 
 | 58 | protected: | 
 
 
 
 
 | 59 | gzfilestream_common(); | 
 
 
 
 
 | 60 |  | 
 
 
 
 
 | 61 | private: | 
 
 
 
 
 | 62 | gzfilebuf *rdbuf(); | 
 
 
 
 
 | 63 |  | 
 
 
 
 
 | 64 | gzfilebuf buffer; | 
 
 
 
 
 | 65 |  | 
 
 
 
 
 | 66 | }; | 
 
 
 
 
 | 67 |  | 
 
 
 
 
 | 68 | class gzifstream : public gzfilestream_common, public istream { | 
 
 
 
 
 | 69 |  | 
 
 
 
 
 | 70 | public: | 
 
 
 
 
 | 71 |  | 
 
 
 
 
 | 72 | gzifstream(); | 
 
 
 
 
 | 73 | gzifstream( const char *name, int io_mode = ios::in ); | 
 
 
 
 
 | 74 | gzifstream( int fd, int io_mode = ios::in ); | 
 
 
 
 
 | 75 |  | 
 
 
 
 
 | 76 | virtual ~gzifstream(); | 
 
 
 
 
 | 77 |  | 
 
 
 
 
 | 78 | }; | 
 
 
 
 
 | 79 |  | 
 
 
 
 
 | 80 | class gzofstream : public gzfilestream_common, public ostream { | 
 
 
 
 
 | 81 |  | 
 
 
 
 
 | 82 | public: | 
 
 
 
 
 | 83 |  | 
 
 
 
 
 | 84 | gzofstream(); | 
 
 
 
 
 | 85 | gzofstream( const char *name, int io_mode = ios::out ); | 
 
 
 
 
 | 86 | gzofstream( int fd, int io_mode = ios::out ); | 
 
 
 
 
 | 87 |  | 
 
 
 
 
 | 88 | virtual ~gzofstream(); | 
 
 
 
 
 | 89 |  | 
 
 
 
 
 | 90 | }; | 
 
 
 
 
 | 91 |  | 
 
 
 
 
 | 92 | template<class T> class gzomanip { | 
 
 
 
 
 | 93 | friend gzofstream &operator<<(gzofstream &, const gzomanip<T> &); | 
 
 
 
 
 | 94 | public: | 
 
 
 
 
 | 95 | gzomanip(gzofstream &(*f)(gzofstream &, T), T v) : func(f), val(v) { } | 
 
 
 
 
 | 96 | private: | 
 
 
 
 
 | 97 | gzofstream &(*func)(gzofstream &, T); | 
 
 
 
 
 | 98 | T val; | 
 
 
 
 
 | 99 | }; | 
 
 
 
 
 | 100 |  | 
 
 
 
 
 | 101 | template<class T> gzofstream &operator<<(gzofstream &s, const gzomanip<T> &m) | 
 
 
 
 
 | 102 | { | 
 
 
 
 
 | 103 | return (*m.func)(s, m.val); | 
 
 
 
 
 | 104 | } | 
 
 
 
 
 | 105 |  | 
 
 
 
 
 | 106 | inline gzofstream &setcompressionlevel( gzofstream &s, int l ) | 
 
 
 
 
 | 107 | { | 
 
 
 
 
 | 108 | (s.rdbuf())->setcompressionlevel(l); | 
 
 
 
 
 | 109 | return s; | 
 
 
 
 
 | 110 | } | 
 
 
 
 
 | 111 |  | 
 
 
 
 
 | 112 | inline gzofstream &setcompressionstrategy( gzofstream &s, int l ) | 
 
 
 
 
 | 113 | { | 
 
 
 
 
 | 114 | (s.rdbuf())->setcompressionstrategy(l); | 
 
 
 
 
 | 115 | return s; | 
 
 
 
 
 | 116 | } | 
 
 
 
 
 | 117 |  | 
 
 
 
 
 | 118 | inline gzomanip<int> setcompressionlevel(int l) | 
 
 
 
 
 | 119 | { | 
 
 
 
 
 | 120 | return gzomanip<int>(&setcompressionlevel,l); | 
 
 
 
 
 | 121 | } | 
 
 
 
 
 | 122 |  | 
 
 
 
 
 | 123 | inline gzomanip<int> setcompressionstrategy(int l) | 
 
 
 
 
 | 124 | { | 
 
 
 
 
 | 125 | return gzomanip<int>(&setcompressionstrategy,l); | 
 
 
 
 
 | 126 | } | 
 
 
 
 
 | 127 |  | 
 
 
 
 
 | 128 | #endif |