10 #ifndef EIGEN_SPARSEASSIGN_H
11 #define EIGEN_SPARSEASSIGN_H
15 template<
typename Derived>
16 template<
typename OtherDerived>
17 Derived& SparseMatrixBase<Derived>::operator=(
const EigenBase<OtherDerived> &other)
19 internal::call_assignment_no_alias(derived(), other.derived());
23 template<
typename Derived>
24 template<
typename OtherDerived>
25 Derived& SparseMatrixBase<Derived>::operator=(
const ReturnByValue<OtherDerived>& other)
28 other.evalTo(derived());
32 template<
typename Derived>
33 template<
typename OtherDerived>
34 inline Derived& SparseMatrixBase<Derived>::operator=(
const SparseMatrixBase<OtherDerived>& other)
37 internal::Assignment<Derived,OtherDerived,internal::assign_op<Scalar> >
38 ::run(derived(), other.derived(), internal::assign_op<Scalar>());
42 template<
typename Derived>
43 inline Derived& SparseMatrixBase<Derived>::operator=(
const Derived& other)
45 internal::call_assignment_no_alias(derived(), other.derived());
52 struct storage_kind_to_evaluator_kind<Sparse> {
53 typedef IteratorBased Kind;
57 struct storage_kind_to_shape<Sparse> {
58 typedef SparseShape Shape;
61 struct Sparse2Sparse {};
62 struct Sparse2Dense {};
64 template<>
struct AssignmentKind<SparseShape, SparseShape> {
typedef Sparse2Sparse Kind; };
65 template<>
struct AssignmentKind<SparseShape, SparseTriangularShape> {
typedef Sparse2Sparse Kind; };
66 template<>
struct AssignmentKind<DenseShape, SparseShape> {
typedef Sparse2Dense Kind; };
69 template<
typename DstXprType,
typename SrcXprType>
70 void assign_sparse_to_sparse(DstXprType &dst,
const SrcXprType &src)
72 typedef typename DstXprType::Scalar Scalar;
73 typedef internal::evaluator<DstXprType> DstEvaluatorType;
74 typedef internal::evaluator<SrcXprType> SrcEvaluatorType;
76 SrcEvaluatorType srcEvaluator(src);
79 const Index outerEvaluationSize = (SrcEvaluatorType::Flags&
RowMajorBit) ? src.rows() : src.cols();
80 if ((!transpose) && src.isRValue())
83 dst.resize(src.rows(), src.cols());
85 dst.reserve((std::max)(src.rows(),src.cols())*2);
86 for (Index j=0; j<outerEvaluationSize; ++j)
89 for (
typename SrcEvaluatorType::InnerIterator it(srcEvaluator, j); it; ++it)
91 Scalar v = it.value();
92 dst.insertBackByOuterInner(j,it.index()) = v;
100 eigen_assert(( ((internal::traits<DstXprType>::SupportedAccessPatterns & OuterRandomAccessPattern)==OuterRandomAccessPattern) ||
101 (!((DstEvaluatorType::Flags & RowMajorBit) != (SrcEvaluatorType::Flags & RowMajorBit)))) &&
102 "the transpose operation is supposed to be handled in SparseMatrix::operator=");
104 enum { Flip = (DstEvaluatorType::Flags &
RowMajorBit) != (SrcEvaluatorType::Flags & RowMajorBit) };
107 DstXprType temp(src.rows(), src.cols());
109 temp.reserve((std::max)(src.rows(),src.cols())*2);
110 for (Index j=0; j<outerEvaluationSize; ++j)
113 for (
typename SrcEvaluatorType::InnerIterator it(srcEvaluator, j); it; ++it)
115 Scalar v = it.value();
116 temp.insertBackByOuterInner(Flip?it.index():j,Flip?j:it.index()) = v;
121 dst = temp.markAsRValue();
126 template<
typename DstXprType,
typename SrcXprType,
typename Functor,
typename Scalar>
127 struct Assignment<DstXprType, SrcXprType, Functor, Sparse2Sparse, Scalar>
129 static void run(DstXprType &dst,
const SrcXprType &src,
const internal::assign_op<typename DstXprType::Scalar> &)
131 assign_sparse_to_sparse(dst.derived(), src.derived());
136 template<
typename DstXprType,
typename SrcXprType,
typename Functor,
typename Scalar>
137 struct Assignment<DstXprType, SrcXprType, Functor, Sparse2Dense, Scalar>
139 static void run(DstXprType &dst,
const SrcXprType &src,
const Functor &func)
141 eigen_assert(dst.rows() == src.rows() && dst.cols() == src.cols());
143 internal::evaluator<SrcXprType> srcEval(src);
144 internal::evaluator<DstXprType> dstEval(dst);
145 const Index outerEvaluationSize = (internal::evaluator<SrcXprType>::Flags&
RowMajorBit) ? src.rows() : src.cols();
146 for (Index j=0; j<outerEvaluationSize; ++j)
147 for (
typename internal::evaluator<SrcXprType>::InnerIterator i(srcEval,j); i; ++i)
148 func.assignCoeff(dstEval.coeffRef(i.row(),i.col()), i.value());
152 template<
typename DstXprType,
typename SrcXprType,
typename Scalar>
153 struct Assignment<DstXprType, SrcXprType,
internal::assign_op<typename DstXprType::Scalar>, Sparse2Dense, Scalar>
155 static void run(DstXprType &dst,
const SrcXprType &src,
const internal::assign_op<typename DstXprType::Scalar> &)
157 eigen_assert(dst.rows() == src.rows() && dst.cols() == src.cols());
160 internal::evaluator<SrcXprType> srcEval(src);
161 internal::evaluator<DstXprType> dstEval(dst);
162 const Index outerEvaluationSize = (internal::evaluator<SrcXprType>::Flags&
RowMajorBit) ? src.rows() : src.cols();
163 for (Index j=0; j<outerEvaluationSize; ++j)
164 for (
typename internal::evaluator<SrcXprType>::InnerIterator i(srcEval,j); i; ++i)
165 dstEval.coeffRef(i.row(),i.col()) = i.value();
171 template<
typename DstXprType,
typename DecType,
typename RhsType,
typename Scalar>
172 struct Assignment<DstXprType, Solve<DecType,RhsType>,
internal::assign_op<Scalar>, Sparse2Sparse, Scalar>
174 typedef Solve<DecType,RhsType> SrcXprType;
175 static void run(DstXprType &dst,
const SrcXprType &src,
const internal::assign_op<Scalar> &)
177 src.dec()._solve_impl(src.rhs(), dst);
181 struct Diagonal2Sparse {};
183 template<>
struct AssignmentKind<SparseShape,DiagonalShape> {
typedef Diagonal2Sparse Kind; };
185 template<
typename DstXprType,
typename SrcXprType,
typename Functor,
typename Scalar>
186 struct Assignment<DstXprType, SrcXprType, Functor, Diagonal2Sparse, Scalar>
188 typedef typename DstXprType::StorageIndex StorageIndex;
189 typedef Array<StorageIndex,Dynamic,1> ArrayXI;
190 typedef Array<Scalar,Dynamic,1> ArrayXS;
191 template<
int Options>
192 static void run(SparseMatrix<Scalar,Options,StorageIndex> &dst,
const SrcXprType &src,
const internal::assign_op<typename DstXprType::Scalar> &)
194 Index size = src.diagonal().size();
195 dst.makeCompressed();
196 dst.resizeNonZeros(size);
197 Map<ArrayXI>(dst.innerIndexPtr(), size).setLinSpaced(0,StorageIndex(size)-1);
198 Map<ArrayXI>(dst.outerIndexPtr(), size+1).setLinSpaced(0,StorageIndex(size));
199 Map<ArrayXS>(dst.valuePtr(), size) = src.diagonal();
202 template<
typename DstDerived>
203 static void run(SparseMatrixBase<DstDerived> &dst,
const SrcXprType &src,
const internal::assign_op<typename DstXprType::Scalar> &)
205 dst.diagonal() = src.diagonal();
208 static void run(DstXprType &dst,
const SrcXprType &src,
const internal::add_assign_op<typename DstXprType::Scalar> &)
209 { dst.diagonal() += src.diagonal(); }
211 static void run(DstXprType &dst,
const SrcXprType &src,
const internal::sub_assign_op<typename DstXprType::Scalar> &)
212 { dst.diagonal() -= src.diagonal(); }
218 #endif // EIGEN_SPARSEASSIGN_H
const unsigned int RowMajorBit
Definition: Constants.h:53
Definition: Eigen_Colamd.h:54