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
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
|
#!/usr/bin/env python3
import argparse
import sys
def parse_args():
parser = argparse.ArgumentParser(description="Align structs on single lines in BEGIN...END ALIGNED SECTION blocks")
parser.add_argument('file', metavar="FILE", nargs='?', help="If provided, a file to process. Otherwise stdin")
return parser.parse_args()
def braces_match(line):
open_braces = 0
for c in line:
if c == '{':
open_braces = open_braces + 1
elif c == '}':
if open_braces == 0:
sys.exit("Too many close braces on line:\n{}", line)
open_braces = open_braces - 1
return (open_braces == 0)
assert(braces_match("{{{}}}"))
assert(not braces_match("{{}{}"))
def is_partial_comment(line):
stripped = line.strip()
return stripped.startswith("/*") and not stripped.endswith("*/")
assert(not is_partial_comment("/* */"))
assert(not is_partial_comment("/* /* */"))
assert(is_partial_comment("/* */ /"))
assert(is_partial_comment("/* */ /*"))
def collineate(lines):
ret = []
current_line = ""
for line in lines:
current_line = current_line + line
if braces_match(current_line) and not is_partial_comment(current_line):
ret.append(current_line)
current_line = ""
if current_line != "":
ret.append(current_line)
return ret
assert(collineate(["/*", "...", "*/"]) == ["/*...*/"])
assert(collineate(["{", "/*", "*/", "}"]) == ["{/**/}"])
assert(collineate(["{", "{", "}", "}"]) == ["{{}}"])
assert(collineate(["{", "}", "{", "}"]) == ["{}", "{}"])
def is_line_ignored(line):
stripped = line.lstrip()
return (stripped == "") or stripped.startswith("/*") or stripped.startswith("//")
def pack(line):
if is_line_ignored(line):
return line
ret = ""
in_leading_whitespace = True
for c in line:
if (c == " ") or (c == "\t"):
if in_leading_whitespace:
ret = ret + c
else:
in_leading_whitespace = False
ret = ret + c
return ret
def prettify(line):
if is_line_ignored(line):
return line
ret = ""
for c in line:
if (c == ","):
ret = ret + ", "
elif (c == "="):
ret = ret + " = "
elif (c == "{"):
ret = ret + "{ "
elif (c == "}"):
ret = ret + " }"
else:
ret = ret + c
return ret
def has_an_offset(c):
return (c == "{") or (c == ",")
def get_offsets(line):
if is_line_ignored(line):
return None
ret = []
offset = -1
for c in line:
offset = offset + 1
if has_an_offset(c):
ret.append(offset)
offset = 0
return ret
def check_offsets(lines, offsets_list):
reference_index = -1
for line in lines:
reference_index = reference_index + 1
if not is_line_ignored(line):
break
if reference_index == len(lines):
return # no lines to check
index = -1
for offsets in offsets_list:
index = index + 1
if offsets == None:
continue
if len(offsets) != len(offsets_list[reference_index]):
sys.exit("Lines have differing numbers of offsets:\n" + lines[index] + "\n" + lines[reference_index])
def collect_max_offsets(offsets_list):
max_offsets = None
for offsets in offsets_list:
if offsets == None:
continue
if max_offsets == None:
max_offsets = offsets
continue
for index in range(len(max_offsets)):
if max_offsets[index] < offsets[index]:
max_offsets[index] = offsets[index]
return max_offsets
def fix_offsets(line, target_offsets):
ret = ""
offset_index = 0
offset = -1
for c in line:
offset = offset + 1
ret = ret + c
if has_an_offset(c):
ret = ret + " " * (target_offsets[offset_index] - offset)
offset_index = offset_index + 1
offset = 0
return ret
def align_section(lines, args):
lines = collineate(lines)
lines = [pack(l) for l in lines]
lines = [prettify(l) for l in lines]
lines = [l.rstrip() for l in lines]
offsets_list = [get_offsets(l) for l in lines]
target_offsets = collect_max_offsets(offsets_list)
lines = [fix_offsets(l, target_offsets) for l in lines]
return lines
def align_file(lines, args):
ret = []
lines_to_align = None
for line in lines:
if lines_to_align is None:
if "BEGIN ALIGNED SECTION" in line:
lines_to_align = []
ret.append(line)
else:
if "END ALIGNED SECTION" in line:
ret = ret + align_section(lines_to_align, args)
lines_to_align = None
ret.append(line)
else:
lines_to_align.append(line)
if lines_to_align is not None:
sys.exit("Incomplete aligned section")
return ret
if __name__ == '__main__':
args = parse_args()
lines = []
if args.file:
with open(args.file) as fd:
lines = [l.rstrip() for l in fd.readlines()]
else:
lines = [l.rstrip() for l in sys.stdin.readlines()]
for line in align_file(lines, args):
print(line)
|