Eigen.h

Go to the documentation of this file.

 
#ifndef PBAT_COMMON_EIGEN_H
#define PBAT_COMMON_EIGEN_H
 
#include "Concepts.h"
 
#include <pbat/Aliases.h>
#include <ranges>
 
namespace pbat {
namespace common {

template <CContiguousArithmeticRange R>
auto ToEigen(R&& r)
    -> Eigen::Map<Eigen::Vector<std::ranges::range_value_t<R>, Eigen::Dynamic> const>
{
    namespace rng = std::ranges;
    return Eigen::Map<Eigen::Vector<std::ranges::range_value_t<R>, Eigen::Dynamic> const>(
        rng::data(r),
        static_cast<Eigen::Index>(rng::size(r)));
}

template <CContiguousArithmeticMatrixRange R>
auto ToEigen(R&& r) -> Eigen::Map<Eigen::Matrix<
    typename std::ranges::range_value_t<R>::Scalar,
    Eigen::Dynamic,
    Eigen::Dynamic> const>
{
    namespace rng    = std::ranges;
    using ValueType  = rng::range_value_t<R>;
    using ScalarType = typename ValueType::Scalar;
    auto rows        = ValueType::RowsAtCompileTime;
    auto cols        = ValueType::ColsAtCompileTime;
    if constexpr (ValueType::Flags & Eigen::RowMajorBit)
        std::swap(rows, cols);
 
    return Eigen::Map<Eigen::Matrix<ScalarType, Eigen::Dynamic, Eigen::Dynamic> const>(
        static_cast<ScalarType const*>(std::addressof(rng::data(r)[0][0])),
        rows,
        static_cast<Eigen::Index>(rng::size(r)) * static_cast<int>(cols));
}
 
namespace detail {
 
template <std::ranges::random_access_range R>
struct Slice
{
    Slice(R&& _r) : r(std::forward<R>(_r)) {}
 
    Index size() const { return static_cast<Index>(std::ranges::size(r)); }
    Index operator[](Index i) const { return static_cast<Index>(r[i]); }
 
    std::remove_cvref_t<R> r;
};
 
} // namespace detail

template <std::ranges::random_access_range R>
auto Slice(R&& r)
{
    return detail::Slice<R>(std::forward<R>(r));
}
 
} // namespace common
} // namespace pbat
 
#endif // PBAT_COMMON_EIGEN_H