summaryrefslogtreecommitdiff
path: root/dist/s_label_loop.py
blob: 5cc222a4250050a5a2f7c9278259ebecce4bf6ca (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
# Mark outer loop boundaries with {@ and }@ .  Nested loops are not marked.
# Each input line is the content of a C function.
import re, sys

p = re.compile('((for |while |_FOREACH|FOREACH_BEGIN)\([^{)]*\)|do) {')
for line in sys.stdin:
    matched = 0
    m = p.search(line)
    while m != None:
        matched = 1
        pos = m.end()
        out = line[:pos] + "@"
        level = 1
        length = len(line)
        while level > 0 and pos < length:
            c = line[pos:pos+1]
            pos += 1
            out += c
            if c == "}":
                level -= 1
            elif c == "{":
                level += 1
        out += "@"
        sys.stdout.write(out)
        line = line[pos:]
        m = p.search(line)
    if matched != 0:
        sys.stdout.write(line)