setup.py

from __future__ import print_function
from setuptools import setup, Extension
import numpy
import subprocess
import sys
import os

try:
    from Cython.Distutils import build_ext
except ImportError:
    raise Exception("*** cython is needed to build this extension.")

cmdclass = {}
ext_modules = []


def getstatusoutput(command):
    process = subprocess.Popen(command, stdout=subprocess.PIPE)
    out, _ = process.communicate()
    return (process.returncode, out)


def find_dependencies():
    kw = {}

    flag_map = {'-I': 'include_dirs', '-L': 'library_dirs', '-l': 'libraries'}

    #
    # find atlas library (try pkgconfig, if that fails look at usual places)
    #

    print("looking for atlas library, trying pkg-config first...")

    # Try pky-config values for atlas
    status, output = getstatusoutput(
        ["pkg-config", "--libs", "--cflags", "blas-atlas"])

    # Try pkg-config values for atlas
    if status != 0:
        status, output = getstatusoutput(
            ["pkg-config", "--libs", "--cflags", "atlas"])

    if status != 0:

        print("looking for atlas library, trying hard-coded paths...")

        found = False

        for libdir in ['/usr/lib', '/usr/lib64', '/usr/lib/x86_64-linux-gnu',
                       '/usr/lib/i386-linux-gnu']:
            if os.path.isfile('%s/libatlas.so.3' % libdir):
                found = True
                break
        if not found:
            raise Exception('Failed to find libatlas.so.3 on your system.')

        kw.setdefault('libraries', []).append('%s/atlas.so.3' % libdir)
        kw.setdefault('libraries', []).append('%s/cblas.so.3' % libdir)
        kw.setdefault('libraries', []).append('%s/f77blas.so.3' % libdir)
        kw.setdefault('libraries', []).append('%s/lapack_atlas.so.3' % libdir)

        include_path = None
        include_found = False
        for include_dir in ['/usr/include/atlas',
                            '/usr/include/x86_64-linux-gnu/atlas']:
            if os.path.isdir(include_dir):
                include_found = True
                include_path = include_dir

        if not include_found:
            raise Exception('Failed to find atlas includes your system.')

        kw.setdefault('include_dirs', []).append(include_path)

        print("looking for atlas library, found it.")
    else:
        print("looking for atlas library, pkg-config found it")
        for token in output.split():
            token = token.decode('utf8')
            kw.setdefault(flag_map.get(token[:2]), []).append(token[2:])

    #
    # pkgconfig: kaldi-asr
    #

    status, output = getstatusoutput(
        ["pkg-config", "--libs", "--cflags", "kaldi-asr"])

    if status != 0:
        raise Exception("*** failed to find pkgconfig for kaldi-asr")

    for token in output.split():
        token = token.decode('utf8')

        prefix = token[:2]
        arg = token[2:]

        # print(repr(token))
        # print(repr(prefix))

        kw.setdefault(flag_map.get(prefix), []).append(arg)

    # print (repr(kw))

    return kw


# CFLAGS = -Wall -pthread -std=c++11 -DKALDI_DOUBLEPRECISION=0 -Wno-sign-compare \
#          -Wno-unused-local-typedefs -Winit-self -DHAVE_EXECINFO_H=1 -DHAVE_CXXABI_H -DHAVE_ATLAS \
#          `pkg-config --cflags kaldi-asr` -g

ext_modules += [
    Extension("kaldiasr.nnet3",
              sources=["kaldiasr/nnet3.pyx", "kaldiasr/nnet3_wrappers.cpp"],
              language="c++",
              extra_compile_args=['-Wall', '-pthread', '-std=c++11',
                                  '-DKALDI_DOUBLEPRECISION=0',
                                  '-Wno-sign-compare',
                                  '-Wno-unused-local-typedefs', '-Winit-self',
                                  '-DHAVE_EXECINFO_H=1', '-DHAVE_CXXABI_H',
                                  '-DHAVE_ATLAS', '-g'],
              **find_dependencies()),
    Extension("kaldiasr.gmm",
              sources=["kaldiasr/gmm.pyx", "kaldiasr/gmm_wrappers.cpp"],
              language="c++",
              extra_compile_args=['-Wall', '-pthread', '-std=c++11',
                                  '-DKALDI_DOUBLEPRECISION=0',
                                  '-Wno-sign-compare',
                                  '-Wno-unused-local-typedefs', '-Winit-self',
                                  '-DHAVE_EXECINFO_H=1', '-DHAVE_CXXABI_H',
                                  '-DHAVE_ATLAS', '-g'],
              **find_dependencies()),
]
cmdclass.update({'build_ext': build_ext})
for e in ext_modules:
    e.cython_directives = {'language_level': "3"}
setup(
    name='py-kaldi-asr',
    version='0.5.2',
    description='Simple Python/Cython interface to kaldi-asr nnet3/chain and gmm decoders',
    long_description=open('README.md').read(),
    author='Guenter Bartsch',
    author_email='guenter@zamia.org',
    maintainer='Guenter Bartsch',
    maintainer_email='guenter@zamia.org',
    url='https://github.com/gooofy/py-kaldi-asr',
    packages=['kaldiasr'],
    cmdclass=cmdclass,
    ext_modules=ext_modules,
    include_dirs=[numpy.get_include()],
    classifiers=[
        'Operating System :: POSIX :: Linux',
        'License :: OSI Approved :: Apache Software License',
        'Programming Language :: Python :: 2',
        'Programming Language :: Python :: 2.7',
        'Programming Language :: Python :: 3',
        'Programming Language :: Python :: 3.5',
        'Programming Language :: Cython',
        'Programming Language :: C++',
        'Intended Audience :: Developers',
        'Topic :: Software Development :: Libraries :: Python Modules',
        'Topic :: Multimedia :: Sound/Audio :: Speech'
    ],
    license='Apache',
    keywords='kaldi asr',
    include_package_data=True,
    install_requires=['numpy', 'cython', ],
)