ShapeMatrices-impl.h

Go to the documentation of this file.

 
namespace NumLib
{
namespace detail
{
/*
 * zero reset functions for Eigen
 */
template <class T>
void setMatrixZero(T& mat)
{
    // mat.setZero();
    const std::size_t n = mat.rows() * mat.cols();
    auto* v = mat.data();
    for (std::size_t i = 0; i < n; i++)
    {
        v[i] = .0;
    }
}
 
template <class T>
void setVectorZero(T& vec)
{
    // vec.setZero();
    const std::size_t n = vec.size();
    auto* v = vec.data();
    for (std::size_t i = 0; i < n; i++)
    {
        v[i] = .0;
    }
}
 
/*
 * "tag dispatching" technique is used below to emulate explicit specialization
 * of a template function in a template class
 */
template <ShapeMatrixType FIELD_TYPE>
struct ShapeDataFieldType
{
};
 
template <class T_N, class T_DNDR, class T_J, class T_DNDX>
inline void setZero(ShapeMatrices<T_N, T_DNDR, T_J, T_DNDX>& shape,
                    ShapeDataFieldType<ShapeMatrixType::N> /*unused*/)
{
    setVectorZero(shape.N);
}
 
template <class T_N, class T_DNDR, class T_J, class T_DNDX>
inline void setZero(ShapeMatrices<T_N, T_DNDR, T_J, T_DNDX>& shape,
                    ShapeDataFieldType<ShapeMatrixType::DNDR> /*unused*/)
{
    setMatrixZero(shape.dNdr);
}
 
template <class T_N, class T_DNDR, class T_J, class T_DNDX>
inline void setZero(ShapeMatrices<T_N, T_DNDR, T_J, T_DNDX>& shape,
                    ShapeDataFieldType<ShapeMatrixType::DNDR_J> /*unused*/)
{
    setZero(shape, ShapeDataFieldType<ShapeMatrixType::DNDR>());
    setMatrixZero(shape.J);
    shape.detJ = .0;
    shape.integralMeasure = 0.0;
}
 
template <class T_N, class T_DNDR, class T_J, class T_DNDX>
inline void setZero(ShapeMatrices<T_N, T_DNDR, T_J, T_DNDX>& shape,
                    ShapeDataFieldType<ShapeMatrixType::N_J> /*unused*/)
{
    setZero(shape, ShapeDataFieldType<ShapeMatrixType::N>());
    setZero(shape, ShapeDataFieldType<ShapeMatrixType::DNDR_J>());
}
 
template <class T_N, class T_DNDR, class T_J, class T_DNDX>
inline void setZero(ShapeMatrices<T_N, T_DNDR, T_J, T_DNDX>& shape,
                    ShapeDataFieldType<ShapeMatrixType::DNDX> /*unused*/)
{
    setZero(shape, ShapeDataFieldType<ShapeMatrixType::DNDR_J>());
    setMatrixZero(shape.invJ);
    setMatrixZero(shape.dNdx);
}
 
template <class T_N, class T_DNDR, class T_J, class T_DNDX>
inline void setZero(ShapeMatrices<T_N, T_DNDR, T_J, T_DNDX>& shape,
                    ShapeDataFieldType<ShapeMatrixType::ALL> /*unused*/)
{
    setZero(shape, ShapeDataFieldType<ShapeMatrixType::N>());
    setZero(shape, ShapeDataFieldType<ShapeMatrixType::DNDX>());
}
 
}  // namespace detail
 
template <class T_N, class T_DNDR, class T_J, class T_DNDX>
inline void ShapeMatrices<T_N, T_DNDR, T_J, T_DNDX>::setZero()
{
    detail::setZero(*this, detail::ShapeDataFieldType<ShapeMatrixType::ALL>());
}
 
template <class T_N, class T_DNDR, class T_J, class T_DNDX>
template <ShapeMatrixType T_SHAPE_MATRIX_TYPE>
inline void ShapeMatrices<T_N, T_DNDR, T_J, T_DNDX>::setZero()
{
    detail::setZero(*this, detail::ShapeDataFieldType<T_SHAPE_MATRIX_TYPE>());
}
 
template <class T_N, class T_DNDR, class T_J, class T_DNDX>
void ShapeMatrices<T_N, T_DNDR, T_J, T_DNDX>::write(std::ostream& out) const
{
    out << "N   :\n" << N << "\n";
    out << "dNdr:\n" << dNdr << "\n";
    out << "J   :\n" << J << "\n";
    out << "|J| : " << detJ << "\n";
    out << "invJ:\n" << invJ << "\n";
    out << "dNdx:\n" << dNdx << "\n";
}
 
template <class T_N, class T_DNDR, class T_J, class T_DNDX>
std::ostream& operator<<(std::ostream& os,
                         const ShapeMatrices<T_N, T_DNDR, T_J, T_DNDX>& shape)
{
    shape.write(os);
    return os;
}
 
}  // namespace NumLib