File: enum_processor.py

package info (click to toggle)
ghc 9.6.6-4
  • links: PTS, VCS
  • area: main
  • in suites: trixie
  • size: 158,216 kB
  • sloc: haskell: 648,228; ansic: 81,656; cpp: 11,808; javascript: 8,444; sh: 5,831; fortran: 3,527; python: 3,277; asm: 2,523; makefile: 2,298; yacc: 1,570; lisp: 532; xml: 196; perl: 145; csh: 2
file content (28 lines) | stat: -rwxr-xr-x 823 bytes parent folder | download | duplicates (4)
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
#!/usr/bin/env python3

# The rough equivalent of the traditional CPP:
#   #define printTest(x) (do{ putStr ( "    " ++ "x" ++ " = " ) ; print (x) })
# which is not portable to clang.

import sys

def process(s):
    while True:
        start = s.find('printTest')
        if start == -1:
            return s
        j0 = j = s.index('(', start) + 1
        depth = 1
        while depth > 0:
            if s[j] == '(':
                depth += 1
            if s[j] == ')':
                depth -= 1
            j += 1
        argument = s[j0:j-1]
        expansion = '(do{ putStr ( "    " ++ "%s" ++ " = " ) ; print (%s) })' \
                    % (argument, argument)
        s = s[:start] + expansion + s[j:]

_, _, inputFile, outputFile = sys.argv
open(outputFile, 'w').write(process(open(inputFile, 'r').read()))