1 | #ifndef MARS_MZlib
|
---|
2 | #define MARS_MZlib
|
---|
3 |
|
---|
4 | #ifndef ROOT_TObject
|
---|
5 | #include <TObject.h>
|
---|
6 | #endif
|
---|
7 |
|
---|
8 | #ifdef __CINT__
|
---|
9 | typedef void *gzFile;
|
---|
10 | #else
|
---|
11 | #include <zlib.h>
|
---|
12 | #endif
|
---|
13 |
|
---|
14 | #include <iostream> // base classes for MLog
|
---|
15 | #include <fstream>
|
---|
16 |
|
---|
17 | class MZlib : public std::streambuf, virtual public std::istream, public TObject
|
---|
18 | {
|
---|
19 | private:
|
---|
20 | static const int fgBufferSize = 47+256; // size of data buff totals 512 bytes under g++ for igzstream at the end.
|
---|
21 |
|
---|
22 | gzFile fFile; // file handle for compressed file
|
---|
23 | char fBuffer[fgBufferSize]; // data buffer
|
---|
24 |
|
---|
25 | int underflow();
|
---|
26 | int fill_buffer(int putback=0);
|
---|
27 |
|
---|
28 | public:
|
---|
29 | MZlib() : istream(this), fFile(0)
|
---|
30 | {
|
---|
31 | setg(fBuffer+4, fBuffer+4, fBuffer+4);
|
---|
32 | }
|
---|
33 | MZlib(const char *name) : istream(this), fFile(0)
|
---|
34 | {
|
---|
35 | setg(fBuffer+4, fBuffer+4, fBuffer+4);
|
---|
36 | open(name);
|
---|
37 | }
|
---|
38 | ~MZlib() { MZlib::close(); }
|
---|
39 |
|
---|
40 | int is_open() { return fFile!=0; }
|
---|
41 |
|
---|
42 | void open(const char* name);
|
---|
43 | void close();
|
---|
44 |
|
---|
45 | # if (__GNUC__>2)
|
---|
46 | std::streambuf::pos_type seekoff(std::streambuf::off_type, std::ios_base::seekdir,
|
---|
47 | std::ios_base::openmode = std::ios_base::in);
|
---|
48 | std::streambuf::pos_type seekpos(std::streambuf::pos_type,
|
---|
49 | std::ios_base::openmode = std::ios_base::in);
|
---|
50 | # else
|
---|
51 | std::streampos seekoff(std::streamoff, int, int = std::ios::in);
|
---|
52 | std::streampos seekpos(std::streampos, int = std::ios::in);
|
---|
53 | # endif
|
---|
54 |
|
---|
55 | ClassDef(MZlib, 0) // A C++ wrapper to istream zlib files
|
---|
56 | };
|
---|
57 |
|
---|
58 | #endif
|
---|