1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
|
/*
* Copyright (C) 2010 Regents of the University of Michigan
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#ifndef __GZFILETYPE_H__
#define __GZFILETYPE_H__
#ifdef __ZLIB_AVAILABLE__
#if defined(_WIN32)
#include <stdio.h> // for NULL!
#endif
#include <stdlib.h>
#include <zlib.h>
#include "FileType.h"
//#include <iostream>
class GzipFileType : public FileType
{
public:
GzipFileType()
{
gzHandle = NULL;
}
virtual ~GzipFileType()
{
}
GzipFileType(const char * filename, const char * mode);
bool operator == (void * rhs)
{
// No two file pointers are the same, so if rhs is not NULL, then
// the two pointers are different (false).
if (rhs != NULL)
return false;
return (gzHandle == rhs);
}
bool operator != (void * rhs)
{
// No two file pointers are the same, so if rhs is not NULL, then
// the two pointers are different (true).
if (rhs != NULL)
return true;
return (gzHandle != rhs);
}
// Close the file.
inline int close()
{
int result = gzclose(gzHandle);
gzHandle = NULL;
return result;
}
// Reset to the beginning of the file.
inline void rewind()
{
// Just call rewind to move to the beginning of the file.
gzrewind(gzHandle);
}
// Check to see if we have reached the EOF.
inline int eof()
{
// check the file for eof.
return gzeof(gzHandle);
}
// Check to see if the file is open.
virtual inline bool isOpen()
{
if (gzHandle != NULL)
{
// gzHandle is not null, so the file is open.
return(true);
}
return(false);
}
// Write to the file
inline unsigned int write(const void * buffer, unsigned int size)
{
return gzwrite(gzHandle, buffer, size);
}
// Read into a buffer from the file. Since the buffer is passed in and
// this would bypass the fileBuffer used by this class, this method must
// be protected.
inline int read(void * buffer, unsigned int size)
{
unsigned int numBytes = gzread(gzHandle, buffer, size);
// if(numBytes != size)
// {
// std::cerr << "Error reading. Read "<< numBytes << " instead of "<< size<<std::endl;
// int error_code = 0;
// const char* errorMsg = gzerror(gzHandle, &error_code);
// std::cerr << "ERROR Code: " << error_code << "; Error Msg: " << errorMsg << std::endl;
// }
return numBytes;
}
// Get current position in the file.
// -1 return value indicates an error.
virtual inline int64_t tell()
{
return gztell(gzHandle);
}
// Seek to the specified offset from the origin.
// origin can be any of the following:
// Note: not all are valid for all filetypes.
// SEEK_SET - Beginning of file
// SEEK_CUR - Current position of the file pointer
// SEEK_END - End of file
// Returns true on successful seek and false on a failed seek.
virtual inline bool seek(int64_t offset, int origin)
{
int64_t returnVal = gzseek(gzHandle, offset, origin);
// Check for failure.
if (returnVal == -1)
{
return false;
}
// Successful.
return true;
}
protected:
// A gzFile is used.
gzFile gzHandle;
};
#endif
#endif
|
