block.hpp

Go to the documentation of this file.

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