stride.hpp

Go to the documentation of this file.

// SPDX-License-Identifier: BSD-2-Clause
 
#ifndef pRC_CORE_TENSOR_FUNCTIONS_STRIDE_H
#define pRC_CORE_TENSOR_FUNCTIONS_STRIDE_H
 
#include <prc/core/functors/view.hpp>
#include <prc/core/tensor/declarations.hpp>
#include <prc/core/tensor/views/stride.hpp>
 
namespace pRC
{
    template<Size... Ss, class X, IsConvertible<Index>... Os,
        IsTensorish R = RemoveReference<X>>
        requires(sizeof...(Ss) == R::Dimension &&
            sizeof...(Os) == R::Dimension &&
            typename R::Sizes() % Sizes<Ss...>() == Constant<Size, 0>())
    static inline constexpr auto stride(X &&a, Os const... offsets)
    {
        if constexpr(IsInvocable<View, X>)
        {
            using V = RemoveReference<decltype(view(forward<X>(a)))>;
            return TensorViews::Stride<typename R::Type,
                decltype(typename R::Sizes() / Sizes<Ss...>()), Sizes<Ss...>,
                V>(view(forward<X>(a)), Subscripts<Ss...>(offsets...));
        }
        else
        {
            return eval(stride<Ss...>(a, offsets...));
        }
    }
}
#endif // pRC_CORE_TENSOR_FUNCTIONS_STRIDE_H