nes-py/__setup.py at master · FragileTech/nes-py · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
"""The setup script for installing and distributing the nes-py package."""
from glob import glob
import os
from setuptools import setup, find_packages, Extension


# set the compiler for the C++ framework
os.environ["CC"] = "g++"
os.environ["CCX"] = "g++"


# read the contents from the README file
with open("README.md") as README_file:
    README = README_file.read()


# The prefix name for the .so library to build. It will follow the format
# lib_nes_env.*.so where the * changes depending on the build system
LIB_NAME = "nes_py.lib_nes_env"
# The source files for building the extension. Globs locate all the cpp files
# used by the LaiNES subproject. MANIFEST.in has to include the blanket
# "cpp" directory to ensure that the .inc file gets included too
SOURCES = glob("nes_py/nes/src/*.cpp") + glob("nes_py/nes/src/mappers/*.cpp")
# The directory pointing to header files used by the LaiNES cpp files.
# This directory has to be included using MANIFEST.in too to include the
# headers with sdist
INCLUDE_DIRS = ["nes_py/nes/include"]
# Build arguments to pass to the compiler
EXTRA_COMPILE_ARGS = ["-std=c++1y", "-march=native", "-pipe", "-O3", "-pthread"]
# The official extension using the name, source, headers, and build args
LIB_NES_ENV = Extension(
    LIB_NAME,
    sources=SOURCES,
    include_dirs=INCLUDE_DIRS,
    extra_compile_args=EXTRA_COMPILE_ARGS,
    extra_link_args=["-pthread"],
)


setup(
    name="fragile_nes_py",
    version="10.0.0",
    description="An NES Emulator and OpenAI Gym interface",
    long_description=README,
    long_description_content_type="text/markdown",
    keywords="NES Emulator OpenAI-Gym",
    classifiers=[
        "Development Status :: 5 - Production/Stable",
        "Intended Audience :: Developers",
        "Intended Audience :: Science/Research",
        "License :: OSI Approved :: MIT License",
        "Operating System :: MacOS :: MacOS X",
        "Operating System :: POSIX :: Linux",
        "Operating System :: Microsoft :: Windows",
        "Programming Language :: C++",
        "Programming Language :: Python :: 3 :: Only",
        "Programming Language :: Python :: 3.5",
        "Programming Language :: Python :: 3.6",
        "Programming Language :: Python :: 3.7",
        "Programming Language :: Python :: 3.8",
        "Programming Language :: Python :: 3.9",
        "Topic :: Games/Entertainment",
        "Topic :: Software Development :: Libraries :: Python Modules",
        "Topic :: System :: Emulators",
    ],
    url="https://github.com/FragileTech/nes-py",
    author="Christian Kauten",
    author_email="kautencreations@gmail.com",
    license="MIT",
    packages=find_packages(exclude=["tests", "*.tests", "*.tests.*"]),
    ext_modules=[LIB_NES_ENV],
    zip_safe=False,
    install_requires=[
        "fragile-gym",
        "numpy>=1.18.5",
        "pyglet<=1.5.11,>=1.4.0",
        "tqdm>=4.48.2",
    ],
    entry_points={
        "console_scripts": [
            "nes_py = nes_py.app.cli:main",
        ],
    },
)