interpolate_QP_position.hpp

Go to the documentation of this file.

#pragma once
 
#include <Kokkos_Core.hpp>
 
namespace kynema::beams {
 
template <typename DeviceType>
struct InterpolateQPPosition {
    template <typename ValueType>
    using View = Kokkos::View<ValueType, DeviceType>;
    template <typename ValueType>
    using ConstView = typename View<ValueType>::const_type;
 
    ConstView<size_t*> num_nodes_per_element;
    ConstView<size_t*> num_qps_per_element;
    ConstView<double***> shape_interpolation;  //< num_elem x num_nodes x num_qps
    ConstView<double** [7]>
        node_position_rotation;      //< node position vector/generalized coordinates in global csys
    View<double** [3]> qp_position;  //< quadrature point position - x, y, z (computed)
 
    KOKKOS_FUNCTION
    void operator()(int element) const {
        using Kokkos::Array;
 
        const auto num_nodes = num_nodes_per_element(element);
        const auto num_qps = num_qps_per_element(element);
        for (auto qp = 0U; qp < num_qps; ++qp) {
            auto local_result = Array<double, 3>{};
            for (auto node = 0U; node < num_nodes; ++node) {
                const auto phi = shape_interpolation(element, node, qp);
                for (auto component = 0U; component < 3U; ++component) {
                    local_result[component] +=
                        node_position_rotation(element, node, component) * phi;
                }
            }
            for (auto component = 0U; component < 3U; ++component) {
                qp_position(element, qp, component) = local_result[component];
            }
        }
    }
};
 
}  // namespace kynema::beams