gridfunction.hh

// -*- tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*-
// vi: set et ts=4 sw=2 sts=2:
#ifndef DUNE_FUNCTIONS_GRIDFUNCTIONS_GRID_FUNCTION_HH
#define DUNE_FUNCTIONS_GRIDFUNCTIONS_GRID_FUNCTION_HH
 
#include <type_traits>
 
#include <dune/common/typeutilities.hh>
 
#include <dune/functions/common/typeerasure.hh>
#include <dune/functions/common/defaultderivativetraits.hh>
#include <dune/functions/common/differentiablefunction.hh>
#include <dune/functions/common/localfunction.hh>
#include <dune/functions/common/functionconcepts.hh>
#include <dune/functions/gridfunctions/localderivativetraits.hh>
#include <dune/functions/gridfunctions/gridfunction_imp.hh>
 
 
 
namespace Dune {
namespace Functions {
 
 
 
/*
 * Default implementation is empty
 * The actual implementation is only given if Signature is an type
 * describing a function signature as Range(Domain).
 */
template<class Signature, class EntitySet, template<class> class DerivativeTraits=DefaultDerivativeTraits, size_t bufferSize=56>
class GridFunction
{};
 
 
 
namespace Imp
{
 
  template<class S, class ES, template<class> class DerivativeTraits, size_t bufferSize>
  struct GridFunctionTraits :
    DifferentiableFunctionTraits<S, DerivativeTraits, bufferSize>
  {
  protected:
    using Base=DifferentiableFunctionTraits<S, DerivativeTraits, bufferSize>;
 
  public:
    using EntitySet = ES;
 
    using Element = typename EntitySet::Element;
 
    using DerivativeSignature = typename Base::DerivativeSignature;
 
    using DerivativeInterface = GridFunction<DerivativeSignature, ES, DerivativeTraits, bufferSize>;
 
    using LocalSignature = typename Base::Range(typename EntitySet::LocalCoordinate);
 
    template<class R>
    using LocalDerivativeTraits = typename Dune::Functions::LocalDerivativeTraits<EntitySet, DerivativeTraits>::template Traits<R>;
 
    using LocalFunctionTraits = typename Dune::Functions::Imp::LocalFunctionTraits<LocalSignature, Element, LocalDerivativeTraits, bufferSize>;
 
    using LocalFunctionInterface = LocalFunction<LocalSignature, Element, LocalDerivativeTraits, bufferSize>;
 
    using Concept = GridFunctionWrapperInterface<S, DerivativeInterface, LocalFunctionInterface, ES>;
 
    template<class B>
    using Model = GridFunctionWrapperImplementation<S, DerivativeInterface, LocalFunctionInterface, ES, B>;
  };
}
 
 
 
template<class Range, class Domain, class ES, template<class> class DerivativeTraits, size_t bufferSize>
class GridFunction<Range(Domain), ES, DerivativeTraits, bufferSize> :
  public TypeErasureBase<
    typename Imp::GridFunctionTraits<Range(Domain), ES, DerivativeTraits, bufferSize>::Concept,
    Imp::GridFunctionTraits<Range(Domain), ES, DerivativeTraits, bufferSize>::template Model>
{
  using Traits = Imp::GridFunctionTraits<Range(Domain), ES, DerivativeTraits, bufferSize>;
 
  using Base = TypeErasureBase<typename Traits::Concept, Traits::template Model>;
 
  using DerivativeInterface = typename Traits::DerivativeInterface;
 
  using LocalFunctionInterface = typename Traits::LocalFunctionInterface;
 
  using EntitySet = typename Traits::EntitySet;
 
public:
 
  template<class F, disableCopyMove<GridFunction, F> = 0 >
  GridFunction(F&& f) :
    Base(std::forward<F>(f))
  {
    static_assert(Dune::Functions::Concept::isGridFunction<F, Range(Domain), EntitySet>(), "Trying to construct a GridFunction from type that does not model the GridFunction concept");
  }
 
  GridFunction() = default;
 
  Range operator() (const Domain& x) const
  {
    return this->asInterface().operator()(x);
  }
 
  friend DerivativeInterface derivative(const GridFunction& t)
  {
    return t.asInterface().derivative();
  }
 
  friend LocalFunctionInterface localFunction(const GridFunction& t)
  {
    return t.asInterface().wrappedLocalFunction();
  }
 
  const EntitySet& entitySet() const
  {
    return this->asInterface().wrappedEntitySet();
  }
};
 
 
 
}} // namespace Dune::Functions
 
 
 
#endif // DUNE_FUNCTIONS_GRIDFUNCTIONS_GRID_FUNCTION_HH