-
Notifications
You must be signed in to change notification settings - Fork 0
/
StreamLineReader.h
78 lines (75 loc) · 1.75 KB
/
StreamLineReader.h
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
#pragma once
#include "FileStream.h"
#include <string>
#include <Windows.h>
class StreamLineReader
{
private:
static const DWORD kDefaultBufferSize = 32768;
public:
enum class Encoding
{
UTF8
};
StreamLineReader(std::wstring filepath) : StreamLineReader(filepath, Encoding::UTF8)
{
}
StreamLineReader(std::wstring filepath, Encoding encoding)
: filestream_(FileStream(filepath, FileStream::Mode::Open)), encoding_(encoding), buffersize_(kDefaultBufferSize)
{
AllocateBuffer();
}
StreamLineReader(FileStream&& filestream) : StreamLineReader(std::forward<FileStream>(filestream), Encoding::UTF8)
{
}
StreamLineReader(FileStream&& filestream, Encoding encoding) : filestream_(std::move(filestream)), encoding_(encoding), buffersize_(kDefaultBufferSize)
{
AllocateBuffer();
}
StreamLineReader(StreamLineReader&& other)
: filestream_(std::move(other.filestream_)), encoding_(other.encoding_)
{
buffersize_ = other.buffersize_;
readindex_ = other.readindex_;
readlength_ = other.readlength_;
buffer_ = other.buffer_;
other.buffer_ = NULL;
}
StreamLineReader& operator=(StreamLineReader&& other)
{
if (this != &other)
{
filestream_ = std::move(other.filestream_);
encoding_ = other.encoding_;
buffersize_ = other.buffersize_;
readindex_ = other.readindex_;
readlength_ = other.readlength_;
buffer_ = other.buffer_;
other.buffer_ = NULL;
}
return *this;
}
~StreamLineReader()
{
FreeBuffer();
}
std::wstring ReadLine();
BOOL EndOfStream()
{
return ReadBytes() == 0;
}
void Close()
{
filestream_.Close();
}
private:
void AllocateBuffer();
void FreeBuffer();
DWORD ReadBytes();
FileStream filestream_;
Encoding encoding_;
PBYTE buffer_;
DWORD buffersize_;
DWORD readindex_{};
DWORD readlength_{};
};