Spaces:
Runtime error
Runtime error
File size: 4,486 Bytes
a550c4e | 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 | #!/usr/bin/env python3
# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: Apache-2.0
"""Setup script for correct_motion standalone package."""
import os
import shutil
import subprocess
import sys
from pathlib import Path
from setuptools import Extension, setup
from setuptools.command.build_ext import build_ext
class CMakeExtension(Extension):
def __init__(self, name, sourcedir=""):
Extension.__init__(self, name, sources=[])
self.sourcedir = os.path.abspath(sourcedir)
class CMakeBuild(build_ext):
def run(self):
try:
subprocess.check_output(["cmake", "--version"])
except OSError:
raise RuntimeError("CMake must be installed to build this package")
for ext in self.extensions:
self.build_extension(ext)
def build_extension(self, ext):
# import pdb; pdb.set_trace() # Debug build process
extdir = os.path.abspath(os.path.dirname(self.get_ext_fullpath(ext.name)))
cmake_args = [
f"-DCMAKE_LIBRARY_OUTPUT_DIRECTORY={extdir}",
f"-DPYTHON_EXECUTABLE={sys.executable}",
]
cfg = "Debug" if self.debug else "Release"
build_args = ["--config", cfg]
cmake_args += [f"-DCMAKE_BUILD_TYPE={cfg}"]
use_mingw = False
mingw_bin = None
if sys.platform == "win32":
generator = os.environ.get("CMAKE_GENERATOR", "")
if generator:
cmake_args = ["-G", generator] + cmake_args
if "mingw" in generator.lower():
use_mingw = True
else:
cmake_args += [f"-DCMAKE_LIBRARY_OUTPUT_DIRECTORY_{cfg.upper()}={extdir}"]
else:
# Try MinGW Makefiles as default on Windows
try:
subprocess.check_output(["g++", "--version"], stderr=subprocess.STDOUT)
use_mingw = True
cmake_args = ["-G", "MinGW Makefiles"] + cmake_args
build_args = [] # MinGW Makefiles do not accept --config
except (OSError, subprocess.CalledProcessError):
# If g++ is not found, let CMake use its default (Visual Studio)
cmake_args += [f"-DCMAKE_LIBRARY_OUTPUT_DIRECTORY_{cfg.upper()}={extdir}"]
if use_mingw:
gxx_path = shutil.which("g++")
if gxx_path:
mingw_bin = Path(gxx_path).parent
else:
build_args += ["--", "-j4"]
env = os.environ.copy()
env["CXXFLAGS"] = f'{env.get("CXXFLAGS", "")} -DVERSION_INFO=\\"{self.distribution.get_version()}\\"'
if not os.path.exists(self.build_temp):
os.makedirs(self.build_temp)
subprocess.check_call(["cmake", ext.sourcedir] + cmake_args, cwd=self.build_temp, env=env)
subprocess.check_call(["cmake", "--build", "."] + build_args, cwd=self.build_temp)
if use_mingw and mingw_bin is not None:
runtime_libs = [
"libstdc++-6.dll",
"libgcc_s_seh-1.dll",
"libwinpthread-1.dll",
]
extdir_path = Path(extdir)
extdir_path.mkdir(parents=True, exist_ok=True)
for lib_name in runtime_libs:
src_path = mingw_bin / lib_name
if src_path.exists():
shutil.copy2(src_path, extdir_path / lib_name)
else:
self.announce(
f"Warning: Expected MinGW runtime DLL '{lib_name}' not found next to g++ (looked in {mingw_bin}). "
"The built extension may fail to import if the DLL is not on PATH.",
level=3,
)
setup(
name="motion_correction",
version="1.0.0",
author="NVIDIA",
description="Standalone correct_motion function",
long_description="",
packages=["motion_correction"],
package_dir={"": "python"},
ext_modules=[CMakeExtension("motion_correction._motion_correction")],
cmdclass={"build_ext": CMakeBuild},
zip_safe=False,
python_requires=">=3.8",
install_requires=[
"torch>=1.10.0",
"numpy>=1.19.0",
# 'cmake' # can install this via pip if the windows system does not have it. But need to run this by yourself before build, not in here.
],
)
|