python312Packages.fenics-dolfinx: init at 0.9.0.post1

2025-05-09 07:42:39 +08:00
parent f76bdc0af6
commit 2bbc507c01
2 changed files with 159 additions and 0 deletions
@@ -0,0 +1,157 @@
+{
+  lib,
+  stdenv,
+  toPythonModule,
+  fetchFromGitHub,
+  buildPythonPackage,
+
+  # build-system
+  scikit-build-core,
+  nanobind,
+
+  # nativeBuildInputs
+  cmake,
+  ninja,
+  pkg-config,
+
+  # buildInputs
+  dolfinx,
+
+  # dependency
+  numpy,
+  cffi,
+  mpi4py,
+  petsc4py,
+  slepc4py,
+  adios2,
+  kahip,
+  fenics-ffcx,
+  fenics-basix,
+  fenics-ufl,
+
+  # nativeCheckInputs
+  scipy,
+  matplotlib,
+  pytest-xdist,
+  pytestCheckHook,
+  writableTmpDirAsHomeHook,
+  mpiCheckPhaseHook,
+
+  # custom options
+  withParmetis ? false,
+
+  # passthru.tests
+  fenics-dolfinx,
+  mpich,
+}:
+
+let
+  fenicsPackages = petsc4py.petscPackages.overrideScope (
+    final: prev: {
+      slepc = final.callPackage slepc4py.override { };
+      adios2 = final.callPackage adios2.override { };
+      kahip = final.callPackage kahip.override { };
+      dolfinx = final.callPackage dolfinx.override { inherit withParmetis; };
+    }
+  );
+in
+buildPythonPackage rec {
+  inherit (dolfinx)
+    version
+    src
+    ;
+  pname = "fenics-dolfinx";
+  pyproject = true;
+
+  pythonRelaxDeps = [
+    "cffi"
+    "fenics-ufl"
+  ];
+
+  preConfigure = ''
+    cd python
+  '';
+
+  dontUseCmakeConfigure = true;
+
+  build-system = [
+    scikit-build-core
+    nanobind
+  ];
+
+  nativeBuildInputs = [
+    cmake
+    ninja
+    pkg-config
+    fenicsPackages.mpi
+  ];
+
+  buildInputs = [
+    fenicsPackages.dolfinx
+  ];
+
+  dependencies = [
+    numpy
+    cffi
+    fenics-basix
+    fenics-ffcx
+    fenics-ufl
+    petsc4py
+    fenicsPackages.slepc
+    fenicsPackages.adios2
+    fenicsPackages.kahip
+    (mpi4py.override { inherit (fenicsPackages) mpi; })
+  ];
+
+  doCheck = true;
+
+  nativeCheckInputs = [
+    scipy
+    matplotlib
+    pytest-xdist
+    pytestCheckHook
+    writableTmpDirAsHomeHook
+    mpiCheckPhaseHook
+  ];
+
+  preCheck = ''
+    rm -rf dolfinx
+  '';
+
+  pythonImportsCheck = [
+    "dolfinx"
+  ];
+
+  disabledTests = [
+    # require cffi<1.17
+    "test_cffi_expression"
+    "test_hexahedron_mesh"
+  ];
+
+  passthru = {
+    tests =
+      {
+        complex = fenics-dolfinx.override {
+          petsc4py = petsc4py.override { scalarType = "complex"; };
+        };
+      }
+      // lib.optionalAttrs stdenv.hostPlatform.isLinux {
+        mpich = fenics-dolfinx.override {
+          petsc4py = petsc4py.override { mpi = mpich; };
+        };
+      };
+  };
+
+  meta = {
+    homepage = "https://fenicsproject.org";
+    downloadPage = "https://github.com/fenics/dolfinx";
+    description = "Computational environment of FEniCSx and implements the FEniCS Problem Solving Environment in C++ and Python";
+    changelog = "https://github.com/fenics/dolfinx/releases/tag/${src.tag}";
+    license = with lib.licenses; [
+      bsd2
+      lgpl3Plus
+    ];
+    platforms = lib.platforms.unix;
+    maintainers = with lib.maintainers; [ qbisi ];
+  };
+}
@@ -4861,6 +4861,8 @@ self: super: with self; {

  fenics-basix = callPackage ../development/python-modules/fenics-basix { };

+  fenics-dolfinx = callPackage ../development/python-modules/fenics-dolfinx { };
+
  fenics-ffcx = callPackage ../development/python-modules/fenics-ffcx { };

  fenics-ufl = callPackage ../development/python-modules/fenics-ufl { };