ShapeValidation.hpp

#pragma once
 
#include <cstddef>
 
#include <stratax/core/Exceptions.hpp>
 
namespace stratax::core::validation {

inline std::size_t nonnegative_shape_dimension(std::ptrdiff_t value, const char* message)
{
    if (value < 0)
    {
        throw Exceptions::ShapeError(message);
    }
 
    return static_cast<std::size_t>(value);
}

inline std::size_t positive_shape_dimension(std::ptrdiff_t value, const char* message)
{
    if (value <= 0)
    {
        throw Exceptions::ShapeError(message);
    }
 
    return static_cast<std::size_t>(value);
}

inline void require_positive_shape_dimension(std::size_t value, const char* message)
{
    if (value == 0)
    {
        throw Exceptions::ShapeError(message);
    }
}

template<typename Lhs, typename Rhs>
[[nodiscard]] bool same_shape(const Lhs& lhs, const Rhs& rhs)
{
    return lhs.size() == rhs.size() && lhs.shape() == rhs.shape();
}

template<typename Lhs, typename Rhs>
void require_same_shape(const Lhs& lhs, const Rhs& rhs, const char* message)
{
    if (!same_shape(lhs, rhs))
    {
        throw Exceptions::ShapeError(message);
    }
}

inline void require_equal_size(std::size_t lhs, std::size_t rhs, const char* message)
{
    if (lhs != rhs)
    {
        throw Exceptions::ShapeError(message);
    }
}
 
}