diff options
| author | Victor Chahuneau <vchahune@cs.cmu.edu> | 2012-06-14 09:27:53 -0400 |
|---|---|---|
| committer | Victor Chahuneau <vchahune@cab.ark.cs.cmu.edu> | 2012-06-14 09:28:49 -0400 |
| commit | c3619d27a6e64d2f13fc16d8959ad7ea70486eb8 (patch) | |
| tree | 57138a0c48a942ee5e05db4efd31ed9d7931c629 /python/setup.py | |
| parent | 3acdf1e4b37637d6df86a7b54fb0f1b0464c172b (diff) | |
Detect boost location before building the python module
Diffstat (limited to 'python/setup.py')
| -rw-r--r-- | python/setup.py | 13 |
1 files changed, 11 insertions, 2 deletions
diff --git a/python/setup.py b/python/setup.py index 756de088..49a1ec8f 100644 --- a/python/setup.py +++ b/python/setup.py @@ -1,13 +1,22 @@ from distutils.core import setup from distutils.extension import Extension from Cython.Distutils import build_ext +import os + +INC = ['..', 'src/', '../decoder', '../utils'] +LIB = ['../decoder', '../utils', '../mteval', '../klm/lm', '../klm/util'] + +BOOST_ROOT = os.getenv('BOOST_ROOT') +if BOOST_ROOT: + INC.append(os.path.join(BOOST_ROOT, 'include/')) + LIB.append(os.path.join(BOOST_ROOT, 'lib/')) ext_modules = [ Extension(name='_cdec', sources=['src/_cdec.pyx'], language='C++', - include_dirs=['..', 'src/', '../decoder', '../utils'], - library_dirs=['../decoder', '../utils', '../mteval', '../klm/lm', '../klm/util'], + include_dirs=INC, + library_dirs=LIB, libraries=['boost_program_options-mt', 'z', 'cdec', 'utils', 'mteval', 'klm', 'klm_util']) ] |