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 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193
|
# ----------------------------------------------------------------------------
# - Open3D: www.open3d.org -
# ----------------------------------------------------------------------------
# Copyright (c) 2018-2024 www.open3d.org
# SPDX-License-Identifier: MIT
# ----------------------------------------------------------------------------
import os
import sys
import platform
import ctypes
from setuptools import setup, find_packages
from setuptools.command.install import install as _install
from wheel.bdist_wheel import bdist_wheel as _bdist_wheel
data_files_spec = [
("share/jupyter/nbextensions/open3d", "open3d/nbextension", "*.*"),
("share/jupyter/labextensions/open3d", "open3d/labextension", "**"),
("share/jupyter/labextensions/open3d", ".", "install.json"),
("etc/jupyter/nbconfig/notebook.d", ".", "open3d.json"),
]
if "@BUILD_JUPYTER_EXTENSION@" == "ON":
try:
from jupyter_packaging import (
create_cmdclass,
install_npm,
ensure_targets,
combine_commands,
)
# ipywidgets and jupyterlab are required to package JS code properly.
# They are not used in setup.py.
import ipywidgets # noqa # pylint: disable=unused-import
import jupyterlab # noqa # pylint: disable=unused-import
except ImportError as error:
print(error.__class__.__name__ + ": " + error.message)
print("Run `pip install -r requirements-jupyter-build.txt`.")
here = os.path.dirname(os.path.abspath(__file__))
js_dir = os.path.join(here, "js")
# Representative files that should exist after a successful build.
js_targets = [
os.path.join(js_dir, "dist", "index.js"),
]
cmdclass = create_cmdclass("jsdeps", data_files_spec=data_files_spec)
cmdclass["jsdeps"] = combine_commands(
install_npm(js_dir, npm=["yarn"], build_cmd="build:prod"),
ensure_targets(js_targets),
)
else:
cmdclass = {}
# Force platform specific wheel.
# https://stackoverflow.com/a/45150383/1255535
class bdist_wheel(_bdist_wheel):
def finalize_options(self):
_bdist_wheel.finalize_options(self)
self.root_is_pure = False
# https://github.com/Yelp/dumb-init/blob/57f7eebef694d780c1013acd410f2f0d3c79f6c6/setup.py#L25
def get_tag(self):
python, abi, plat = _bdist_wheel.get_tag(self)
if plat[:5] == "linux":
libc = ctypes.CDLL("libc.so.6")
libc.gnu_get_libc_version.restype = ctypes.c_char_p
GLIBC_VER = libc.gnu_get_libc_version().decode("utf8").split(".")
plat = f"manylinux_{GLIBC_VER[0]}_{GLIBC_VER[1]}{plat[5:]}"
elif plat[:6] == "macosx":
# If the Python interpreter is an universal2 app the resulting wheel is tagged as
# universal2 instead of the current architecture. This is a workaround to fix it.
plat = plat.replace("universal2", platform.machine())
return python, abi, plat
cmdclass["bdist_wheel"] = bdist_wheel
# Force use of "platlib" dir for auditwheel to recognize this is a non-pure
# build
# http://lxr.yanyahua.com/source/llvmlite/setup.py
class install(_install):
def finalize_options(self):
_install.finalize_options(self)
self.install_libbase = self.install_platlib
self.install_lib = self.install_platlib
cmdclass["install"] = install
# Read requirements.
with open("requirements.txt", "r") as f:
install_requires = [line.strip() for line in f.readlines() if line]
# Read requirements for ML.
if "@BUNDLE_OPEN3D_ML@" == "ON":
with open("@OPEN3D_ML_ROOT@/requirements.txt", "r") as f:
install_requires += [line.strip() for line in f.readlines() if line]
entry_points = {
"console_scripts": ["open3d = @PYPI_PACKAGE_NAME@.tools.cli:main",]
}
if sys.platform != "darwin": # Remove check when off main thread GUI works
entry_points.update({
"tensorboard_plugins": [
"Open3D = @PYPI_PACKAGE_NAME@.visualization.tensorboard_plugin"
".plugin:Open3DPlugin",
]
})
classifiers = [
# https://pypi.org/pypi?%3Aaction=list_classifiers
"Development Status :: 3 - Alpha",
"Environment :: MacOS X",
"Environment :: Win32 (MS Windows)",
"Environment :: X11 Applications",
"Environment :: GPU :: NVIDIA CUDA",
"Intended Audience :: Developers",
"Intended Audience :: Education",
"Intended Audience :: Other Audience",
"Intended Audience :: Science/Research",
"License :: OSI Approved :: MIT License",
"Natural Language :: English",
"Operating System :: POSIX :: Linux",
"Operating System :: MacOS :: MacOS X",
"Operating System :: Microsoft :: Windows",
"Programming Language :: C",
"Programming Language :: C++",
"Programming Language :: Python :: 3",
"Programming Language :: Python :: 3.8",
"Programming Language :: Python :: 3.9",
"Programming Language :: Python :: 3.10",
"Programming Language :: Python :: 3.11",
"Programming Language :: Python :: 3.12",
"Topic :: Education",
"Topic :: Multimedia :: Graphics :: 3D Modeling",
"Topic :: Multimedia :: Graphics :: 3D Rendering",
"Topic :: Multimedia :: Graphics :: Capture",
"Topic :: Multimedia :: Graphics :: Graphics Conversion",
"Topic :: Multimedia :: Graphics :: Viewers",
"Topic :: Scientific/Engineering",
"Topic :: Scientific/Engineering :: Mathematics",
"Topic :: Scientific/Engineering :: Visualization",
"Topic :: Software Development :: Libraries :: Python Modules",
"Topic :: Utilities",
]
name = "@PYPI_PACKAGE_NAME@"
with open("README.rst") as readme:
long_description = readme.read()
# open3d-cpu wheel for Linux x86_64
if sys.platform.startswith("linux") and platform.machine() in (
'i386', 'x86_64', 'AMD64'
) and "@BUILD_CUDA_MODULE@" == "OFF" and "@BUILD_SYCL_MODULE@" == "OFF":
name += "-cpu"
long_description += ("\n\nThis wheel only contains CPU functionality. "
"Use the open3d wheel for full functionality.")
classifiers.remove("Environment :: GPU :: NVIDIA CUDA")
setup_args = dict(
name=name,
version="@PROJECT_VERSION@",
python_requires=">=3.8",
include_package_data=True,
install_requires=install_requires,
packages=find_packages(),
entry_points=entry_points,
zip_safe=False,
cmdclass=cmdclass,
author="Open3D Team",
author_email="@PROJECT_EMAIL@",
url="@PROJECT_HOMEPAGE_URL@",
project_urls={
"Documentation": "@PROJECT_DOCS@",
"Source code": "@PROJECT_CODE@",
"Issues": "@PROJECT_ISSUES@",
},
classifiers=classifiers,
keywords="3D reconstruction point cloud mesh RGB-D visualization",
license="MIT",
description="@PROJECT_DESCRIPTION@",
long_description=long_description,
long_description_content_type="text/x-rst",
# Metadata below is valid but currently ignored by pip (<=v23)
obsoletes=["open3d_python"],
provides=["open3d", "open3d_cpu"], # For open3d-cpu
)
setup(**setup_args)
|