mixin.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/common/tuplevector.hh>
 
#include <ikarus/finiteelements/fetraits.hh>
 
namespace Ikarus {
template <typename PreFE, template <typename, typename> class... Skills>
struct FEMixin : Skills<PreFE, typename PreFE::template FE<Skills...>>...
{
  explicit FEMixin(typename Skills<PreFE, typename PreFE::template FE<Skills...>>::Pre&&... skillsArgs)
      : Skills<PreFE, typename PreFE::template FE<Skills...>>(
            std::forward<typename Skills<PreFE, typename PreFE::template FE<Skills...>>::Pre>(skillsArgs))... {}
 
  template <template <typename, typename> class Skill>
  static consteval bool hasSkill() {
    return Ikarus::traits::hasType<Skill<PreFE, typename PreFE::template FE<Skills...>>,
                                   std::tuple<Skills<PreFE, typename PreFE::template FE<Skills...>>...>>::value;
  }
 
private:
  template <typename T>
  consteval static auto computeSupportedResultTypes() {
    if constexpr (requires { typename T::SupportedResultTypes; })
      return typename T::SupportedResultTypes();
    else
      return std::tuple();
  }
 
public:
  using SupportedResultTypes =
      decltype(std::tuple_cat(computeSupportedResultTypes<Skills<PreFE, typename PreFE::template FE<Skills...>>>()...));
 
  using Traits                  = PreFE::Traits;
  using FERequirementType       = typename Traits::FERequirementType;
  using LocalView               = typename Traits::LocalView;
  static constexpr int worldDim = Traits::worlddim;
 
  friend auto calculateScalar(const FEMixin& self, const FERequirementType& par) {
    return self.template calculateScalarImpl<double>(par);
  }
 
  friend void calculateVector(const FEMixin& self, const FERequirementType& par,
                              typename Traits::template VectorType<> force) {
    self.template calculateVectorImpl<double>(par, force);
  }
 
  friend void calculateMatrix(const FEMixin& self, const FERequirementType& par,
                              typename Traits::template MatrixType<> K) {
    self.template calculateMatrixImpl<double>(par, K);
  }
 
  using Skills<PreFE, typename PreFE::template FE<Skills...>>::calculateAtImpl...;
 
  template <template <typename, int, int> class RT>
  requires requires(FEMixin m, const FERequirementType& req, const Dune::FieldVector<double, Traits::mydim>& local) {
    m.template calculateAtImpl<RT>(req, local, Dune::PriorityTag<10>());
  }
  auto calculateAt(const FERequirementType& req, const Dune::FieldVector<double, Traits::mydim>& local) const {
    return this->template calculateAtImpl<RT>(req, local, Dune::PriorityTag<10>());
  }
 
private:
  template <typename Sk>
  auto invokeBind() {
    if constexpr (requires { this->Sk::bindImpl(); })
      Sk::bindImpl();
  }
 
  static constexpr bool implementsCalculateScalarImpl =
      (requires(FEMixin m, const FERequirementType& par,
                const std::optional<std::reference_wrapper<const Eigen::VectorX<double>>>& dx) {
        m.Skills<PreFE, typename PreFE::template FE<Skills...>>::calculateScalarImpl(par, dx);
      } ||
       ...);
 
public:
  void bind() { (invokeBind<Skills<PreFE, typename PreFE::template FE<Skills...>>>(), ...); }
 
  template <typename ScalarType = double>
  requires implementsCalculateScalarImpl
  auto calculateScalarImpl(
      const FERequirementType& par,
      const std::optional<std::reference_wrapper<const Eigen::VectorX<ScalarType>>>& dx = std::nullopt) const {
    return (Skills<PreFE, typename PreFE::template FE<Skills...>>::template calculateScalarImpl<ScalarType>(par, dx) +
            ... + ScalarType{0});
  }
 
private:
  static constexpr bool implementsCalculateVectorImpl =
      (requires(FEMixin m, const FERequirementType& par, typename Traits::template VectorType<double> force,
                const std::optional<std::reference_wrapper<const Eigen::VectorX<double>>>& dx) {
        m.Skills<PreFE, typename PreFE::template FE<Skills...>>::calculateVectorImpl(par, force, dx);
      } ||
       ...);
 
public:
  template <typename ScalarType>
  requires implementsCalculateVectorImpl
  void calculateVectorImpl(
      const FERequirementType& par, typename Traits::template VectorType<ScalarType> force,
      const std::optional<std::reference_wrapper<const Eigen::VectorX<ScalarType>>>& dx = std::nullopt) const {
    (Skills<PreFE, typename PreFE::template FE<Skills...>>::template calculateVectorImpl<ScalarType>(par, force, dx),
     ...);
  }
 
private:
  static constexpr bool implementsCalculateMatrixImpl =
      (requires(FEMixin m, const FERequirementType& par, typename Traits::template MatrixType<double> K,
                const std::optional<std::reference_wrapper<const Eigen::VectorX<double>>>& dx) {
        m.Skills<PreFE, typename PreFE::template FE<Skills...>>::calculateMatrixImpl(par, K, dx);
      } ||
       ...);
 
public:
  template <typename ScalarType>
  requires implementsCalculateMatrixImpl
  void calculateMatrixImpl(
      const FERequirementType& par, typename Traits::template MatrixType<ScalarType> K,
      const std::optional<std::reference_wrapper<const Eigen::VectorX<ScalarType>>>& dx = std::nullopt) const {
    (Skills<PreFE, typename PreFE::template FE<Skills...>>::template calculateMatrixImpl<ScalarType>(par, K, dx), ...);
  }
 
protected:
  const auto& underlying() const { return static_cast<const typename PreFE::template FE<Skills...>&>(*this); }
  auto& underlying() { return static_cast<typename PreFE::template FE<Skills...>&>(*this); }
};
 
template <typename... ARGS>
struct Skills
{
  using Args = std::tuple<ARGS...>;
  Args args;
};
 
template <typename... Args>
auto skills(const Args&... args) {
  return Skills<std::remove_cvref_t<Args>...>{std::forward_as_tuple(std::remove_cvref_t<Args>(args)...)};
}
 
template <typename... Args1, typename... Args2>
auto merge(const Skills<Args1...>& sk1, const Skills<Args2...>& sk2) {
  return Skills<std::remove_cvref_t<Args1>..., std::remove_cvref_t<Args2>...>{std::tuple_cat(sk1.args, sk2.args)};
}
 
} // namespace Ikarus