Function petsc_nls_module

python/dolfinx/wrappers/petsc.cpp:275–353

}

void petsc_nls_module(nb::module_& m)
{
  // dolfinx::NewtonSolver
  nb::class_<dolfinx::nls::petsc::NewtonSolver>(m, "NewtonSolver")
      .def(
          "__init__",
          [](dolfinx::nls::petsc::NewtonSolver* ns,
             const dolfinx_wrappers::MPICommWrapper comm)
          {
            new (ns) dolfinx::nls::petsc::NewtonSolver(comm.get());
            std::cerr << "NewtonSolver is deprecated, and will be removed in a "
                         "future release.\n";
          },
          nb::arg("comm"))
      .def_prop_ro("krylov_solver",
                   [](const dolfinx::nls::petsc::NewtonSolver& self)
                   {
                     KSP ksp = self.get_krylov_solver().ksp();
                     PyObject* obj = PyPetscKSP_New(ksp);
                     return nb::steal(obj);
                   })
      .def("setF", &dolfinx::nls::petsc::NewtonSolver::setF, nb::arg("F"),
           nb::arg("b"))
      .def("setJ", &dolfinx::nls::petsc::NewtonSolver::setJ, nb::arg("J"),
           nb::arg("Jmat"))
      .def("setP", &dolfinx::nls::petsc::NewtonSolver::setP, nb::arg("P"),
           nb::arg("Pmat"))
      .def(
          "set_update",
          [](dolfinx::nls::petsc::NewtonSolver& self,
             const std::function<void(
                 const dolfinx::nls::petsc::NewtonSolver* solver, const Vec,
                 Vec)>&
                 update) // See
                         // https://github.com/wjakob/nanobind/discussions/361
                         // on why we pass NewtonSolver* rather than
                         // NewtonSolver&
          {
            self.set_update(
                [update](const dolfinx::nls::petsc::NewtonSolver& solver,
                         const Vec dx, Vec x) { update(&solver, dx, x); });
          },
          nb::arg("update"))
      .def(
          "set_convergence_check",
          [](dolfinx::nls::petsc::NewtonSolver& self,
             const std::function<std::pair<double, bool>(
                 const dolfinx::nls::petsc::NewtonSolver* solver, const Vec)>&
                 convergence_check) // See
                                    // https://github.com/wjakob/nanobind/discussions/361
                                    // on why we pass NewtonSolver* rather than
                                    // NewtonSolver&
          {
            self.set_convergence_check(
                [convergence_check](
                    const dolfinx::nls::petsc::NewtonSolver& solver,
                    const Vec r) { return convergence_check(&solver, r); });
          },