view_expression_builder.hpp

Go to the documentation of this file.

//=======================================================================
// Copyright (c) 2014-2023 Baptiste Wicht
// Distributed under the terms of the MIT License.
// (See accompanying file LICENSE or copy at
//  http://opensource.org/licenses/MIT)
//=======================================================================

#pragma once

namespace etl {

template <size_t D, etl_expr E>
auto dim(E&& value, size_t i) -> detail::identity_helper<E, dim_view<detail::build_identity_type<E>, D>> {
    return detail::identity_helper<E, dim_view<detail::build_identity_type<E>, D>>{{value, i}};
}

template <etl_expr E>
auto row(E&& value, size_t i) -> detail::identity_helper<E, dim_view<detail::build_identity_type<E>, 1>> {
    return detail::identity_helper<E, dim_view<detail::build_identity_type<E>, 1>>{{value, i}};
}

template <etl_expr E>
auto col(E&& value, size_t i) -> detail::identity_helper<E, dim_view<detail::build_identity_type<E>, 2>> {
    return detail::identity_helper<E, dim_view<detail::build_identity_type<E>, 2>>{{value, i}};
}

template <matrix E>
auto sub(E&& value, size_t i) -> sub_view<detail::build_identity_type<E>, false> {
    return {value, i};
}

template <etl_2d E>
auto sub(E&& value, size_t i, size_t j, size_t m, size_t n) -> sub_matrix_2d<detail::build_identity_type<E>, false> {
    return {value, i, j, m, n};
}

template <etl_3d E>
auto sub(E&& value, size_t i, size_t j, size_t k, size_t m, size_t n, size_t o) -> sub_matrix_3d<detail::build_identity_type<E>, false> {
    return {value, i, j, k, m, n, o};
}

template <etl_4d E>
auto sub(E&& value, size_t i, size_t j, size_t k, size_t l, size_t m, size_t n, size_t o, size_t p) -> sub_matrix_4d<detail::build_identity_type<E>, false> {
    return {value, i, j, k, l, m, n, o, p};
}

template <etl_expr E>
auto slice(E&& value, size_t first, size_t last) -> slice_view<detail::build_identity_type<E>> {
    return {value, first, last};
}

template <size_t... Dims, etl_expr E>
auto reshape(E&& value) -> fast_matrix_view<detail::build_identity_type<E>, is_dma<E>, Dims...> {
    cpp_assert(decay_traits<E>::is_generator || etl::size(value) == (Dims * ...), "Invalid size for reshape");

    return fast_matrix_view<detail::build_identity_type<E>, is_dma<E>, Dims...>{value};
}

template <etl_expr E, typename... S>
auto reshape(E&& value, S... sizes) -> dyn_matrix_view<detail::build_identity_type<E>, sizeof...(sizes)> {
    using ret_type = dyn_matrix_view<detail::build_identity_type<E>, sizeof...(sizes)>;

    cpp_assert(decay_traits<E>::is_generator || etl::size(value) == util::size(sizes...), "Invalid size for reshape");

    return ret_type{value, size_t(sizes)...};
}

// Virtual Views that returns rvalues

template <typename D = double>
auto magic(size_t i) -> detail::virtual_helper<D, magic_view<D>> {
    return detail::virtual_helper<D, magic_view<D>>{magic_view<D>{i}};
}

template <size_t N, typename D = double>
auto magic() -> detail::virtual_helper<D, fast_magic_view<D, N>> {
    return detail::virtual_helper<D, fast_magic_view<D, N>>{{}};
}

} //end of namespace etl