freedom_signature.hpp

Go to the documentation of this file.

#pragma once
 
#include <type_traits>
 
#include <Kokkos_Core.hpp>
 
namespace kynema::dof {
 
enum class FreedomSignature : std::uint8_t {
    AllComponents = 0b00111111,  //< Enable all 6 degrees of freedom
    JustPosition = 0b00111000,   //< Enable only translational DOFs
    JustRotation = 0b00000111,   //< Enable only rotational DOFs
    NoComponents = 0b00000000    //< Disable all degrees of freedom
};
 
KOKKOS_INLINE_FUNCTION
FreedomSignature operator|(FreedomSignature x, FreedomSignature y) {
    using T = std::underlying_type_t<FreedomSignature>;
    return static_cast<FreedomSignature>(static_cast<T>(x) | static_cast<T>(y));
}
 
KOKKOS_INLINE_FUNCTION
size_t count_active_dofs(FreedomSignature x) {
    using T = std::underlying_type_t<FreedomSignature>;
    auto count = 0UL;
    constexpr auto zero = T{0};
    constexpr auto one = T{1};
    for (auto value = static_cast<T>(x); value > zero; value = value >> 1) {
        count += value & one;
    }
    return count;
}
 
}  // namespace kynema::dof