home *** CD-ROM | disk | FTP | other *** search
/ Freelog 33 / Freelog033.iso / Progr / Python-2.2.1.exe / H2PY.PY < prev    next >
Encoding:
Python Source  |  2001-12-06  |  5.3 KB  |  157 lines
  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. filedict = {}
  42. importable = {}
  43.  
  44. try:
  45.     searchdirs=os.environ['include'].split(';')
  46. except KeyError:
  47.     try:
  48.         searchdirs=os.environ['INCLUDE'].split(';')
  49.     except KeyError:
  50.         try:
  51.             if  sys.platform.find("beos") == 0:
  52.                 searchdirs=os.environ['BEINCLUDES'].split(';')
  53.             else:
  54.                 raise KeyError
  55.         except KeyError:
  56.             searchdirs=['/usr/include']
  57.  
  58. def main():
  59.     global filedict
  60.     opts, args = getopt.getopt(sys.argv[1:], 'i:')
  61.     for o, a in opts:
  62.         if o == '-i':
  63.             ignores.append(re.compile(a))
  64.     if not args:
  65.         args = ['-']
  66.     for filename in args:
  67.         if filename == '-':
  68.             sys.stdout.write('# Generated by h2py from stdin\n')
  69.             process(sys.stdin, sys.stdout)
  70.         else:
  71.             fp = open(filename, 'r')
  72.             outfile = os.path.basename(filename)
  73.             i = outfile.rfind('.')
  74.             if i > 0: outfile = outfile[:i]
  75.             modname = outfile.upper()
  76.             outfile = modname + '.py'
  77.             outfp = open(outfile, 'w')
  78.             outfp.write('# Generated by h2py from %s\n' % filename)
  79.             filedict = {}
  80.             for dir in searchdirs:
  81.                 if filename[:len(dir)] == dir:
  82.                     filedict[filename[len(dir)+1:]] = None  # no '/' trailing
  83.                     importable[filename[len(dir)+1:]] = modname
  84.                     break
  85.             process(fp, outfp)
  86.             outfp.close()
  87.             fp.close()
  88.  
  89. def process(fp, outfp, env = {}):
  90.     lineno = 0
  91.     while 1:
  92.         line = fp.readline()
  93.         if not line: break
  94.         lineno = lineno + 1
  95.         match = p_define.match(line)
  96.         if match:
  97.             # gobble up continuation lines
  98.             while line[-2:] == '\\\n':
  99.                 nextline = fp.readline()
  100.                 if not nextline: break
  101.                 lineno = lineno + 1
  102.                 line = line + nextline
  103.             name = match.group(1)
  104.             body = line[match.end():]
  105.             # replace ignored patterns by spaces
  106.             for p in ignores:
  107.                 body = p.sub(' ', body)
  108.             # replace char literals by ord(...)
  109.             body = p_char.sub('ord(\\0)', body)
  110.             stmt = '%s = %s\n' % (name, body.strip())
  111.             ok = 0
  112.             try:
  113.                 exec stmt in env
  114.             except:
  115.                 sys.stderr.write('Skipping: %s' % stmt)
  116.             else:
  117.                 outfp.write(stmt)
  118.         match = p_macro.match(line)
  119.         if match:
  120.             macro, arg = match.group(1, 2)
  121.             body = line[match.end():]
  122.             for p in ignores:
  123.                 body = p.sub(' ', body)
  124.             body = p_char.sub('ord(\\0)', body)
  125.             stmt = 'def %s(%s): return %s\n' % (macro, arg, body)
  126.             try:
  127.                 exec stmt in env
  128.             except:
  129.                 sys.stderr.write('Skipping: %s' % stmt)
  130.             else:
  131.                 outfp.write(stmt)
  132.         match = p_include.match(line)
  133.         if match:
  134.             regs = match.regs
  135.             a, b = regs[1]
  136.             filename = line[a:b]
  137.             if importable.has_key(filename):
  138.                 outfp.write('from %s import *\n' % importable[filename])
  139.             elif not filedict.has_key(filename):
  140.                 filedict[filename] = None
  141.                 inclfp = None
  142.                 for dir in searchdirs:
  143.                     try:
  144.                         inclfp = open(dir + '/' + filename)
  145.                         break
  146.                     except IOError:
  147.                         pass
  148.                 if inclfp:
  149.                     outfp.write(
  150.                             '\n# Included from %s\n' % filename)
  151.                     process(inclfp, outfp, env)
  152.                 else:
  153.                     sys.stderr.write('Warning - could not find file %s\n' %
  154.                                      filename)
  155.  
  156. main()
  157.