summaryrefslogtreecommitdiff
path: root/ZAPD/OutputFormatter.cpp
blob: 362ef98fc9b0b7e18a7ad6a83b105bf010e46358 (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
111
112
113
114
115
116
117
118
119
120
121
122
123
#include "OutputFormatter.h"
#include <Globals.h>

void OutputFormatter::Flush()
{
	//if (!Globals::Instance->otrMode) // OTRTODO: MULTITHREADING
	{
		if (col > lineLimit && !Globals::Instance->otrMode)
		{
			str.append(1, '\n');
			str.append(currentIndent, ' ');

			uint32_t newCol = currentIndent + (wordP - word);

			for (uint32_t 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)
{
	// OTRTODO
	//if (!Globals::Instance->singleThreaded)
		//return 0;

	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;
}

int OutputFormatter::Write(const std::string& buf)
{
	return Write(buf.data(), buf.size());
}

thread_local 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(uint32_t tabSize, uint32_t indentation, uint32_t lineLimit)
	: tabSize{tabSize}, lineLimit{lineLimit}, col{0}, nest{0}, nestIndent{indentation},
	  currentIndent{indentation}, wordNests(0), wordP{word}, spaceP{space}
{
}

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

	return std::move(str);
}