-
Notifications
You must be signed in to change notification settings - Fork 23
/
setup.py
30 lines (24 loc) · 862 Bytes
/
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
"""Setup script for the DEODR project."""
import os
import re
import numpy as np
from Cython.Build import cythonize
from setuptools import setup
extensions = "deodr/differentiable_renderer_cython.pyx"
my_modules = cythonize(extensions, annotate=True, language="c++")
with open(os.path.join(os.path.dirname(__file__), "deodr", "__init__.py")) as fp:
for line in fp:
m = re.search(r'^\s*__version__\s*=\s*([\'"])([^\'"]+)\1\s*$', line)
if m:
version = m.group(2)
break
else:
raise RuntimeError("Unable to find own __version__ string")
print(f"version = {version}")
setup(
version=version,
url="https://github.com/martinResearch/DEODR",
data_files=[("C++", ["C++/DifferentiableRenderer.h"])],
ext_modules=my_modules, # additional source file(s)),
include_dirs=[np.get_include()],
)