home *** CD-ROM | disk | FTP | other *** search
/ PC World 2001 April / PCWorld_2001-04_cd.bin / Software / TemaCD / webclean / !!!python!!! / BeOpen-Python-2.0.exe / H2PY.PY < prev    next >
Encoding:
Python Source  |  2000-09-28  |  4.2 KB  |  151 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, regex, regsub, string, getopt, os
  25.  
  26. p_define = regex.compile('^[\t ]*#[\t ]*define[\t ]+\([a-zA-Z0-9_]+\)[\t ]+')
  27.  
  28. p_macro = regex.compile(
  29.   '^[\t ]*#[\t ]*define[\t ]+'
  30.   '\([a-zA-Z0-9_]+\)(\([_a-zA-Z][_a-zA-Z0-9]*\))[\t ]+')
  31.  
  32. p_include = regex.compile('^[\t ]*#[\t ]*include[\t ]+<\([a-zA-Z0-9_/\.]+\)')
  33.  
  34. p_comment = regex.compile('/\*\([^*]+\|\*+[^/]\)*\(\*+/\)?')
  35. p_cpp_comment = regex.compile('//.*')
  36.  
  37. ignores = [p_comment, p_cpp_comment]
  38.  
  39. p_char = regex.compile("'\(\\\\.[^\\\\]*\|[^\\\\]\)'")
  40.  
  41. filedict = {}
  42.  
  43. try:
  44.     searchdirs=string.splitfields(os.environ['include'],';')
  45. except KeyError:
  46.     try:
  47.         searchdirs=string.splitfields(os.environ['INCLUDE'],';')
  48.     except KeyError:
  49.         try:
  50.             if string.find( sys.platform, "beos" ) == 0:
  51.                 searchdirs=string.splitfields(os.environ['BEINCLUDES'],';')
  52.             else:
  53.                 raise KeyError
  54.         except KeyError:
  55.             searchdirs=['/usr/include']
  56.  
  57. def main():
  58.     global filedict
  59.     opts, args = getopt.getopt(sys.argv[1:], 'i:')
  60.     for o, a in opts:
  61.         if o == '-i':
  62.             ignores.append(regex.compile(a))
  63.     if not args:
  64.         args = ['-']
  65.     for filename in args:
  66.         if filename == '-':
  67.             sys.stdout.write('# Generated by h2py from stdin\n')
  68.             process(sys.stdin, sys.stdout)
  69.         else:
  70.             fp = open(filename, 'r')
  71.             outfile = os.path.basename(filename)
  72.             i = string.rfind(outfile, '.')
  73.             if i > 0: outfile = outfile[:i]
  74.             outfile = string.upper(outfile)
  75.             outfile = outfile + '.py'
  76.             outfp = open(outfile, 'w')
  77.             outfp.write('# Generated by h2py from %s\n' % filename)
  78.             filedict = {}
  79.             for dir in searchdirs:
  80.                 if filename[:len(dir)] == dir:
  81.                     filedict[filename[len(dir)+1:]] = None    # no '/' trailing
  82.                     break
  83.             process(fp, outfp)
  84.             outfp.close()
  85.             fp.close()
  86.  
  87. def process(fp, outfp, env = {}):
  88.     lineno = 0
  89.     while 1:
  90.         line = fp.readline()
  91.         if not line: break
  92.         lineno = lineno + 1
  93.         n = p_define.match(line)
  94.         if n >= 0:
  95.             # gobble up continuation lines
  96.             while line[-2:] == '\\\n':
  97.                 nextline = fp.readline()
  98.                 if not nextline: break
  99.                 lineno = lineno + 1
  100.                 line = line + nextline
  101.             name = p_define.group(1)
  102.             body = line[n:]
  103.             # replace ignored patterns by spaces
  104.             for p in ignores:
  105.                 body = regsub.gsub(p, ' ', body)
  106.             # replace char literals by ord(...)
  107.             body = regsub.gsub(p_char, 'ord(\\0)', body)
  108.             stmt = '%s = %s\n' % (name, string.strip(body))
  109.             ok = 0
  110.             try:
  111.                 exec stmt in env
  112.             except:
  113.                 sys.stderr.write('Skipping: %s' % stmt)
  114.             else:
  115.                 outfp.write(stmt)
  116.         n =p_macro.match(line)
  117.         if n >= 0:
  118.             macro, arg = p_macro.group(1, 2)
  119.             body = line[n:]
  120.             for p in ignores:
  121.                 body = regsub.gsub(p, ' ', body)
  122.             body = regsub.gsub(p_char, 'ord(\\0)', body)
  123.             stmt = 'def %s(%s): return %s\n' % (macro, arg, body)
  124.             try:
  125.                 exec stmt in env
  126.             except:
  127.                 sys.stderr.write('Skipping: %s' % stmt)
  128.             else:
  129.                 outfp.write(stmt)
  130.         if p_include.match(line) >= 0:
  131.             regs = p_include.regs
  132.             a, b = regs[1]
  133.             filename = line[a:b]
  134.             if not filedict.has_key(filename):
  135.                 filedict[filename] = None
  136.                 inclfp = None
  137.                 for dir in searchdirs:
  138.                     try:
  139.                         inclfp = open(dir + '/' + filename, 'r')
  140.                         break
  141.                     except IOError:
  142.                         pass
  143.                 if inclfp:
  144.                     outfp.write(
  145.                         '\n# Included from %s\n' % filename)
  146.                     process(inclfp, outfp, env)
  147.                 else:
  148.                     sys.stderr.write('Warning - could not find file %s' % filename)
  149.  
  150. main()
  151.