refactor shared components of python tools, and add emmaken.py

1 files changed, 146 insertions, 0 deletions

diff --git a/tools/emmaken.py b/tools/emmaken.py
new file mode 100755
index 00000000..1b29f7a9
--- /dev/null
+++ b/tools/emmaken.py
@@ -0,0 +1,146 @@
+#!/usr/bin/env python
+
+'''
+emmaken - the emscripten make proxy tool
+========================================
+
+Tell your build system to use this instead of the compiler, linker, ar and
+ranlib. All the normal build commands will be sent to this script, which
+will proxy them to the appropriate LLVM build commands, in order to
+generate proper code for Emscripten to later process.
+
+For example, compilation will be translated into calls to llvm-gcc
+with -emit-llvm, and linking will be translated into calls to llvm-link,
+and so forth.
+
+Example uses:
+
+ * With configure, do something like
+
+    RANLIB=PATH/emmaken.py AR=PATH/emmaken.py CXX=PATH/emmaken.py CC=PATH/emmaken.py ./configure [options]
+
+   where PATH is the path to this file.
+
+ * With CMake, do something like
+
+    SET(CMAKE_C_COMPILER "PATH/emmaken.py")
+    SET(CMAKE_CXX_COMPILER "PATH/emmaken.py")
+    SET(CMAKE_LINKER "PATH/emmaken.py")
+    SET(CMAKE_CXX_LINKER "PATH/emmaken.py")
+    SET(CMAKE_C_LINK_EXECUTABLE "PATH/emmaken.py")
+    SET(CMAKE_CXX_LINK_EXECUTABLE "PATH/emmaken.py")
+    SET(CMAKE_AR "PATH/emmaken.py")
+    SET(CMAKE_RANLIB "PATH/emmaken.py")
+
+After setting that up, run your build system normally. It should generate
+LLVM instead of the normal output, and end up with .ll files that you can
+give to Emscripten. Note that this tool doesn't run Emscripten itself. Note
+also that you may need to do some manual fiddling later, for example to
+link files that weren't linked, and them llvm-dis them.
+'''
+
+import sys
+import os
+import subprocess
+
+abspath = os.path.abspath(os.path.dirname(__file__))
+def path_from_root(*pathelems):
+  return os.path.join(os.path.sep, *(abspath.split(os.sep)[:-1] + list(pathelems)))
+exec(open(path_from_root('tools', 'shared.py'), 'r').read())
+
+CONFIG_FILE = os.path.expanduser('~/.emscripten')
+assert os.path.exists(CONFIG_FILE)
+exec(open(CONFIG_FILE, 'r').read())
+
+try:
+  print >> sys.stderr, 'emmaken.py: ', ' '.join(sys.argv)
+
+  CC='/home/alon/Dev/llvm-gcc-4.2-2.8.source/cbuild/install/bin/llvm-g++'
+  CC_ARG_SKIP = ['-g', '-O1', '-O2', '-O3']
+  CC_ADDITIONAL_ARGS = ['-m32', '-U__i386__', '-U__x86_64__', '-UX87_DOUBLE_ROUNDING', '-UHAVE_GCC_ASM_FOR_X87']
+
+  #CC='llvm-gcc'
+  #CC_ARG_SKIP = ['-g', '-O1', '-O2', '-O3']
+  #CC_ADDITIONAL_ARGS = ['-U__i386__', '-U__x86_64__']
+
+  LLVM_LINK = '/home/alon/Dev/llvm-2.8/cbuild/Release/bin/llvm-link'
+  LLVM_DIS = '/home/alon/Dev/llvm-2.8/cbuild/Release/bin/llvm-dis'
+  ALLOWED_LINK_ARGS = ['-f', '-help', '-o', '-print-after', '-print-after-all', '-print-before',
+                       '-print-before-all', '-time-passes', '-v', '-verify-dom-info', '-version' ]  
+  #LINK_ARG_SKIP = ['-pthread', '-DNDEBUG', '-g', '-O3', '-Wall', '-Wstrict-prototypes',
+  #                 '-lpthread', '-ldl', '-lutil', '-Xlinker', '-export-dynamic', '-lm', '-shared']
+
+  # ----------------  End configs -------------
+
+  if len(sys.argv) == 2 and 'conftest' not in ' '.join(sys.argv): # Avoid messing with configure, see below too
+    # ranlib
+    os.execvp(LLVM_DIS, ['-show-annotations', sys.argv[1]])
+    sys.exit(0)
+  if sys.argv[1] in ['x', 't']:
+    # noop ar
+    sys.exit(0)
+
+  use_linker = True
+  #use_linker = False
+
+  opts = []
+  files = []
+  for arg in sys.argv[1:]:
+      if arg.startswith('-'):
+          opts.append(arg)
+      else:
+          files.append(arg)
+          if arg.endswith(('.c', '.cc', '.cpp')):
+              use_linker = False
+              
+  if '--version' in opts:
+      use_linker = False
+
+  if sys.argv[1] == 'cru': # ar
+    sys.argv = sys.argv[:1] + sys.argv[3:] + ['-o='+sys.argv[2]]
+    assert use_linker, 'Linker should be used in this case'
+
+  if use_linker:
+      call = LLVM_LINK
+      newargs = []
+      found_o = False
+      for arg in sys.argv[1:]:
+          if os.path.basename(sys.argv[0])=='arproxy.py':
+              if arg.endswith('.a'):
+                  newargs.append('-o=%s' % arg)
+              elif arg.endswith('.o'):
+                  newargs.append(arg)
+              else:
+                  pass
+              continue
+          if found_o:
+              newargs.append('-o=%s' % arg)
+              found_o = False
+              continue
+          if arg.startswith('-'):
+              if arg == '-o':
+                  found_o = True
+                  continue
+              prefix = arg.split('=')[0]
+              if prefix in ALLOWED_LINK_ARGS:
+                  newargs.append(arg)
+          elif arg.endswith('.so'):
+              continue # .so's do not exist yet, in many cases
+          else:
+              # not option, so just append
+              newargs.append(arg)
+  else:
+      call = CC
+      newargs = [ arg for arg in sys.argv[1:] if arg not in CC_ARG_SKIP ] + CC_ADDITIONAL_ARGS
+      if 'conftest.c' not in files:
+          newargs.append('-emit-llvm')
+          if CC=='llvm-gcc':
+              newargs.append('-c') 
+
+  print >> sys.stderr, "Running:", call, ' '.join(newargs)
+
+  os.execvp(call, [call] + newargs)
+except:
+  print 'Error in emmaken.py. Is the config file ~/.emscripten set up properly?'
+  raise
+