forked from hungpham2511/toppra
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathsetup.py
51 lines (43 loc) · 1.67 KB
/
setup.py
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
from setuptools import setup, Extension
from Cython.Distutils import build_ext
from Cython.Build import cythonize
import numpy as np
NAME = "toppra"
VERSION = "0.2"
DESCR = "An implementation of TOPP-RA (TOPP via Reachability Analysis) for time-parametrizing" \
"trajectories for robots subject to kinematic (velocity and acceleration) and dynamic" \
"(torque) constraints. Some other kinds of constraints are also supported."
URL = "https://github.com/hungpham2511/toppra"
REQUIRES = ['numpy', 'cython', "scipy>0.18", "coloredlogs", "enum", "cvxpy>=0.4.11, <1.0.0", "pytest"]
AUTHOR = "Hung Pham"
EMAIL = "[email protected]"
LICENSE = "MIT"
SRC_DIR = "toppra"
PACKAGES = ["toppra",
"toppra.constraint",
"toppra.algorithm",
"toppra.algorithm.reachabilitybased",
"toppra.solverwrapper"]
ext_1 = Extension(SRC_DIR + "._CythonUtils",
[SRC_DIR + "/_CythonUtils.pyx"],
libraries=[],
include_dirs=[np.get_include()])
ext_2 = Extension(SRC_DIR + ".solverwrapper.cy_seidel_solverwrapper",
[SRC_DIR + "/solverwrapper/cy_seidel_solverwrapper.pyx"],
extra_compile_args=['-O1'],
include_dirs=[np.get_include()])
EXTENSIONS = [ext_1, ext_2]
if __name__ == "__main__":
setup(install_requires=REQUIRES,
packages=PACKAGES,
zip_safe=False,
name=NAME,
version=VERSION,
description=DESCR,
author=AUTHOR,
author_email=EMAIL,
url=URL,
license=LICENSE,
cmdclass={"build_ext": build_ext},
ext_modules=cythonize(EXTENSIONS)
)