strade/transform.cpp at master · CuWO4/strade · GitHub

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
#include "include/transform.h"
#include "include/data.h"

#include <iostream>

/* @return whether ch is '{' or ';' or '}' */
static bool is_control_char(char ch) {
    return ch == '{' || ch == ';' || ch == '}';
}

void transform(void) {
    std::string tmp = "";
    int in_bracket = 0;
    bool in_single_quote = false, in_double_quote = false;
    bool at_begin = true;
    for (unsigned i = 0; i < g_text.length(); i++) { // handle the comment
        if (g_text[i] == '(') in_bracket++;
        if (g_text[i] == ')') in_bracket--;
        // Note: The two conditions i>0 and g_text[i - 1] != '\\' take advantage of short-circuit evaluation techniques
        if (g_text[i] == '\'' && i > 0 && g_text[i - 1] != '\\') in_single_quote = !in_single_quote;
        if (g_text[i] == '\"' && i > 0 && g_text[i - 1] != '\\') in_double_quote = !in_double_quote;
        if (i + 1 < g_text.length()) {
            if (g_text[i] == '/' && g_text[i + 1] == '/') {
                if(!at_begin) {
                    tmp += '\n';
                    for (int j = 0; j < g_space_count[g_line]; j++) tmp += ' ';
                }

                for (; i < g_text.length() && g_text[i] != '\n'; i++) {
                    tmp += g_text[i];
                    g_char_count[g_line]++;
                }
            }
            if (g_text[i] == '/' && g_text[i + 1] == '*') {
                if(!at_begin) {
                    tmp += '\n';
                    for (int j = 0; j < g_space_count[g_line]; j++) tmp += ' ';
                }

                for (; i < g_text.length() - 1 &&
                    !(g_text[i] == '*' && g_text[i + 1] == '/'); i++) {
                    tmp += g_text[i];
                    if (g_text[i] == '\n') g_line++;
                    else g_char_count[g_line]++;
                }

            }
        }
        if (g_text[i] != ' ') at_begin = false;
        if (is_control_char(g_text[i])) {
            if (in_single_quote || in_double_quote) {
                tmp += g_text[i];
                g_char_count[g_line]++;
                continue;
            }
            if (g_text[i] == ';' && in_bracket) {
                tmp += g_text[i];
                g_char_count[g_line]++;
                continue;
            }

            while (true) {
                if (tmp.length() == 0) break;

                char c = tmp[tmp.length() - 1];
                if (c == '\n') {
                    g_line--;
                    tmp.pop_back();
                }
                else if (c == ' ') {
                    g_char_count[g_line]--;
                    tmp.pop_back();
                }
                else break;
            }

            std::string spaces = "";
            for (int j = g_char_count[g_line]; j <= g_max_width; j++) {
                spaces += ' ';
                g_char_count[g_line]++;
            }
            tmp += spaces;

            tmp += g_text[i];
            g_char_count[g_line]++;
        }
        else if (g_text[i] == '\n') {
            g_line++;
            g_char_count[g_line] = 0;
            at_begin = true;
            tmp += '\n';
        }
        else {
            g_char_count[g_line]++;
            tmp += g_text[i];
        }
    }
    g_text = tmp;
}