blas1_swap_elements.hpp

Go to the documentation of this file.

// SPDX-License-Identifier: BSL-1.0
// SPDX-FileCopyrightText: Copyright (C) 2023 Tobias Hienzsch
 
#ifndef TETL_LINALG_BLAS1_SWAP_ELEMENTS_HPP
#define TETL_LINALG_BLAS1_SWAP_ELEMENTS_HPP
 
#include <etl/_contracts/check.hpp>
#include <etl/_linalg/exposition.hpp>
#include <etl/_utility/swap.hpp>
 
namespace etl::linalg {
 
/// \ingroup linalg
template <inout_object InOutObj1, inout_object InOutObj2>
    requires(InOutObj1::rank() == InOutObj1::rank())
constexpr auto swap_elements(InOutObj1 x, InOutObj2 y) -> void
{
    TETL_PRECONDITION(x.extents() == y.extents());
 
    using index_type = detail::common_index_type_t<InOutObj1, InOutObj2>;
 
    if constexpr (InOutObj1::rank() == 1) {
        static_assert(detail::compatible_static_extents<InOutObj1, InOutObj2>(0, 0));
 
        for (index_type i{0}; i < static_cast<index_type>(x.extent(0)); ++i) {
            using etl::swap;
            swap(x(i), y(i));
        }
    } else {
        static_assert(InOutObj1::rank() == 2);
        static_assert(detail::compatible_static_extents<InOutObj1, InOutObj2>(0, 0));
        static_assert(detail::compatible_static_extents<InOutObj1, InOutObj2>(1, 1));
 
        for (index_type i{0}; i < static_cast<index_type>(x.extent(0)); ++i) {
            for (index_type j{0}; j < static_cast<index_type>(x.extent(1)); ++j) {
                using etl::swap;
                swap(x(i, j), y(i, j));
            }
        }
    }
}
 
} // namespace etl::linalg
 
#endif // TETL_LINALG_BLAS1_SWAP_ELEMENTS_HPP