From ca1a7e5809dd9dbc1ac3486ff601b5eb1b58184b Mon Sep 17 00:00:00 2001
From: Simon Rit <simon.rit@creatis.insa-lyon.fr>
Date: Thu, 5 Sep 2024 17:24:32 +0200
Subject: [PATCH] ENH: Only extract required directories from the
 ITKPythonBuilds tar

The selection of directories is based on the compiled python versions.
---
 ...n-download-cache-and-build-module-wheels.sh | 18 ++++++++++++++++--
 1 file changed, 16 insertions(+), 2 deletions(-)

diff --git a/scripts/macpython-download-cache-and-build-module-wheels.sh b/scripts/macpython-download-cache-and-build-module-wheels.sh
index 9e14bf1..6b7773f 100755
--- a/scripts/macpython-download-cache-and-build-module-wheels.sh
+++ b/scripts/macpython-download-cache-and-build-module-wheels.sh
@@ -48,13 +48,27 @@ else
   tarball_arch=""
 fi
 # Fetch ITKPythonBuilds archive containing ITK build artifacts
+rm -fr ITKPythonPackage
 echo "Fetching https://github.com/InsightSoftwareConsortium/ITKPythonBuilds/releases/download/${ITK_PACKAGE_VERSION:=v5.4.0}/ITKPythonBuilds-macosx${tarball_arch}.tar.zst"
 if [[ ! -f ITKPythonBuilds-macosx${tarball_arch}.tar.zst ]]; then
   aria2c -c --file-allocation=none -o ITKPythonBuilds-macosx${tarball_arch}.tar.zst -s 10 -x 10 https://github.com/InsightSoftwareConsortium/ITKPythonBuilds/releases/download/${ITK_PACKAGE_VERSION:=v5.4.0}/ITKPythonBuilds-macosx${tarball_arch}.tar.zst
 fi
 unzstd --long=31 ITKPythonBuilds-macosx${tarball_arch}.tar.zst -o ITKPythonBuilds-macosx${tarball_arch}.tar
 PATH="$(dirname $(brew list gnu-tar | grep gnubin)):$PATH"
-gtar xf ITKPythonBuilds-macosx${tarball_arch}.tar --checkpoint=10000 --checkpoint-action=dot
+gtar xf ITKPythonBuilds-macosx${tarball_arch}.tar --checkpoint=10000 --checkpoint-action=dot \
+  ITKPythonPackage/ITK-source \
+  ITKPythonPackageRequiredExtractionDir.txt \
+  ITKPythonPackage/scripts
+
+# Extract subdirectories specific to the compiled python versions
+args=( "$@"  )
+source ITKPythonPackage/scripts/macpython-build-common.sh
+for version in "$PYTHON_VERSIONS"; do
+  gtar xf ITKPythonBuilds-macosx${tarball_arch}.tar --checkpoint=10000 --checkpoint-action=dot \
+    --wildcards "ITKPythonPackage/ITK-${version}-macosx*" \
+    "ITKPythonPackage/venvs/${version}"
+done
+
 rm ITKPythonBuilds-macosx${tarball_arch}.tar
 
 # Optional: Update build scripts
@@ -82,4 +96,4 @@ if [[ ! ${ITK_USE_LOCAL_PYTHON} ]]; then
 fi
 
 echo "Building module wheels"
-/Users/svc-dashboard/D/P/ITKPythonPackage/scripts/macpython-build-module-wheels.sh "$@"
+/Users/svc-dashboard/D/P/ITKPythonPackage/scripts/macpython-build-module-wheels.sh "${args[@]}"