xref: /aosp_15_r20/external/eigen/Eigen/src/Core/Swap.h (revision bf2c37156dfe67e5dfebd6d394bad8b2ab5804d4)
1 // This file is part of Eigen, a lightweight C++ template library
2 // for linear algebra.
3 //
4 // Copyright (C) 2006-2008 Benoit Jacob <[email protected]>
5 //
6 // This Source Code Form is subject to the terms of the Mozilla
7 // Public License v. 2.0. If a copy of the MPL was not distributed
8 // with this file, You can obtain one at http://mozilla.org/MPL/2.0/.
9 
10 #ifndef EIGEN_SWAP_H
11 #define EIGEN_SWAP_H
12 
13 namespace Eigen {
14 
15 namespace internal {
16 
17 // Overload default assignPacket behavior for swapping them
18 template<typename DstEvaluatorTypeT, typename SrcEvaluatorTypeT>
19 class generic_dense_assignment_kernel<DstEvaluatorTypeT, SrcEvaluatorTypeT, swap_assign_op<typename DstEvaluatorTypeT::Scalar>, Specialized>
20  : public generic_dense_assignment_kernel<DstEvaluatorTypeT, SrcEvaluatorTypeT, swap_assign_op<typename DstEvaluatorTypeT::Scalar>, BuiltIn>
21 {
22 protected:
23   typedef generic_dense_assignment_kernel<DstEvaluatorTypeT, SrcEvaluatorTypeT, swap_assign_op<typename DstEvaluatorTypeT::Scalar>, BuiltIn> Base;
24   using Base::m_dst;
25   using Base::m_src;
26   using Base::m_functor;
27 
28 public:
29   typedef typename Base::Scalar Scalar;
30   typedef typename Base::DstXprType DstXprType;
31   typedef swap_assign_op<Scalar> Functor;
32 
33   EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
generic_dense_assignment_kernel(DstEvaluatorTypeT & dst,const SrcEvaluatorTypeT & src,const Functor & func,DstXprType & dstExpr)34   generic_dense_assignment_kernel(DstEvaluatorTypeT &dst, const SrcEvaluatorTypeT &src, const Functor &func, DstXprType& dstExpr)
35     : Base(dst, src, func, dstExpr)
36   {}
37 
38   template<int StoreMode, int LoadMode, typename PacketType>
assignPacket(Index row,Index col)39   EIGEN_STRONG_INLINE void assignPacket(Index row, Index col)
40   {
41     PacketType tmp = m_src.template packet<LoadMode,PacketType>(row,col);
42     const_cast<SrcEvaluatorTypeT&>(m_src).template writePacket<LoadMode>(row,col, m_dst.template packet<StoreMode,PacketType>(row,col));
43     m_dst.template writePacket<StoreMode>(row,col,tmp);
44   }
45 
46   template<int StoreMode, int LoadMode, typename PacketType>
assignPacket(Index index)47   EIGEN_STRONG_INLINE void assignPacket(Index index)
48   {
49     PacketType tmp = m_src.template packet<LoadMode,PacketType>(index);
50     const_cast<SrcEvaluatorTypeT&>(m_src).template writePacket<LoadMode>(index, m_dst.template packet<StoreMode,PacketType>(index));
51     m_dst.template writePacket<StoreMode>(index,tmp);
52   }
53 
54   // TODO find a simple way not to have to copy/paste this function from generic_dense_assignment_kernel, by simple I mean no CRTP (Gael)
55   template<int StoreMode, int LoadMode, typename PacketType>
assignPacketByOuterInner(Index outer,Index inner)56   EIGEN_STRONG_INLINE void assignPacketByOuterInner(Index outer, Index inner)
57   {
58     Index row = Base::rowIndexByOuterInner(outer, inner);
59     Index col = Base::colIndexByOuterInner(outer, inner);
60     assignPacket<StoreMode,LoadMode,PacketType>(row, col);
61   }
62 };
63 
64 } // namespace internal
65 
66 } // end namespace Eigen
67 
68 #endif // EIGEN_SWAP_H
69