llvmpy/setup.py
2008-06-07 13:44:25 +00:00

97 lines
2.4 KiB
Python

#!/usr/bin/env python
import sys, os
from distutils.core import setup, Extension
def _run(cmd):
return os.popen(cmd).read().rstrip()
def get_libs_and_objs(llvm_config, components):
parts = _run(llvm_config + ' --libs ' + ' '.join(components)).split()
libs = []
objs = []
for part in parts:
if part.startswith('-l'):
libs.append(part[2:])
else:
assert part.endswith('.o')
# objs.append(part[:-2])
objs.append(part) # eh, looks like we need the .o after all
return (libs, objs)
def get_llvm_config():
# get from command-line, or use default
lc = 'llvm-config'
for i in xrange(0, len(sys.argv)):
arg = sys.argv[i]
if arg.startswith('--llvm-config='):
del sys.argv[i]
lc = arg.split('=')[1]
# see if it works
version = _run(lc + ' --version')
if version == '':
return (lc, False) # didn't work
return (lc, True)
def call_setup(llvm_config):
incdir = _run(llvm_config + ' --includedir')
libdir = _run(llvm_config + ' --libdir')
ldflags = _run(llvm_config + ' --ldflags')
libs_core, objs_core = get_libs_and_objs(llvm_config,
['core', 'analysis', 'scalaropts', 'executionengine',
'jit', 'native'])
std_libs = [ 'pthread', 'dl', 'm' ]
ext_core = Extension(
'llvm._core',
['llvm/_core.c', 'llvm/wrap.c', 'llvm/extra.cpp'],
define_macros = [
('__STDC_LIMIT_MACROS', None),
('_GNU_SOURCE', None)],
include_dirs = [incdir],
library_dirs = [libdir],
libraries = std_libs + libs_core,
extra_objects = objs_core)
setup(
name='llvm-py',
version='0.2',
description='Python Bindings for LLVM',
author='Mahadevan R',
author_email='mdevan.foobar@gmail.com',
url='http://code.google.com/p/llvm-py/',
packages=['llvm'],
py_modules = [ 'llvm.core' ],
ext_modules = [ ext_core ],)
def main():
# get llvm config
llvm_config, is_good = get_llvm_config()
if is_good:
print "Using llvm-config=" + llvm_config
else:
print "Cannot invoke llvm-config (tried '%s')." % llvm_config
print "Try again with --llvm-config=/path/to/llvm-config."
return 1
# setup
call_setup(llvm_config)
# done
return 0
ev = main()
sys.exit(ev)