magsac/setup.py at master · danini/magsac · 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
"""Setup for pymagsac."""

import sys

from setuptools import setup, find_packages

try:
    from skbuild import setup
except ImportError:
    print(
        "Please update pip to pip 10 or greater, or a manually install the PEP 518 requirements in pyproject.toml",
        file=sys.stderr,
    )
    raise

cmake_args = []
debug = False
cfg = "Debug" if debug else "Release"
cmake_args += ["-DCMAKE_BUILD_TYPE=" + cfg]
cmake_args += ["-DCREATE_SAMPLE_PROJECT=OFF"]  # <-- Disable the sample project

setup(
    name="pymagsac",
    version="0.3.dev0",
    author="Daniel Barath, Dmytro Mishkin",
    author_email="barath.daniel@sztaki.hu",
    description="MAGSAC and MAGSAC++",
    long_description=open("README.md").read(),
    long_description_content_type="text/markdown",
    packages=find_packages("src"),
    license_file="LICENSE",
    package_dir={"": "src"},
    cmake_args=cmake_args,
    cmake_install_dir="src/pymagsac",
    cmake_install_target="install",
    zip_safe=False,
    install_requires="numpy",
)