10#ifndef XTENSOR_VIEW_HPP
11#define XTENSOR_VIEW_HPP
20#include <xtl/xclosure.hpp>
21#include <xtl/xmeta_utils.hpp>
22#include <xtl/xsequence.hpp>
23#include <xtl/xtype_traits.hpp>
25#include "../containers/xarray.hpp"
26#include "../containers/xcontainer.hpp"
27#include "../containers/xtensor.hpp"
28#include "../core/xaccessible.hpp"
29#include "../core/xiterable.hpp"
30#include "../core/xsemantic.hpp"
31#include "../core/xtensor_config.hpp"
32#include "../core/xtensor_forward.hpp"
33#include "../views/xbroadcast.hpp"
34#include "../views/xslice.hpp"
35#include "../views/xview_utils.hpp"
46 template <
class Tag,
class CT,
class... S>
49 template <
class CT,
class... S>
55 template <
class CT,
class... S>
60 template <
class CT,
class... S>
61 using xview_base_t =
typename xview_base<CT, S...>::type;
68 template <
bool is_const,
class CT,
class... S>
71 template <
class ST,
class... S>
78 struct is_xrange : std::false_type
83 struct is_xrange<xrange<T>> : std::true_type
88 struct is_xall_slice : std::false_type
93 struct is_xall_slice<xall<T>> : std::true_type
97 template <layout_type L,
bool val
id,
bool all_seen,
bool range_seen,
class V>
98 struct is_contiguous_view_impl
100 static constexpr bool value =
false;
104 struct static_dimension
106 static constexpr std::ptrdiff_t value = -1;
109 template <
class T, std::
size_t N>
110 struct static_dimension<std::array<T, N>>
112 static constexpr std::ptrdiff_t value =
static_cast<std::ptrdiff_t
>(N);
115 template <
class T, std::
size_t N>
116 struct static_dimension<xt::const_array<T, N>>
118 static constexpr std::ptrdiff_t value =
static_cast<std::ptrdiff_t
>(N);
121 template <std::size_t... I>
122 struct static_dimension<xt::fixed_shape<I...>>
124 static constexpr std::ptrdiff_t value =
sizeof...(I);
129 template <
class CT,
class... S>
130 struct is_xscalar_impl<xview<CT, S...>>
132 static constexpr bool value =
static_cast<std::ptrdiff_t
>(integral_count<S...>()
133 ) == static_dimension<
typename std::decay_t<CT>::shape_type>::value
139 struct is_strided_slice_impl : std::true_type
144 struct is_strided_slice_impl<xkeep_slice<T>> : std::false_type
149 struct is_strided_slice_impl<xdrop_slice<T>> : std::false_type
154 template <
class E,
class... S>
155 struct is_strided_view
156 : std::integral_constant<
158 std::conjunction<has_data_interface<E>, is_strided_slice_impl<std::decay_t<S>>...>::value>
172 template <
bool val
id,
bool all_seen,
bool range_seen,
class V>
175 using slice = xtl::mpl::front_t<V>;
176 static constexpr bool is_range_slice = is_xrange<slice>::value;
177 static constexpr bool is_int_slice = xtl::is_integral<slice>::value;
178 static constexpr bool is_all_slice = is_xall_slice<slice>::value;
179 static constexpr bool have_all_seen = all_seen || is_all_slice;
180 static constexpr bool have_range_seen = is_range_slice;
182 static constexpr bool is_valid = valid
185 : (!range_seen && (is_int_slice || is_range_slice)));
188 have_all_seen, range_seen || is_range_slice,
189 xtl::mpl::pop_front_t < V >> ::value;
192 template <
bool val
id,
bool all_seen,
bool range_seen>
193 struct is_contiguous_view_impl<
layout_type::
row_major, valid, all_seen, range_seen, xtl::mpl::vector<>>
195 static constexpr bool value = valid;
201 template <
bool val
id,
bool int_seen,
bool range_seen,
class V>
204 using slice = xtl::mpl::front_t<V>;
205 static constexpr bool is_range_slice = is_xrange<slice>::value;
206 static constexpr bool is_int_slice = xtl::is_integral<slice>::value;
207 static constexpr bool is_all_slice = is_xall_slice<slice>::value;
209 static constexpr bool have_int_seen = int_seen || is_int_slice;
211 static constexpr bool is_valid = valid
214 : (!range_seen && (is_all_slice || is_range_slice)));
216 have_int_seen, is_range_slice || range_seen,
217 xtl::mpl::pop_front_t < V >> ::value;
220 template <
bool val
id,
bool int_seen,
bool range_seen>
223 static constexpr bool value = valid;
227 template <
class E,
class... S>
228 struct is_contiguous_view
229 : std::integral_constant<
231 has_data_interface<E>::value
233 E::static_layout == layout_type::column_major
234 && static_cast<std::size_t>(static_dimension<typename E::shape_type>::value) != sizeof...(S)
236 && is_contiguous_view_impl<E::static_layout, true, false, false, xtl::mpl::vector<S...>>::value>
240 template <layout_type L,
class T, std::ptrdiff_t offset>
241 struct unwrap_offset_container
246 template <
class T, std::ptrdiff_t offset>
249 using type = sequence_view<T, offset, static_dimension<T>::value>;
252 template <
class T, std::ptrdiff_t start, std::ptrdiff_t end, std::ptrdiff_t offset>
255 using type = sequence_view<T, start + offset, end>;
258 template <
class T, std::ptrdiff_t offset>
261 using type = sequence_view<T, 0, static_dimension<T>::value - offset>;
264 template <
class T, std::ptrdiff_t start, std::ptrdiff_t end, std::ptrdiff_t offset>
267 using type = sequence_view<T, start, end - offset>;
270 template <
class E,
class... S>
271 struct get_contigous_shape_type
274 using type = std::conditional_t<
275 std::disjunction<is_xrange<S>...>::value,
276 typename xview_shape_type<
typename E::shape_type, S...>::type,
278 typename unwrap_offset_container<E::static_layout,
typename E::inner_shape_type, integral_count<S...>()>::type>;
282 struct is_sequence_view : std::integral_constant<bool, false>
286 template <
class T, std::ptrdiff_t S, std::ptrdiff_t E>
287 struct is_sequence_view<sequence_view<T, S, E>> : std::integral_constant<bool, true>
292 template <
class CT,
class... S>
295 using xexpression_type = std::decay_t<CT>;
296 using reference = inner_reference_t<CT>;
297 using const_reference =
typename xexpression_type::const_reference;
298 using size_type =
typename xexpression_type::size_type;
299 using temporary_type = view_temporary_type_t<xexpression_type, S...>;
301 static constexpr layout_type layout = detail::is_contiguous_view<xexpression_type, S...>::value
302 ? xexpression_type::static_layout
305 static constexpr bool is_const = std::is_const<std::remove_reference_t<CT>>::value;
307 using extract_storage_type = xtl::mpl::eval_if_t<
309 detail::expr_storage_type<xexpression_type>,
311 using storage_type = std::conditional_t<is_const, const extract_storage_type, extract_storage_type>;
314 template <
class CT,
class... S>
317 using xexpression_type = std::decay_t<CT>;
319 static constexpr bool is_strided_view = detail::is_strided_view<xexpression_type, S...>::value;
320 static constexpr bool is_contiguous_view = detail::is_contiguous_view<xexpression_type, S...>::value;
322 using inner_shape_type = std::conditional_t<
324 typename detail::get_contigous_shape_type<xexpression_type, S...>::type,
325 typename xview_shape_type<
typename xexpression_type::shape_type, S...>::type>;
327 using stepper = std::conditional_t<
332 using const_stepper = std::conditional_t<
352 template <
class CT,
class... S>
353 class xview :
public xview_semantic<xview<CT, S...>>,
354 public std::conditional_t<
355 detail::is_contiguous_view<std::decay_t<CT>, S...>::value,
356 xcontiguous_iterable<xview<CT, S...>>,
357 xiterable<xview<CT, S...>>>,
358 public xaccessible<xview<CT, S...>>,
359 public extension::xview_base_t<CT, S...>
363 using self_type =
xview<CT, S...>;
365 using xexpression_type = std::decay_t<CT>;
366 using semantic_base = xview_semantic<self_type>;
369 using accessible_base = xaccessible<self_type>;
370 using extension_base = extension::xview_base_t<CT, S...>;
371 using expression_tag =
typename extension_base::expression_tag;
373 static constexpr bool is_const = std::is_const<std::remove_reference_t<CT>>::value;
374 using value_type =
typename xexpression_type::value_type;
375 using simd_value_type = xt_simd::simd_type<value_type>;
376 using bool_load_type =
typename xexpression_type::bool_load_type;
377 using reference =
typename inner_types::reference;
378 using const_reference =
typename inner_types::const_reference;
379 using pointer = std::
380 conditional_t<is_const, typename xexpression_type::const_pointer, typename xexpression_type::pointer>;
381 using const_pointer =
typename xexpression_type::const_pointer;
382 using size_type =
typename inner_types::size_type;
383 using difference_type =
typename xexpression_type::difference_type;
385 static constexpr layout_type static_layout = inner_types::layout;
388 static constexpr bool is_strided_view = detail::is_strided_view<xexpression_type, S...>::value;
389 static constexpr bool is_contiguous_view = contiguous_layout;
392 using inner_shape_type =
typename iterable_base::inner_shape_type;
393 using shape_type =
typename xview_shape_type<
typename xexpression_type::shape_type, S...>::type;
395 using xexpression_inner_strides_type = xtl::mpl::eval_if_t<
397 detail::expr_inner_strides_type<xexpression_type>,
400 using xexpression_inner_backstrides_type = xtl::mpl::eval_if_t<
402 detail::expr_inner_backstrides_type<xexpression_type>,
405 using storage_type =
typename inner_types::storage_type;
407 static constexpr bool has_trivial_strides = is_contiguous_view
408 && !std::disjunction<detail::is_xrange<S>...>::value;
409 using inner_strides_type = std::conditional_t<
411 typename detail::unwrap_offset_container<
412 xexpression_type::static_layout,
413 xexpression_inner_strides_type,
414 integral_count<S...>()>::type,
415 get_strides_t<shape_type>>;
417 using inner_backstrides_type = std::conditional_t<
419 typename detail::unwrap_offset_container<
420 xexpression_type::static_layout,
421 xexpression_inner_backstrides_type,
422 integral_count<S...>()>::type,
423 get_strides_t<shape_type>>;
425 using strides_type = get_strides_t<shape_type>;
426 using backstrides_type = strides_type;
429 using slice_type = std::tuple<S...>;
431 using stepper =
typename iterable_base::stepper;
432 using const_stepper =
typename iterable_base::const_stepper;
434 using linear_iterator = std::conditional_t<
436 std::conditional_t<is_const, typename xexpression_type::const_linear_iterator, typename xexpression_type::linear_iterator>,
437 typename iterable_base::linear_iterator>;
438 using const_linear_iterator = std::conditional_t<
440 typename xexpression_type::const_linear_iterator,
441 typename iterable_base::const_linear_iterator>;
443 using reverse_linear_iterator = std::reverse_iterator<linear_iterator>;
444 using const_reverse_linear_iterator = std::reverse_iterator<const_linear_iterator>;
446 using container_iterator = pointer;
447 using const_container_iterator = const_pointer;
448 static constexpr std::size_t rank = SIZE_MAX;
452 template <
class CTA,
class FSL,
class... SL>
453 explicit xview(CTA&& e, FSL&& first_slice, SL&&...
slices)
noexcept;
456 self_type& operator=(
const xview& rhs);
462 disable_xexpression<E, self_type>& operator=(
const E& e);
464 const inner_shape_type&
shape() const noexcept;
465 const slice_type&
slices() const noexcept;
467 bool is_contiguous() const noexcept;
468 using accessible_base::
shape;
473 template <class... Args>
474 reference operator()(Args... args);
475 template <class... Args>
476 reference unchecked(Args... args);
478 reference element(It first, It last);
480 template <class... Args>
481 const_reference operator()(Args... args) const;
482 template <class... Args>
483 const_reference unchecked(Args... args) const;
485 const_reference element(It first, It last) const;
496 template <class ST,
bool Enable = is_strided_view>
497 std::enable_if_t<!Enable, stepper> stepper_begin(const ST&
shape);
498 template <class ST,
bool Enable = is_strided_view>
499 std::enable_if_t<!Enable, stepper> stepper_end(const ST&
shape,
layout_type l);
501 template <class ST,
bool Enable = is_strided_view>
502 std::enable_if_t<!Enable, const_stepper> stepper_begin(const ST&
shape) const;
503 template <class ST,
bool Enable = is_strided_view>
504 std::enable_if_t<!Enable, const_stepper> stepper_end(const ST&
shape,
layout_type l) const;
506 template <class ST,
bool Enable = is_strided_view>
507 std::enable_if_t<Enable, stepper> stepper_begin(const ST&
shape);
508 template <class ST,
bool Enable = is_strided_view>
509 std::enable_if_t<Enable, stepper> stepper_end(const ST&
shape,
layout_type l);
511 template <class ST,
bool Enable = is_strided_view>
512 std::enable_if_t<Enable, const_stepper> stepper_begin(const ST&
shape) const;
513 template <class ST,
bool Enable = is_strided_view>
514 std::enable_if_t<Enable, const_stepper> stepper_end(const ST&
shape,
layout_type l) const;
516 template <class T = xexpression_type>
519 template <class T = xexpression_type>
522 template <class T = xexpression_type>
523 std::enable_if_t<
has_data_interface<T>::value && is_strided_view, linear_iterator> linear_begin();
525 template <class T = xexpression_type>
526 std::enable_if_t<
has_data_interface<T>::value && is_strided_view, linear_iterator> linear_end();
528 template <class T = xexpression_type>
529 std::enable_if_t<
has_data_interface<T>::value && is_strided_view, const_linear_iterator>
530 linear_begin() const;
532 template <class T = xexpression_type>
533 std::enable_if_t<
has_data_interface<T>::value && is_strided_view, const_linear_iterator>
536 template <class T = xexpression_type>
537 std::enable_if_t<
has_data_interface<T>::value && is_strided_view, const_linear_iterator>
538 linear_cbegin() const;
540 template <class T = xexpression_type>
541 std::enable_if_t<
has_data_interface<T>::value && is_strided_view, const_linear_iterator>
544 template <class T = xexpression_type>
545 std::enable_if_t<
has_data_interface<T>::value && is_strided_view, reverse_linear_iterator>
548 template <class T = xexpression_type>
549 std::enable_if_t<
has_data_interface<T>::value && is_strided_view, reverse_linear_iterator>
552 template <class T = xexpression_type>
553 std::enable_if_t<
has_data_interface<T>::value && is_strided_view, const_reverse_linear_iterator>
554 linear_rbegin() const;
556 template <class T = xexpression_type>
557 std::enable_if_t<
has_data_interface<T>::value && is_strided_view, const_reverse_linear_iterator>
560 template <class T = xexpression_type>
561 std::enable_if_t<
has_data_interface<T>::value && is_strided_view, const_reverse_linear_iterator>
562 linear_crbegin() const;
564 template <class T = xexpression_type>
565 std::enable_if_t<
has_data_interface<T>::value && is_strided_view, const_reverse_linear_iterator>
566 linear_crend() const;
568 template <class T = xexpression_type>
569 std::enable_if_t<
has_data_interface<T>::value && is_strided_view, const inner_strides_type&>
572 template <class T = xexpression_type>
573 std::enable_if_t<
has_data_interface<T>::value && is_strided_view, const inner_strides_type&>
576 template <class T = xexpression_type>
577 std::enable_if_t<
has_data_interface<T>::value && is_strided_view, const_pointer> data() const;
579 template <class T = xexpression_type>
582 template <class T = xexpression_type>
584 data_offset() const noexcept;
587 inline It data_xbegin_impl(It begin) const noexcept;
590 inline It data_xend_impl(It begin,
layout_type l, size_type offset) const noexcept;
591 inline container_iterator data_xbegin() noexcept;
592 inline const_container_iterator data_xbegin() const noexcept;
593 inline container_iterator data_xend(
layout_type l, size_type offset) noexcept;
595 inline const_container_iterator data_xend(
layout_type l, size_type offset) const noexcept;
598 template <class ST = self_type, class = std::enable_if_t<is_xscalar<std::decay_t<ST>>::value,
int>>
604 template <
class ST = self_type,
class = std::enable_if_t<is_xscalar<std::decay_t<ST>>::value,
int>>
605 operator const_reference()
const
610 size_type underlying_size(size_type dim)
const;
612 xtl::xclosure_pointer<self_type&> operator&() &;
613 xtl::xclosure_pointer<const self_type&> operator&() const&;
614 xtl::xclosure_pointer<self_type> operator&() &&;
618 class T = xexpression_type,
619 class = std::enable_if_t<has_data_interface<T>::value && is_contiguous_view,
int>>
620 void assign_to(xexpression<E>& e,
bool force_resize) const;
623 using rebind_t =
xview<E, S...>;
626 rebind_t<E> build_view(E&& e) const;
632 template <class requested_type>
633 using simd_return_type = xt_simd::simd_return_type<value_type, requested_type>;
635 template <class T, class R>
636 using enable_simd_interface = std::enable_if_t<has_simd_interface<T>::value && is_strided_view, R>;
638 template <class align, class simd, class T = xexpression_type>
639 enable_simd_interface<T,
void> store_simd(size_type i, const simd& e);
643 class requested_type = value_type,
644 std::
size_t N = xt_simd::simd_traits<requested_type>::
size,
645 class T = xexpression_type>
646 enable_simd_interface<T, simd_return_type<requested_type>> load_simd(size_type i) const;
648 template <class T = xexpression_type>
649 enable_simd_interface<T, reference> data_element(size_type i);
651 template <class T = xexpression_type>
652 enable_simd_interface<T, const_reference> data_element(size_type i) const;
654 template <class T = xexpression_type>
655 enable_simd_interface<T, reference> flat(size_type i);
657 template <class T = xexpression_type>
658 enable_simd_interface<T, const_reference> flat(size_type i) const;
663 template <std::
size_t I>
664 struct lesser_condition
666 static constexpr bool value = (I + newaxis_count_before<S...>(I + 1) <
sizeof...(S));
671 inner_shape_type m_shape;
672 mutable inner_strides_type m_strides;
673 mutable inner_backstrides_type m_backstrides;
674 mutable std::size_t m_data_offset;
675 mutable bool m_strides_computed;
677 template <
class CTA,
class FSL,
class... SL>
678 explicit xview(std::true_type, CTA&& e, FSL&& first_slice, SL&&...
slices)
noexcept;
680 template <
class CTA,
class FSL,
class... SL>
681 explicit xview(std::false_type, CTA&& e, FSL&& first_slice, SL&&...
slices)
noexcept;
683 template <
class... Args>
684 auto make_index_sequence(Args... args)
const noexcept;
686 void compute_strides(std::true_type)
const;
687 void compute_strides(std::false_type)
const;
691 template <
class Arg,
class... Args>
692 reference access(Arg
arg, Args... args);
694 const_reference access()
const;
696 template <
class Arg,
class... Args>
697 const_reference access(Arg
arg, Args... args)
const;
699 template <typename std::decay_t<CT>::size_type... I,
class... Args>
700 reference unchecked_impl(std::index_sequence<I...>, Args... args);
702 template <typename std::decay_t<CT>::size_type... I,
class... Args>
703 const_reference unchecked_impl(std::index_sequence<I...>, Args... args)
const;
705 template <typename std::decay_t<CT>::size_type... I,
class... Args>
706 reference access_impl(std::index_sequence<I...>, Args... args);
708 template <typename std::decay_t<CT>::size_type... I,
class... Args>
709 const_reference access_impl(std::index_sequence<I...>, Args... args)
const;
711 template <typename std::decay_t<CT>::size_type I,
class... Args>
712 std::enable_if_t<lesser_condition<I>::value, size_type> index(Args... args)
const;
714 template <typename std::decay_t<CT>::size_type I,
class... Args>
715 std::enable_if_t<!lesser_condition<I>::value, size_type> index(Args... args)
const;
717 template <typename std::decay_t<CT>::size_type,
class T>
718 size_type sliced_access(const xslice<T>& slice)
const;
720 template <typename std::decay_t<CT>::size_type I,
class T, class Arg, class... Args>
721 size_type sliced_access(const xslice<T>& slice, Arg
arg, Args... args)
const;
723 template <typename std::decay_t<CT>::size_type I,
class T, class... Args>
724 disable_xslice<T, size_type> sliced_access(const T&
squeeze, Args...)
const;
726 using base_index_type = xindex_type_t<typename xexpression_type::shape_type>;
729 base_index_type make_index(It first, It last)
const;
731 void assign_temporary_impl(temporary_type&& tmp);
733 template <std::size_t... I>
734 std::size_t data_offset_impl(std::index_sequence<I...>)
const noexcept;
736 template <std::size_t... I>
737 auto compute_strides_impl(std::index_sequence<I...>)
const noexcept;
739 inner_shape_type compute_shape(std::true_type)
const;
740 inner_shape_type compute_shape(std::false_type)
const;
742 template <
class E, std::size_t... I>
743 rebind_t<E> build_view_impl(E&& e, std::index_sequence<I...>)
const;
745 friend class xview_semantic<
xview<CT, S...>>;
748 template <
class E,
class... S>
749 auto view(E&& e, S&&... slices);
752 auto row(E&& e, std::ptrdiff_t index);
755 auto col(E&& e, std::ptrdiff_t index);
764 struct get_stepper_impl
766 using xexpression_type =
typename V::xexpression_type;
767 using type =
typename xexpression_type::stepper;
771 struct get_stepper_impl<const V>
773 using xexpression_type =
typename V::xexpression_type;
774 using type =
typename xexpression_type::const_stepper;
779 using get_stepper =
typename detail::get_stepper_impl<V>::type;
781 template <
bool is_const,
class CT,
class... S>
786 using view_type = std::conditional_t<is_const,
const xview<CT, S...>,
xview<CT, S...>>;
787 using substepper_type = get_stepper<view_type>;
789 using value_type =
typename substepper_type::value_type;
790 using reference =
typename substepper_type::reference;
791 using pointer =
typename substepper_type::pointer;
792 using difference_type =
typename substepper_type::difference_type;
793 using size_type =
typename view_type::size_type;
795 using shape_type =
typename substepper_type::shape_type;
797 xview_stepper() =
default;
806 reference operator*()
const;
808 void step(size_type dim);
809 void step_back(size_type dim);
810 void step(size_type dim, size_type n);
811 void step_back(size_type dim, size_type n);
812 void reset(size_type dim);
813 void reset_back(size_type dim);
820 bool is_newaxis_slice(size_type index)
const noexcept;
824 void common_step_forward(size_type dim, F f);
826 void common_step_backward(size_type dim, F f);
829 void common_step_forward(size_type dim, size_type n, F f);
831 void common_step_backward(size_type dim, size_type n, F f);
834 void common_reset(size_type dim, F f,
bool backwards);
837 substepper_type m_it;
839 std::array<std::size_t,
sizeof...(S)> m_index_keeper;
843 template <
class ST,
class... S>
849 template <
class I, std::size_t L,
class... S>
852 using type = std::array<I, L - integral_count<S...>() + newaxis_count<S...>()>;
855 template <std::size_t... I,
class... S>
858 using type =
typename xview_shape_type<std::array<std::size_t,
sizeof...(I)>, S...>::type;
879 template <
class CT,
class... S>
880 template <
class CTA,
class FSL,
class... SL>
883 std::integral_constant<bool, has_trivial_strides>{},
884 std::forward<CTA>(e),
885 std::forward<FSL>(first_slice),
886 std::forward<SL>(slices)...
892 template <
class CT,
class... S>
893 template <
class CTA,
class FSL,
class... SL>
895 : m_e(std::forward<CTA>(e))
896 , m_slices(std::forward<FSL>(first_slice), std::forward<SL>(slices)...)
897 , m_shape(compute_shape(detail::is_sequence_view<inner_shape_type>{}))
898 , m_strides(m_e.strides())
899 , m_backstrides(m_e.backstrides())
900 , m_data_offset(data_offset_impl(std::make_index_sequence<
sizeof...(S)>()))
901 , m_strides_computed(
true)
905 template <
class CT,
class... S>
906 template <
class CTA,
class FSL,
class... SL>
908 : m_e(std::forward<CTA>(e))
909 , m_slices(std::forward<FSL>(first_slice), std::forward<SL>(slices)...)
910 , m_shape(compute_shape(std::false_type{}))
911 , m_strides_computed(
false)
917 template <
class CT,
class... S>
918 inline auto xview<CT, S...>::operator=(
const xview& rhs) -> self_type&
921 return this->assign_temporary(std::move(tmp));
931 template <
class CT,
class... S>
935 return semantic_base::operator=(e);
940 template <
class CT,
class... S>
942 inline auto xview<CT, S...>::operator=(
const E& e) -> disable_xexpression<E, self_type>&
955 template <
class CT,
class... S>
964 template <
class CT,
class... S>
973 template <
class CT,
class... S>
976 if constexpr (is_strided_view)
980 return static_layout;
984 bool strides_match = do_strides_match(
shape(),
strides(), m_e.layout(),
true);
994 template <
class CT,
class... S>
995 inline bool xview<CT, S...>::is_contiguous() const noexcept
1011 template <
class CT,
class... S>
1017 std::fill(linear_begin(), linear_end(), value);
1021 std::fill(this->begin(), this->end(), value);
1031 template <
class CT,
class... S>
1032 template <
class... Args>
1033 inline auto xview<CT, S...>::operator()(Args... args) -> reference
1035 XTENSOR_TRY(check_index(
shape(), args...));
1036 XTENSOR_CHECK_DIMENSION(
shape(), args...);
1039 return access(
static_cast<size_type
>(args)...);
1061 template <
class CT,
class... S>
1062 template <
class... Args>
1063 inline auto xview<CT, S...>::unchecked(Args... args) -> reference
1065 return unchecked_impl(make_index_sequence(args...),
static_cast<size_type
>(args)...);
1068 template <
class CT,
class... S>
1070 inline auto xview<CT, S...>::element(It first, It last) -> reference
1072 XTENSOR_TRY(check_element_index(shape(), first, last));
1074 auto index = make_index(first, last);
1075 return m_e.element(index.cbegin(), index.cend());
1084 template <
class CT,
class... S>
1085 template <
class... Args>
1086 inline auto xview<CT, S...>::operator()(Args... args)
const -> const_reference
1088 XTENSOR_TRY(check_index(
shape(), args...));
1089 XTENSOR_CHECK_DIMENSION(
shape(), args...);
1092 return access(
static_cast<size_type
>(args)...);
1114 template <
class CT,
class... S>
1115 template <
class... Args>
1116 inline auto xview<CT, S...>::unchecked(Args... args)
const -> const_reference
1118 return unchecked_impl(make_index_sequence(args...),
static_cast<size_type
>(args)...);
1121 template <
class CT,
class... S>
1123 inline auto xview<CT, S...>::element(It first, It last)
const -> const_reference
1126 auto index = make_index(first, last);
1127 return m_e.element(index.cbegin(), index.cend());
1133 template <
class CT,
class... S>
1142 template <
class CT,
class... S>
1153 template <
class CT,
class... S>
1155 inline auto xview<CT, S...>::storage() -> std::enable_if_t<has_data_interface<T>::value, storage_type&>
1157 return m_e.storage();
1160 template <
class CT,
class... S>
1162 inline auto xview<CT, S...>::storage() const
1165 return m_e.storage();
1168 template <
class CT,
class... S>
1170 auto xview<CT, S...>::linear_begin()
1171 -> std::enable_if_t<has_data_interface<T>::value && is_strided_view, linear_iterator>
1173 return m_e.storage().begin() + data_offset();
1176 template <
class CT,
class... S>
1178 auto xview<CT, S...>::linear_end()
1179 -> std::enable_if_t<has_data_interface<T>::value && is_strided_view, linear_iterator>
1181 return m_e.storage().begin() + data_offset() + this->size();
1184 template <
class CT,
class... S>
1186 auto xview<CT, S...>::linear_begin() const
1187 -> std::enable_if_t<
has_data_interface<T>::value && is_strided_view, const_linear_iterator>
1189 return linear_cbegin();
1192 template <
class CT,
class... S>
1194 auto xview<CT, S...>::linear_end() const
1195 -> std::enable_if_t<
has_data_interface<T>::value && is_strided_view, const_linear_iterator>
1197 return linear_cend();
1200 template <
class CT,
class... S>
1202 auto xview<CT, S...>::linear_cbegin() const
1203 -> std::enable_if_t<
has_data_interface<T>::value && is_strided_view, const_linear_iterator>
1205 return m_e.storage().cbegin() + data_offset();
1208 template <
class CT,
class... S>
1210 auto xview<CT, S...>::linear_cend() const
1211 -> std::enable_if_t<
has_data_interface<T>::value && is_strided_view, const_linear_iterator>
1213 return m_e.storage().cbegin() + data_offset() + this->size();
1216 template <
class CT,
class... S>
1218 auto xview<CT, S...>::linear_rbegin()
1219 -> std::enable_if_t<has_data_interface<T>::value && is_strided_view, reverse_linear_iterator>
1221 return reverse_linear_iterator(linear_end());
1224 template <
class CT,
class... S>
1226 auto xview<CT, S...>::linear_rend()
1227 -> std::enable_if_t<has_data_interface<T>::value && is_strided_view, reverse_linear_iterator>
1229 return reverse_linear_iterator(linear_begin());
1232 template <
class CT,
class... S>
1234 auto xview<CT, S...>::linear_rbegin() const
1235 -> std::enable_if_t<
has_data_interface<T>::value && is_strided_view, const_reverse_linear_iterator>
1237 return linear_crbegin();
1240 template <
class CT,
class... S>
1242 auto xview<CT, S...>::linear_rend() const
1243 -> std::enable_if_t<
has_data_interface<T>::value && is_strided_view, const_reverse_linear_iterator>
1245 return linear_crend();
1248 template <
class CT,
class... S>
1250 auto xview<CT, S...>::linear_crbegin() const
1251 -> std::enable_if_t<
has_data_interface<T>::value && is_strided_view, const_reverse_linear_iterator>
1253 return const_reverse_linear_iterator(linear_end());
1256 template <
class CT,
class... S>
1258 auto xview<CT, S...>::linear_crend() const
1259 -> std::enable_if_t<
has_data_interface<T>::value && is_strided_view, const_reverse_linear_iterator>
1261 return const_reverse_linear_iterator(linear_begin());
1267 template <
class CT,
class... S>
1269 inline auto xview<CT, S...>::strides() const
1270 -> std::enable_if_t<
has_data_interface<T>::value && is_strided_view, const inner_strides_type&>
1272 if (!m_strides_computed)
1275 m_strides_computed =
true;
1280 template <
class CT,
class... S>
1282 inline auto xview<CT, S...>::backstrides() const
1283 -> std::enable_if_t<
has_data_interface<T>::value && is_strided_view, const inner_strides_type&>
1285 if (!m_strides_computed)
1288 m_strides_computed =
true;
1290 return m_backstrides;
1296 template <
class CT,
class... S>
1298 inline auto xview<CT, S...>::data() const
1304 template <
class CT,
class... S>
1306 inline auto xview<CT, S...>::data()
1307 -> std::enable_if_t<has_data_interface<T>::value && is_strided_view, pointer>
1312 template <
class CT,
class... S>
1313 template <std::size_t... I>
1314 inline std::size_t xview<CT, S...>::data_offset_impl(std::index_sequence<I...>)
const noexcept
1316 auto temp = std::array<std::ptrdiff_t,
sizeof...(S)>(
1317 {(
static_cast<ptrdiff_t
>(xt::value(std::get<I>(m_slices), 0)))...}
1320 std::ptrdiff_t result = 0;
1322 for (; i < std::min(
sizeof...(S), m_e.strides().size()); ++i)
1324 result += temp[i] * m_e.strides()[i - newaxis_count_before<S...>(i)];
1326 for (; i <
sizeof...(S); ++i)
1330 return static_cast<std::size_t
>(result) + m_e.data_offset();
1336 template <
class CT,
class... S>
1338 inline auto xview<CT, S...>::data_offset() const noexcept
1341 if (!m_strides_computed)
1344 m_strides_computed =
true;
1346 return m_data_offset;
1351 template <
class CT,
class... S>
1352 inline auto xview<CT, S...>::underlying_size(size_type dim)
const -> size_type
1354 return m_e.shape()[dim];
1357 template <
class CT,
class... S>
1358 inline auto xview<CT, S...>::operator&() & -> xtl::xclosure_pointer<self_type&>
1360 return xtl::closure_pointer(*
this);
1363 template <
class CT,
class... S>
1364 inline auto xview<CT, S...>::operator&() const& -> xtl::xclosure_pointer<const self_type&>
1366 return xtl::closure_pointer(*
this);
1369 template <
class CT,
class... S>
1370 inline auto xview<CT, S...>::operator&() && -> xtl::xclosure_pointer<self_type>
1372 return xtl::closure_pointer(std::move(*
this));
1385 template <
class CT,
class... S>
1389 return xt::broadcast_shape(m_shape,
shape);
1397 template <
class CT,
class... S>
1401 if constexpr (is_strided_view)
1403 return str.size() ==
strides().size() && std::equal(str.cbegin(), str.cend(), strides().begin());
1413 template <
class CT,
class... S>
1415 inline It xview<CT, S...>::data_xbegin_impl(It begin)
const noexcept
1417 return begin + data_offset();
1420 template <
class CT,
class... S>
1422 inline It xview<CT, S...>::data_xend_impl(It begin,
layout_type l, size_type offset)
const noexcept
1424 return strided_data_end(*
this, begin, l, offset);
1427 template <
class CT,
class... S>
1428 inline auto xview<CT, S...>::data_xbegin() noexcept -> container_iterator
1430 return data_xbegin_impl(data());
1433 template <
class CT,
class... S>
1434 inline auto xview<CT, S...>::data_xbegin() const noexcept -> const_container_iterator
1436 return data_xbegin_impl(data());
1439 template <
class CT,
class... S>
1440 inline auto xview<CT, S...>::data_xend(
layout_type l, size_type offset)
noexcept -> container_iterator
1442 return data_xend_impl(data() + data_offset(), l, offset);
1445 template <
class CT,
class... S>
1446 inline auto xview<CT, S...>::data_xend(
layout_type l, size_type offset)
const noexcept
1447 -> const_container_iterator
1449 return data_xend_impl(data() + data_offset(), l, offset);
1453 template <
class CT,
class... S>
1454 template <
class E,
class T,
class>
1455 void xview<CT, S...>::assign_to(
xexpression<E>& e,
bool force_resize)
const
1457 auto& de = e.derived_cast();
1458 de.resize(shape(), force_resize);
1459 std::copy(data() + data_offset(), data() + data_offset() + de.size(), de.template begin<static_layout>());
1462 template <
class CT,
class... S>
1463 template <
class E, std::size_t... I>
1464 inline auto xview<CT, S...>::build_view_impl(E&& e, std::index_sequence<I...>)
const -> rebind_t<E>
1466 return rebind_t<E>(std::forward<E>(e), std::get<I>(m_slices)...);
1469 template <
class CT,
class... S>
1471 inline auto xview<CT, S...>::build_view(E&& e)
const -> rebind_t<E>
1473 return build_view_impl(std::forward<E>(e), std::make_index_sequence<
sizeof...(S)>());
1476 template <
class CT,
class... S>
1477 template <
class align,
class simd,
class T>
1478 inline auto xview<CT, S...>::store_simd(size_type i,
const simd& e) -> enable_simd_interface<T, void>
1480 return m_e.template store_simd<xt_simd::unaligned_mode>(data_offset() + i, e);
1483 template <
class CT,
class... S>
1484 template <
class align,
class requested_type, std::
size_t N,
class T>
1485 inline auto xview<CT, S...>::load_simd(size_type i)
const
1486 -> enable_simd_interface<T, simd_return_type<requested_type>>
1488 return m_e.template load_simd<xt_simd::unaligned_mode, requested_type>(data_offset() + i);
1491 template <
class CT,
class... S>
1493 inline auto xview<CT, S...>::data_element(size_type i) -> enable_simd_interface<T, reference>
1495 return m_e.data_element(data_offset() + i);
1498 template <
class CT,
class... S>
1500 inline auto xview<CT, S...>::data_element(size_type i)
const -> enable_simd_interface<T, const_reference>
1502 return m_e.data_element(data_offset() + i);
1505 template <
class CT,
class... S>
1507 inline auto xview<CT, S...>::flat(size_type i) -> enable_simd_interface<T, reference>
1509 XTENSOR_ASSERT(is_contiguous());
1510 return m_e.flat(data_offset() + i);
1513 template <
class CT,
class... S>
1515 inline auto xview<CT, S...>::flat(size_type i)
const -> enable_simd_interface<T, const_reference>
1517 XTENSOR_ASSERT(is_contiguous());
1518 return m_e.flat(data_offset() + i);
1521 template <
class CT,
class... S>
1522 template <
class... Args>
1523 inline auto xview<CT, S...>::make_index_sequence(Args...) const noexcept
1525 return std::make_index_sequence<
1526 (
sizeof...(Args) + integral_count<S...>() > newaxis_count<S...>()
1527 ?
sizeof...(Args) + integral_count<S...>() - newaxis_count<S...>()
1531 template <
class CT,
class... S>
1532 template <std::size_t... I>
1533 inline auto xview<CT, S...>::compute_strides_impl(std::index_sequence<I...>)
const noexcept
1535 std::size_t original_dim = m_e.dimension();
1536 return std::array<std::ptrdiff_t,
sizeof...(I)>(
1537 {(
static_cast<std::ptrdiff_t
>(xt::step_size(std::get<integral_skip<S...>(I)>(m_slices), 1))
1538 * ((integral_skip<S...>(I) - newaxis_count_before<S...>(integral_skip<S...>(I))) < original_dim
1539 ? m_e.strides()[integral_skip<S...>(I) - newaxis_count_before<S...>(integral_skip<S...>(I))]
1544 template <
class CT,
class... S>
1545 inline void xview<CT, S...>::compute_strides(std::false_type)
const
1547 m_strides = xtl::make_sequence<inner_strides_type>(this->dimension(), 0);
1548 m_backstrides = xtl::make_sequence<inner_strides_type>(this->dimension(), 0);
1550 constexpr std::size_t n_strides =
sizeof...(S) - integral_count<S...>();
1552 auto slice_strides = compute_strides_impl(std::make_index_sequence<n_strides>());
1554 for (std::size_t i = 0; i < n_strides; ++i)
1556 m_strides[i] = slice_strides[i];
1558 detail::adapt_strides(shape(), m_strides, &m_backstrides, i);
1560 for (std::size_t i = n_strides; i < this->dimension(); ++i)
1562 m_strides[i] = m_e.strides()[i + integral_count<S...>() - newaxis_count<S...>()];
1563 detail::adapt_strides(shape(), m_strides, &m_backstrides, i);
1566 m_data_offset = data_offset_impl(std::make_index_sequence<
sizeof...(S)>());
1569 template <
class CT,
class... S>
1570 inline void xview<CT, S...>::compute_strides(std::true_type)
const
1574 template <
class CT,
class... S>
1575 inline auto xview<CT, S...>::access() -> reference
1577 return access_impl(make_index_sequence());
1580 template <
class CT,
class... S>
1581 template <
class Arg,
class... Args>
1582 inline auto xview<CT, S...>::access(Arg
arg, Args... args) -> reference
1584 if (
sizeof...(Args) >= this->dimension())
1586 return access(args...);
1588 return access_impl(make_index_sequence(
arg, args...),
arg, args...);
1591 template <
class CT,
class... S>
1592 inline auto xview<CT, S...>::access() const -> const_reference
1594 return access_impl(make_index_sequence());
1597 template <
class CT,
class... S>
1598 template <
class Arg,
class... Args>
1599 inline auto xview<CT, S...>::access(Arg
arg, Args... args)
const -> const_reference
1601 if (
sizeof...(Args) >= this->dimension())
1603 return access(args...);
1605 return access_impl(make_index_sequence(
arg, args...),
arg, args...);
1608 template <
class CT,
class... S>
1609 template <typename std::decay_t<CT>::size_type... I,
class... Args>
1610 inline auto xview<CT, S...>::unchecked_impl(std::index_sequence<I...>, Args... args) -> reference
1612 return m_e.unchecked(index<I>(args...)...);
1615 template <
class CT,
class... S>
1616 template <typename std::decay_t<CT>::size_type... I,
class... Args>
1617 inline auto xview<CT, S...>::unchecked_impl(std::index_sequence<I...>, Args... args)
const
1620 return m_e.unchecked(index<I>(args...)...);
1623 template <
class CT,
class... S>
1624 template <typename std::decay_t<CT>::size_type... I,
class... Args>
1625 inline auto xview<CT, S...>::access_impl(std::index_sequence<I...>, Args... args) -> reference
1627 return m_e(index<I>(args...)...);
1630 template <
class CT,
class... S>
1631 template <typename std::decay_t<CT>::size_type... I,
class... Args>
1632 inline auto xview<CT, S...>::access_impl(std::index_sequence<I...>, Args... args)
const -> const_reference
1634 return m_e(index<I>(args...)...);
1637 template <
class CT,
class... S>
1638 template <typename std::decay_t<CT>::size_type I,
class... Args>
1639 inline auto xview<CT, S...>::index(Args... args)
const
1640 -> std::enable_if_t<lesser_condition<I>::value, size_type>
1642 return sliced_access<I - integral_count_before<S...>(I) + newaxis_count_before<S...>(I + 1)>(
1643 std::get<I + newaxis_count_before<S...>(I + 1)>(m_slices),
1648 template <
class CT,
class... S>
1649 template <typename std::decay_t<CT>::size_type I,
class... Args>
1650 inline auto xview<CT, S...>::index(Args... args)
const
1651 -> std::enable_if_t<!lesser_condition<I>::value, size_type>
1653 return argument<I - integral_count<S...>() + newaxis_count<S...>()>(args...);
1656 template <
class CT,
class... S>
1657 template <typename std::decay_t<CT>::size_type I,
class T>
1658 inline auto
xview<CT, S...>::sliced_access(const xslice<T>& slice) const -> size_type
1660 return static_cast<size_type
>(slice.derived_cast()(0));
1663 template <
class CT,
class... S>
1664 template <typename std::decay_t<CT>::size_type I,
class T, class Arg, class... Args>
1665 inline auto
xview<CT, S...>::sliced_access(const xslice<T>& slice, Arg arg, Args... args) const -> size_type
1667 using ST =
typename T::size_type;
1668 return static_cast<size_type
>(
1669 slice.derived_cast()(argument<I>(
static_cast<ST
>(arg),
static_cast<ST
>(args)...))
1673 template <
class CT,
class... S>
1674 template <typename std::decay_t<CT>::size_type I,
class T, class... Args>
1675 inline auto
xview<CT, S...>::sliced_access(const T& squeeze, Args...) const -> disable_xslice<T, size_type>
1677 return static_cast<size_type
>(
squeeze);
1680 template <
class CT,
class... S>
1682 inline auto xview<CT, S...>::make_index(It first, It last)
const -> base_index_type
1684 auto index = xtl::make_sequence<base_index_type>(m_e.dimension(), 0);
1685 using diff_type =
typename std::iterator_traits<It>::difference_type;
1686 using ivalue_type =
typename base_index_type::value_type;
1687 auto func1 = [&first](
const auto& s)
noexcept
1689 return get_slice_value(s, first);
1691 auto func2 = [](
const auto& s)
noexcept
1693 return xt::value(s, 0);
1696 auto s =
static_cast<diff_type
>(
1697 (std::min)(
static_cast<size_type
>(std::distance(first, last)), this->dimension())
1699 auto first_copy = last - s;
1700 for (size_type i = 0; i != m_e.dimension(); ++i)
1702 size_type k = newaxis_skip<S...>(i);
1706 std::advance(first,
static_cast<diff_type
>(k - xt::integral_count_before<S...>(i)));
1710 index[i] = k <
sizeof...(S) ? apply<size_type>(k, func1, m_slices)
1711 :
static_cast<ivalue_type
>(*first);
1715 index[i] = k <
sizeof...(S) ? apply<size_type>(k, func2, m_slices) : ivalue_type(0);
1721 template <
class CT,
class... S>
1722 inline auto xview<CT, S...>::compute_shape(std::true_type)
const -> inner_shape_type
1724 return inner_shape_type(m_e.shape());
1727 template <
class CT,
class... S>
1728 inline auto xview<CT, S...>::compute_shape(std::false_type)
const -> inner_shape_type
1730 std::size_t dim = m_e.dimension() - integral_count<S...>() + newaxis_count<S...>();
1731 auto shape = xtl::make_sequence<inner_shape_type>(dim, 0);
1732 auto func = [](
const auto& s)
noexcept
1736 for (size_type i = 0; i != dim; ++i)
1738 size_type index = integral_skip<S...>(i);
1739 shape[i] = index <
sizeof...(S) ? apply<size_type>(index, func, m_slices)
1740 : m_e.shape()[index - newaxis_count_before<S...>(index)];
1745 namespace xview_detail
1747 template <
class V,
class T>
1748 inline void run_assign_temporary_impl(V& v,
const T& t, std::true_type )
1750 strided_loop_assigner<true>::run(v, t);
1753 template <
class V,
class T>
1755 run_assign_temporary_impl(V& v,
const T& t, std::false_type )
1757 std::copy(t.cbegin(), t.cend(), v.begin());
1761 template <
class CT,
class... S>
1762 inline void xview<CT, S...>::assign_temporary_impl(
temporary_type&& tmp)
1764 constexpr bool fast_assign = detail::is_strided_view<xexpression_type, S...>::value
1766 xview_detail::run_assign_temporary_impl(*
this, tmp, std::integral_constant<bool, fast_assign>{});
1771 template <
class E,
class... S>
1772 inline std::size_t get_underlying_shape_index(std::size_t I)
1774 return I - newaxis_count_before<get_slice_type<E, S>...>(I);
1777 template <
class... S>
1781 struct check_slice<>
1783 using type = void_t<>;
1786 template <
class S,
class... SL>
1787 struct check_slice<S, SL...>
1789 static_assert(!std::is_same<S, xellipsis_tag>::value,
"ellipsis not supported vith xview");
1790 using type =
typename check_slice<SL...>::type;
1793 template <
class E, std::size_t... I,
class... S>
1794 inline auto make_view_impl(E&& e, std::index_sequence<I...>, S&&... slices)
1797 using view_type = xview<xtl::closure_type_t<E>, get_slice_type<std::decay_t<E>, S>...>;
1800 get_slice_implementation(
1802 std::forward<S>(slices),
1803 get_underlying_shape_index<std::decay_t<E>, S...>(I)
1818 template <
class E,
class... S>
1819 inline auto view(E&& e, S&&... slices)
1821 return detail::make_view_impl(
1823 std::make_index_sequence<
sizeof...(S)>(),
1824 std::forward<S>(slices)...
1835 inline static auto make(E&& e,
const std::ptrdiff_t index)
1837 const auto shape = e.shape();
1838 check_dimension(shape);
1845 inline static void check_dimension(
const S& shape)
1847 if (shape.size() != 2)
1850 std::invalid_argument,
1851 "A row can only be accessed on an expression with exact two dimensions"
1856 template <
class T, std::
size_t N>
1857 inline static void check_dimension(
const std::array<T, N>&)
1859 static_assert(N == 2,
"A row can only be accessed on an expression with exact two dimensions");
1868 inline static auto make(E&& e,
const std::ptrdiff_t index)
1870 const auto shape = e.shape();
1871 check_dimension(shape);
1878 inline static void check_dimension(
const S& shape)
1880 if (shape.size() != 2)
1883 std::invalid_argument,
1884 "A column can only be accessed on an expression with exact two dimensions"
1889 template <
class T, std::
size_t N>
1890 inline static void check_dimension(
const std::array<T, N>&)
1892 static_assert(N == 2,
"A column can only be accessed on an expression with exact two dimensions");
1907 inline auto row(E&& e, std::ptrdiff_t index)
1909 return detail::row_impl::make(e, index);
1922 inline auto col(E&& e, std::ptrdiff_t index)
1924 return detail::column_impl::make(e, index);
1931 template <
class CT,
class... S>
1932 template <
class ST,
bool Enable>
1933 inline auto xview<CT, S...>::stepper_begin(
const ST& shape) -> std::enable_if_t<!Enable, stepper>
1935 size_type offset = shape.size() - this->dimension();
1936 return stepper(
this, m_e.stepper_begin(m_e.shape()), offset);
1939 template <
class CT,
class... S>
1940 template <
class ST,
bool Enable>
1941 inline auto xview<CT, S...>::stepper_end(
const ST& shape,
layout_type l)
1942 -> std::enable_if_t<!Enable, stepper>
1944 size_type offset = shape.size() - this->dimension();
1945 return stepper(
this, m_e.stepper_end(m_e.shape(), l), offset,
true, l);
1948 template <
class CT,
class... S>
1949 template <
class ST,
bool Enable>
1950 inline auto xview<CT, S...>::stepper_begin(
const ST& shape)
const
1951 -> std::enable_if_t<!Enable, const_stepper>
1953 size_type offset = shape.size() - this->dimension();
1954 const xexpression_type& e = m_e;
1955 return const_stepper(
this, e.stepper_begin(m_e.shape()), offset);
1958 template <
class CT,
class... S>
1959 template <
class ST,
bool Enable>
1960 inline auto xview<CT, S...>::stepper_end(
const ST& shape,
layout_type l)
const
1961 -> std::enable_if_t<!Enable, const_stepper>
1963 size_type offset = shape.size() - this->dimension();
1964 const xexpression_type& e = m_e;
1965 return const_stepper(
this, e.stepper_end(m_e.shape(), l), offset,
true, l);
1968 template <
class CT,
class... S>
1969 template <
class ST,
bool Enable>
1970 inline auto xview<CT, S...>::stepper_begin(
const ST& shape) -> std::enable_if_t<Enable, stepper>
1972 size_type offset = shape.size() - this->dimension();
1973 return stepper(
this, data_xbegin(), offset);
1976 template <
class CT,
class... S>
1977 template <
class ST,
bool Enable>
1978 inline auto xview<CT, S...>::stepper_end(
const ST& shape,
layout_type l)
1979 -> std::enable_if_t<Enable, stepper>
1981 size_type offset = shape.size() - this->dimension();
1982 return stepper(
this, data_xend(l, offset), offset);
1985 template <
class CT,
class... S>
1986 template <
class ST,
bool Enable>
1987 inline auto xview<CT, S...>::stepper_begin(
const ST& shape)
const
1988 -> std::enable_if_t<Enable, const_stepper>
1990 size_type offset = shape.size() - this->dimension();
1991 return const_stepper(
this, data_xbegin(), offset);
1994 template <
class CT,
class... S>
1995 template <
class ST,
bool Enable>
1996 inline auto xview<CT, S...>::stepper_end(
const ST& shape,
layout_type l)
const
1997 -> std::enable_if_t<Enable, const_stepper>
1999 size_type offset = shape.size() - this->dimension();
2000 return const_stepper(
this, data_xend(l, offset), offset);
2007 template <
bool is_const,
class CT,
class... S>
2008 inline xview_stepper<is_const, CT, S...>::xview_stepper(
2021 std::fill(m_index_keeper.begin(), m_index_keeper.end(), 0);
2022 auto func = [](
const auto& s)
noexcept
2024 return xt::value(s, 0);
2026 for (size_type i = 0; i <
sizeof...(S); ++i)
2028 if (!is_newaxis_slice(i))
2030 size_type s = apply<size_type>(i, func, p_view->slices());
2031 size_type index = i - newaxis_count_before<S...>(i);
2032 m_it.step(index, s);
2042 template <
bool is_const,
class CT,
class... S>
2043 inline auto xview_stepper<is_const, CT, S...>::operator*() const -> reference
2048 template <
bool is_const,
class CT,
class... S>
2049 inline void xview_stepper<is_const, CT, S...>::step(size_type dim)
2051 auto func = [
this](size_type index, size_type offset)
2053 m_it.step(index, offset);
2055 common_step_forward(dim, func);
2058 template <
bool is_const,
class CT,
class... S>
2059 inline void xview_stepper<is_const, CT, S...>::step_back(size_type dim)
2061 auto func = [
this](size_type index, size_type offset)
2063 m_it.step_back(index, offset);
2065 common_step_backward(dim, func);
2068 template <
bool is_const,
class CT,
class... S>
2069 inline void xview_stepper<is_const, CT, S...>::step(size_type dim, size_type n)
2071 auto func = [
this](size_type index, size_type offset)
2073 m_it.step(index, offset);
2075 common_step_forward(dim, n, func);
2078 template <
bool is_const,
class CT,
class... S>
2079 inline void xview_stepper<is_const, CT, S...>::step_back(size_type dim, size_type n)
2081 auto func = [
this](size_type index, size_type offset)
2083 m_it.step_back(index, offset);
2085 common_step_backward(dim, n, func);
2088 template <
bool is_const,
class CT,
class... S>
2089 inline void xview_stepper<is_const, CT, S...>::reset(size_type dim)
2091 auto func = [
this](size_type index, size_type offset)
2093 m_it.step_back(index, offset);
2095 common_reset(dim, func,
false);
2098 template <
bool is_const,
class CT,
class... S>
2099 inline void xview_stepper<is_const, CT, S...>::reset_back(size_type dim)
2101 auto func = [
this](size_type index, size_type offset)
2103 m_it.step(index, offset);
2105 common_reset(dim, func,
true);
2108 template <
bool is_const,
class CT,
class... S>
2109 inline void xview_stepper<is_const, CT, S...>::to_begin()
2111 std::fill(m_index_keeper.begin(), m_index_keeper.end(), 0);
2115 template <
bool is_const,
class CT,
class... S>
2116 inline void xview_stepper<is_const, CT, S...>::to_end(layout_type l)
2122 template <
bool is_const,
class CT,
class... S>
2123 inline bool xview_stepper<is_const, CT, S...>::is_newaxis_slice(size_type index)
const noexcept
2126 return newaxis_count_before<S...>(index + 1) != newaxis_count_before<S...>(index);
2129 template <
bool is_const,
class CT,
class... S>
2130 inline void xview_stepper<is_const, CT, S...>::to_end_impl(layout_type l)
2132 auto func = [](
const auto& s)
noexcept
2134 return xt::value(s, get_size(s) - 1);
2136 auto size_func = [](
const auto& s)
noexcept
2141 for (size_type i = 0; i <
sizeof...(S); ++i)
2143 if (!is_newaxis_slice(i))
2145 size_type s = apply<size_type>(i, func, p_view->slices());
2146 size_type ix = apply<size_type>(i, size_func, p_view->slices());
2147 m_index_keeper[i] = ix - size_type(1);
2148 size_type index = i - newaxis_count_before<S...>(i);
2149 s = p_view->underlying_size(index) - 1 - s;
2150 m_it.step_back(index, s);
2153 if (l == layout_type::row_major)
2155 for (size_type i =
sizeof...(S); i > 0; --i)
2157 if (!is_newaxis_slice(i - 1))
2159 m_index_keeper[i - 1]++;
2164 else if (l == layout_type::column_major)
2166 for (size_type i = 0; i <
sizeof...(S); ++i)
2168 if (!is_newaxis_slice(i))
2170 m_index_keeper[i]++;
2177 XTENSOR_THROW(std::runtime_error,
"Iteration only allowed in row or column major.");
2181 template <
bool is_const,
class CT,
class... S>
2183 void xview_stepper<is_const, CT, S...>::common_step_forward(size_type dim, F f)
2185 if (dim >= m_offset)
2187 auto func = [&dim,
this](
const auto& s)
noexcept
2189 return step_size(s, this->m_index_keeper[dim]++, 1);
2191 size_type index = integral_skip<S...>(dim);
2192 if (!is_newaxis_slice(index))
2194 size_type step_size = index <
sizeof...(S) ? apply<size_type>(index, func, p_view->slices())
2196 index -= newaxis_count_before<S...>(index);
2197 f(index, step_size);
2202 template <
bool is_const,
class CT,
class... S>
2204 void xview_stepper<is_const, CT, S...>::common_step_forward(size_type dim, size_type n, F f)
2206 if (dim >= m_offset)
2208 auto func = [&dim, &n,
this](
const auto& s)
noexcept
2210 auto st_size = step_size(s, this->m_index_keeper[dim], n);
2211 this->m_index_keeper[dim] += n;
2212 return size_type(st_size);
2215 size_type index = integral_skip<S...>(dim);
2216 if (!is_newaxis_slice(index))
2218 size_type step_size = index <
sizeof...(S) ? apply<size_type>(index, func, p_view->slices())
2220 index -= newaxis_count_before<S...>(index);
2221 f(index, step_size);
2226 template <
bool is_const,
class CT,
class... S>
2228 void xview_stepper<is_const, CT, S...>::common_step_backward(size_type dim, F f)
2230 if (dim >= m_offset)
2232 auto func = [&dim,
this](
const auto& s)
noexcept
2234 this->m_index_keeper[dim]--;
2235 return step_size(s, this->m_index_keeper[dim], 1);
2237 size_type index = integral_skip<S...>(dim);
2238 if (!is_newaxis_slice(index))
2240 size_type step_size = index <
sizeof...(S) ? apply<size_type>(index, func, p_view->slices())
2242 index -= newaxis_count_before<S...>(index);
2243 f(index, step_size);
2248 template <
bool is_const,
class CT,
class... S>
2250 void xview_stepper<is_const, CT, S...>::common_step_backward(size_type dim, size_type n, F f)
2252 if (dim >= m_offset)
2254 auto func = [&dim, &n,
this](
const auto& s)
noexcept
2256 this->m_index_keeper[dim] -= n;
2257 return step_size(s, this->m_index_keeper[dim], n);
2260 size_type index = integral_skip<S...>(dim);
2261 if (!is_newaxis_slice(index))
2263 size_type step_size = index <
sizeof...(S) ? apply<size_type>(index, func, p_view->slices())
2265 index -= newaxis_count_before<S...>(index);
2266 f(index, step_size);
2271 template <
bool is_const,
class CT,
class... S>
2273 void xview_stepper<is_const, CT, S...>::common_reset(size_type dim, F f,
bool backwards)
2275 auto size_func = [](
const auto& s)
noexcept
2279 auto end_func = [](
const auto& s)
noexcept
2281 return xt::value(s, get_size(s) - 1) - xt::value(s, 0);
2284 size_type index = integral_skip<S...>(dim);
2285 if (!is_newaxis_slice(index))
2287 if (dim < m_index_keeper.size())
2289 size_type size = index <
sizeof...(S) ? apply<size_type>(index, size_func, p_view->slices())
2290 : p_view->shape()[dim];
2291 m_index_keeper[dim] = backwards ? size - 1 : 0;
2294 size_type reset_n = index <
sizeof...(S) ? apply<size_type>(index, end_func, p_view->slices())
2295 : p_view->shape()[dim] - 1;
2296 index -= newaxis_count_before<S...>(index);
Fixed shape implementation for compile time defined arrays.
size_type size() const noexcept
Base class for xexpressions.
Base class for multidimensional iterable expressions.
Multidimensional view with tensor semantic.
xview(CTA &&e, FSL &&first_slice, SL &&... slices) noexcept
Constructs a view on the specified xexpression.
const slice_type & slices() const noexcept
bool has_linear_assign(const ST &strides) const
const inner_shape_type & shape() const noexcept
Returns the shape of the view.
bool broadcast_shape(ST &shape, bool reuse_cache=false) const
xexpression_type & expression() noexcept
void fill(const T &value)
layout_type layout() const noexcept
auto arg(E &&e) noexcept
Calculates the phase angle (in radians) elementwise for the complex numbers in e.
auto squeeze(E &&e)
Returns a squeeze view of the given expression.
std::size_t compute_strides(const shape_type &shape, layout_type l, strides_type &strides)
Compute the strides given the shape and the layout of an array.
auto strides(const E &e, stride_type type=stride_type::normal) noexcept
Get strides of an object.
standard mathematical functions for xexpressions
auto all() noexcept
Returns a slice representing a full dimension, to be used as an argument of view function.
auto row(E &&e, std::ptrdiff_t index)
Constructs and returns a row (sliced view) on the specified expression.
auto col(E &&e, std::ptrdiff_t index)
Constructs and returns a column (sliced view) on the specified expression.
auto view(E &&e, S &&... slices)
Constructs and returns a view on the specified xexpression.