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
|
#include "stdafx.h"
#include "Indent.h"
static const wchar_t indentStr[] = L" ";
Indent::Indent(std::wostream &stream) :
owner(stream),
to(stream.rdbuf()),
startOfLine(true) {
owner.rdbuf(this);
setp(buffer, buffer + bufferSize - 1);
}
Indent::~Indent() {
sync();
owner.rdbuf(to);
}
int Indent::sync() {
overflow(traits_type::eof());
to->pubsync();
return 0;
}
Indent::int_type Indent::overflow(int_type ch) {
wchar_t *begin = pbase();
wchar_t *end = pptr();
if (ch != traits_type::eof())
*end++ = wchar(ch);
for (wchar_t *at = begin; at != end; at++) {
if (*at == '\n') {
startOfLine = true;
} else if (startOfLine) {
startOfLine = false;
if (at != begin)
to->sputn(begin, at - begin);
to->sputn(indentStr, ARRAY_COUNT(indentStr) - 1);
begin = at;
}
}
if (begin != end)
to->sputn(begin, end - begin);
setp(buffer, buffer + bufferSize - 1);
return 0;
}
|
