11 #ifndef EIGEN_DENSESTORAGEBASE_H
12 #define EIGEN_DENSESTORAGEBASE_H
14 #if defined(EIGEN_INITIALIZE_MATRICES_BY_ZERO)
15 # define EIGEN_INITIALIZE_COEFFS
16 # define EIGEN_INITIALIZE_COEFFS_IF_THAT_OPTION_IS_ENABLED for(int i=0;i<base().size();++i) coeffRef(i)=Scalar(0);
17 #elif defined(EIGEN_INITIALIZE_MATRICES_BY_NAN)
18 # define EIGEN_INITIALIZE_COEFFS
19 # define EIGEN_INITIALIZE_COEFFS_IF_THAT_OPTION_IS_ENABLED for(int i=0;i<base().size();++i) coeffRef(i)=std::numeric_limits<Scalar>::quiet_NaN();
21 # undef EIGEN_INITIALIZE_COEFFS
22 # define EIGEN_INITIALIZE_COEFFS_IF_THAT_OPTION_IS_ENABLED
29 template<
int MaxSizeAtCompileTime>
struct check_rows_cols_for_overflow {
30 template<
typename Index>
32 static EIGEN_ALWAYS_INLINE
void run(
Index,
Index)
37 template<>
struct check_rows_cols_for_overflow<
Dynamic> {
38 template<
typename Index>
40 static EIGEN_ALWAYS_INLINE
void run(
Index rows,
Index cols)
44 Index max_index = (std::size_t(1) << (8 *
sizeof(
Index) - 1)) - 1;
45 bool error = (rows == 0 || cols == 0) ?
false
46 : (rows > max_index / cols);
48 throw_std_bad_alloc();
52 template <
typename Derived,
53 typename OtherDerived = Derived,
54 bool IsVector = bool(Derived::IsVectorAtCompileTime) && bool(OtherDerived::IsVectorAtCompileTime)>
55 struct conservative_resize_like_impl;
57 template<
typename MatrixTypeA,
typename MatrixTypeB,
bool SwapPo
inters>
struct matrix_swap_impl;
61 #ifdef EIGEN_PARSED_BY_DOXYGEN
73 template<
typename _Scalar,
int _Rows,
int _Cols,
int _Options,
int _MaxRows,
int _MaxCols>
77 template<
typename _Scalar,
int _Rows,
int _Cols,
int _Options,
int _MaxRows,
int _MaxCols>
94 template<
typename Derived>
97 template<typename Derived>
102 enum { Options = internal::traits<Derived>::Options };
103 typedef typename internal::dense_xpr_base<Derived>::type Base;
105 typedef typename internal::traits<Derived>::StorageKind StorageKind;
106 typedef typename internal::traits<Derived>::Scalar Scalar;
108 typedef typename internal::packet_traits<Scalar>::type PacketScalar;
110 typedef Derived DenseType;
112 using Base::RowsAtCompileTime;
113 using Base::ColsAtCompileTime;
114 using Base::SizeAtCompileTime;
115 using Base::MaxRowsAtCompileTime;
116 using Base::MaxColsAtCompileTime;
117 using Base::MaxSizeAtCompileTime;
118 using Base::IsVectorAtCompileTime;
121 template<
typename PlainObjectType,
int MapOptions,
typename Str
ideType>
friend class Eigen::Map;
126 #if EIGEN_MAX_ALIGN_BYTES>0
129 friend class Eigen::Map<const Derived, AlignedMax>;
139 DenseStorage<Scalar, Base::MaxSizeAtCompileTime, Base::RowsAtCompileTime, Base::ColsAtCompileTime, Options> m_storage;
142 enum { NeedsToAlign = (SizeAtCompileTime !=
Dynamic) && (internal::traits<Derived>::Alignment>0) };
143 EIGEN_MAKE_ALIGNED_OPERATOR_NEW_IF(NeedsToAlign)
146 Base& base() {
return *
static_cast<Base*
>(
this); }
148 const Base& base()
const {
return *
static_cast<const Base*
>(
this); }
151 EIGEN_STRONG_INLINE
Index rows()
const {
return m_storage.rows(); }
153 EIGEN_STRONG_INLINE
Index cols()
const {
return m_storage.cols(); }
163 return m_storage.data()[colId + rowId * m_storage.cols()];
165 return m_storage.data()[rowId + colId * m_storage.rows()];
175 return m_storage.data()[index];
186 return m_storage.data()[colId + rowId * m_storage.cols()];
188 return m_storage.data()[rowId + colId * m_storage.rows()];
198 return m_storage.data()[index];
207 return m_storage.data()[colId + rowId * m_storage.cols()];
209 return m_storage.data()[rowId + colId * m_storage.rows()];
217 return m_storage.data()[index];
221 template<
int LoadMode>
222 EIGEN_STRONG_INLINE PacketScalar packet(
Index rowId,
Index colId)
const
224 return internal::ploadt<PacketScalar, LoadMode>
226 ? colId + rowId * m_storage.cols()
227 : rowId + colId * m_storage.rows()));
231 template<
int LoadMode>
232 EIGEN_STRONG_INLINE PacketScalar packet(
Index index)
const
234 return internal::ploadt<PacketScalar, LoadMode>(m_storage.data() + index);
238 template<
int StoreMode>
239 EIGEN_STRONG_INLINE
void writePacket(
Index rowId,
Index colId,
const PacketScalar& val)
241 internal::pstoret<Scalar, PacketScalar, StoreMode>
243 ? colId + rowId * m_storage.cols()
244 : rowId + colId * m_storage.rows()), val);
248 template<
int StoreMode>
249 EIGEN_STRONG_INLINE
void writePacket(
Index index,
const PacketScalar& val)
251 internal::pstoret<Scalar, PacketScalar, StoreMode>(m_storage.data() + index, val);
255 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
const Scalar *
data()
const
256 {
return m_storage.data(); }
260 {
return m_storage.data(); }
281 eigen_assert( EIGEN_IMPLIES(RowsAtCompileTime!=
Dynamic,rows==RowsAtCompileTime)
282 && EIGEN_IMPLIES(ColsAtCompileTime!=
Dynamic,cols==ColsAtCompileTime)
283 && EIGEN_IMPLIES(RowsAtCompileTime==
Dynamic && MaxRowsAtCompileTime!=
Dynamic,rows<=MaxRowsAtCompileTime)
284 && EIGEN_IMPLIES(ColsAtCompileTime==
Dynamic && MaxColsAtCompileTime!=
Dynamic,cols<=MaxColsAtCompileTime)
285 && rows>=0 && cols>=0 &&
"Invalid sizes when resizing a matrix or array.");
286 internal::check_rows_cols_for_overflow<MaxSizeAtCompileTime>::run(rows, cols);
287 #ifdef EIGEN_INITIALIZE_COEFFS
288 Index size = rows*cols;
289 bool size_changed = size != this->size();
290 m_storage.resize(size, rows, cols);
291 if(size_changed) EIGEN_INITIALIZE_COEFFS_IF_THAT_OPTION_IS_ENABLED
293 m_storage.resize(rows*cols, rows, cols);
312 eigen_assert(((SizeAtCompileTime ==
Dynamic && (MaxSizeAtCompileTime==
Dynamic || size<=MaxSizeAtCompileTime)) || SizeAtCompileTime == size) && size>=0);
313 #ifdef EIGEN_INITIALIZE_COEFFS
314 bool size_changed = size != this->size();
316 if(RowsAtCompileTime == 1)
317 m_storage.resize(size, 1, size);
319 m_storage.resize(size, size, 1);
320 #ifdef EIGEN_INITIALIZE_COEFFS
321 if(size_changed) EIGEN_INITIALIZE_COEFFS_IF_THAT_OPTION_IS_ENABLED
360 template<
typename OtherDerived>
364 const OtherDerived& other = _other.
derived();
365 internal::check_rows_cols_for_overflow<MaxSizeAtCompileTime>::run(other.rows(), other.cols());
366 const Index othersize = other.rows()*other.cols();
367 if(RowsAtCompileTime == 1)
369 eigen_assert(other.rows() == 1 || other.cols() == 1);
372 else if(ColsAtCompileTime == 1)
374 eigen_assert(other.rows() == 1 || other.cols() == 1);
377 else resize(other.rows(), other.cols());
392 internal::conservative_resize_like_impl<Derived>::run(*
this, rows, cols);
434 internal::conservative_resize_like_impl<Derived>::run(*
this, size);
446 template<
typename OtherDerived>
450 internal::conservative_resize_like_impl<Derived,OtherDerived>::run(*
this, other);
463 template<
typename OtherDerived>
468 return Base::lazyAssign(other.
derived());
471 template<
typename OtherDerived>
473 EIGEN_STRONG_INLINE Derived&
operator=(
const ReturnByValue<OtherDerived>& func)
475 resize(func.rows(), func.cols());
476 return Base::operator=(func);
484 EIGEN_STRONG_INLINE PlainObjectBase() : m_storage()
490 #ifndef EIGEN_PARSED_BY_DOXYGEN
494 explicit PlainObjectBase(internal::constructor_without_unaligned_array_assert)
495 : m_storage(internal::constructor_without_unaligned_array_assert())
501 #if EIGEN_HAS_RVALUE_REFERENCES
503 PlainObjectBase(PlainObjectBase&& other) EIGEN_NOEXCEPT
504 : m_storage( std::move(other.m_storage) )
509 PlainObjectBase&
operator=(PlainObjectBase&& other) EIGEN_NOEXCEPT
512 swap(m_storage, other.m_storage);
520 : Base(), m_storage(other.m_storage) { }
523 : m_storage(size, rows, cols)
530 template<
typename OtherDerived>
535 _check_template_params();
541 template<
typename OtherDerived>
546 _check_template_params();
551 template<
typename OtherDerived>
555 _check_template_params();
557 resize(other.rows(), other.cols());
558 other.evalTo(this->derived());
566 template<
typename OtherDerived>
571 Base::operator=(other.
derived());
572 return this->derived();
613 template<
int Outer,
int Inner>
615 {
return typename StridedConstMapType<Stride<Outer, Inner> >::type(
data, stride); }
616 template<
int Outer,
int Inner>
618 {
return typename StridedMapType<Stride<Outer, Inner> >::type(
data, stride); }
619 template<
int Outer,
int Inner>
621 {
return typename StridedConstMapType<Stride<Outer, Inner> >::type(
data, size, stride); }
622 template<
int Outer,
int Inner>
624 {
return typename StridedMapType<Stride<Outer, Inner> >::type(
data, size, stride); }
625 template<
int Outer,
int Inner>
627 {
return typename StridedConstMapType<Stride<Outer, Inner> >::type(
data, rows, cols, stride); }
628 template<
int Outer,
int Inner>
630 {
return typename StridedMapType<Stride<Outer, Inner> >::type(
data, rows, cols, stride); }
632 template<
int Outer,
int Inner>
634 {
return typename StridedConstAlignedMapType<Stride<Outer, Inner> >::type(
data, stride); }
635 template<
int Outer,
int Inner>
637 {
return typename StridedAlignedMapType<Stride<Outer, Inner> >::type(
data, stride); }
638 template<
int Outer,
int Inner>
640 {
return typename StridedConstAlignedMapType<Stride<Outer, Inner> >::type(
data, size, stride); }
641 template<
int Outer,
int Inner>
643 {
return typename StridedAlignedMapType<Stride<Outer, Inner> >::type(
data, size, stride); }
644 template<
int Outer,
int Inner>
646 {
return typename StridedConstAlignedMapType<Stride<Outer, Inner> >::type(
data, rows, cols, stride); }
647 template<
int Outer,
int Inner>
649 {
return typename StridedAlignedMapType<Stride<Outer, Inner> >::type(
data, rows, cols, stride); }
652 using Base::setConstant;
664 using Base::setRandom;
668 #ifdef EIGEN_PLAINOBJECTBASE_PLUGIN
669 #include EIGEN_PLAINOBJECTBASE_PLUGIN
680 template<
typename OtherDerived>
684 #ifdef EIGEN_NO_AUTOMATIC_RESIZING
685 eigen_assert((this->size()==0 || (IsVectorAtCompileTime ? (this->size() == other.
size())
686 : (rows() == other.
rows() && cols() == other.
cols())))
687 &&
"Size mismatch. Automatic resizing is disabled because EIGEN_NO_AUTOMATIC_RESIZING is defined");
688 EIGEN_ONLY_USED_FOR_DEBUG(other);
710 template<
typename OtherDerived>
714 internal::call_assignment(this->derived(), other.
derived());
715 return this->derived();
723 template<
typename OtherDerived>
732 internal::call_assignment_no_alias(this->derived(), other.
derived(), internal::assign_op<Scalar,typename OtherDerived::Scalar>());
733 return this->derived();
736 template<
typename T0,
typename T1>
738 EIGEN_STRONG_INLINE
void _init2(
Index rows,
Index cols,
typename internal::enable_if<Base::SizeAtCompileTime!=2,T0>::type* = 0)
740 const bool t0_is_integer_alike = internal::is_valid_index_type<T0>::value;
741 const bool t1_is_integer_alike = internal::is_valid_index_type<T1>::value;
742 EIGEN_STATIC_ASSERT(t0_is_integer_alike &&
744 FLOATING_POINT_ARGUMENT_PASSED__INTEGER_WAS_EXPECTED)
748 template<
typename T0,
typename T1>
750 EIGEN_STRONG_INLINE
void _init2(
const T0& val0,
const T1& val1,
typename internal::enable_if<Base::SizeAtCompileTime==2,T0>::type* = 0)
753 m_storage.data()[0] =
Scalar(val0);
754 m_storage.data()[1] =
Scalar(val1);
757 template<
typename T0,
typename T1>
760 typename internal::enable_if< (!internal::is_same<Index,Scalar>::value)
761 && (internal::is_same<T0,Index>::value)
762 && (internal::is_same<T1,Index>::value)
763 && Base::SizeAtCompileTime==2,T1>::type* = 0)
766 m_storage.data()[0] =
Scalar(val0);
767 m_storage.data()[1] =
Scalar(val1);
774 EIGEN_STRONG_INLINE
void _init1(
Index size,
typename internal::enable_if< (Base::SizeAtCompileTime!=1 || !internal::is_convertible<T, Scalar>::value)
775 && ((!internal::is_same<
typename internal::traits<Derived>::XprKind,
ArrayXpr>::value || Base::SizeAtCompileTime==
Dynamic)),T>::type* = 0)
778 const bool is_integer_alike = internal::is_valid_index_type<T>::value;
779 EIGEN_UNUSED_VARIABLE(is_integer_alike);
780 EIGEN_STATIC_ASSERT(is_integer_alike,
781 FLOATING_POINT_ARGUMENT_PASSED__INTEGER_WAS_EXPECTED)
788 EIGEN_STRONG_INLINE
void _init1(
const Scalar& val0,
typename internal::enable_if<Base::SizeAtCompileTime==1 && internal::is_convertible<T, Scalar>::value,T>::type* = 0)
791 m_storage.data()[0] = val0;
798 typename internal::enable_if< (!internal::is_same<Index,Scalar>::value)
799 && (internal::is_same<Index,T>::value)
800 && Base::SizeAtCompileTime==1
801 && internal::is_convertible<T, Scalar>::value,T*>::type* = 0)
804 m_storage.data()[0] =
Scalar(val0);
815 template<
typename T,
typename OtherDerived>
824 EIGEN_STRONG_INLINE
void _init1(
const Derived& other){
829 template<
typename T,
typename OtherDerived>
832 this->derived() = other;
835 template<
typename T,
typename OtherDerived>
837 EIGEN_STRONG_INLINE
void _init1(
const ReturnByValue<OtherDerived>& other)
839 resize(other.rows(), other.cols());
840 other.evalTo(this->derived());
843 template<
typename T,
typename OtherDerived,
int ColsAtCompileTime>
854 typename internal::enable_if< Base::SizeAtCompileTime!=
Dynamic
855 && Base::SizeAtCompileTime!=1
856 && internal::is_convertible<T, Scalar>::value
857 && internal::is_same<
typename internal::traits<Derived>::XprKind,
ArrayXpr>::value,T>::type* = 0)
859 Base::setConstant(val0);
866 typename internal::enable_if< (!internal::is_same<Index,Scalar>::value)
867 && (internal::is_same<Index,T>::value)
868 && Base::SizeAtCompileTime!=
Dynamic
869 && Base::SizeAtCompileTime!=1
870 && internal::is_convertible<T, Scalar>::value
871 && internal::is_same<
typename internal::traits<Derived>::XprKind,
ArrayXpr>::value,T*>::type* = 0)
873 Base::setConstant(val0);
876 template<
typename MatrixTypeA,
typename MatrixTypeB,
bool SwapPo
inters>
877 friend struct internal::matrix_swap_impl;
881 #ifndef EIGEN_PARSED_BY_DOXYGEN
886 template<
typename OtherDerived>
890 enum { SwapPointers = internal::is_same<Derived, OtherDerived>::value && Base::SizeAtCompileTime==
Dynamic };
891 internal::matrix_swap_impl<Derived, OtherDerived, bool(SwapPointers)>::run(this->derived(), other.
derived());
897 template<
typename OtherDerived>
899 void swap(DenseBase<OtherDerived>
const & other)
900 { Base::swap(other.derived()); }
903 static EIGEN_STRONG_INLINE
void _check_template_params()
905 EIGEN_STATIC_ASSERT((EIGEN_IMPLIES(MaxRowsAtCompileTime==1 && MaxColsAtCompileTime!=1, (Options&
RowMajor)==
RowMajor)
906 && EIGEN_IMPLIES(MaxColsAtCompileTime==1 && MaxRowsAtCompileTime!=1, (Options&
RowMajor)==0)
907 && ((RowsAtCompileTime ==
Dynamic) || (RowsAtCompileTime >= 0))
908 && ((ColsAtCompileTime ==
Dynamic) || (ColsAtCompileTime >= 0))
909 && ((MaxRowsAtCompileTime ==
Dynamic) || (MaxRowsAtCompileTime >= 0))
910 && ((MaxColsAtCompileTime ==
Dynamic) || (MaxColsAtCompileTime >= 0))
911 && (MaxRowsAtCompileTime == RowsAtCompileTime || RowsAtCompileTime==
Dynamic)
912 && (MaxColsAtCompileTime == ColsAtCompileTime || ColsAtCompileTime==
Dynamic)
914 INVALID_MATRIX_TEMPLATE_PARAMETERS)
917 enum { IsPlainObjectBase = 1 };
923 template <
typename Derived,
typename OtherDerived,
bool IsVector>
924 struct conservative_resize_like_impl
926 static void run(DenseBase<Derived>& _this,
Index rows,
Index cols)
928 if (_this.rows() == rows && _this.cols() == cols)
return;
929 EIGEN_STATIC_ASSERT_DYNAMIC_SIZE(Derived)
931 if ( ( Derived::IsRowMajor && _this.cols() == cols) ||
932 (!Derived::IsRowMajor && _this.rows() == rows) )
934 internal::check_rows_cols_for_overflow<Derived::MaxSizeAtCompileTime>::run(rows, cols);
935 _this.derived().m_storage.conservativeResize(rows*cols,rows,cols);
940 typename Derived::PlainObject tmp(rows,cols);
941 const Index common_rows = numext::mini(rows, _this.rows());
942 const Index common_cols = numext::mini(cols, _this.cols());
943 tmp.block(0,0,common_rows,common_cols) = _this.block(0,0,common_rows,common_cols);
944 _this.derived().swap(tmp);
948 static void run(DenseBase<Derived>& _this,
const DenseBase<OtherDerived>& other)
950 if (_this.rows() == other.rows() && _this.cols() == other.cols())
return;
957 EIGEN_STATIC_ASSERT_DYNAMIC_SIZE(Derived)
958 EIGEN_STATIC_ASSERT_DYNAMIC_SIZE(OtherDerived)
960 if ( ( Derived::IsRowMajor && _this.cols() == other.cols()) ||
961 (!Derived::IsRowMajor && _this.rows() == other.rows()) )
963 const Index new_rows = other.rows() - _this.rows();
964 const Index new_cols = other.cols() - _this.cols();
965 _this.derived().m_storage.conservativeResize(other.size(),other.rows(),other.cols());
967 _this.bottomRightCorner(new_rows, other.cols()) = other.bottomRows(new_rows);
969 _this.bottomRightCorner(other.rows(), new_cols) = other.rightCols(new_cols);
974 typename Derived::PlainObject tmp(other);
975 const Index common_rows = numext::mini(tmp.rows(), _this.rows());
976 const Index common_cols = numext::mini(tmp.cols(), _this.cols());
977 tmp.block(0,0,common_rows,common_cols) = _this.block(0,0,common_rows,common_cols);
978 _this.derived().swap(tmp);
985 template <
typename Derived,
typename OtherDerived>
986 struct conservative_resize_like_impl<Derived,OtherDerived,true>
987 : conservative_resize_like_impl<Derived,OtherDerived,false>
989 using conservative_resize_like_impl<Derived,OtherDerived,false>::run;
991 static void run(DenseBase<Derived>& _this,
Index size)
993 const Index new_rows = Derived::RowsAtCompileTime==1 ? 1 : size;
994 const Index new_cols = Derived::RowsAtCompileTime==1 ? size : 1;
995 _this.derived().m_storage.conservativeResize(size,new_rows,new_cols);
998 static void run(DenseBase<Derived>& _this,
const DenseBase<OtherDerived>& other)
1000 if (_this.rows() == other.rows() && _this.cols() == other.cols())
return;
1002 const Index num_new_elements = other.size() - _this.size();
1004 const Index new_rows = Derived::RowsAtCompileTime==1 ? 1 : other.rows();
1005 const Index new_cols = Derived::RowsAtCompileTime==1 ? other.cols() : 1;
1006 _this.derived().m_storage.conservativeResize(other.size(),new_rows,new_cols);
1008 if (num_new_elements > 0)
1009 _this.tail(num_new_elements) = other.tail(num_new_elements);
1013 template<
typename MatrixTypeA,
typename MatrixTypeB,
bool SwapPo
inters>
1014 struct matrix_swap_impl
1017 static inline void run(MatrixTypeA& a, MatrixTypeB& b)
1023 template<
typename MatrixTypeA,
typename MatrixTypeB>
1024 struct matrix_swap_impl<MatrixTypeA, MatrixTypeB, true>
1027 static inline void run(MatrixTypeA& a, MatrixTypeB& b)
1029 static_cast<typename MatrixTypeA::Base&
>(a).m_storage.swap(
static_cast<typename MatrixTypeB::Base&
>(b).m_storage);
1037 #endif // EIGEN_DENSESTORAGEBASE_H