Method get_range

cpp/nd/array.hpp:969–986

template <typename I>
array array::concrete_holder_<I>::get_range(int64_t start, int64_t end) const
{
    if constexpr (impl::has_data_member_function_v<I> && !impl::is_scalar_v<I> &&
                  !impl::has_is_none_member_variable_v<I>) {
        auto s = impl_.shape();
        ASSERT(!s.empty());
        auto v =
            std::accumulate(s.begin() + 1, s.end(), 1UL, std::multiplies<uint64_t>()) * nd::dtype_bytes(impl_.dtype());
        auto d = impl_.data().data();
        auto sh = icm::small_vector<int64_t>(s.begin(), s.end());
        sh[0] = end - start;
        static_assert(impl::has_owner_member_function_v<I>);
        return array(impl::std_span_array_nd(
            impl_.owner(), std::span<const uint8_t>(d + start * v, (end - start) * v), icm::shape(sh), impl_.dtype()));
    } else {
        throw invalid_operation("range() method is not implemented for this array.");
    }
}

} // namespace nd