swap.hpp

Go to the documentation of this file.

// SPDX-License-Identifier: BSD-2-Clause
 
#ifndef pRC_CORE_TENSOR_FUNCTIONS_SWAP_H
#define pRC_CORE_TENSOR_FUNCTIONS_SWAP_H
 
#include <prc/core/basic/range.hpp>
#include <prc/core/functors/swap.hpp>
#include <prc/core/functors/view.hpp>
#include <prc/core/tensor/declarations.hpp>
 
namespace pRC
{
    template<class XA, class XB, IsTensorish RA = RemoveReference<XA>,
        IsTensorish RB = RemoveReference<XB>,
        class EA = ResultOf<XA, typename RA::Subscripts>,
        class EB = ResultOf<XB, typename RB::Subscripts>>
        requires IsSame<typename RA::Sizes, typename RB::Sizes> &&
        IsInvocable<Swap, EA, EB> && IsInvocable<View, XA> &&
        IsInvocable<View, XB>
    static inline constexpr auto swap(XA &&a, XB &&b)
    {
        range<typename RA::Sizes>(
            [&a, &b](auto const... indices)
            {
                swap(forward<XA>(a)(indices...), forward<XB>(b)(indices...));
            });
    }
}
#endif // pRC_CORE_TENSOR_FUNCTIONS_SWAP_H