| 1 | #! /usr/bin/env python
|
|---|
| 2 |
|
|---|
| 3 | # Read #define's and translate to Python code.
|
|---|
| 4 | # Handle #include statements.
|
|---|
| 5 | # Handle #define macros with one argument.
|
|---|
| 6 | # Anything that isn't recognized or doesn't translate into valid
|
|---|
| 7 | # Python is ignored.
|
|---|
| 8 |
|
|---|
| 9 | # Without filename arguments, acts as a filter.
|
|---|
| 10 | # If one or more filenames are given, output is written to corresponding
|
|---|
| 11 | # filenames in the local directory, translated to all uppercase, with
|
|---|
| 12 | # the extension replaced by ".py".
|
|---|
| 13 |
|
|---|
| 14 | # By passing one or more options of the form "-i regular_expression"
|
|---|
| 15 | # you can specify additional strings to be ignored. This is useful
|
|---|
| 16 | # e.g. to ignore casts to u_long: simply specify "-i '(u_long)'".
|
|---|
| 17 |
|
|---|
| 18 | # XXX To do:
|
|---|
| 19 | # - turn trailing C comments into Python comments
|
|---|
| 20 | # - turn C Boolean operators "&& || !" into Python "and or not"
|
|---|
| 21 | # - what to do about #if(def)?
|
|---|
| 22 | # - what to do about macros with multiple parameters?
|
|---|
| 23 |
|
|---|
| 24 | import sys, re, getopt, os
|
|---|
| 25 |
|
|---|
| 26 | p_define = re.compile('^[\t ]*#[\t ]*define[\t ]+([a-zA-Z0-9_]+)[\t ]+')
|
|---|
| 27 |
|
|---|
| 28 | p_macro = re.compile(
|
|---|
| 29 | '^[\t ]*#[\t ]*define[\t ]+'
|
|---|
| 30 | '([a-zA-Z0-9_]+)\(([_a-zA-Z][_a-zA-Z0-9]*)\)[\t ]+')
|
|---|
| 31 |
|
|---|
| 32 | p_include = re.compile('^[\t ]*#[\t ]*include[\t ]+<([a-zA-Z0-9_/\.]+)')
|
|---|
| 33 |
|
|---|
| 34 | p_comment = re.compile(r'/\*([^*]+|\*+[^/])*(\*+/)?')
|
|---|
| 35 | p_cpp_comment = re.compile('//.*')
|
|---|
| 36 |
|
|---|
| 37 | ignores = [p_comment, p_cpp_comment]
|
|---|
| 38 |
|
|---|
| 39 | p_char = re.compile(r"'(\\.[^\\]*|[^\\])'")
|
|---|
| 40 |
|
|---|
| 41 | p_hex = re.compile(r"0x([0-9a-fA-F]+)L?")
|
|---|
| 42 |
|
|---|
| 43 | filedict = {}
|
|---|
| 44 | importable = {}
|
|---|
| 45 |
|
|---|
| 46 | try:
|
|---|
| 47 | searchdirs=os.environ['include'].split(';')
|
|---|
| 48 | except KeyError:
|
|---|
| 49 | try:
|
|---|
| 50 | searchdirs=os.environ['INCLUDE'].split(';')
|
|---|
| 51 | except KeyError:
|
|---|
| 52 | try:
|
|---|
| 53 | if sys.platform.find("beos") == 0:
|
|---|
| 54 | searchdirs=os.environ['BEINCLUDES'].split(';')
|
|---|
| 55 | elif sys.platform.startswith("atheos"):
|
|---|
| 56 | searchdirs=os.environ['C_INCLUDE_PATH'].split(':')
|
|---|
| 57 | else:
|
|---|
| 58 | raise KeyError
|
|---|
| 59 | except KeyError:
|
|---|
| 60 | searchdirs=['/usr/include']
|
|---|
| 61 |
|
|---|
| 62 | def main():
|
|---|
| 63 | global filedict
|
|---|
| 64 | opts, args = getopt.getopt(sys.argv[1:], 'i:')
|
|---|
| 65 | for o, a in opts:
|
|---|
| 66 | if o == '-i':
|
|---|
| 67 | ignores.append(re.compile(a))
|
|---|
| 68 | if not args:
|
|---|
| 69 | args = ['-']
|
|---|
| 70 | for filename in args:
|
|---|
| 71 | if filename == '-':
|
|---|
| 72 | sys.stdout.write('# Generated by h2py from stdin\n')
|
|---|
| 73 | process(sys.stdin, sys.stdout)
|
|---|
| 74 | else:
|
|---|
| 75 | fp = open(filename, 'r')
|
|---|
| 76 | outfile = os.path.basename(filename)
|
|---|
| 77 | i = outfile.rfind('.')
|
|---|
| 78 | if i > 0: outfile = outfile[:i]
|
|---|
| 79 | modname = outfile.upper()
|
|---|
| 80 | outfile = modname + '.py'
|
|---|
| 81 | outfp = open(outfile, 'w')
|
|---|
| 82 | outfp.write('# Generated by h2py from %s\n' % filename)
|
|---|
| 83 | filedict = {}
|
|---|
| 84 | for dir in searchdirs:
|
|---|
| 85 | if filename[:len(dir)] == dir:
|
|---|
| 86 | filedict[filename[len(dir)+1:]] = None # no '/' trailing
|
|---|
| 87 | importable[filename[len(dir)+1:]] = modname
|
|---|
| 88 | break
|
|---|
| 89 | process(fp, outfp)
|
|---|
| 90 | outfp.close()
|
|---|
| 91 | fp.close()
|
|---|
| 92 |
|
|---|
| 93 | def pytify(body):
|
|---|
| 94 | # replace ignored patterns by spaces
|
|---|
| 95 | for p in ignores:
|
|---|
| 96 | body = p.sub(' ', body)
|
|---|
| 97 | # replace char literals by ord(...)
|
|---|
| 98 | body = p_char.sub('ord(\\0)', body)
|
|---|
| 99 | # Compute negative hexadecimal constants
|
|---|
| 100 | start = 0
|
|---|
| 101 | UMAX = 2*(sys.maxint+1)
|
|---|
| 102 | while 1:
|
|---|
| 103 | m = p_hex.search(body, start)
|
|---|
| 104 | if not m: break
|
|---|
| 105 | s,e = m.span()
|
|---|
| 106 | val = long(body[slice(*m.span(1))], 16)
|
|---|
| 107 | if val > sys.maxint:
|
|---|
| 108 | val -= UMAX
|
|---|
| 109 | body = body[:s] + "(" + str(val) + ")" + body[e:]
|
|---|
| 110 | start = s + 1
|
|---|
| 111 | return body
|
|---|
| 112 |
|
|---|
| 113 | def process(fp, outfp, env = {}):
|
|---|
| 114 | lineno = 0
|
|---|
| 115 | while 1:
|
|---|
| 116 | line = fp.readline()
|
|---|
| 117 | if not line: break
|
|---|
| 118 | lineno = lineno + 1
|
|---|
| 119 | match = p_define.match(line)
|
|---|
| 120 | if match:
|
|---|
| 121 | # gobble up continuation lines
|
|---|
| 122 | while line[-2:] == '\\\n':
|
|---|
| 123 | nextline = fp.readline()
|
|---|
| 124 | if not nextline: break
|
|---|
| 125 | lineno = lineno + 1
|
|---|
| 126 | line = line + nextline
|
|---|
| 127 | name = match.group(1)
|
|---|
| 128 | body = line[match.end():]
|
|---|
| 129 | body = pytify(body)
|
|---|
| 130 | ok = 0
|
|---|
| 131 | stmt = '%s = %s\n' % (name, body.strip())
|
|---|
| 132 | try:
|
|---|
| 133 | exec stmt in env
|
|---|
| 134 | except:
|
|---|
| 135 | sys.stderr.write('Skipping: %s' % stmt)
|
|---|
| 136 | else:
|
|---|
| 137 | outfp.write(stmt)
|
|---|
| 138 | match = p_macro.match(line)
|
|---|
| 139 | if match:
|
|---|
| 140 | macro, arg = match.group(1, 2)
|
|---|
| 141 | body = line[match.end():]
|
|---|
| 142 | body = pytify(body)
|
|---|
| 143 | stmt = 'def %s(%s): return %s\n' % (macro, arg, body)
|
|---|
| 144 | try:
|
|---|
| 145 | exec stmt in env
|
|---|
| 146 | except:
|
|---|
| 147 | sys.stderr.write('Skipping: %s' % stmt)
|
|---|
| 148 | else:
|
|---|
| 149 | outfp.write(stmt)
|
|---|
| 150 | match = p_include.match(line)
|
|---|
| 151 | if match:
|
|---|
| 152 | regs = match.regs
|
|---|
| 153 | a, b = regs[1]
|
|---|
| 154 | filename = line[a:b]
|
|---|
| 155 | if importable.has_key(filename):
|
|---|
| 156 | outfp.write('from %s import *\n' % importable[filename])
|
|---|
| 157 | elif not filedict.has_key(filename):
|
|---|
| 158 | filedict[filename] = None
|
|---|
| 159 | inclfp = None
|
|---|
| 160 | for dir in searchdirs:
|
|---|
| 161 | try:
|
|---|
| 162 | inclfp = open(dir + '/' + filename)
|
|---|
| 163 | break
|
|---|
| 164 | except IOError:
|
|---|
| 165 | pass
|
|---|
| 166 | if inclfp:
|
|---|
| 167 | outfp.write(
|
|---|
| 168 | '\n# Included from %s\n' % filename)
|
|---|
| 169 | process(inclfp, outfp, env)
|
|---|
| 170 | else:
|
|---|
| 171 | sys.stderr.write('Warning - could not find file %s\n' %
|
|---|
| 172 | filename)
|
|---|
| 173 |
|
|---|
| 174 | if __name__ == '__main__':
|
|---|
| 175 | main()
|
|---|
