setup.py

# -*- coding: utf-8 -*-
import os
import sys
from itertools import chain

from setuptools import setup, Extension, find_packages

try:
    from Cython.Build import cythonize
except ImportError:
    # A 'cythonize' stub is needed so that build, develop and install can
    # start before Cython is installed.
    cythonize = lambda extensions, **kwargs: extensions  # noqa
    USE_CYTHON = False
else:
    USE_CYTHON = True


_CYTHONIZE_WITH_COVERAGE = os.environ.get("_CYTHONIZE_WITH_COVERAGE", False)

if _CYTHONIZE_WITH_COVERAGE and not USE_CYTHON:
    raise RuntimeError(
        "Configured to build using Cython "
        "and coverage but Cython not available."
    )


def read(filename):
    with open(filename, 'r') as file_handle:
        return file_handle.read()


def get_version(version_tuple):
    if not isinstance(version_tuple[-1], int):
        return '.'.join(map(str, version_tuple[:-1])) + version_tuple[-1]
    return '.'.join(map(str, version_tuple))


init = os.path.join(os.path.dirname(__file__), 'implot', '__init__.py')
version_line = list(filter(lambda l: l.startswith('VERSION'), open(init)))[0]

VERSION = get_version(eval(version_line.split('=')[-1]))
README = os.path.join(os.path.dirname(__file__), 'README.md')


if sys.platform in ('cygwin', 'win32'):  # windows
    # note: `/FI` means forced include in VC++/VC
    # note: may be obsoleted in future if ImGui gets patched
    os_specific_flags = ['/FIpy_imconfig.h']
    # placeholder for future
    os_specific_macros = []
else:  # OS X and Linux
    # note: `-include` means forced include in GCC/clang
    # note: may be obsoleted in future if ImGui gets patched
    # placeholder for future
    os_specific_flags = ['-includeconfig-cpp/py_imconfig.h']
    os_specific_macros = []


if _CYTHONIZE_WITH_COVERAGE:
    compiler_directives = {
        'linetrace': True,
    }
    cythonize_opts = {
        'gdb_debug': True,
    }
    general_macros = [('CYTHON_TRACE_NOGIL', '1')]
else:
    compiler_directives = {}
    cythonize_opts = {}
    general_macros = []


def extension_sources(path):
    sources = ["{0}{1}".format(path, '.pyx' if USE_CYTHON else '.cpp')]

    if not USE_CYTHON:
        # note: Cython will pick these files automatically but when building
        #       a plain C++ sdist without Cython we need to explicitly mark
        #       these files for compilation and linking.
        sources += [
            'imgui-cpp/imgui.cpp',
            'imgui-cpp/imgui_draw.cpp',
            'imgui-cpp/imgui_demo.cpp',
            'imgui-cpp/imgui_widgets.cpp',
            'imgui-cpp/imgui_tables.cpp',
            'implot-cpp/implot.cpp',
            'implot-cpp/implot_items.cpp',
            'implot-cpp/implot_demo.cpp',
            'config-cpp/py_imconfig.cpp'
        ]

    return sources


def backend_extras(*requirements):
    """Construct list of requirements for backend integration.

    All built-in backends depend on PyOpenGL so add it as default requirement.
    """
    return ["PyOpenGL"] + list(requirements)

EXTRAS_REQUIRE = {
    'Cython':  ['Cython>=0.24,<0.30'],
    'cocos2d': backend_extras(
        "cocos2d",
        "pyglet>=1.5.6; sys_platform == 'darwin'",
    ),
    'sdl2': backend_extras('PySDL2'),
    'glfw': backend_extras('glfw'),
    'pygame': backend_extras('pygame'),
    'opengl': backend_extras(),
    'pyglet': backend_extras(
        "pyglet; sys_platform != 'darwin'",
        "pyglet>=1.5.6; sys_platform == 'darwin'",
    )
}

def imgui_location():
    imgui_path = ''
    try:
        import imgui
    except ImportError:
        print('pyimgui module is required to build pyimplot')
        exit(1)
    finally:
        imgui_path = imgui.__path__[0]
        print(imgui_path)
    return imgui_path

# construct special 'full' extra that adds requirements for all built-in
# backend integrations and additional extra features.
EXTRAS_REQUIRE['full'] = list(set(chain(*EXTRAS_REQUIRE.values())))

EXTENSIONS = [
    # Extension(
    #     "imgui.core", extension_sources("imgui/core"),
    #     extra_compile_args=os_specific_flags,
    #     define_macros=[
    #         # note: for raising custom exceptions directly in ImGui code
    #         ('PYIMGUI_CUSTOM_EXCEPTION', None)
    #     ] + os_specific_macros + general_macros,
    #     include_dirs=['imgui', 'config-cpp', 'imgui-cpp', 'ansifeed-cpp'],
    # ),
    # Extension(
    #     "imgui.internal", extension_sources("imgui/internal"),
    #     extra_compile_args=os_specific_flags,
    #     define_macros=[
    #         # note: for raising custom exceptions directly in ImGui code
    #         ('PYIMGUI_CUSTOM_EXCEPTION', None)
    #     ] + os_specific_macros + general_macros,
    #     include_dirs=['imgui', 'config-cpp', 'imgui-cpp', 'ansifeed-cpp'],
    # ),
    Extension(
        "implot.plot", extension_sources("implot/plot"),
        extra_compile_args=os_specific_flags,
        # XXX: handle Windows/MacOS
        extra_link_args=["-Wl,-rpath,$ORIGIN/implotcpp"],
        define_macros=[
            # note: for raising custom exceptions directly in ImGui code
            ('PYIMGUI_CUSTOM_EXCEPTION', None)
        ] + os_specific_macros + general_macros,
        # include_dirs=['imgui', 'implot-cpp'],
        # include_dirs=['implot', 'config-cpp', 'imgui-cpp', 'ansifeed-cpp', 'implot-cpp'],
        # include_dirs=['implot', 'config-cpp', 'imgui-cpp', 'implot-cpp'],
        # include_dirs=['implot', 'config-cpp', 'imgui-cpp', 'implot-cpp'],
        include_dirs=['implot', 'config-cpp', imgui_location()+'/imguicpp', 'implot-cpp'],
        library_dirs=["implot/implotcpp", imgui_location()+'/imguicpp'],
        # order matters; libimplot needs to preceede libimgui!
        libraries=["implot", "imgui"],
    ),
]


setup(
    name='implot',
    version=VERSION,
    packages=find_packages('.'),

    author=u'Hinko Kocevar',
    author_email='hinxx@protonmail.com',

    description="Cython-based Python bindings for ImPlot",
    long_description=read(README),
    long_description_content_type="text/markdown",

    url="https://github.com/hinxx/pyimplot",

    ext_modules=cythonize(
        EXTENSIONS,
        compiler_directives=compiler_directives, **cythonize_opts
    ),
    extras_require=EXTRAS_REQUIRE,
    include_package_data=True,

    license='BSD',
    classifiers=[
        'Intended Audience :: Developers',
        'License :: OSI Approved :: BSD License',

        'Programming Language :: Cython',
        'Programming Language :: Python :: 2',
        'Programming Language :: Python :: 2.7',

        'Programming Language :: Python :: 3',
        'Programming Language :: Python :: 3.3',
        'Programming Language :: Python :: 3.4',
        'Programming Language :: Python :: 3.5',
        'Programming Language :: Python :: 3.6',

        'Programming Language :: Python :: Implementation :: CPython',
        'Programming Language :: Cython',

        'Operating System :: MacOS :: MacOS X',
        'Operating System :: POSIX :: Linux',
        'Operating System :: Microsoft :: Windows',

        'Topic :: Games/Entertainment',
    ],
)