dida/Common/GzipStream.h at master · BirolLab/dida · GitHub - github.com

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
#ifndef _ZIPSTREAM_H_
#define _ZIPSTREAM_H_

#include "Options.h"
#include <gzstream.h>
#include <cassert>

static inline std::istream* openInputStream(std::string path)
{
	std::istream* s;
#if HAVE_LIBZ
	if (opt::gzip)
		s = new igzstream(path.c_str());
	else
		s = new std::ifstream(path.c_str());
#else
	s = new std::ifstream(path.c_str());
#endif
	assert(*s);
	return s;
}

static inline void closeInputStream(std::istream* s)
{
	assert(s != NULL);
#if HAVE_LIBZ
	if (opt::gzip)
		((igzstream*)s)->close();
	else
		((std::ifstream*)s)->close();
#else
	((std::ifstream*)s)->close();
#endif
	delete s;
}

static inline std::ostream* openOutputStream(std::string path)
{
	std::ostream* s;
#if HAVE_LIBZ
	if (opt::gzip)
		s = new ogzstream(path.c_str());
	else
		s = new std::ofstream(path.c_str());
#else
	s = new std::ofstream(path.c_str());
#endif
	assert(*s);
	return s;
}

static inline void closeOutputStream(std::ostream* s)
{
	assert(s != NULL);
#if HAVE_LIBZ
	if (opt::gzip)
		((ogzstream*)s)->close();
	else
		((std::ofstream*)s)->close();
#else
	((std::ofstream*)s)->close();
#endif
	delete s;
}

#endif