outsideoperator.hh

Go to the documentation of this file.

// -*- tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*-
// vi: set et ts=4 sw=2 sts=2:
 
// SPDX-FileCopyrightText: Copyright © DUNE-FUFEM Project contributors, see file AUTHORS.md
// SPDX-License-Identifier: LicenseRef-GPL-2.0-only-with-DUNE-exception OR LGPL-3.0-or-later
 
#ifndef DUNE_FUFEM_FORMS_OUTSIDEOPERATOR_HH
#define DUNE_FUFEM_FORMS_OUTSIDEOPERATOR_HH
 
#include <cstddef>
#include <type_traits>
#include <tuple>
#include <utility>
#include <functional>
 
#include <dune/common/indices.hh>
#include <dune/common/typelist.hh>
#include <dune/common/typetraits.hh>
 
#include <dune/fufem/forms/baseclass.hh>
#include <dune/fufem/forms/tensors.hh>
#include <dune/fufem/forms/localfunctionadaptor.hh>
 
 
 
namespace Dune::Fufem::Forms::Impl {
 
  template<class Operator, int arity = Operator::arity>
  struct OutsideOperatorBaseTraits
  {
    using type = MultilinearOperator<arity>;
  };
 
  template<class Operator>
  struct OutsideOperatorBaseTraits<Operator, 1>
  {
    using type = UnaryOperator<Operator::argIndex>;
  };
 
} // namespace Dune::Fufem::Forms::Impl
 
 
 
namespace Dune::Fufem::Forms {
 
 
 
  template<class BaseOperator>
  class OutsideOperator
    : public Impl::OutsideOperatorBaseTraits<BaseOperator>::type
  {
    using Base = typename Impl::OutsideOperatorBaseTraits<BaseOperator>::type;
    using BaseLocalOperator = decltype(localOperator(std::declval<BaseOperator>()));
 
    static_assert(Base::arity <= 1, "Only unary and nullary outside operators can be constructed");
 
  public:
 
    using Element = typename BaseOperator::Element;
    using Range = typename BaseOperator::Range;
 
    OutsideOperator(const BaseOperator& baseOperator) :
      baseOperator_(baseOperator)
    {}
 
    class LocalOperator : public BaseLocalOperator
    {
      using BaseTensor = decltype(std::declval<BaseLocalOperator>()(0));
    public:
 
      using Element = typename BaseLocalOperator::Element;
      using Intersection = typename BaseLocalOperator::Intersection;
      using CacheManager = typename BaseLocalOperator::CacheManager;
      using Range = typename BaseLocalOperator::Range;
 
      LocalOperator(BaseLocalOperator&& baseLocalOperator) :
        BaseLocalOperator(std::move(baseLocalOperator))
      {}
 
      auto operator()(std::size_t index) const
      {
        using namespace Impl::Tensor;
        using namespace Dune::Indices;
        auto y = BaseLocalOperator::operator()(index);
        if constexpr (Base::arity == 0)
          return y;
        else if constexpr (Base::arity == 1)
          return SparseTensor(y.rank(),
            [&, y](auto&& f) {
              sparseForEach(y, [&](const auto& y_i, auto i0, auto... i) {
                f(y_i, _1, i...);
              });
            },
            y.nnz(),
            Dune::MetaType<std::conditional_t<IsLazyTensor<BaseTensor>::value,
                typename BaseLocalOperator::Range&&,
                typename BaseLocalOperator::Range
              >>{}
          );
      }
 
      auto quadratureRuleKey() const
      {
        return BaseLocalOperator::quadratureRuleKey();
      }
 
      template<class... LV>
      void registerLocalViews(const LV&... lvs)
      {
        BaseLocalOperator::registerOutsideLocalViews(lvs...);
      }
 
      template<class... LV>
      void registerOutsideLocalViews(const LV&... lvs)
      {
        BaseLocalOperator::registerLocalViews(lvs...);
      }
 
      void registerCaches(CacheManager& cacheManager)
      {
        BaseLocalOperator::registerOutsideCaches(cacheManager);
      }
 
      void registerOutsideCaches(CacheManager& cacheManager)
      {
        BaseLocalOperator::registerCaches(cacheManager);
      }
 
      template<class Dummy=void>
      void bind(const Element&)
      {
        static_assert(Dune::AlwaysFalse<Dummy>::value, "Outside operators can only be used for forms integrated over the skeleton");
      }
 
      void bind(const Intersection& intersection, const Element& element, const Element& otherElement)
      {
        BaseLocalOperator::bind(intersection, otherElement, element);
      }
 
      template<class Dummy=void>
      void bindToCaches(CacheManager& cacheManager)
      {
        static_assert(Dune::AlwaysFalse<Dummy>::value, "Outside operators can only be used for forms integrated over the skeleton");
      }
 
      void bindToCaches(CacheManager& cacheManager, CacheManager& outsideCacheManager)
      {
        BaseLocalOperator::bindToCaches(outsideCacheManager, cacheManager);
      }
 
    };
 
    friend LocalOperator localOperator(const OutsideOperator& outsideOperator)
    {
      return LocalOperator(localOperator(outsideOperator.baseOperator()));
    }
 
    auto basis() const
    {
      return baseOperator_.basis();
    }
 
    auto treePath() const
    {
      return baseOperator_.treePath();
    }
 
    const BaseOperator& baseOperator() const
    {
      return baseOperator_;
    }
 
    friend auto jacobian(const OutsideOperator& outsideOperator)
    {
      using BaseOperatorGradient = decltype(jacobian(outsideOperator.baseOperator()));
      return OutsideOperator<BaseOperatorGradient>(jacobian(outsideOperator.baseOperator()));
    }
 
    friend auto gradient(const OutsideOperator& outsideOperator)
    {
      using BaseOperatorGradient = decltype(gradient(outsideOperator.baseOperator()));
      return OutsideOperator<BaseOperatorGradient>(gradient(outsideOperator.baseOperator()));
    }
 
    friend auto grad(const OutsideOperator& outsideOperator)
    {
      return gradient(outsideOperator);
    }
 
    friend auto divergence(const OutsideOperator& outsideOperator)
    {
      using BaseOperatorGradient = decltype(divergence(outsideOperator.baseOperator()));
      return OutsideOperator<BaseOperatorGradient>(divergence(outsideOperator.baseOperator()));
    }
 
    friend auto div(const OutsideOperator& outsideOperator)
    {
      return divergence(outsideOperator);
    }
 
    friend auto curl(const OutsideOperator& outsideOperator)
    {
      using BaseOperatorGradient = decltype(curl(outsideOperator.baseOperator()));
      return OutsideOperator<BaseOperatorGradient>(curl(outsideOperator.baseOperator()));
    }
 
    friend auto rot(const OutsideOperator& outsideOperator)
    {
      return curl(outsideOperator);
    }
 
  private:
    BaseOperator baseOperator_;
  };
 
 
 
} // namespace Dune::Fufem::Forms
 
 
#endif // DUNE_FUFEM_FORMS_OUTSIDEOPERATOR_HH