Please, help us to better know about our user community by answering the following short survey: https://forms.gle/wpyrxWi18ox9Z5ae9
 Eigen  3.3.9
DenseCoeffsBase.h
1 // This file is part of Eigen, a lightweight C++ template library
2 // for linear algebra.
3 //
4 // Copyright (C) 2006-2010 Benoit Jacob <jacob.benoit.1@gmail.com>
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_DENSECOEFFSBASE_H
11 #define EIGEN_DENSECOEFFSBASE_H
12
13 namespace Eigen {
14
15 namespace internal {
17 {
18  typedef typename conditional<is_arithmetic<T>::value, T, typename add_const_on_value_type<T>::type>::type type;
19 };
20 }
21
33 template<typename Derived>
34 class DenseCoeffsBase<Derived,ReadOnlyAccessors> : public EigenBase<Derived>
35 {
36  public:
37  typedef typename internal::traits<Derived>::StorageKind StorageKind;
38  typedef typename internal::traits<Derived>::Scalar Scalar;
39  typedef typename internal::packet_traits<Scalar>::type PacketScalar;
40
41  // Explanation for this CoeffReturnType typedef.
42  // - This is the return type of the coeff() method.
43  // - The LvalueBit means exactly that we can offer a coeffRef() method, which means exactly that we can get references
44  // to coeffs, which means exactly that we can have coeff() return a const reference (as opposed to returning a value).
45  // - The is_artihmetic check is required since "const int", "const double", etc. will cause warnings on some systems
46  // while the declaration of "const T", where T is a non arithmetic type does not. Always returning "const Scalar&" is
47  // not possible, since the underlying expressions might not offer a valid address the reference could be referring to.
48  typedef typename internal::conditional<bool(internal::traits<Derived>::Flags&LvalueBit),
49  const Scalar&,
50  typename internal::conditional<internal::is_arithmetic<Scalar>::value, Scalar, const Scalar>::type
51  >::type CoeffReturnType;
52
54  typename internal::packet_traits<Scalar>::type
55  >::type PacketReturnType;
56
57  typedef EigenBase<Derived> Base;
58  using Base::rows;
59  using Base::cols;
60  using Base::size;
61  using Base::derived;
62
63  EIGEN_DEVICE_FUNC
64  EIGEN_STRONG_INLINE Index rowIndexByOuterInner(Index outer, Index inner) const
65  {
66  return int(Derived::RowsAtCompileTime) == 1 ? 0
67  : int(Derived::ColsAtCompileTime) == 1 ? inner
68  : int(Derived::Flags)&RowMajorBit ? outer
69  : inner;
70  }
71
72  EIGEN_DEVICE_FUNC
73  EIGEN_STRONG_INLINE Index colIndexByOuterInner(Index outer, Index inner) const
74  {
75  return int(Derived::ColsAtCompileTime) == 1 ? 0
76  : int(Derived::RowsAtCompileTime) == 1 ? inner
77  : int(Derived::Flags)&RowMajorBit ? inner
78  : outer;
79  }
80
95  EIGEN_DEVICE_FUNC
96  EIGEN_STRONG_INLINE CoeffReturnType coeff(Index row, Index col) const
97  {
98  eigen_internal_assert(row >= 0 && row < rows()
99  && col >= 0 && col < cols());
100  return internal::evaluator<Derived>(derived()).coeff(row,col);
101  }
102
103  EIGEN_DEVICE_FUNC
104  EIGEN_STRONG_INLINE CoeffReturnType coeffByOuterInner(Index outer, Index inner) const
105  {
106  return coeff(rowIndexByOuterInner(outer, inner),
107  colIndexByOuterInner(outer, inner));
108  }
109
114  EIGEN_DEVICE_FUNC
115  EIGEN_STRONG_INLINE CoeffReturnType operator()(Index row, Index col) const
116  {
117  eigen_assert(row >= 0 && row < rows()
118  && col >= 0 && col < cols());
119  return coeff(row, col);
120  }
121
137  EIGEN_DEVICE_FUNC
138  EIGEN_STRONG_INLINE CoeffReturnType
139  coeff(Index index) const
140  {
141  EIGEN_STATIC_ASSERT(internal::evaluator<Derived>::Flags & LinearAccessBit,
142  THIS_COEFFICIENT_ACCESSOR_TAKING_ONE_ACCESS_IS_ONLY_FOR_EXPRESSIONS_ALLOWING_LINEAR_ACCESS)
143  eigen_internal_assert(index >= 0 && index < size());
144  return internal::evaluator<Derived>(derived()).coeff(index);
145  }
146
147
156  EIGEN_DEVICE_FUNC
157  EIGEN_STRONG_INLINE CoeffReturnType
158  operator[](Index index) const
159  {
160  EIGEN_STATIC_ASSERT(Derived::IsVectorAtCompileTime,
162  eigen_assert(index >= 0 && index < size());
163  return coeff(index);
164  }
165
176  EIGEN_DEVICE_FUNC
177  EIGEN_STRONG_INLINE CoeffReturnType
178  operator()(Index index) const
179  {
180  eigen_assert(index >= 0 && index < size());
181  return coeff(index);
182  }
183
186  EIGEN_DEVICE_FUNC
187  EIGEN_STRONG_INLINE CoeffReturnType
188  x() const { return (*this)[0]; }
189
192  EIGEN_DEVICE_FUNC
193  EIGEN_STRONG_INLINE CoeffReturnType
194  y() const
195  {
196  EIGEN_STATIC_ASSERT(Derived::SizeAtCompileTime==-1 || Derived::SizeAtCompileTime>=2, OUT_OF_RANGE_ACCESS);
197  return (*this)[1];
198  }
199
202  EIGEN_DEVICE_FUNC
203  EIGEN_STRONG_INLINE CoeffReturnType
204  z() const
205  {
206  EIGEN_STATIC_ASSERT(Derived::SizeAtCompileTime==-1 || Derived::SizeAtCompileTime>=3, OUT_OF_RANGE_ACCESS);
207  return (*this)[2];
208  }
209
212  EIGEN_DEVICE_FUNC
213  EIGEN_STRONG_INLINE CoeffReturnType
214  w() const
215  {
216  EIGEN_STATIC_ASSERT(Derived::SizeAtCompileTime==-1 || Derived::SizeAtCompileTime>=4, OUT_OF_RANGE_ACCESS);
217  return (*this)[3];
218  }
219
231  EIGEN_STRONG_INLINE PacketReturnType packet(Index row, Index col) const
232  {
233  typedef typename internal::packet_traits<Scalar>::type DefaultPacketType;
234  eigen_internal_assert(row >= 0 && row < rows() && col >= 0 && col < cols());
236  }
237
238
241  EIGEN_STRONG_INLINE PacketReturnType packetByOuterInner(Index outer, Index inner) const
242  {
244  colIndexByOuterInner(outer, inner));
245  }
246
258  EIGEN_STRONG_INLINE PacketReturnType packet(Index index) const
259  {
260  EIGEN_STATIC_ASSERT(internal::evaluator<Derived>::Flags & LinearAccessBit,
261  THIS_COEFFICIENT_ACCESSOR_TAKING_ONE_ACCESS_IS_ONLY_FOR_EXPRESSIONS_ALLOWING_LINEAR_ACCESS)
262  typedef typename internal::packet_traits<Scalar>::type DefaultPacketType;
263  eigen_internal_assert(index >= 0 && index < size());
265  }
266
267  protected:
268  // explanation: DenseBase is doing "using ..." on the methods from DenseCoeffsBase.
269  // But some methods are only available in the DirectAccess case.
270  // So we add dummy methods here with these names, so that "using... " doesn't fail.
271  // It's not private so that the child class DenseBase can access them, and it's not public
272  // either since it's an implementation detail, so has to be protected.
273  void coeffRef();
274  void coeffRefByOuterInner();
275  void writePacket();
276  void writePacketByOuterInner();
277  void copyCoeff();
278  void copyCoeffByOuterInner();
279  void copyPacket();
280  void copyPacketByOuterInner();
281  void stride();
282  void innerStride();
283  void outerStride();
284  void rowStride();
285  void colStride();
286 };
287
299 template<typename Derived>
300 class DenseCoeffsBase<Derived, WriteAccessors> : public DenseCoeffsBase<Derived, ReadOnlyAccessors>
301 {
302  public:
303
305
306  typedef typename internal::traits<Derived>::StorageKind StorageKind;
307  typedef typename internal::traits<Derived>::Scalar Scalar;
308  typedef typename internal::packet_traits<Scalar>::type PacketScalar;
309  typedef typename NumTraits<Scalar>::Real RealScalar;
310
311  using Base::coeff;
312  using Base::rows;
313  using Base::cols;
314  using Base::size;
315  using Base::derived;
316  using Base::rowIndexByOuterInner;
317  using Base::colIndexByOuterInner;
318  using Base::operator[];
319  using Base::operator();
320  using Base::x;
321  using Base::y;
322  using Base::z;
323  using Base::w;
324
339  EIGEN_DEVICE_FUNC
340  EIGEN_STRONG_INLINE Scalar& coeffRef(Index row, Index col)
341  {
342  eigen_internal_assert(row >= 0 && row < rows()
343  && col >= 0 && col < cols());
344  return internal::evaluator<Derived>(derived()).coeffRef(row,col);
345  }
346
347  EIGEN_DEVICE_FUNC
348  EIGEN_STRONG_INLINE Scalar&
349  coeffRefByOuterInner(Index outer, Index inner)
350  {
351  return coeffRef(rowIndexByOuterInner(outer, inner),
352  colIndexByOuterInner(outer, inner));
353  }
354
360  EIGEN_DEVICE_FUNC
361  EIGEN_STRONG_INLINE Scalar&
363  {
364  eigen_assert(row >= 0 && row < rows()
365  && col >= 0 && col < cols());
366  return coeffRef(row, col);
367  }
368
369
385  EIGEN_DEVICE_FUNC
386  EIGEN_STRONG_INLINE Scalar&
388  {
389  EIGEN_STATIC_ASSERT(internal::evaluator<Derived>::Flags & LinearAccessBit,
390  THIS_COEFFICIENT_ACCESSOR_TAKING_ONE_ACCESS_IS_ONLY_FOR_EXPRESSIONS_ALLOWING_LINEAR_ACCESS)
391  eigen_internal_assert(index >= 0 && index < size());
392  return internal::evaluator<Derived>(derived()).coeffRef(index);
393  }
394
402  EIGEN_DEVICE_FUNC
403  EIGEN_STRONG_INLINE Scalar&
405  {
406  EIGEN_STATIC_ASSERT(Derived::IsVectorAtCompileTime,
408  eigen_assert(index >= 0 && index < size());
409  return coeffRef(index);
410  }
411
421  EIGEN_DEVICE_FUNC
422  EIGEN_STRONG_INLINE Scalar&
424  {
425  eigen_assert(index >= 0 && index < size());
426  return coeffRef(index);
427  }
428
431  EIGEN_DEVICE_FUNC
432  EIGEN_STRONG_INLINE Scalar&
433  x() { return (*this)[0]; }
434
437  EIGEN_DEVICE_FUNC
438  EIGEN_STRONG_INLINE Scalar&
439  y()
440  {
441  EIGEN_STATIC_ASSERT(Derived::SizeAtCompileTime==-1 || Derived::SizeAtCompileTime>=2, OUT_OF_RANGE_ACCESS);
442  return (*this)[1];
443  }
444
447  EIGEN_DEVICE_FUNC
448  EIGEN_STRONG_INLINE Scalar&
449  z()
450  {
451  EIGEN_STATIC_ASSERT(Derived::SizeAtCompileTime==-1 || Derived::SizeAtCompileTime>=3, OUT_OF_RANGE_ACCESS);
452  return (*this)[2];
453  }
454
457  EIGEN_DEVICE_FUNC
458  EIGEN_STRONG_INLINE Scalar&
459  w()
460  {
461  EIGEN_STATIC_ASSERT(Derived::SizeAtCompileTime==-1 || Derived::SizeAtCompileTime>=4, OUT_OF_RANGE_ACCESS);
462  return (*this)[3];
463  }
464 };
465
477 template<typename Derived>
478 class DenseCoeffsBase<Derived, DirectAccessors> : public DenseCoeffsBase<Derived, ReadOnlyAccessors>
479 {
480  public:
481
483  typedef typename internal::traits<Derived>::Scalar Scalar;
484  typedef typename NumTraits<Scalar>::Real RealScalar;
485
486  using Base::rows;
487  using Base::cols;
488  using Base::size;
489  using Base::derived;
490
495  EIGEN_DEVICE_FUNC
496  inline Index innerStride() const
497  {
498  return derived().innerStride();
499  }
500
506  EIGEN_DEVICE_FUNC
507  inline Index outerStride() const
508  {
509  return derived().outerStride();
510  }
511
512  // FIXME shall we remove it ?
513  inline Index stride() const
514  {
515  return Derived::IsVectorAtCompileTime ? innerStride() : outerStride();
516  }
517
522  EIGEN_DEVICE_FUNC
523  inline Index rowStride() const
524  {
525  return Derived::IsRowMajor ? outerStride() : innerStride();
526  }
527
532  EIGEN_DEVICE_FUNC
533  inline Index colStride() const
534  {
535  return Derived::IsRowMajor ? innerStride() : outerStride();
536  }
537 };
538
550 template<typename Derived>
551 class DenseCoeffsBase<Derived, DirectWriteAccessors>
552  : public DenseCoeffsBase<Derived, WriteAccessors>
553 {
554  public:
555
557  typedef typename internal::traits<Derived>::Scalar Scalar;
558  typedef typename NumTraits<Scalar>::Real RealScalar;
559
560  using Base::rows;
561  using Base::cols;
562  using Base::size;
563  using Base::derived;
564
569  EIGEN_DEVICE_FUNC
570  inline Index innerStride() const
571  {
572  return derived().innerStride();
573  }
574
580  EIGEN_DEVICE_FUNC
581  inline Index outerStride() const
582  {
583  return derived().outerStride();
584  }
585
586  // FIXME shall we remove it ?
587  inline Index stride() const
588  {
589  return Derived::IsVectorAtCompileTime ? innerStride() : outerStride();
590  }
591
596  EIGEN_DEVICE_FUNC
597  inline Index rowStride() const
598  {
599  return Derived::IsRowMajor ? outerStride() : innerStride();
600  }
601
606  EIGEN_DEVICE_FUNC
607  inline Index colStride() const
608  {
609  return Derived::IsRowMajor ? innerStride() : outerStride();
610  }
611 };
612
613 namespace internal {
614
615 template<int Alignment, typename Derived, bool JustReturnZero>
616 struct first_aligned_impl
617 {
618  static inline Index run(const Derived&)
619  { return 0; }
620 };
621
622 template<int Alignment, typename Derived>
623 struct first_aligned_impl<Alignment, Derived, false>
624 {
625  static inline Index run(const Derived& m)
626  {
627  return internal::first_aligned<Alignment>(m.data(), m.size());
628  }
629 };
630
638 template<int Alignment, typename Derived>
639 static inline Index first_aligned(const DenseBase<Derived>& m)
640 {
641  enum { ReturnZero = (int(evaluator<Derived>::Alignment) >= Alignment) || !(Derived::Flags & DirectAccessBit) };
642  return first_aligned_impl<Alignment, Derived, ReturnZero>::run(m.derived());
643 }
644
645 template<typename Derived>
646 static inline Index first_default_aligned(const DenseBase<Derived>& m)
647 {
648  typedef typename Derived::Scalar Scalar;
649  typedef typename packet_traits<Scalar>::type DefaultPacketType;
650  return internal::first_aligned<int(unpacket_traits<DefaultPacketType>::alignment),Derived>(m);
651 }
652
653 template<typename Derived, bool HasDirectAccess = has_direct_access<Derived>::ret>
654 struct inner_stride_at_compile_time
655 {
656  enum { ret = traits<Derived>::InnerStrideAtCompileTime };
657 };
658
659 template<typename Derived>
660 struct inner_stride_at_compile_time<Derived, false>
661 {
662  enum { ret = 0 };
663 };
664
665 template<typename Derived, bool HasDirectAccess = has_direct_access<Derived>::ret>
666 struct outer_stride_at_compile_time
667 {
668  enum { ret = traits<Derived>::OuterStrideAtCompileTime };
669 };
670
671 template<typename Derived>
672 struct outer_stride_at_compile_time<Derived, false>
673 {
674  enum { ret = 0 };
675 };
676
677 } // end namespace internal
678
679 } // end namespace Eigen
680
681 #endif // EIGEN_DENSECOEFFSBASE_H
Eigen::DenseCoeffsBase< Derived, DirectAccessors >::colStride
Index colStride() const
Definition: DenseCoeffsBase.h:533
CoeffReturnType operator[](Index index) const
Definition: DenseCoeffsBase.h:158
Eigen
Namespace containing all symbols from the Eigen library.
Definition: Core:309
Eigen::EigenBase::Index
Eigen::Index Index
The interface type of indices.
Definition: EigenBase.h:38
Eigen::DenseCoeffsBase< Derived, WriteAccessors >::operator()
Scalar & operator()(Index index)
Definition: DenseCoeffsBase.h:423
Eigen::DenseCoeffsBase< Derived, DirectWriteAccessors >::colStride
Index colStride() const
Definition: DenseCoeffsBase.h:607
Eigen::EigenBase
Definition: EigenBase.h:30
Eigen::DenseCoeffsBase< Derived, DirectAccessors >::innerStride
Index innerStride() const
Definition: DenseCoeffsBase.h:496
Eigen::RowMajorBit
const unsigned int RowMajorBit
Definition: Constants.h:61
Eigen::DenseCoeffsBase< Derived, DirectAccessors >::outerStride
Index outerStride() const
Definition: DenseCoeffsBase.h:507
Eigen::DenseCoeffsBase< Derived, WriteAccessors >::x
Scalar & x()
Definition: DenseCoeffsBase.h:433
Definition: DenseCoeffsBase.h:35
CoeffReturnType operator()(Index index) const
Definition: DenseCoeffsBase.h:178
Eigen::DirectAccessBit
const unsigned int DirectAccessBit
Definition: Constants.h:150
Eigen::DirectAccessors
@ DirectAccessors
Definition: Constants.h:370
Eigen::DenseCoeffsBase< Derived, WriteAccessors >::operator[]
Scalar & operator[](Index index)
Definition: DenseCoeffsBase.h:404
Eigen::LvalueBit
const unsigned int LvalueBit
Definition: Constants.h:139
CoeffReturnType coeff(Index index) const
Definition: DenseCoeffsBase.h:139
CoeffReturnType x() const
Definition: DenseCoeffsBase.h:188
CoeffReturnType operator()(Index row, Index col) const
Definition: DenseCoeffsBase.h:115
Definition: Constants.h:366
Eigen::DenseCoeffsBase< Derived, WriteAccessors >::w
Scalar & w()
Definition: DenseCoeffsBase.h:459
CoeffReturnType w() const
Definition: DenseCoeffsBase.h:214
Eigen::LinearAccessBit
const unsigned int LinearAccessBit
Definition: Constants.h:125
Eigen::DenseCoeffsBase< Derived, WriteAccessors >::z
Scalar & z()
Definition: DenseCoeffsBase.h:449
Eigen::DenseCoeffsBase< Derived, DirectWriteAccessors >::outerStride
Index outerStride() const
Definition: DenseCoeffsBase.h:581
Eigen::DenseCoeffsBase< Derived, WriteAccessors >::coeffRef
Scalar & coeffRef(Index row, Index col)
Definition: DenseCoeffsBase.h:340
Eigen::DirectWriteAccessors
@ DirectWriteAccessors
Definition: Constants.h:372
Eigen::DenseCoeffsBase< Derived, DirectWriteAccessors >::innerStride
Index innerStride() const
Definition: DenseCoeffsBase.h:570
CoeffReturnType y() const
Definition: DenseCoeffsBase.h:194
Eigen::DenseCoeffsBase< Derived, DirectAccessors >::rowStride
Index rowStride() const
Definition: DenseCoeffsBase.h:523
CoeffReturnType z() const
Definition: DenseCoeffsBase.h:204
CoeffReturnType coeff(Index row, Index col) const
Definition: DenseCoeffsBase.h:96
Eigen::DenseCoeffsBase< Derived, WriteAccessors >::coeffRef
Scalar & coeffRef(Index index)
Definition: DenseCoeffsBase.h:387
Eigen::WriteAccessors
@ WriteAccessors
Definition: Constants.h:368
Eigen::DenseCoeffsBase< Derived, WriteAccessors >::operator()
Scalar & operator()(Index row, Index col)
Definition: DenseCoeffsBase.h:362
Eigen::DenseCoeffsBase< Derived, DirectWriteAccessors >::rowStride
Index rowStride() const
Definition: DenseCoeffsBase.h:597
Eigen::NumTraits
Holds information about the various numeric (i.e. scalar) types allowed by Eigen.
Definition: NumTraits.h:151
Eigen::DenseCoeffsBase< Derived, WriteAccessors >