nonlinearsolverfactory.hh

Go to the documentation of this file.

// SPDX-FileCopyrightText: 2021-2025 The Ikarus Developers mueller@ibb.uni-stuttgart.de
// SPDX-License-Identifier: LGPL-3.0-or-later
 
#pragma once
 
#include <utility>
 
#include <ikarus/assembler/dirichletbcenforcement.hh>
#include <ikarus/utils/defaultfunctions.hh>
#include <ikarus/utils/differentiablefunctionfactory.hh>
 
namespace Ikarus {
 
template <typename NLSSetting>
struct NonlinearSolverFactory
{
  NonlinearSolverFactory(NLSSetting s)
      : settings(s) {}
 
  NLSSetting settings;
 
  template <typename Assembler>
  requires Concepts::FlatAssembler<typename std::remove_cvref_t<Assembler>::element_type>
  auto create(Assembler&& assembler) const {
    auto f        = DifferentiableFunctionFactory::op(assembler);
    using fTraits = typename decltype(f)::Traits;
 
    using CorrectionType = typename fTraits::template Range<1>;
    using Domain         = typename fTraits::Domain;
    auto updateF         = [assembler, setting = settings]<typename D, typename C>(D& x, const C& b) {
      if (assembler->dBCOption() == DBCOption::Reduced) {
        setting.updateFunction(x, assembler->createFullVector(b));
      } else
        setting.updateFunction(x, b);
    };
    auto settingsNew = settings.rebindUpdateFunction(std::move(updateF));
    return createNonlinearSolver(std::move(settingsNew), std::move(f));
  }
};
}; // namespace Ikarus