beams.hpp

Go to the documentation of this file.

#pragma once
 
#include <Kokkos_Core.hpp>
 
#include "dof_management/freedom_signature.hpp"
 
namespace kynema {
 
template <typename DeviceType>
struct Beams {
    template <typename ValueType>
    using View = Kokkos::View<ValueType, DeviceType>;
 
    size_t num_elems;       // Total number of element
    size_t max_elem_nodes;  // Maximum number of nodes per element
    size_t max_elem_qps;    // Maximum number of quadrature points per element
 
    // Element-based data
    View<size_t*> num_nodes_per_element;
    View<size_t*> num_qps_per_element;
    View<size_t**> node_state_indices;  // State row index for each node
    View<dof::FreedomSignature**> element_freedom_signature;
    View<size_t** [6]> element_freedom_table;
    View<double* [6]> element_mu;  // Damping coefficients per element
 
    View<double[3]> gravity;
 
    // Node-based data
    View<double** [7]> node_x0;      // Initial position/rotation
    View<double** [7]> node_u;       // State: translation/rotation displacement
    View<double** [6]> node_u_dot;   // State: translation/rotation velocity
    View<double** [6]> node_u_ddot;  // State: translation/rotation acceleration
    View<double** [6]> node_FX;      // External forces
 
    // Quadrature point data
    View<double**> qp_weight;           // Integration weights
    View<double**> qp_jacobian;         // Jacobian vector
    View<double** [6][6]> qp_Mstar;     // Mass matrix in material frame
    View<double** [6][6]> qp_Cstar;     // Stiffness matrix in material frame
    View<double** [7]> qp_x;            // Current position/orientation
    View<double** [3]> qp_x0;           // Initial position
    View<double** [3]> qp_x0_prime;     // Initial position derivative
    View<double** [4]> qp_r0;           // Initial rotation
    View<double** [3]> qp_u;            // State: translation displacement
    View<double** [3]> qp_u_prime;      // State: translation displacement derivative
    View<double** [3]> qp_u_dot;        // State: translation velocity
    View<double** [3]> qp_u_ddot;       // State: translation acceleration
    View<double** [4]> qp_r;            // State: rotation
    View<double** [4]> qp_r_prime;      // State: rotation derivative
    View<double** [3]> qp_omega;        // State: angular velocity
    View<double** [3]> qp_omega_dot;    // State: position/rotation
    View<double** [3]> qp_deformation;  // Deformation relative to rigid body motion
    View<double** [3][4]> qp_E;         // Quaternion derivative
    View<double** [6]> qp_Fe;           // External force
 
    View<double** [6]> residual_vector_terms;
    View<double*** [6][6]> system_matrix_terms;
 
    // Shape Function data
    View<double***> shape_interp;  // Shape function values
    View<double***> shape_deriv;   // Shape function derivatives
 
    // Constructor which initializes views based on given sizes
    Beams(const size_t n_beams, const size_t max_e_nodes, const size_t max_e_qps)
        : num_elems(n_beams),
          max_elem_nodes(max_e_nodes),
          max_elem_qps(max_e_qps),
          // Element Data
          num_nodes_per_element(
              Kokkos::view_alloc("num_nodes_per_element", Kokkos::WithoutInitializing), num_elems
          ),
          num_qps_per_element(
              Kokkos::view_alloc("num_qps_per_element", Kokkos::WithoutInitializing), num_elems
          ),
          node_state_indices(
              Kokkos::view_alloc("node_state_indices", Kokkos::WithoutInitializing), num_elems,
              max_elem_nodes
          ),
          element_freedom_signature(
              Kokkos::view_alloc("element_freedom_signature", Kokkos::WithoutInitializing),
              num_elems, max_elem_nodes
          ),
          element_freedom_table(
              Kokkos::view_alloc("element_freedom_table", Kokkos::WithoutInitializing), num_elems,
              max_elem_nodes
          ),
          element_mu(Kokkos::view_alloc("mu", Kokkos::WithoutInitializing), num_elems),
          gravity(Kokkos::view_alloc("gravity", Kokkos::WithoutInitializing)),
          // Node Data
          node_x0(
              Kokkos::view_alloc("node_x0", Kokkos::WithoutInitializing), num_elems, max_elem_nodes
          ),
          node_u(
              Kokkos::view_alloc("node_u", Kokkos::WithoutInitializing), num_elems, max_elem_nodes
          ),
          node_u_dot(
              Kokkos::view_alloc("node_u_dot", Kokkos::WithoutInitializing), num_elems,
              max_elem_nodes
          ),
          node_u_ddot(
              Kokkos::view_alloc("node_u_ddot", Kokkos::WithoutInitializing), num_elems,
              max_elem_nodes
          ),
          node_FX(
              Kokkos::view_alloc("node_force_external", Kokkos::WithoutInitializing), num_elems,
              max_elem_nodes
          ),
          // Quadrature Point data
          qp_weight(
              Kokkos::view_alloc("qp_weight", Kokkos::WithoutInitializing), num_elems, max_elem_qps
          ),
          qp_jacobian(
              Kokkos::view_alloc("qp_jacobian", Kokkos::WithoutInitializing), num_elems, max_elem_qps
          ),
          qp_Mstar(
              Kokkos::view_alloc("qp_Mstar", Kokkos::WithoutInitializing), num_elems, max_elem_qps
          ),
          qp_Cstar(
              Kokkos::view_alloc("qp_Cstar", Kokkos::WithoutInitializing), num_elems, max_elem_qps
          ),
          qp_x(Kokkos::view_alloc("qp_x", Kokkos::WithoutInitializing), num_elems, max_elem_qps),
          qp_x0(Kokkos::view_alloc("qp_x0", Kokkos::WithoutInitializing), num_elems, max_elem_qps),
          qp_x0_prime(
              Kokkos::view_alloc("qp_x0_prime", Kokkos::WithoutInitializing), num_elems, max_elem_qps
          ),
          qp_r0(Kokkos::view_alloc("qp_r0", Kokkos::WithoutInitializing), num_elems, max_elem_qps),
          qp_u(Kokkos::view_alloc("qp_u", Kokkos::WithoutInitializing), num_elems, max_elem_qps),
          qp_u_prime(
              Kokkos::view_alloc("qp_u_prime", Kokkos::WithoutInitializing), num_elems, max_elem_qps
          ),
          qp_u_dot(
              Kokkos::view_alloc("qp_u_dot", Kokkos::WithoutInitializing), num_elems, max_elem_qps
          ),
          qp_u_ddot(
              Kokkos::view_alloc("qp_u_ddot", Kokkos::WithoutInitializing), num_elems, max_elem_qps
          ),
          qp_r(Kokkos::view_alloc("qp_r", Kokkos::WithoutInitializing), num_elems, max_elem_qps),
          qp_r_prime(
              Kokkos::view_alloc("qp_r_prime", Kokkos::WithoutInitializing), num_elems, max_elem_qps
          ),
          qp_omega(
              Kokkos::view_alloc("qp_omega", Kokkos::WithoutInitializing), num_elems, max_elem_qps
          ),
          qp_omega_dot(
              Kokkos::view_alloc("qp_omega_dot", Kokkos::WithoutInitializing), num_elems,
              max_elem_qps
          ),
          qp_deformation(
              Kokkos::view_alloc("qp_deformation", Kokkos::WithoutInitializing), num_elems,
              max_elem_qps
          ),
          qp_Fe(Kokkos::view_alloc("qp_Fe", Kokkos::WithoutInitializing), num_elems, max_elem_qps),
          residual_vector_terms(
              Kokkos::view_alloc("residual_vector_terms", Kokkos::WithoutInitializing), num_elems,
              max_elem_nodes
          ),
          system_matrix_terms(
              Kokkos::view_alloc("system_matrix_terms", Kokkos::WithoutInitializing), num_elems,
              max_elem_nodes, max_elem_nodes
          ),
          // Shape Function data
          shape_interp(
              Kokkos::view_alloc("shape_interp", Kokkos::WithoutInitializing), num_elems,
              max_elem_nodes, max_elem_qps
          ),
          shape_deriv(
              Kokkos::view_alloc("deriv_interp", Kokkos::WithoutInitializing), num_elems,
              max_elem_nodes, max_elem_qps
          ) {
        Kokkos::deep_copy(element_freedom_signature, dof::FreedomSignature::AllComponents);
    }
};
 
}  // namespace kynema