python/dirichletvalues/dirichletvalues.hh

Go to the documentation of this file.

// SPDX-FileCopyrightText: 2021-2024 The Ikarus Developers mueller@ibb.uni-stuttgart.de
// SPDX-License-Identifier: LGPL-3.0-or-later
 
#pragma once
 
#include <dune/functions/functionspacebases/lagrangebasis.hh>
#include <dune/functions/functionspacebases/powerbasis.hh>
#include <dune/grid/yaspgrid.hh>
#include <dune/python/common/typeregistry.hh>
#include <dune/python/functions/globalbasis.hh>
#include <dune/python/pybind11/eigen.h>
#include <dune/python/pybind11/functional.h>
#include <dune/python/pybind11/pybind11.h>
#include <dune/python/pybind11/stl.h>
#include <dune/python/pybind11/stl_bind.h>
 
#include <ikarus/finiteelements/ferequirements.hh>
// PYBIND11_MAKE_OPAQUE(std::vector<bool>);
namespace Ikarus::Python {
 
  template <class DirichletValues, class... options>
  void registerDirichletValues(pybind11::handle scope, pybind11::class_<DirichletValues, options...> cls) {
    using pybind11::operator""_a;
 
    using Basis        = typename DirichletValues::Basis;
    using BackendType  = typename DirichletValues::BackendType;
    using FlagsType    = typename DirichletValues::FlagsType;
    using MultiIndex   = typename Basis::MultiIndex;
    using LocalView    = typename Basis::LocalView;
    using Intersection = typename Basis::GridView::Intersection;
 
    pybind11::module scopedf = pybind11::module::import("dune.functions");
    typedef Dune::Python::LocalViewWrapper<Basis> LocalViewWrapper;
    auto includes = Dune::Python::IncludeFiles{"dune/python/functions/globalbasis.hh"};
    auto lv       = Dune::Python::insertClass<LocalViewWrapper>(
                  scopedf, "LocalView",
                  Dune::Python::GenerateTypeName("Dune::Python::LocalViewWrapper", Dune::MetaType<Basis>()), includes)
                  .first;
 
    cls.def(pybind11::init([](const Basis& basis) { return new DirichletValues(basis); }),
            pybind11::keep_alive<1, 2>());
 
    // Eigen::Ref needed due to https://pybind11.readthedocs.io/en/stable/advanced/cast/eigen.html#pass-by-reference
    cls.def("fixBoundaryDOFs",
            [](DirichletValues& self, const std::function<void(Eigen::Ref<Eigen::VectorX<bool>>, int)>& f) {
              auto lambda = [&](BackendType& vec, const MultiIndex& indexGlobal) {
                // we explicitly only allow flat indices
                f(vec.vector(), indexGlobal[0]);
              };
              self.fixBoundaryDOFs(lambda);
            });
 
    cls.def("fixBoundaryDOFsUsingLocalView",
            [](DirichletValues& self,
               const std::function<void(Eigen::Ref<Eigen::VectorX<bool>>, int, LocalViewWrapper&)>& f) {
              auto lambda = [&](BackendType& vec, int localIndex, LocalView& lv) {
                auto lvWrapper = LocalViewWrapper(lv.globalBasis());
                // this can be simplified when
                // https://gitlab.dune-project.org/staging/dune-functions/-/merge_requests/418 becomes available
                pybind11::object obj = pybind11::cast(lv.element());
                lvWrapper.bind(obj);
                f(vec.vector(), localIndex, lvWrapper);
              };
              self.fixBoundaryDOFs(lambda);
            });
 
    cls.def("fixBoundaryDOFsUsingLocalViewAndIntersection",
            [](DirichletValues& self,
               const std::function<void(Eigen::Ref<Eigen::VectorX<bool>>, int, LocalViewWrapper&, const Intersection&)>&
                   f) {
              auto lambda = [&](BackendType& vec, int localIndex, LocalView& lv, const Intersection& intersection) {
                auto lvWrapper = LocalViewWrapper(lv.globalBasis());
                // this can be simplified when
                // https://gitlab.dune-project.org/staging/dune-functions/-/merge_requests/418 becomes available
                pybind11::object obj = pybind11::cast(lv.element());
                lvWrapper.bind(obj);
                f(vec.vector(), localIndex, lvWrapper, intersection);
              };
              self.fixBoundaryDOFs(lambda);
            });
 
    cls.def("fixDOFs",
            [](DirichletValues& self, const std::function<void(Eigen::Ref<Eigen::VectorX<bool>>, const Basis&)>& f) {
              self.fixBoundaryDOFs(f);
            });
  }
 
}  // namespace Ikarus::Python