123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466 |
- #ifndef ZFSTREAM_H
- #define ZFSTREAM_H
- #include <istream>
- #include <ostream>
- #include "zlib.h"
- class gzfilebuf : public std::streambuf
- {
- public:
-
- gzfilebuf();
-
- virtual
- ~gzfilebuf();
-
- int
- setcompression(int comp_level,
- int comp_strategy = Z_DEFAULT_STRATEGY);
-
- bool
- is_open() const { return (file != NULL); }
-
- gzfilebuf*
- open(const char* name,
- std::ios_base::openmode mode);
-
- gzfilebuf*
- attach(int fd,
- std::ios_base::openmode mode);
-
- gzfilebuf*
- close();
- protected:
-
- bool
- open_mode(std::ios_base::openmode mode,
- char* c_mode) const;
-
- virtual std::streamsize
- showmanyc();
-
- virtual int_type
- underflow();
-
- virtual int_type
- overflow(int_type c = traits_type::eof());
-
- virtual std::streambuf*
- setbuf(char_type* p,
- std::streamsize n);
-
- virtual int
- sync();
- private:
-
- void
- enable_buffer();
-
- void
- disable_buffer();
-
- gzFile file;
-
- std::ios_base::openmode io_mode;
-
- bool own_fd;
-
- char_type* buffer;
-
- std::streamsize buffer_size;
-
- bool own_buffer;
- };
- class gzifstream : public std::istream
- {
- public:
-
- gzifstream();
-
- explicit
- gzifstream(const char* name,
- std::ios_base::openmode mode = std::ios_base::in);
-
- explicit
- gzifstream(int fd,
- std::ios_base::openmode mode = std::ios_base::in);
-
- gzfilebuf*
- rdbuf() const
- { return const_cast<gzfilebuf*>(&sb); }
-
- bool
- is_open() { return sb.is_open(); }
-
- void
- open(const char* name,
- std::ios_base::openmode mode = std::ios_base::in);
-
- void
- attach(int fd,
- std::ios_base::openmode mode = std::ios_base::in);
-
- void
- close();
- private:
-
- gzfilebuf sb;
- };
- class gzofstream : public std::ostream
- {
- public:
-
- gzofstream();
-
- explicit
- gzofstream(const char* name,
- std::ios_base::openmode mode = std::ios_base::out);
-
- explicit
- gzofstream(int fd,
- std::ios_base::openmode mode = std::ios_base::out);
-
- gzfilebuf*
- rdbuf() const
- { return const_cast<gzfilebuf*>(&sb); }
-
- bool
- is_open() { return sb.is_open(); }
-
- void
- open(const char* name,
- std::ios_base::openmode mode = std::ios_base::out);
-
- void
- attach(int fd,
- std::ios_base::openmode mode = std::ios_base::out);
-
- void
- close();
- private:
-
- gzfilebuf sb;
- };
- template<typename T1, typename T2>
- class gzomanip2
- {
- public:
-
- template <typename Ta, typename Tb>
- friend gzofstream&
- operator<<(gzofstream&,
- const gzomanip2<Ta,Tb>&);
-
- gzomanip2(gzofstream& (*f)(gzofstream&, T1, T2),
- T1 v1,
- T2 v2);
- private:
-
- gzofstream&
- (*func)(gzofstream&, T1, T2);
-
- T1 val1;
- T2 val2;
- };
- inline gzofstream&
- setcompression(gzofstream &gzs, int l, int s = Z_DEFAULT_STRATEGY)
- {
- (gzs.rdbuf())->setcompression(l, s);
- return gzs;
- }
- template<typename T1, typename T2>
- inline
- gzomanip2<T1,T2>::gzomanip2(gzofstream &(*f)(gzofstream &, T1, T2),
- T1 v1,
- T2 v2)
- : func(f), val1(v1), val2(v2)
- { }
- template<typename T1, typename T2>
- inline gzofstream&
- operator<<(gzofstream& s, const gzomanip2<T1,T2>& m)
- { return (*m.func)(s, m.val1, m.val2); }
- inline gzomanip2<int,int>
- setcompression(int l, int s = Z_DEFAULT_STRATEGY)
- { return gzomanip2<int,int>(&setcompression, l, s); }
- #endif
|