python312Packages.itk: init at 5.4.0

pkgs/development/libraries/itk/generic.nix

@@ -10,6 +10,8 @@
   fetchFromGitHub,
   fetchpatch,
   cmake,
+  castxml,
+  swig,
   expat,
   fftw,
   gdcm,
@@ -20,13 +22,21 @@
   libpng,
   libX11,
   libuuid,
+  patchelf,
+  python ? null,
+  numpy ? null,
   xz,
   vtk,
+  which,
   zlib,
   Cocoa,
+  enablePython ? false,
 }:
 
 let
+  # Python wrapper contains its own VTK support incompatible with MODULE_ITKVtkGlue
+  withVtk = !enablePython;
+
   itkGenericLabelInterpolatorSrc = fetchFromGitHub {
     owner = "InsightSoftwareConsortium";
     repo = "ITKGenericLabelInterpolator";
@@ -47,6 +57,19 @@ let
     rev = "bb896868fc6480835495d0da4356d5db009592a6";
     hash = "sha256-MfaIA0xxA/pzUBSwnAevr17iR23Bo5iQO2cSyknS3o4=";
   };
+
+  # remove after next swig update:
+  swigUnstable = swig.overrideAttrs ({
+    version = "4.2.1-unstable-2024-08-19";
+
+    src = fetchFromGitHub {
+      owner = "swig";
+      repo = "swig";
+      rev = "5ac5d90f970759fbe705fae551d0743a7c63c67e";
+      hash = "sha256-32EFLHpP4l04nqrc8dt4Qsr8deTBqLt8lUlhnNnaIGU=";
+    };
+
+  });
 in
 
 stdenv.mkDerivation {
@@ -77,7 +100,8 @@ stdenv.mkDerivation {
     ln -sr ${itkSimpleITKFiltersSrc} Modules/External/ITKSimpleITKFilters
   '';
 
-  cmakeFlags = [
+  cmakeFlags =
+    [
       "-DBUILD_EXAMPLES=OFF"
       "-DBUILD_SHARED_LIBS=ON"
       "-DITK_FORBID_DOWNLOADS=ON"
@@ -94,22 +118,38 @@ stdenv.mkDerivation {
       "-DModule_ITKIOMINC=ON"
       "-DModule_ITKIOTransformMINC=ON"
       "-DModule_SimpleITKFilters=ON"
-    "-DModule_ITKVtkGlue=ON"
       "-DModule_ITKReview=ON"
       "-DModule_MGHIO=ON"
       "-DModule_AdaptiveDenoising=ON"
       "-DModule_GenericLabelInterpolator=ON"
-  ];
+    ]
+    ++ lib.optionals enablePython [
+      "-DITK_WRAP_PYTHON=ON"
+      "-DITK_USE_SYSTEM_CASTXML=ON"
+      "-DITK_USE_SYSTEM_SWIG=ON"
+      "-DPY_SITE_PACKAGES_PATH=${placeholder "out"}/${python.sitePackages}"
+    ]
+    ++ lib.optionals withVtk [ "-DModule_ITKVtkGlue=ON" ];
 
-  nativeBuildInputs = [
+  nativeBuildInputs =
+    [
       cmake
       xz
+    ]
+    ++ lib.optionals enablePython [
+      castxml
+      swigUnstable
+      which
     ];
-  buildInputs = [
+
+  buildInputs =
+    [
       libX11
       libuuid
-    vtk
-  ] ++ lib.optionals stdenv.isDarwin [ Cocoa ];
+    ]
+    ++ lib.optionals stdenv.isDarwin [ Cocoa ]
+    ++ lib.optionals enablePython [ python ]
+    ++ lib.optionals withVtk [ vtk ];
   # Due to ITKVtkGlue=ON and the additional dependencies needed to configure VTK 9
   # (specifically libGL and libX11 on Linux),
   # it's now seemingly necessary for packages that configure ITK to
@@ -130,12 +170,30 @@ stdenv.mkDerivation {
     libpng
     libtiff
     zlib
-  ] ++ vtk.propagatedBuildInputs;
+  ] ++ lib.optionals withVtk vtk.propagatedBuildInputs ++ lib.optionals enablePython [ numpy ];
+
+  postInstall = lib.optionalString enablePython ''
+    substitute \
+      ${./itk.egg-info} \
+      $out/${python.sitePackages}/itk-${version}.egg-info \
+      --subst-var-by ITK_VER "${version}"
+  '';
+
+  # remove forbidden reference to /build which occur when building the Python wrapping
+  # (also remove a copy of itkTestDriver with incorrect permissions/RPATH):
+  preFixup = lib.optionals enablePython ''
+    rm $out/${python.sitePackages}/itk/itkTestDriver
+    find $out/${python.sitePackages}/itk -type f -name '*.so*' -exec \
+      patchelf {} --shrink-rpath --allowed-rpath-prefixes "$NIX_STORE" \;
+  '';
 
   meta = {
     description = "Insight Segmentation and Registration Toolkit";
     homepage = "https://www.itk.org";
     license = lib.licenses.asl20;
     maintainers = with lib.maintainers; [ bcdarwin ];
+    # aarch64-linux Python wrapping fails with "error: unknown type name '_Float128'" and similar;
+    # compilation runs slowly and times out on Darwin
+    platforms = with lib.platforms; if enablePython then [ "x86_64-linux" ] else unix;
   };
 }

pkgs/development/libraries/itk/itk.egg-info

@@ -0,0 +1,5 @@
+Metadata-Version: 2.1
+Name: itk
+Version: @ITK_VER@
+Summary: Insight Segmentation and Registration Toolkit
+Platform: UNKNOWN

pkgs/top-level/python-packages.nix

@@ -6206,6 +6206,12 @@ self: super: with self; {
 
   itanium-demangler = callPackage ../development/python-modules/itanium-demangler { };
 
+  itk = toPythonModule (pkgs.itk.override {
+    inherit python numpy;
+    enablePython = true;
+  });
+
+
   item-synchronizer = callPackage ../development/python-modules/item-synchronizer { };
 
   itemadapter = callPackage ../development/python-modules/itemadapter { };
@@ -14477,7 +14483,7 @@ self: super: with self; {
   simplehound = callPackage ../development/python-modules/simplehound { };
 
   simpleitk = callPackage ../development/python-modules/simpleitk {
-    inherit (pkgs) simpleitk;
+    inherit (pkgs) itk simpleitk;
   };
 
   simplejson = callPackage ../development/python-modules/simplejson { };