# Copyright (c) 2020, NVIDIA CORPORATION.  All rights reserved.
#
# NVIDIA CORPORATION and its licensors retain all intellectual property
# and proprietary rights in and to this software, related documentation
# and any modifications thereto.  Any use, reproduction, disclosure or
# distribution of this software and related documentation without an express
# license agreement from NVIDIA CORPORATION is strictly prohibited.

import nvdiffrast
import setuptools
import os

with open("README.md", "r") as fh:
    long_description = fh.read()

setuptools.setup(
    name="nvdiffrast",
    version=nvdiffrast.__version__,
    author="Samuli Laine",
    author_email="slaine@nvidia.com",
    description="nvdiffrast - modular primitives for high-performance differentiable rendering",
    long_description=long_description,
    long_description_content_type="text/markdown",
    url="https://github.com/NVlabs/nvdiffrast",
    packages=setuptools.find_packages(),
    package_data={
        "nvdiffrast": [
            "common/*.h",
            "common/*.inl",
            "common/*.cu",
            "common/*.cpp",
            "common/cudaraster/*.hpp",
            "common/cudaraster/impl/*.cpp",
            "common/cudaraster/impl/*.hpp",
            "common/cudaraster/impl/*.inl",
            "common/cudaraster/impl/*.cu",
            "lib/*.h",
            "torch/*.h",
            "torch/*.inl",
            "torch/*.cpp",
            "tensorflow/*.cu",
        ]
        + (["lib/*.lib"] if os.name == "nt" else [])
    },
    include_package_data=True,
    install_requires=[
        "numpy"
    ],  # note: can't require torch here as it will install torch even for a TensorFlow container
    classifiers=[
        "Programming Language :: Python :: 3",
        "Operating System :: OS Independent",
    ],
    python_requires=">=3.6",
)