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,typename OtherDerived::Scalar> >
38 ::run(derived(), other.derived(), internal::assign_op<Scalar,typename OtherDerived::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; };
67 template<>
struct AssignmentKind<DenseShape, SparseTriangularShape> {
typedef Sparse2Dense Kind; };
70 template<
typename DstXprType,
typename SrcXprType>
71 void assign_sparse_to_sparse(DstXprType &dst,
const SrcXprType &src)
73 typedef typename DstXprType::Scalar Scalar;
74 typedef internal::evaluator<DstXprType> DstEvaluatorType;
75 typedef internal::evaluator<SrcXprType> SrcEvaluatorType;
77 SrcEvaluatorType srcEvaluator(src);
80 const Index outerEvaluationSize = (SrcEvaluatorType::Flags&
RowMajorBit) ? src.rows() : src.cols();
81 if ((!transpose) && src.isRValue())
84 dst.resize(src.rows(), src.cols());
86 dst.reserve((std::min)(src.rows()*src.cols(), (std::max)(src.rows(),src.cols())*2));
87 for (
Index j=0; j<outerEvaluationSize; ++j)
90 for (
typename SrcEvaluatorType::InnerIterator it(srcEvaluator, j); it; ++it)
92 Scalar v = it.value();
93 dst.insertBackByOuterInner(j,it.index()) = v;
101 eigen_assert(( ((internal::traits<DstXprType>::SupportedAccessPatterns & OuterRandomAccessPattern)==OuterRandomAccessPattern) ||
102 (!((DstEvaluatorType::Flags & RowMajorBit) != (SrcEvaluatorType::Flags & RowMajorBit)))) &&
103 "the transpose operation is supposed to be handled in SparseMatrix::operator=");
105 enum { Flip = (DstEvaluatorType::Flags &
RowMajorBit) != (SrcEvaluatorType::Flags & RowMajorBit) };
108 DstXprType temp(src.rows(), src.cols());
110 temp.reserve((std::min)(src.rows()*src.cols(), (std::max)(src.rows(),src.cols())*2));
111 for (
Index j=0; j<outerEvaluationSize; ++j)
114 for (
typename SrcEvaluatorType::InnerIterator it(srcEvaluator, j); it; ++it)
116 Scalar v = it.value();
117 temp.insertBackByOuterInner(Flip?it.index():j,Flip?j:it.index()) = v;
122 dst = temp.markAsRValue();
127 template<
typename DstXprType,
typename SrcXprType,
typename Functor>
128 struct Assignment<DstXprType, SrcXprType, Functor, Sparse2Sparse>
130 static void run(DstXprType &dst,
const SrcXprType &src,
const internal::assign_op<typename DstXprType::Scalar,typename SrcXprType::Scalar> &)
132 assign_sparse_to_sparse(dst.derived(), src.derived());
137 template<
typename DstXprType,
typename SrcXprType,
typename Functor,
typename Weak>
138 struct Assignment<DstXprType, SrcXprType, Functor, Sparse2Dense, Weak>
140 static void run(DstXprType &dst,
const SrcXprType &src,
const Functor &func)
142 if(internal::is_same<Functor,internal::assign_op<typename DstXprType::Scalar,typename SrcXprType::Scalar> >::value)
145 internal::evaluator<SrcXprType> srcEval(src);
146 resize_if_allowed(dst, src, func);
147 internal::evaluator<DstXprType> dstEval(dst);
149 const Index outerEvaluationSize = (internal::evaluator<SrcXprType>::Flags&
RowMajorBit) ? src.rows() : src.cols();
150 for (
Index j=0; j<outerEvaluationSize; ++j)
151 for (
typename internal::evaluator<SrcXprType>::InnerIterator i(srcEval,j); i; ++i)
152 func.assignCoeff(dstEval.coeffRef(i.row(),i.col()), i.value());
157 template<
typename DstXprType,
typename Func1,
typename Func2>
158 struct assignment_from_dense_op_sparse
160 template<
typename SrcXprType,
typename InitialFunc>
161 static EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
162 void run(DstXprType &dst,
const SrcXprType &src,
const InitialFunc& )
164 #ifdef EIGEN_SPARSE_ASSIGNMENT_FROM_DENSE_OP_SPARSE_PLUGIN 165 EIGEN_SPARSE_ASSIGNMENT_FROM_DENSE_OP_SPARSE_PLUGIN
168 call_assignment_no_alias(dst, src.lhs(), Func1());
169 call_assignment_no_alias(dst, src.rhs(), Func2());
173 template<
typename Lhs,
typename Rhs,
typename Scalar>
174 static EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
175 typename internal::enable_if<internal::is_same<typename internal::evaluator_traits<Rhs>::Shape,DenseShape>::value>::type
176 run(DstXprType &dst,
const CwiseBinaryOp<internal::scalar_sum_op<Scalar,Scalar>,
const Lhs,
const Rhs> &src,
177 const internal::assign_op<typename DstXprType::Scalar,Scalar>& )
179 #ifdef EIGEN_SPARSE_ASSIGNMENT_FROM_SPARSE_ADD_DENSE_PLUGIN 180 EIGEN_SPARSE_ASSIGNMENT_FROM_SPARSE_ADD_DENSE_PLUGIN
184 call_assignment_no_alias(dst, src.rhs(), Func1());
185 call_assignment_no_alias(dst, src.lhs(), Func2());
189 template<
typename Lhs,
typename Rhs,
typename Scalar>
190 static EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
191 typename internal::enable_if<internal::is_same<typename internal::evaluator_traits<Rhs>::Shape,DenseShape>::value>::type
192 run(DstXprType &dst,
const CwiseBinaryOp<internal::scalar_difference_op<Scalar,Scalar>,
const Lhs,
const Rhs> &src,
193 const internal::assign_op<typename DstXprType::Scalar,Scalar>& )
195 #ifdef EIGEN_SPARSE_ASSIGNMENT_FROM_SPARSE_SUB_DENSE_PLUGIN 196 EIGEN_SPARSE_ASSIGNMENT_FROM_SPARSE_SUB_DENSE_PLUGIN
200 call_assignment_no_alias(dst, -src.rhs(), Func1());
201 call_assignment_no_alias(dst, src.lhs(), add_assign_op<typename DstXprType::Scalar,typename Lhs::Scalar>());
205 #define EIGEN_CATCH_ASSIGN_DENSE_OP_SPARSE(ASSIGN_OP,BINOP,ASSIGN_OP2) \ 206 template< typename DstXprType, typename Lhs, typename Rhs, typename Scalar> \ 207 struct Assignment<DstXprType, CwiseBinaryOp<internal::BINOP<Scalar,Scalar>, const Lhs, const Rhs>, internal::ASSIGN_OP<typename DstXprType::Scalar,Scalar>, \ 209 typename internal::enable_if< internal::is_same<typename internal::evaluator_traits<Lhs>::Shape,DenseShape>::value \ 210 || internal::is_same<typename internal::evaluator_traits<Rhs>::Shape,DenseShape>::value>::type> \ 211 : assignment_from_dense_op_sparse<DstXprType, internal::ASSIGN_OP<typename DstXprType::Scalar,typename Lhs::Scalar>, internal::ASSIGN_OP2<typename DstXprType::Scalar,typename Rhs::Scalar> > \ 214 EIGEN_CATCH_ASSIGN_DENSE_OP_SPARSE(assign_op, scalar_sum_op,add_assign_op);
215 EIGEN_CATCH_ASSIGN_DENSE_OP_SPARSE(add_assign_op,scalar_sum_op,add_assign_op);
216 EIGEN_CATCH_ASSIGN_DENSE_OP_SPARSE(sub_assign_op,scalar_sum_op,sub_assign_op);
218 EIGEN_CATCH_ASSIGN_DENSE_OP_SPARSE(assign_op, scalar_difference_op,sub_assign_op);
219 EIGEN_CATCH_ASSIGN_DENSE_OP_SPARSE(add_assign_op,scalar_difference_op,sub_assign_op);
220 EIGEN_CATCH_ASSIGN_DENSE_OP_SPARSE(sub_assign_op,scalar_difference_op,add_assign_op);
225 template<
typename DstXprType,
typename DecType,
typename RhsType,
typename Scalar>
226 struct Assignment<DstXprType, Solve<DecType,RhsType>,
internal::assign_op<Scalar,Scalar>, Sparse2Sparse>
228 typedef Solve<DecType,RhsType> SrcXprType;
229 static void run(DstXprType &dst,
const SrcXprType &src,
const internal::assign_op<Scalar,Scalar> &)
231 Index dstRows = src.rows();
232 Index dstCols = src.cols();
233 if((dst.rows()!=dstRows) || (dst.cols()!=dstCols))
234 dst.resize(dstRows, dstCols);
236 src.dec()._solve_impl(src.rhs(), dst);
240 struct Diagonal2Sparse {};
242 template<>
struct AssignmentKind<SparseShape,DiagonalShape> {
typedef Diagonal2Sparse Kind; };
244 template<
typename DstXprType,
typename SrcXprType,
typename Functor>
245 struct Assignment<DstXprType, SrcXprType, Functor, Diagonal2Sparse>
247 typedef typename DstXprType::StorageIndex StorageIndex;
248 typedef typename DstXprType::Scalar Scalar;
250 template<
int Options,
typename AssignFunc>
251 static void run(SparseMatrix<Scalar,Options,StorageIndex> &dst,
const SrcXprType &src,
const AssignFunc &func)
252 { dst.assignDiagonal(src.diagonal(), func); }
254 template<
typename DstDerived>
255 static void run(SparseMatrixBase<DstDerived> &dst,
const SrcXprType &src,
const internal::assign_op<typename DstXprType::Scalar,typename SrcXprType::Scalar> &)
256 { dst.derived().diagonal() = src.diagonal(); }
258 template<
typename DstDerived>
259 static void run(SparseMatrixBase<DstDerived> &dst,
const SrcXprType &src,
const internal::add_assign_op<typename DstXprType::Scalar,typename SrcXprType::Scalar> &)
260 { dst.derived().diagonal() += src.diagonal(); }
262 template<
typename DstDerived>
263 static void run(SparseMatrixBase<DstDerived> &dst,
const SrcXprType &src,
const internal::sub_assign_op<typename DstXprType::Scalar,typename SrcXprType::Scalar> &)
264 { dst.derived().diagonal() -= src.diagonal(); }
270 #endif // EIGEN_SPARSEASSIGN_H Namespace containing all symbols from the Eigen library.
Definition: Core:141
const unsigned int RowMajorBit
Definition: Constants.h:66
EIGEN_DEFAULT_DENSE_INDEX_TYPE Index
The Index type as used for the API.
Definition: Meta.h:74
Definition: Eigen_Colamd.h:50