polynomials_rannacher_turek.h

// ---------------------------------------------------------------------
//
// Copyright (C) 2015 - 2018 by the deal.II authors
//
// This file is part of the deal.II library.
//
// The deal.II library is free software; you can use it, redistribute
// it, and/or modify it under the terms of the GNU Lesser General
// Public License as published by the Free Software Foundation; either
// version 2.1 of the License, or (at your option) any later version.
// The full text of the license can be found in the file LICENSE at
// the top level of the deal.II distribution.
//
// ---------------------------------------------------------------------


#ifndef dealii_polynomials_rannacher_turek_h
#define dealii_polynomials_rannacher_turek_h

#include <deal.II/base/point.h>
#include <deal.II/base/tensor.h>
#include <vector>

DEAL_II_NAMESPACE_OPEN


template <int dim>
class PolynomialsRannacherTurek
{
public:
  static const unsigned int dimension = dim;

  PolynomialsRannacherTurek();

  double compute_value(const unsigned int i,
                       const Point<dim> &p) const;

  template <int order>
  Tensor<order,dim> compute_derivative (const unsigned int i,
                                        const Point<dim> &p) const;

  Tensor<1, dim> compute_grad(const unsigned int i,
                              const Point<dim> &p) const;

  Tensor<2, dim> compute_grad_grad(const unsigned int i,
                                   const Point<dim> &p) const;

  void compute(const Point<dim> &unit_point,
               std::vector<double> &values,
               std::vector<Tensor<1, dim> > &grads,
               std::vector<Tensor<2,dim> > &grad_grads,
               std::vector<Tensor<3,dim> > &third_derivatives,
               std::vector<Tensor<4,dim> > &fourth_derivatives) const;
};


namespace internal
{
  namespace PolynomialsRannacherTurekImplementation
  {
    template <int order, int dim>
    inline
    Tensor<order,dim>
    compute_derivative (const unsigned int,
                        const Point<dim> &)
    {
      Assert (dim == 2, ExcNotImplemented());
      return Tensor<order,dim>();
    }


    template <int order>
    inline
    Tensor<order,2>
    compute_derivative (const unsigned int i,
                        const Point<2> &p)
    {
      const unsigned int dim = 2;

      Tensor<order,dim> derivative;
      switch (order)
        {
        case 1:
        {
          Tensor<1,dim> &grad = *reinterpret_cast<Tensor<1,dim>*>(&derivative);
          if (i == 0)
            {
              grad[0] = -2.5 + 3*p(0);
              grad[1] = 1.5 - 3*p(1);
            }
          else if (i == 1)
            {
              grad[0] = -0.5 + 3.0*p(0);
              grad[1] = 1.5 - 3.0*p(1);
            }
          else if (i == 2)
            {
              grad[0] = 1.5 - 3.0*p(0);
              grad[1] = -2.5 + 3.0*p(1);
            }
          else if (i == 3)
            {
              grad[0] = 1.5 - 3.0*p(0);
              grad[1] = -0.5 + 3.0*p(1);
            }
          else
            {
              Assert(false, ExcNotImplemented());
            }
          return derivative;
        }
        case 2:
        {
          Tensor<2,dim> &grad_grad = *reinterpret_cast<Tensor<2,dim>*>(&derivative);
          if (i == 0)
            {
              grad_grad[0][0] = 3;
              grad_grad[0][1] = 0;
              grad_grad[1][0] = 0;
              grad_grad[1][1] = -3;
            }
          else if (i == 1)
            {
              grad_grad[0][0] = 3;
              grad_grad[0][1] = 0;
              grad_grad[1][0] = 0;
              grad_grad[1][1] = -3;
            }
          else if (i == 2)
            {
              grad_grad[0][0] = -3;
              grad_grad[0][1] = 0;
              grad_grad[1][0] = 0;
              grad_grad[1][1] = 3;
            }
          else if (i == 3)
            {
              grad_grad[0][0] = -3;
              grad_grad[0][1] = 0;
              grad_grad[1][0] = 0;
              grad_grad[1][1] = 3;
            }
          return derivative;
        }
        default:
        {
          // higher derivatives are all zero
          return Tensor<order,dim>();
        }
        }
    }
  }
}



// template functions
template <int dim>
template <int order>
Tensor<order,dim>
PolynomialsRannacherTurek<dim>::compute_derivative (const unsigned int i,
                                                    const Point<dim> &p) const
{
  return internal::PolynomialsRannacherTurekImplementation::compute_derivative<order> (i, p);
}


DEAL_II_NAMESPACE_CLOSE

#endif