Quad4.hpp

Go to the documentation of this file.

#pragma once
// Auto-generated by tools/gen_shape_functions -- DO NOT EDIT
// Element: Quad4
// Regenerate: /usr/bin/python3 -m tools.gen_shape_functions.generate
 
#include "DNDS/Defines.hpp"
#include "Geom/Geometric.hpp"
#include "Geom/ElemEnum.hpp"
#include "Geom/ElementTraitsBase.hpp"
 
namespace DNDS::Geom::Elem
{
 
    // Forward declaration (primary template is in ElementTraitsBase.hpp)
    template <ElemType>
    struct ShapeFuncImpl;
 
    // <GEN_SHAPE_FUNCS_BEGIN>
    template <>
    struct ShapeFuncImpl<Quad4>
    {
        template <class TPoint, class TArray>
        DNDS_DEVICE_CALLABLE static void Diff0(const TPoint &p, TArray &&v)
        {
            t_real xi = p[0];
            t_real et = p[1];
            const t_real _t0 = xi - 1;
            const t_real _t1 = ((0.25)) * et + (-0.25);
            const t_real _t2 = xi + 1;
            const t_real _t3 = ((0.25)) * et + (0.25);
            v(0, 0) = _t0 * _t1;
            v(0, 1) = -_t1 * _t2;
            v(0, 2) = _t2 * _t3;
            v(0, 3) = -_t0 * _t3;
        }
 
        template <class TPoint, class TArray>
        DNDS_DEVICE_CALLABLE static void Diff1(const TPoint &p, TArray &&v)
        {
            t_real xi = p[0];
            t_real et = p[1];
            const t_real _t0 = ((0.25)) * et;
            const t_real _t1 = _t0 + (-0.25);
            const t_real _t2 = _t0 + (0.25);
            const t_real _t3 = ((0.25)) * xi;
            const t_real _t4 = _t3 + (-0.25);
            const t_real _t5 = _t3 + (0.25);
            v(0, 0) = _t1;
            v(0, 1) = -_t1;
            v(0, 2) = _t2;
            v(0, 3) = -_t2;
            v(1, 0) = _t4;
            v(1, 1) = -_t5;
            v(1, 2) = _t5;
            v(1, 3) = -_t4;
        }
 
        template <class TPoint, class TArray>
        DNDS_DEVICE_CALLABLE static void Diff2(const TPoint &p, TArray &&v)
        {
            t_real xi = p[0];
            t_real et = p[1];
            v(1, 0) = (0.25);
            v(1, 1) = (-0.25);
            v(1, 2) = (0.25);
            v(1, 3) = (-0.25);
        }
 
        template <class TPoint, class TArray>
        DNDS_DEVICE_CALLABLE static void Diff3(const TPoint &p, TArray &&v)
        {
            t_real xi = p[0];
            t_real et = p[1];
            // all zero
        }
    };
    // <GEN_SHAPE_FUNCS_END>
 
    // <GEN_ELEM_TRAITS_BEGIN>
 
    template <>
    struct ElementTraits<Quad4>
    {
        static constexpr ElemType elemType = Quad4;
        static constexpr int dim = 2;
        static constexpr int order = 1;
        static constexpr int numVertices = 4;
        static constexpr int numNodes = 4;
        static constexpr int numFaces = 4;
        static constexpr ParamSpace paramSpace = QuadSpace;
        static constexpr t_real paramSpaceVol = 4.0;
        // 3 * NNodes is a compile-time constant; no overflow possible.
        // NOLINTNEXTLINE(bugprone-implicit-widening-of-multiplication-result)
        static constexpr std::array<t_real, 3 * 4> standardCoords = {
            -1, -1, 0, // Node 0: vertex
            1, -1, 0,  // Node 1: vertex
            1, 1, 0,   // Node 2: vertex
            -1, 1, 0}; // Node 3: vertex
 
        static constexpr ElemType GetFaceType(t_index /*iFace*/) { return Line2; }
 
        static constexpr std::array<std::array<t_index, 10>, 4> faceNodes = {{{0, 1},
                                                                              {1, 2},
                                                                              {2, 3},
                                                                              {3, 0}}};
 
        static constexpr ElemType elevatedType = Quad9;
        static constexpr int numElevNodes = 5;
 
        static constexpr std::array<tElevSpan, 5> elevSpans = {{{0, 1},
                                                                {1, 2},
                                                                {2, 3},
                                                                {3, 0},
                                                                {0, 1, 2, 3}}};
 
        static constexpr std::array<ElemType, 5> elevNodeSpanTypes = {
            Line2, Line2, Line2, Line2, Quad4};
 
        static constexpr int vtkCellType = 9;
 
        static constexpr std::array<int, 4> vtkNodeOrder = {0, 1, 2, 3};
    };
    // <GEN_ELEM_TRAITS_END>
 
} // namespace DNDS::Geom::Elem