beams_input.hpp

Go to the documentation of this file.

#pragma once
 
#include <algorithm>
#include <array>
#include <numeric>
#include <span>
#include <vector>
 
#include "beam_element.hpp"
 
namespace kynema {
 
struct BeamsInput {
    std::vector<BeamElement> elements;  //< Elements in the beam
    std::array<double, 3> gravity;      //< Gravity vector
 
    BeamsInput(std::span<const BeamElement> elems, std::span<const double, 3> g)
        : gravity({g[0], g[1], g[2]}) {
        elements.assign(std::begin(elems), std::end(elems));
    }
 
    [[nodiscard]] size_t NumElements() const { return elements.size(); }
 
    template <typename Accessor>
    [[nodiscard]] size_t ComputeSum(Accessor accessor) const {
        return std::transform_reduce(
            elements.begin(), elements.end(), size_t{0U}, std::plus{},
            [&accessor](const BeamElement& e) {
                return accessor(e);
            }
        );
    }
 
    template <typename Accessor>
    [[nodiscard]] size_t ComputeMax(Accessor accessor) const {
        return std::transform_reduce(
            elements.begin(), elements.end(), size_t{0U},
            [](auto a, auto b) {
                return std::max(a, b);
            },
            [&accessor](const auto& e) {
                return accessor(e);
            }
        );
    }
 
    [[nodiscard]] size_t NumNodes() const {
        return ComputeSum([](const BeamElement& e) {
            return e.node_ids.size();
        });
    }
 
    [[nodiscard]] size_t NumQuadraturePoints() const {
        return ComputeSum([](const BeamElement& e) {
            return e.quadrature.size();
        });
    }
 
    [[nodiscard]] size_t MaxElemNodes() const {
        return ComputeMax([](const BeamElement& e) {
            return e.node_ids.size();
        });
    }
 
    [[nodiscard]] size_t MaxElemQuadraturePoints() const {
        return ComputeMax([](const BeamElement& e) {
            return e.quadrature.size();
        });
    }
};
 
}  // namespace kynema