DirichletBC.inc.hpp

#pragma once

#include <type_traits>

#include <dune/functions/functionspacebases/subentitydofs.hh>

#include <amdis/functions/Interpolate.hpp>
#include <amdis/LinearAlgebra.hpp>

namespace AMDiS {

template <class B, class V>
void DirichletBC<B, V>::
init()
{
  dirichletNodes_.assign(basis_.dimension(), false);

  auto localView = basis_.localView();
  auto seDOFs = Dune::Functions::subEntityDOFs(basis_);
  auto const& gridView = basis_.gridView();
  for (auto const& element : elements(gridView, typename BackendTraits<B>::PartitionSet{})) {
    if (element.hasBoundaryIntersections()) {
      localView.bind(element);
      for(auto const& intersection: intersections(gridView, element))
        if (intersection.boundary())
          for(auto localIndex: seDOFs.bind(localView,intersection))
            dirichletNodes_[localView.index(localIndex)]
              = dirichletNodes_[localView.index(localIndex)] || boundarySubset_(intersection);
    }
  }

}


template <class B, class V>
  template <class Mat, class Sol, class Rhs>
void DirichletBC<B, V>::
apply(Mat& matrix, Sol& solution, Rhs& rhs)
{
  AMDiS::interpolate(basis_, solution, values_, tag::defaulted{}, tag::defaulted{}, dirichletNodes_);
  dirichletBC(matrix, solution, rhs, dirichletNodes_);
}

} // end namespace AMDiS