sqrt.hh

// -*- tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*-
// vi: set et ts=8 sw=2 sts=2:
#ifndef DUNE_PDELAB_FUNCTION_SQRT_HH
#define DUNE_PDELAB_FUNCTION_SQRT_HH
 
#include <dune/pdelab/common/function.hh>
 
namespace Dune {
  namespace PDELab {
 
 
    template<typename GF>
    class SqrtGridFunctionAdapter
      : public GridFunctionBase<typename GF::Traits,
                                SqrtGridFunctionAdapter<GF> >
    {
      static_assert(GF::Traits::dimRange == 1, "Dimension of range must "
                    "be 1 to take the sqrt");
 
      typedef typename GF::Traits T;
      typedef GridFunctionBase<T, SqrtGridFunctionAdapter<GF> >
              Base;
      typedef typename T::RangeFieldType RF;
 
      GF& gf;
 
    public:
      typedef typename Base::Traits Traits;
 
      SqrtGridFunctionAdapter(GF& gf_)
        : gf(gf_)
      { }
 
      void evaluate(const typename Traits::ElementType &e,
                    const typename Traits::DomainType &x,
                    typename Traits::RangeType &y) const {
        using std::sqrt;
        gf.evaluate(e,x,y);
        y[0] = sqrt(y[0]);
      }
 
      const typename Traits::GridViewType& getGridView() const {
        return gf.getGridView();
      }
 
      template<typename Time>
      void setTime(Time time) { gf.setTime(time); }
    };
 
  } // namspace PDELab
} // namspace Dune
 
#endif // DUNE_PDELAB_FUNCTION_SQRT_HH