ENH: Allow specifying ITK branch, tag, or hash for build-all scripts

Add --itk-ref option to both shell and Python build-all-latest-wheels
scripts. Defaults to "main" but accepts any git ref (branch name, tag
like v6.0b02, or commit hash). The ITK clone is now a full clone (not
shallow) to support arbitrary refs.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

Author: hjmjohnson

scripts/build-all-latest-wheels.sh

@@ -1,24 +1,46 @@
 #!/bin/bash
-# Build ITK + all remote module Python wheels from latest main branches.
-# Usage: ./scripts/build-all-latest-wheels.sh [platform-env]
-# Example: ./scripts/build-all-latest-wheels.sh linux-py311
+# Build ITK + all remote module Python wheels.
+# Usage: ./scripts/build-all-latest-wheels.sh [options]
+#   --platform-env ENV    Pixi environment (default: linux-py311)
+#   --itk-ref REF         ITK branch, tag, or commit hash (default: main)
+#   --itk-repo URL        ITK git URL
+#   --ipp-branch BRANCH   ITKPythonPackage branch (default: python-build-system)
+#   --ipp-repo URL        ITKPythonPackage git URL
+# Example:
+#   ./scripts/build-all-latest-wheels.sh --itk-ref v6.0b02 --platform-env linux-py311
 set -euo pipefail
 
-PLATFORM_ENV="${1:-linux-py311}"
-TIMESTAMP=$(date +%Y%m%d%H%M%S)
-WORKDIR="/tmp/${TIMESTAMP}_LatestITKPython"
-DIST_DIR="${WORKDIR}/dist"
+# Defaults
+PLATFORM_ENV="linux-py311"
+ITK_REF="main"
 ITK_REPO="https://github.com/InsightSoftwareConsortium/ITK.git"
 IPP_REPO="https://github.com/BRAINSia/ITKPythonPackage.git"
 IPP_BRANCH="python-build-system"
 
+while [[ $# -gt 0 ]]; do
+  case "$1" in
+    --platform-env) PLATFORM_ENV="$2"; shift 2 ;;
+    --itk-ref)      ITK_REF="$2"; shift 2 ;;
+    --itk-repo)     ITK_REPO="$2"; shift 2 ;;
+    --ipp-branch)   IPP_BRANCH="$2"; shift 2 ;;
+    --ipp-repo)     IPP_REPO="$2"; shift 2 ;;
+    *) echo "Unknown option: $1"; exit 1 ;;
+  esac
+done
+
+TIMESTAMP=$(date +%Y%m%d%H%M%S)
+WORKDIR="/tmp/${TIMESTAMP}_LatestITKPython"
+DIST_DIR="${WORKDIR}/dist"
+
 mkdir -p "${DIST_DIR}"
 echo "=== Build directory: ${WORKDIR}"
 echo "=== Platform: ${PLATFORM_ENV}"
+echo "=== ITK ref: ${ITK_REF}"
 
 # 1) Clone ITK
-echo "=== Cloning ITK (main)..."
-git clone --depth 1 --branch main "${ITK_REPO}" "${WORKDIR}/ITK"
+echo "=== Cloning ITK (${ITK_REF})..."
+git clone "${ITK_REPO}" "${WORKDIR}/ITK"
+git -C "${WORKDIR}/ITK" checkout "${ITK_REF}"
 
 # 2) Clone ITKPythonPackage
 echo "=== Cloning ITKPythonPackage (${IPP_BRANCH})..."
@@ -70,7 +92,7 @@ for name in "${module_list[@]}"; do
   echo "=== Building ${name}..."
   if pixi run -e "${PLATFORM_ENV}" -- python scripts/build_wheels.py \
     --platform-env "${PLATFORM_ENV}" \
-    --itk-git-tag main \
+    --itk-git-tag "${ITK_REF}" \
     --itk-source-dir "${WORKDIR}/ITK" \
     --module-source-dir "${MODULES_DIR}/${name}" \
     --no-build-itk-tarball-cache \

scripts/build_all_latest_wheels.py

@@ -21,8 +21,10 @@ def run(cmd: list[str], **kwargs) -> subprocess.CompletedProcess:
     return subprocess.run(cmd, check=True, **kwargs)
 
 
-def clone(repo: str, dest: Path, branch: str | None = None) -> bool:
-    cmd = ["git", "clone", "--depth", "1"]
+def clone(repo: str, dest: Path, branch: str | None = None, depth: int | None = 1) -> bool:
+    cmd = ["git", "clone"]
+    if depth is not None:
+        cmd += ["--depth", str(depth)]
     if branch:
         cmd += ["--branch", branch]
     cmd += [repo, str(dest)]
@@ -50,6 +52,7 @@ def build_wheels(
     platform_env: str,
     build_dir: Path,
     itk_source: Path,
+    itk_ref: str = "main",
     module_source: Path | None = None,
     skip_itk: bool = False,
 ) -> bool:
@@ -64,7 +67,7 @@ def build_wheels(
         "--platform-env",
         platform_env,
         "--itk-git-tag",
-        "main",
+        itk_ref,
         "--itk-source-dir",
         str(itk_source),
         "--no-build-itk-tarball-cache",
@@ -104,6 +107,11 @@ def main():
         default="https://github.com/InsightSoftwareConsortium/ITK.git",
         help="ITK git URL",
     )
+    parser.add_argument(
+        "--itk-ref",
+        default="main",
+        help="ITK branch, tag, or commit hash (default: main)",
+    )
     args = parser.parse_args()
 
     timestamp = datetime.now().strftime("%Y%m%d%H%M%S")
@@ -114,11 +122,13 @@ def main():
 
     print(f"=== Build directory: {workdir}")
     print(f"=== Platform: {args.platform_env}")
+    print(f"=== ITK ref: {args.itk_ref}")
 
     # 1) Clone ITK
-    print("=== Cloning ITK (main)...")
+    print(f"=== Cloning ITK ({args.itk_ref})...")
     itk_dir = workdir / "ITK"
-    clone(args.itk_repo, itk_dir, branch="main")
+    clone(args.itk_repo, itk_dir, depth=None)
+    run(["git", "checkout", args.itk_ref], cwd=itk_dir)
 
     # 2) Clone ITKPythonPackage
     print(f"=== Cloning ITKPythonPackage ({args.ipp_branch})...")
@@ -147,7 +157,7 @@ def main():
 
     # 4) Build ITK wheels
     print("=== Building ITK Python wheels...")
-    if not build_wheels(ipp_dir, args.platform_env, build_dir, itk_dir):
+    if not build_wheels(ipp_dir, args.platform_env, build_dir, itk_dir, args.itk_ref):
         print("FATAL: ITK wheel build failed")
         sys.exit(1)
 
@@ -165,6 +175,7 @@ def main():
             args.platform_env,
             build_dir,
             itk_dir,
+            args.itk_ref,
             module_source=mod_dir,
             skip_itk=True,
         ):