Repository navigation
Expand file tree
/
Copy pathmacos_build_python_instance.py
More file actions
242 lines (210 loc) · 8.78 KB
/
Copy pathmacos_build_python_instance.py
File metadata and controls
242 lines (210 loc) · 8.78 KB
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
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
from __future__ import annotations
import copy
from os import environ
from pathlib import Path
from build_python_instance_base import BuildPythonInstanceBase
from wheel_builder_utils import _remove_tree
from macos_venv_utils import create_macos_venvs
class MacOSBuildPythonInstance(BuildPythonInstanceBase):
def clone(self):
# Pattern for generating a deep copy of the current object state as a new build instance
cls = self.__class__
new = cls.__new__(cls)
new.__dict__ = copy.deepcopy(self.__dict__)
return new
def get_pixi_environment_name(self):
# The pixi environment name is the same as the manylinux version
# and is related to the environment setups defined in pixi.toml
# in the root of this git directory that contains these scripts.
return "macos"
def prepare_build_env(self) -> None:
# #############################################
# ### Setup build tools
self._build_type = "Release"
self._use_tbb: str = "OFF"
self._tbb_dir = None # Reset from the default
# The interpreter is provided; ensure basic tools are available
self.venv_paths()
self.update_venv_itk_build_configurations()
self.cmake_compiler_configurations.set(
"CMAKE_MAKE_PROGRAM:FILEPATH",
f"{self.package_env_config['NINJA_EXECUTABLE']}",
)
macosx_target = self.package_env_config.get("MACOSX_DEPLOYMENT_TARGET", "")
if macosx_target:
self.cmake_compiler_configurations.set(
"CMAKE_OSX_DEPLOYMENT_TARGET:STRING", macosx_target
)
target_arch = (
"arm64" if self.package_env_config["ARCH"] == "arm64" else "x86_64"
)
self.cmake_compiler_configurations.set(
"CMAKE_OSX_ARCHITECTURES:STRING", target_arch
)
itk_binary_build_name: str = (
self.build_dir_root
/ "build"
/ f"ITK-{self.py_env}-{self.get_pixi_environment_name()}_{target_arch}"
)
self.cmake_itk_source_build_configurations.set(
"ITK_BINARY_DIR:PATH", itk_binary_build_name
)
# Keep values consistent with prior quoting behavior
self.cmake_compiler_configurations.set("CMAKE_CXX_FLAGS:STRING", "-O3 -DNDEBUG")
self.cmake_compiler_configurations.set("CMAKE_C_FLAGS:STRING", "-O3 -DNDEBUG")
def post_build_fixup(self) -> None:
# delocate on macOS x86_64 only
if self.package_env_config["ARCH"] == "x86_64":
self.fixup_wheels()
def post_build_cleanup(self) -> None:
"""Clean build artifacts
Actions (leaving dist/ intact):
- remove oneTBB-prefix (symlink or dir)
- remove ITKPythonPackage/, tools/, _skbuild/, build/
- remove top-level *.egg-info
- remove ITK-* build tree and tarballs
- if ITK_MODULE_PREQ is set ("org/name@ref:org2/name2@ref2"), remove cloned module dirs
"""
base = Path(self.package_env_config["IPP_SOURCE_DIR"])
def rm(tree_path: Path):
try:
_remove_tree(tree_path)
except Exception:
pass
# 1) unlink oneTBB-prefix if it's a symlink or file
tbb_link = base / "oneTBB-prefix"
try:
if tbb_link.is_symlink() or tbb_link.is_file():
tbb_link.unlink(missing_ok=True) # type: ignore[arg-type]
elif tbb_link.exists():
# If it is a directory (not expected), remove tree
rm(tbb_link)
except Exception:
pass
# 2) standard build directories
for rel in ("ITKPythonPackage", "tools", "_skbuild", "build"):
rm(base / rel)
# 3) egg-info folders at top-level
for p in base.glob("*.egg-info"):
rm(p)
# 4) ITK build tree and tarballs
target_arch = (
"arm64" if self.package_env_config["ARCH"] == "arm64" else "x86_64"
)
for p in base.glob(f"ITK-*-{self.package_env_config}_{target_arch}"):
rm(p)
# Tarballs
for p in base.glob(f"ITKPythonBuilds-{self.package_env_config}*.tar.zst"):
rm(p)
# 5) Optional module prerequisites cleanup (ITK_MODULE_PREQ)
# Format: "InsightSoftwareConsortium/ITKModuleA@v1.0:Kitware/ITKModuleB@sha" -> remove repo names
itk_preq = self.package_env_config.get("ITK_MODULE_PREQ") or environ.get(
"ITK_MODULE_PREQ", ""
)
if itk_preq:
for entry in itk_preq.split(":"):
entry = entry.strip()
if not entry:
continue
try:
module_name = entry.split("@", 1)[0].split("/", 1)[1]
except Exception:
continue
rm(base / module_name)
def final_import_test(self) -> None:
self._final_import_test_fn(self.py_env, Path(self.dist_dir))
def fixup_wheel(self, filepath, lib_paths: str = "") -> None:
self.remove_apple_double_files()
# macOS fix-up with delocate (only needed for x86_64)
if self.package_env_config["ARCH"] != "arm64":
delocate_listdeps = (
self.venv_info_dict["venv_bin_path"] / "delocate-listdeps"
)
delocate_wheel = self.venv_info_dict["venv_bin_path"] / "delocate-wheel"
self.echo_check_call([str(delocate_listdeps), str(filepath)])
self.echo_check_call([str(delocate_wheel), str(filepath)])
def build_tarball(self):
self.create_posix_tarball()
def remove_apple_double_files(self):
try:
# Optional: clean AppleDouble files if tool is available
self.echo_check_call(
["dot_clean", str(self.package_env_config["IPP_SOURCE_DIR"].name)]
)
except Exception:
# dot_clean may not be available; continue without it
pass
def venv_paths(self) -> None:
"""Resolve virtualenv tool paths.
py_env may be a name under IPP_SOURCE_DIR/venvs or an absolute/relative path to a venv.
"""
# First determine if py_env is the full path to a python environment
_command_line_pip_executable = Path(self.py_env) / "bin" / "pip3"
if _command_line_pip_executable.exists():
venv_dir = Path(self.py_env)
local_pip_executable = _command_line_pip_executable
else:
venv_root_dir: Path = self.build_dir_root / "venvs"
_venvs_dir_list = create_macos_venvs(self.py_env, venv_root_dir)
if len(_venvs_dir_list) != 1:
raise ValueError(
f"Expected exactly one venv for {self.py_env}, found {_venvs_dir_list}"
)
venv_dir = _venvs_dir_list[0]
local_pip_executable = venv_dir / "bin" / "pip3"
self.echo_check_call([local_pip_executable, "install", "--upgrade", "pip"])
self.echo_check_call(
[
local_pip_executable,
"install",
"--upgrade",
"build",
"numpy",
"scikit-build-core",
# os-specific tools below
"delocate",
]
)
# Install dependencies
self.echo_check_call(
[
local_pip_executable,
"install",
"--upgrade",
"-r",
str(self.package_env_config["IPP_SOURCE_DIR"] / "requirements-dev.txt"),
]
)
self._pip_uninstall_itk_wildcard(local_pip_executable)
(
python_executable,
python_include_dir,
python_library,
pip_executable,
venv_bin_path,
venv_base_dir,
) = self.find_unix_exectable_paths(venv_dir)
self.venv_info_dict = {
"python_executable": python_executable,
"python_include_dir": python_include_dir,
"python_library": python_library,
"pip_executable": pip_executable,
"venv_bin_path": venv_bin_path,
"venv_base_dir": venv_base_dir,
}
def discover_python_venvs(
self, platform_os_name: str, platform_architechure: str
) -> list[str]:
names = []
# Discover virtualenvs under project 'venvs' folder
def _discover_ipp_venvs() -> list[str]:
venvs_dir = self.build_dir_root / "venvs"
if not venvs_dir.exists():
return []
names.extend([p.name for p in venvs_dir.iterdir() if p.is_dir()])
# Sort for stable order
return sorted(names)
default_py_envs = _discover_ipp_venvs()
return default_py_envs
def _final_import_test_fn(self, py_env, param):
pass