Please, help us to better know about our user community by answering the following short survey: https://forms.gle/wpyrxWi18ox9Z5ae9
Eigen  3.4.0
CwiseUnaryView.h
1 // This file is part of Eigen, a lightweight C++ template library
2 // for linear algebra.
3 //
4 // Copyright (C) 2009-2010 Gael Guennebaud <gael.guennebaud@inria.fr>
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_CWISE_UNARY_VIEW_H
11 #define EIGEN_CWISE_UNARY_VIEW_H
12 
13 namespace Eigen {
14 
15 namespace internal {
16 template<typename ViewOp, typename MatrixType>
17 struct traits<CwiseUnaryView<ViewOp, MatrixType> >
18  : traits<MatrixType>
19 {
20  typedef typename result_of<
21  ViewOp(const typename traits<MatrixType>::Scalar&)
22  >::type Scalar;
23  typedef typename MatrixType::Nested MatrixTypeNested;
24  typedef typename remove_all<MatrixTypeNested>::type _MatrixTypeNested;
25  enum {
26  FlagsLvalueBit = is_lvalue<MatrixType>::value ? LvalueBit : 0,
27  Flags = traits<_MatrixTypeNested>::Flags & (RowMajorBit | FlagsLvalueBit | DirectAccessBit), // FIXME DirectAccessBit should not be handled by expressions
28  MatrixTypeInnerStride = inner_stride_at_compile_time<MatrixType>::ret,
29  // need to cast the sizeof's from size_t to int explicitly, otherwise:
30  // "error: no integral type can represent all of the enumerator values
31  InnerStrideAtCompileTime = MatrixTypeInnerStride == Dynamic
32  ? int(Dynamic)
33  : int(MatrixTypeInnerStride) * int(sizeof(typename traits<MatrixType>::Scalar) / sizeof(Scalar)),
34  OuterStrideAtCompileTime = outer_stride_at_compile_time<MatrixType>::ret == Dynamic
35  ? int(Dynamic)
36  : outer_stride_at_compile_time<MatrixType>::ret * int(sizeof(typename traits<MatrixType>::Scalar) / sizeof(Scalar))
37  };
38 };
39 }
40 
41 template<typename ViewOp, typename MatrixType, typename StorageKind>
42 class CwiseUnaryViewImpl;
43 
57 template<typename ViewOp, typename MatrixType>
58 class CwiseUnaryView : public CwiseUnaryViewImpl<ViewOp, MatrixType, typename internal::traits<MatrixType>::StorageKind>
59 {
60  public:
61 
62  typedef typename CwiseUnaryViewImpl<ViewOp, MatrixType,typename internal::traits<MatrixType>::StorageKind>::Base Base;
63  EIGEN_GENERIC_PUBLIC_INTERFACE(CwiseUnaryView)
64  typedef typename internal::ref_selector<MatrixType>::non_const_type MatrixTypeNested;
65  typedef typename internal::remove_all<MatrixType>::type NestedExpression;
66 
67  explicit EIGEN_DEVICE_FUNC inline CwiseUnaryView(MatrixType& mat, const ViewOp& func = ViewOp())
68  : m_matrix(mat), m_functor(func) {}
69 
70  EIGEN_INHERIT_ASSIGNMENT_OPERATORS(CwiseUnaryView)
71 
72  EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE EIGEN_CONSTEXPR
73  Index rows() const EIGEN_NOEXCEPT { return m_matrix.rows(); }
74  EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE EIGEN_CONSTEXPR
75  Index cols() const EIGEN_NOEXCEPT { return m_matrix.cols(); }
76 
78  EIGEN_DEVICE_FUNC const ViewOp& functor() const { return m_functor; }
79 
81  EIGEN_DEVICE_FUNC const typename internal::remove_all<MatrixTypeNested>::type&
82  nestedExpression() const { return m_matrix; }
83 
85  EIGEN_DEVICE_FUNC typename internal::remove_reference<MatrixTypeNested>::type&
86  nestedExpression() { return m_matrix; }
87 
88  protected:
89  MatrixTypeNested m_matrix;
90  ViewOp m_functor;
91 };
92 
93 // Generic API dispatcher
94 template<typename ViewOp, typename XprType, typename StorageKind>
95 class CwiseUnaryViewImpl
96  : public internal::generic_xpr_base<CwiseUnaryView<ViewOp, XprType> >::type
97 {
98 public:
99  typedef typename internal::generic_xpr_base<CwiseUnaryView<ViewOp, XprType> >::type Base;
100 };
101 
102 template<typename ViewOp, typename MatrixType>
103 class CwiseUnaryViewImpl<ViewOp,MatrixType,Dense>
104  : public internal::dense_xpr_base< CwiseUnaryView<ViewOp, MatrixType> >::type
105 {
106  public:
107 
108  typedef CwiseUnaryView<ViewOp, MatrixType> Derived;
109  typedef typename internal::dense_xpr_base< CwiseUnaryView<ViewOp, MatrixType> >::type Base;
110 
111  EIGEN_DENSE_PUBLIC_INTERFACE(Derived)
112  EIGEN_INHERIT_ASSIGNMENT_OPERATORS(CwiseUnaryViewImpl)
113 
114  EIGEN_DEVICE_FUNC inline Scalar* data() { return &(this->coeffRef(0)); }
115  EIGEN_DEVICE_FUNC inline const Scalar* data() const { return &(this->coeff(0)); }
116 
117  EIGEN_DEVICE_FUNC EIGEN_CONSTEXPR inline Index innerStride() const
118  {
119  return derived().nestedExpression().innerStride() * sizeof(typename internal::traits<MatrixType>::Scalar) / sizeof(Scalar);
120  }
121 
122  EIGEN_DEVICE_FUNC EIGEN_CONSTEXPR inline Index outerStride() const
123  {
124  return derived().nestedExpression().outerStride() * sizeof(typename internal::traits<MatrixType>::Scalar) / sizeof(Scalar);
125  }
126  protected:
127  EIGEN_DEFAULT_EMPTY_CONSTRUCTOR_AND_DESTRUCTOR(CwiseUnaryViewImpl)
128 };
129 
130 } // end namespace Eigen
131 
132 #endif // EIGEN_CWISE_UNARY_VIEW_H
const internal::remove_all< MatrixTypeNested >::type & nestedExpression() const
Definition: CwiseUnaryView.h:82
const unsigned int DirectAccessBit
Definition: Constants.h:155
const unsigned int LvalueBit
Definition: Constants.h:144
const ViewOp & functor() const
Definition: CwiseUnaryView.h:78
Namespace containing all symbols from the Eigen library.
Definition: Core:141
const unsigned int RowMajorBit
Definition: Constants.h:66
Generic lvalue expression of a coefficient-wise unary operator of a matrix or a vector.
Definition: CwiseUnaryView.h:58
internal::remove_reference< MatrixTypeNested >::type & nestedExpression()
Definition: CwiseUnaryView.h:86
EIGEN_DEFAULT_DENSE_INDEX_TYPE Index
The Index type as used for the API.
Definition: Meta.h:74
Definition: Eigen_Colamd.h:50
const int Dynamic
Definition: Constants.h:22