about summary refs log tree commit diff
path: root/src/base/tl/stream.hpp
blob: c307b968c50e032cb6ce150ca8bc9d40bb11e5a6 (plain)
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
#ifndef TL_FILE_STREAM_HPP
#define TL_FILE_STREAM_HPP

class input_stream
{
public:	
	virtual ~input_stream() {}
	virtual size_t read(void *data, size_t size) = 0;
	virtual size_t size() = 0;
};

class output_stream
{
public:	
	virtual ~output_stream() {}
	virtual size_t write(const void *data, size_t size) = 0;
};


// input wrapping
// RAII style
class file_backend
{
private:
	file_backend(const file_backend &other) { /* no copy allowed */ }
protected:
	IOHANDLE file_handle;
	
	explicit file_backend(const char *filename, int flags)
	{
		file_handle = io_open(filename, flags);
	}
	
	~file_backend()
	{
		if(file_handle)
			io_close(file_handle);
	}
public:
	bool is_open() const { return file_handle != 0; }
};

class file_reader : public input_stream, public file_backend
{
public:	
	explicit file_reader(const char *filename)
	: file_backend(filename, IOFLAG_READ)
	{}
	
	virtual size_t read(void *data, size_t size) { return io_read(file_handle, data, size); }
	virtual size_t size() { return io_length(file_handle); }
};


class file_writer : public output_stream, public file_backend
{
public:	
	explicit file_writer(const char *filename)
	: file_backend(filename, IOFLAG_WRITE)
	{}
	
	virtual size_t write(const void *data, size_t size) { return io_write(file_handle, data, size); }
};

#endif