CwiseBinaryOp.h
Go to the documentation of this file.
1 // This file is part of Eigen, a lightweight C++ template library
2 // for linear algebra.
3 //
4 // Copyright (C) 2008-2014 Gael Guennebaud <gael.guennebaud@inria.fr>
5 // Copyright (C) 2006-2008 Benoit Jacob <jacob.benoit.1@gmail.com>
6 //
7 // This Source Code Form is subject to the terms of the Mozilla
8 // Public License v. 2.0. If a copy of the MPL was not distributed
9 // with this file, You can obtain one at http://mozilla.org/MPL/2.0/.
10 
11 #ifndef EIGEN_CWISE_BINARY_OP_H
12 #define EIGEN_CWISE_BINARY_OP_H
13 
14 namespace Eigen {
15 
16 namespace internal {
17 template<typename BinaryOp, typename Lhs, typename Rhs>
18 struct traits<CwiseBinaryOp<BinaryOp, Lhs, Rhs> >
19 {
20  // we must not inherit from traits<Lhs> since it has
21  // the potential to cause problems with MSVC
22  typedef typename remove_all<Lhs>::type Ancestor;
24  enum {
29  };
30 
31  // even though we require Lhs and Rhs to have the same scalar type (see CwiseBinaryOp constructor),
32  // we still want to handle the case when the result type is different.
33  typedef typename result_of<
34  BinaryOp(
35  const typename Lhs::Scalar&,
36  const typename Rhs::Scalar&
37  )
40  typename traits<Rhs>::StorageKind,
41  BinaryOp>::ret StorageKind;
44  typedef typename Lhs::Nested LhsNested;
45  typedef typename Rhs::Nested RhsNested;
48  enum {
49  Flags = cwise_promote_storage_order<typename traits<Lhs>::StorageKind,typename traits<Rhs>::StorageKind,_LhsNested::Flags & RowMajorBit,_RhsNested::Flags & RowMajorBit>::value
50  };
51 };
52 } // end namespace internal
53 
54 template<typename BinaryOp, typename Lhs, typename Rhs, typename StorageKind>
56 
76 template<typename BinaryOp, typename LhsType, typename RhsType>
77 class CwiseBinaryOp :
78  public CwiseBinaryOpImpl<
79  BinaryOp, LhsType, RhsType,
80  typename internal::cwise_promote_storage_type<typename internal::traits<LhsType>::StorageKind,
81  typename internal::traits<RhsType>::StorageKind,
82  BinaryOp>::ret>,
84 {
85  public:
86 
90 
91  typedef typename CwiseBinaryOpImpl<
92  BinaryOp, LhsType, RhsType,
95  BinaryOp>::ret>::Base Base;
97 
98  typedef typename internal::ref_selector<LhsType>::type LhsNested;
99  typedef typename internal::ref_selector<RhsType>::type RhsNested;
100  typedef typename internal::remove_reference<LhsNested>::type _LhsNested;
101  typedef typename internal::remove_reference<RhsNested>::type _RhsNested;
102 
103  EIGEN_DEVICE_FUNC
104  EIGEN_STRONG_INLINE CwiseBinaryOp(const Lhs& aLhs, const Rhs& aRhs, const BinaryOp& func = BinaryOp())
105  : m_lhs(aLhs), m_rhs(aRhs), m_functor(func)
106  {
107  EIGEN_CHECK_BINARY_COMPATIBILIY(BinaryOp,typename Lhs::Scalar,typename Rhs::Scalar);
108  // require the sizes to match
110  eigen_assert(aLhs.rows() == aRhs.rows() && aLhs.cols() == aRhs.cols());
111  }
112 
113  EIGEN_DEVICE_FUNC
115  // return the fixed size type if available to enable compile time optimizations
116  if (internal::traits<typename internal::remove_all<LhsNested>::type>::RowsAtCompileTime==Dynamic)
117  return m_rhs.rows();
118  else
119  return m_lhs.rows();
120  }
121  EIGEN_DEVICE_FUNC
123  // return the fixed size type if available to enable compile time optimizations
124  if (internal::traits<typename internal::remove_all<LhsNested>::type>::ColsAtCompileTime==Dynamic)
125  return m_rhs.cols();
126  else
127  return m_lhs.cols();
128  }
129 
131  EIGEN_DEVICE_FUNC
132  const _LhsNested& lhs() const { return m_lhs; }
134  EIGEN_DEVICE_FUNC
135  const _RhsNested& rhs() const { return m_rhs; }
137  EIGEN_DEVICE_FUNC
138  const BinaryOp& functor() const { return m_functor; }
139 
140  protected:
143  const BinaryOp m_functor;
144 };
145 
146 // Generic API dispatcher
147 template<typename BinaryOp, typename Lhs, typename Rhs, typename StorageKind>
148 class CwiseBinaryOpImpl
149  : public internal::generic_xpr_base<CwiseBinaryOp<BinaryOp, Lhs, Rhs> >::type
150 {
151 public:
153 };
154 
159 template<typename Derived>
160 template<typename OtherDerived>
161 EIGEN_STRONG_INLINE Derived &
163 {
165  return derived();
166 }
167 
172 template<typename Derived>
173 template<typename OtherDerived>
174 EIGEN_STRONG_INLINE Derived &
176 {
178  return derived();
179 }
180 
181 } // end namespace Eigen
182 
183 #endif // EIGEN_CWISE_BINARY_OP_H
184 
cwise_promote_storage_type< typename traits< Lhs >::StorageKind, typename traits< Rhs >::StorageKind, BinaryOp >::ret StorageKind
Definition: CwiseBinaryOp.h:41
SCALAR Scalar
Definition: bench_gemm.cpp:33
#define EIGEN_GENERIC_PUBLIC_INTERFACE(Derived)
Definition: Macros.h:852
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Derived & operator+=(const MatrixBase< OtherDerived > &other)
#define EIGEN_STRONG_INLINE
Definition: Macros.h:494
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Derived & operator-=(const MatrixBase< OtherDerived > &other)
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Index cols() const
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Index rows() const
internal::remove_all< LhsType >::type Lhs
Definition: CwiseBinaryOp.h:88
Namespace containing all symbols from the Eigen library.
Definition: jet.h:637
CwiseBinaryOpImpl< BinaryOp, LhsType, RhsType, typename internal::cwise_promote_storage_type< typename internal::traits< LhsType >::StorageKind, typename internal::traits< Rhs >::StorageKind, BinaryOp >::ret >::Base Base
Definition: CwiseBinaryOp.h:95
const unsigned int RowMajorBit
Definition: Constants.h:61
EIGEN_DEVICE_FUNC const _LhsNested & lhs() const
internal::generic_xpr_base< CwiseBinaryOp< BinaryOp, Lhs, Rhs > >::type Base
promote_index_type< typename traits< Lhs >::StorageIndex, typename traits< Rhs >::StorageIndex >::type StorageIndex
Definition: CwiseBinaryOp.h:43
internal::remove_all< BinaryOp >::type Functor
Definition: CwiseBinaryOp.h:87
Generic expression where a coefficient-wise binary operator is applied to two expressions.
Definition: CwiseBinaryOp.h:77
result_of< BinaryOp(const typename Lhs::Scalar &, const typename Rhs::Scalar &) >::type Scalar
Definition: CwiseBinaryOp.h:38
EIGEN_DEFAULT_DENSE_INDEX_TYPE Index
The Index type as used for the API.
Definition: Meta.h:33
#define eigen_assert(x)
Definition: Macros.h:579
const BinaryOp m_functor
EIGEN_DEVICE_FUNC const BinaryOp & functor() const
DenseIndex ret
Definition: level1_impl.h:59
#define EIGEN_CHECK_BINARY_COMPATIBILIY(BINOP, LHS, RHS)
Definition: XprHelper.h:815
const int Dynamic
Definition: Constants.h:21
EIGEN_DEVICE_FUNC const _RhsNested & rhs() const
Base class for all dense matrices, vectors, and expressions.
Definition: MatrixBase.h:48
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void call_assignment(Dst &dst, const Src &src)
Definition: pytypes.h:897
internal::remove_all< RhsType >::type Rhs
Definition: CwiseBinaryOp.h:89
#define EIGEN_STATIC_ASSERT_SAME_MATRIX_SIZE(TYPE0, TYPE1)
Definition: StaticAssert.h:189


gtsam
Author(s):
autogenerated on Sat May 8 2021 02:41:54