summaryrefslogtreecommitdiff
path: root/ZAPD/OutputFormatter.cpp
blob: 33fcb54b76b262d9a07a1da13d153bd3de0f0de8 (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
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
#include "OutputFormatter.h"

void OutputFormatter::Flush()
{
	if (col > lineLimit)
	{
		str.append(1, '\n');
		str.append(currentIndent, ' ');

		int newCol = currentIndent + (wordP - word);

		for (int i = 0; i < wordNests; i++)
			nestIndent[nest - i] -= col - newCol;

		col = newCol;
	}
	else
	{
		str.append(space, spaceP - space);
	}
	spaceP = space;

	str.append(word, wordP - word);
	wordP = word;
	wordNests = 0;
}

int OutputFormatter::Write(const char* buf, int count)
{
	for (int i = 0; i < count; i++)
	{
		char c = buf[i];

		if (c == ' ' || c == '\t' || c == '\n')
		{
			if (wordP - word != 0)
			{
				Flush();
			}

			if (c == '\n')
			{
				col = 0;
				*spaceP++ = c;
			}
			else if (c == '\t')
			{
				int n = tabSize - (col % tabSize);
				col += n;
				for (int j = 0; j < n; j++)
					*spaceP++ = ' ';
			}
			else
			{
				col++;
				*spaceP++ = c;
			}

			currentIndent = nestIndent[nest];
		}
		else
		{
			col++;

			if (c == '(')
			{
				nest++;
				nestIndent[nest] = col;
				wordNests++;
			}
			else if (c == ')')
			{
				if (nest > 0)
					nest--;
				if (wordNests > 0)
					wordNests--;
			}

			*wordP++ = c;
		}
	}

	return count;
}

OutputFormatter* OutputFormatter::Instance;

int OutputFormatter::WriteStatic(const char* buf, int count)
{
	return Instance->Write(buf, count);
}

int (*OutputFormatter::StaticWriter())(const char* buf, int count)
{
	Instance = this;
	return &WriteStatic;
}

OutputFormatter::OutputFormatter(int tabSize, int defaultIndent, int lineLimit)
	: tabSize{tabSize}, defaultIndent{defaultIndent}, lineLimit{lineLimit}, col{0}, nest{0},
	  nestIndent{defaultIndent}, currentIndent{defaultIndent}, wordNests(0), wordP{word}, spaceP{space}
{
}

std::string OutputFormatter::GetOutput()
{
	Flush();

	return std::move(str);
}