ShapeFunction.hpp

// This file is part of the imaging2 class library.
//
// University of Innsbruck, Infmath Imaging, 2009.
// http://infmath.uibk.ac.at
//
// All rights reserved.


#ifndef SHAPEFUNCTION_H
#define SHAPEFUNCTION_H

#include <core/imaging2.hpp>


namespace imaging
{
  template<size_t N_NODES, size_t N_FACE_NODES, size_t N>
  class ShapeFunction
  {
  public:
    static const size_t n_element_nodes = N_NODES;
    
    static const size_t n_shape_face_nodes = N_FACE_NODES;
    
    static size_t face_node(size_t face, size_t node);
  
    float_t value(size_t node_index, const ublas::fixed_vector<float_t, N> & in) const;
    
    ublas::fixed_vector<float_t, N> & gradient(size_t node_index, const ublas::fixed_vector<float_t, N> & in, ublas::fixed_vector<float_t, N> & out) const;
  };

  class Bilinear2dShapeFunction : public ShapeFunction<4, 2, 2>
  {
  public:
    static size_t face_node(size_t face, size_t node);
    
    float_t value(size_t node, const ublas::fixed_vector<float_t, 2> & in) const;

    ublas::fixed_vector<float_t, 2> & gradient(size_t node_index, const ublas::fixed_vector<float_t, 2> & in, ublas::fixed_vector<float_t, 2> & out) const;
  };
  

class Trilinear3dShapeFunction : public ShapeFunction<8, 4, 3>
  {
  public:
    static size_t face_node(size_t face, size_t node);
    
    float_t value(size_t node, const ublas::fixed_vector<float_t, 3> & in) const;

    ublas::fixed_vector<float_t, 3> & gradient(size_t node_index, const ublas::fixed_vector<float_t, 3> & in, ublas::fixed_vector<float_t, 3> & out) const;
  };


  class Linear2dShapeFunction : public ShapeFunction<3, 2, 2>
  {
  public:
    static size_t face_node(size_t face, size_t node);
    
    float_t value(size_t node_index, const ublas::fixed_vector<float_t, 2> & in) const;


    ublas::fixed_vector<float_t, 2> & gradient(size_t node_index, const ublas::fixed_vector<float_t, 2> & in, ublas::fixed_vector<float_t, 2> & out) const;

  };
  
  class Linear3dShapeFunction : public ShapeFunction<4, 3, 3>
  {
  public:
    static size_t face_node(size_t face, size_t node);
    float_t value(size_t node_index, const ublas::fixed_vector<float_t, 3> & in) const;

    ublas::fixed_vector<float_t, 3> & gradient(size_t node_index, const ublas::fixed_vector<float_t, 3> & in, ublas::fixed_vector<float_t, 3> & out) const;

  };


  class Linear1dShapeFunction : public ShapeFunction<2, 1, 1>
  {
  public:
    static size_t face_node(size_t face, size_t node);
    
    float_t value(size_t node, const ublas::fixed_vector<float_t, 1> & in) const;


    ublas::fixed_vector<float_t, 1> & gradient(size_t node, const ublas::fixed_vector<float_t, 1> & in, ublas::fixed_vector<float_t, 1> & out) const;
  };

}


#endif

---