create_springs.hpp

Go to the documentation of this file.

#pragma once
#include <span>
 
#include "model/node.hpp"
#include "springs.hpp"
#include "springs_input.hpp"
 
namespace kynema {
 
template <typename DeviceType>
inline Springs<DeviceType> CreateSprings(
    const SpringsInput& springs_input, std::span<const Node> nodes
) {
    using Kokkos::create_mirror_view;
    using Kokkos::deep_copy;
    using Kokkos::WithoutInitializing;
 
    Springs<DeviceType> springs(springs_input.NumElements());
 
    auto host_node_state_indices =
        create_mirror_view(WithoutInitializing, springs.node_state_indices);
    auto host_x0 = create_mirror_view(WithoutInitializing, springs.x0);
    auto host_l_ref = create_mirror_view(WithoutInitializing, springs.l_ref);
    auto host_k = create_mirror_view(WithoutInitializing, springs.k);
 
    for (auto elementIdx : std::views::iota(0U, springs_input.NumElements())) {
        const auto& element = springs_input.elements[elementIdx];
 
        host_node_state_indices(elementIdx, 0U) = static_cast<size_t>(element.node_ids[0]);
        host_node_state_indices(elementIdx, 1U) = static_cast<size_t>(element.node_ids[1]);
 
        for (auto component : std::views::iota(0U, 3U)) {
            host_x0(elementIdx, component) =
                nodes[element.node_ids[1]].x0[component] - nodes[element.node_ids[0]].x0[component];
        }
 
        host_l_ref(elementIdx) = element.undeformed_length;
        host_k(elementIdx) = element.stiffness;
    }
 
    deep_copy(springs.node_state_indices, host_node_state_indices);
    deep_copy(springs.x0, host_x0);
    deep_copy(springs.l_ref, host_l_ref);
    deep_copy(springs.k, host_k);
 
    return springs;
}
 
}  // namespace kynema