gauss_field_view.hpp

#ifndef NIHU_GAUSS_FIELD_VIEW_HPP_INCLUDED
#define NIHU_GAUSS_FIELD_VIEW_HPP_INCLUDED
 
#include "core/field.hpp"
 
#include "line_1_gauss_field.hpp"
#include "quad_1_gauss_shape_set.hpp"
#include "tria_1_gauss_shape_set.hpp"
 
namespace NiHu
{
 
namespace field_option
{
    struct gauss {};
} // namespace NiHu::field_option
 
namespace field_traits
{
 
namespace internal
{
    template <class Domain>
    struct assign_gauss_shape;
 
    template <>
    struct assign_gauss_shape<line_domain> 
    {
        typedef line_1_gauss_shape_set type;
    };
 
    template <>
    struct assign_gauss_shape<tria_domain>
    {
        typedef tria_1_gauss_shape_set type;
    };
 
    template <>
    struct assign_gauss_shape<quad_domain>
    {
        typedef quad_1_gauss_shape_set type;
    };
} // namespace NiHu::field_traits::internal
 
 
template <class ElemType, class Dimension>
struct nset_type<field_view<ElemType, field_option::gauss, Dimension> >
{
    typedef typename internal::assign_gauss_shape<typename ElemType::domain_t>::type type;
};
} // namespace NiHu::field_traits
 
template <class ElemType, class Dimension>
class field_impl<field_view<ElemType, field_option::gauss, Dimension> > :
    public ElemType
{
    typedef field_view<ElemType, field_option::gauss, Dimension> Derived;
    typedef typename field_traits::dof_vector_type<Derived>::type dofs_t;
    typedef typename field_traits::dof_vector_return_type<Derived>::type dofs_return_t;
    typedef typename field_traits::eval_return_type<Derived>::type eval_return_t;
    typedef typename field_traits::nset_type<Derived>::type nset_t;
    typedef typename ElemType::xi_t xi_t;
public:
 
    eval_return_t eval(xi_t const &xi) const
    {
        return nset_t::template eval_shape<0>(xi);
    }
 
 
    ElemType const &get_elem() const
    {
        return *this;   // static cast
    }
 
    dofs_return_t get_dofs() const
    {
        // TODO: in an inhomogeneous mesh, dof assignment is more complex!
        enum { D = Dimension::value * shape_set_traits::num_nodes<nset_t>::value };
        return dofs_t::LinSpaced(D, ElemType::get_id()(0)*D, ElemType::get_id()(0)*D + D - 1);
    }
};
 
template <class Elem, class Dimension = field_dimension::_1d>
field_view<Elem, field_option::gauss, Dimension> const &
gauss_view(element_base<Elem> const &e, Dimension dim = Dimension())
{
    return create_field_view(e, field_option::gauss(), dim);
}
 
} // namespace NiHu
 
#endif /* NIHU_GAUSS_FIELD_VIEW_HPP_INCLUDED */