setup.py

"""
Script for Cython code compilation.

ref: https://docs.python.org/3/distutils/apiref.html
"""

import os
import argparse

from os.path import join, exists, dirname, abspath, expanduser
from Cython.Build import cythonize
from numpy import get_include
from setuptools import setup, Extension, find_packages
import Cython.Compiler.Options
Cython.Compiler.Options.annotate = True

_wd = abspath(os.curdir)
base = abspath(dirname(__file__))
_incl = [get_include()]

# change to the project directory
os.chdir(base)

# first, create an Extension object w/ appropriate name and sources
ext = [
	# ====== DATA STRUCTURES ======
	Extension(
		name='datastructures.heap',
		sources=['datastructures/heap.pyx'],
		include_dirs=_incl,
		extra_compile_args=['-w'],
		extra_link_args=['-w']
	),
	# ====== CO-OCCURRENCE COUNT OF RELATIONS ======
	Extension(
		name='algorithms.relcooc._relcooc',
		sources=['algorithms/relcooc/_relcooc.pyx'],
		include_dirs=_incl,
		extra_compile_args=['-w'],
		extra_link_args=['-w']
	),
	# ====== KNOWLEDGE STREAM (KS) ======
	Extension(
		name='algorithms.mincostflow.ssp_helper',
		sources=['algorithms/mincostflow/ssp_helper.pyx'],
		include_dirs=_incl,
		extra_compile_args=['-w'],
		extra_link_args=['-w']
	),
	############# SM ###################################
	Extension(
		name='algorithms.sm.rel_closure',
		sources=['algorithms/sm/rel_closure.pyx'],
		include_dirs=_incl,
		extra_compile_args=['-w'],
		extra_link_args=['-w']
	),

	Extension(
		name='algorithms.sm.shortest_paths',
		sources=['algorithms/sm/shortest_paths.pyx'],
		include_dirs=_incl,
		extra_compile_args=['-w'],
		extra_link_args=['-w']
	),

	Extension(
		name='algorithms.sm.rel_closure_2',
		sources=['algorithms/sm/rel_closure_2.pyx'],
		include_dirs=_incl,
		extra_compile_args=['-w'],
		extra_link_args=['-w']
	),
    Extension(
		name='algorithms.sm.yenKSP',
		sources=['algorithms/sm/yenKSP.pyx'],
		include_dirs=_incl,
		extra_compile_args=['-w'],
		extra_link_args=['-w']
	),
    Extension(
		name='algorithms.sm.extract_paths',
		sources=['algorithms/sm/extract_paths.pyx'],
		include_dirs=_incl,
		extra_compile_args=['-w'],
		extra_link_args=['-w']
	),
    Extension(
		name='algorithms.sm.extract_paths_par',
		sources=['algorithms/sm/extract_paths_par.pyx'],
		include_dirs=_incl,
		extra_compile_args=['-w'],
		extra_link_args=['-w']
	),
	# ====== RELATIONAL KNOWLEDGE LINKER (KL-REL) ======
	Extension(
		name='algorithms.relklinker.rel_closure',
		sources=['algorithms/relklinker/rel_closure.pyx'],
		include_dirs=_incl,
		extra_compile_args=['-w'],
		extra_link_args=['-w']
	),
	# ====== KNOWLEDGE LINKER (KL) ======
	Extension(
		name='algorithms.klinker.closure',
		sources=['algorithms/klinker/closure.pyx'],
		include_dirs=_incl,
		extra_compile_args=['-w'],
		extra_link_args=['-w']
	),
	# ====== PREDICATE PATH MINING (PREDPATH) ======
	Extension(
		name='algorithms.pathenum',
		sources=['algorithms/pathenum.pyx'],
		include_dirs=_incl,
		extra_compile_args=['-w'],
		extra_link_args=['-w'],
		language='c++'
	),
	# ====== PATH RANKING ALGORITHM (PRA) ======
	Extension(
		name='algorithms.pra.pra_helper',
		sources=['algorithms/pra/pra_helper.pyx'],
		include_dirs=_incl,
		extra_compile_args=['-w'],
		extra_link_args=['-w'],
		language='c++'
	),
	# ====== KATZ (KZ) ======
	Extension(
		name='algorithms.linkpred.pathenum',
		sources=['algorithms/linkpred/pathenum.pyx'],
		include_dirs=_incl,
		extra_compile_args=['-w'],
		extra_link_args=['-w'],
		language='c++'
	),
	# ====== SIMRANK ======
	Extension(
		name='algorithms.linkpred.simrank_helper',
		sources=['algorithms/linkpred/simrank_helper.pyx'],
		include_dirs=_incl,
		extra_compile_args=['-w'],
		extra_link_args=['-w'],
		language='c++'
	),
]

kwargs = dict(
    name="streamminer",
    description='Stream Miner Algorithm',
    version='0.1.1',
    author='Himanshu Ahuja, Alex Michels, original by Prashant Shiralkar and others (see CONTRIBUTORS.md)',
    author_email='',
    packages=[
        'datastructures', 'algorithms', 'algorithms.mincostflow',
        'algorithms.relklinker', 'algorithms.klinker'
    ],
    include_package_data=True,
    install_requires=[
        'numpy',
        'scipy',
    ],
    entry_points={
        'console_scripts': [
            'sm = streamminer:main'
        ]
    },
    ext_modules=cythonize(ext)
)

parser = argparse.ArgumentParser(description=__file__, add_help=False)

if __name__ == '__main__':
    args, rest = parser.parse_known_args()
    kwargs['script_args'] = rest
    setup(**kwargs)

# back to the working directory
os.chdir(_wd)