| 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 | 
|---|