| from __future__ import print_function |
| from distutils.core import setup |
| from distutils.extension import Extension |
|
|
| import sys |
| import os |
|
|
| mosesdir = os.path.abspath('../../') |
| includes = [mosesdir, os.path.join(mosesdir, 'moses'), os.path.join(mosesdir, 'OnDiskPt'), os.path.join(mosesdir, 'moses/TranslationModel')] |
| libdir = os.path.join(mosesdir, 'lib') |
|
|
| |
| |
| available_switches = ['--with-cmph', '--moses-lib', '--cython', '--max-factors', '--max-kenlm-order'] |
| with_cmph = False |
| defines = {'MAX_NUM_FACTORS':'1'} |
| suffix = '.cpp' |
| cmdcls = {} |
| while sys.argv[-1].split('=')[0] in available_switches: |
| param = sys.argv.pop().split('=') |
| if param[0] == '--with-cmph': |
| with_cmph = True |
| if param[0] == '--moses-lib': |
| libdir = param[1] |
| if param[0] == '--cython': |
| print('I will be cythoning your pyx files...', file=sys.stderr) |
| try: |
| from Cython.Distutils import build_ext |
| suffix = '.pyx' |
| cmdcls['build_ext'] = build_ext |
| except ImportError: |
| print('You do not seem to have Cython installed') |
| if param[0] == '--max-factors': |
| defines['MAX_NUM_FACTORS'] = param[1] |
| if param[0] == '--max-kenlm-order': |
| defines['KENLM_MAX_ORDER'] = param[1] |
|
|
| print('mosesdir=%s\nincludes=%s\nlibdir=%s\ncmph=%s' % (mosesdir, includes, libdir, with_cmph), file=sys.stderr) |
|
|
| |
| basic=[] |
| moses = ['moses'] |
| additional = [] |
|
|
| if with_cmph: |
| additional.append('cmph') |
|
|
| extensions = [ |
| Extension(name='moses.dictree', |
| sources=['moses/dictree' + suffix], |
| include_dirs = ['.'] + includes, |
| library_dirs = [libdir], |
| runtime_library_dirs = [libdir], |
| libraries= basic + moses + additional, |
| language='c++', |
| extra_compile_args=['-O3'] + ['-D%s=%s' % (k, v) for k, v in defines.iteritems()] |
| ), |
| ] |
|
|
| setup( |
| cmdclass = cmdcls, |
| name = 'moses', |
| ext_modules= extensions, |
| packages = ['moses'], |
| package_dir = {'':'.'}, |
|
|
| ) |
|
|
