10 #ifndef EIGEN_SPARSE_BLOCK_H 11 #define EIGEN_SPARSE_BLOCK_H 16 template<
typename XprType,
int BlockRows,
int BlockCols>
25 enum { OuterSize = IsRowMajor ? BlockRows : BlockCols };
27 using Base::convert_index;
29 EIGEN_SPARSE_PUBLIC_INTERFACE(BlockType)
32 : m_matrix(xpr), m_outerStart(convert_index(i)), m_outerSize(OuterSize)
36 : m_matrix(xpr), m_outerStart(convert_index(IsRowMajor ? startRow : startCol)), m_outerSize(convert_index(IsRowMajor ? blockRows : blockCols))
39 EIGEN_STRONG_INLINE
Index rows()
const {
return IsRowMajor ? m_outerSize.value() : m_matrix.rows(); }
40 EIGEN_STRONG_INLINE
Index cols()
const {
return IsRowMajor ? m_matrix.cols() : m_outerSize.value(); }
42 Index nonZeros()
const 45 EvaluatorType matEval(m_matrix);
47 Index end = m_outerStart + m_outerSize.value();
48 for(
Index j=m_outerStart; j<end; ++j)
49 for(
typename EvaluatorType::InnerIterator it(matEval, j); it; ++it)
54 inline const Scalar coeff(
Index row,
Index col)
const 56 return m_matrix.coeff(row + (IsRowMajor ? m_outerStart : 0), col + (IsRowMajor ? 0 : m_outerStart));
59 inline const Scalar coeff(
Index index)
const 61 return m_matrix.coeff(IsRowMajor ? m_outerStart : index, IsRowMajor ? index : m_outerStart);
64 inline const XprType& nestedExpression()
const {
return m_matrix; }
65 inline XprType& nestedExpression() {
return m_matrix; }
66 Index startRow()
const {
return IsRowMajor ? m_outerStart : 0; }
67 Index startCol()
const {
return IsRowMajor ? 0 : m_outerStart; }
68 Index blockRows()
const {
return IsRowMajor ? m_outerSize.value() : m_matrix.rows(); }
69 Index blockCols()
const {
return IsRowMajor ? m_matrix.cols() : m_outerSize.value(); }
83 EIGEN_STATIC_ASSERT(
sizeof(
T)==0, THIS_SPARSE_BLOCK_SUBEXPRESSION_IS_READ_ONLY);
95 template<
typename SparseMatrixType,
int BlockRows,
int BlockCols>
102 using Base::convert_index;
105 EIGEN_SPARSE_PUBLIC_INTERFACE(BlockType)
107 typedef typename Base::IndexVector IndexVector;
108 enum { OuterSize = IsRowMajor ? BlockRows : BlockCols };
112 : m_matrix(xpr), m_outerStart(convert_index(i)), m_outerSize(OuterSize)
116 : m_matrix(xpr), m_outerStart(convert_index(IsRowMajor ? startRow : startCol)), m_outerSize(convert_index(IsRowMajor ? blockRows : blockCols))
119 template<
typename OtherDerived>
123 _NestedMatrixType& matrix = m_matrix;
129 eigen_internal_assert(tmp.outerSize()==m_outerSize.value());
132 Index nnz = tmp.nonZeros();
133 Index start = m_outerStart==0 ? 0 : m_matrix.outerIndexPtr()[m_outerStart];
134 Index end = m_matrix.outerIndexPtr()[m_outerStart+m_outerSize.value()];
135 Index block_size = end - start;
136 Index tail_size = m_matrix.outerIndexPtr()[m_matrix.outerSize()] - end;
138 Index free_size = m_matrix.isCompressed()
139 ?
Index(matrix.data().allocatedSize()) + block_size
142 Index tmp_start = tmp.outerIndexPtr()[0];
144 bool update_trailing_pointers =
false;
148 typename SparseMatrixType::Storage newdata(m_matrix.data().allocatedSize() - block_size + nnz);
150 internal::smart_copy(m_matrix.valuePtr(), m_matrix.valuePtr() + start, newdata.valuePtr());
151 internal::smart_copy(m_matrix.innerIndexPtr(), m_matrix.innerIndexPtr() + start, newdata.indexPtr());
153 internal::smart_copy(tmp.valuePtr() + tmp_start, tmp.valuePtr() + tmp_start + nnz, newdata.valuePtr() + start);
154 internal::smart_copy(tmp.innerIndexPtr() + tmp_start, tmp.innerIndexPtr() + tmp_start + nnz, newdata.indexPtr() + start);
156 internal::smart_copy(matrix.valuePtr()+end, matrix.valuePtr()+end + tail_size, newdata.valuePtr()+start+nnz);
157 internal::smart_copy(matrix.innerIndexPtr()+end, matrix.innerIndexPtr()+end + tail_size, newdata.indexPtr()+start+nnz);
159 newdata.resize(m_matrix.outerIndexPtr()[m_matrix.outerSize()] - block_size + nnz);
161 matrix.data().swap(newdata);
163 update_trailing_pointers =
true;
167 if(m_matrix.isCompressed())
170 matrix.data().resize(start + nnz + tail_size);
172 internal::smart_memmove(matrix.valuePtr()+end, matrix.valuePtr() + end+tail_size, matrix.valuePtr() + start+nnz);
173 internal::smart_memmove(matrix.innerIndexPtr()+end, matrix.innerIndexPtr() + end+tail_size, matrix.innerIndexPtr() + start+nnz);
175 update_trailing_pointers =
true;
178 internal::smart_copy(tmp.valuePtr() + tmp_start, tmp.valuePtr() + tmp_start + nnz, matrix.valuePtr() + start);
179 internal::smart_copy(tmp.innerIndexPtr() + tmp_start, tmp.innerIndexPtr() + tmp_start + nnz, matrix.innerIndexPtr() + start);
183 if(IsVectorAtCompileTime)
185 if(!m_matrix.isCompressed())
186 matrix.innerNonZeroPtr()[m_outerStart] = StorageIndex(nnz);
187 matrix.outerIndexPtr()[m_outerStart] = StorageIndex(start);
191 StorageIndex p = StorageIndex(start);
192 for(
Index k=0; k<m_outerSize.value(); ++k)
194 StorageIndex nnz_k = internal::convert_index<StorageIndex>(tmp.innerVector(k).nonZeros());
195 if(!m_matrix.isCompressed())
196 matrix.innerNonZeroPtr()[m_outerStart+k] = nnz_k;
197 matrix.outerIndexPtr()[m_outerStart+k] = p;
202 if(update_trailing_pointers)
204 StorageIndex offset = internal::convert_index<StorageIndex>(nnz - block_size);
205 for(
Index k = m_outerStart + m_outerSize.value(); k<=matrix.outerSize(); ++k)
207 matrix.outerIndexPtr()[k] += offset;
214 inline BlockType& operator=(
const BlockType& other)
216 return operator=<BlockType>(other);
219 inline const Scalar* valuePtr()
const 220 {
return m_matrix.valuePtr(); }
221 inline Scalar* valuePtr()
222 {
return m_matrix.valuePtr(); }
224 inline const StorageIndex* innerIndexPtr()
const 225 {
return m_matrix.innerIndexPtr(); }
226 inline StorageIndex* innerIndexPtr()
227 {
return m_matrix.innerIndexPtr(); }
229 inline const StorageIndex* outerIndexPtr()
const 230 {
return m_matrix.outerIndexPtr() + m_outerStart; }
231 inline StorageIndex* outerIndexPtr()
232 {
return m_matrix.outerIndexPtr() + m_outerStart; }
234 inline const StorageIndex* innerNonZeroPtr()
const 235 {
return isCompressed() ? 0 : (m_matrix.innerNonZeroPtr()+m_outerStart); }
236 inline StorageIndex* innerNonZeroPtr()
237 {
return isCompressed() ? 0 : (m_matrix.innerNonZeroPtr()+m_outerStart); }
239 bool isCompressed()
const {
return m_matrix.innerNonZeroPtr()==0; }
243 return m_matrix.coeffRef(row + (IsRowMajor ? m_outerStart : 0), col + (IsRowMajor ? 0 : m_outerStart));
246 inline const Scalar coeff(
Index row,
Index col)
const 248 return m_matrix.coeff(row + (IsRowMajor ? m_outerStart : 0), col + (IsRowMajor ? 0 : m_outerStart));
251 inline const Scalar coeff(
Index index)
const 253 return m_matrix.coeff(IsRowMajor ? m_outerStart : index, IsRowMajor ? index : m_outerStart);
256 const Scalar& lastCoeff()
const 259 eigen_assert(Base::nonZeros()>0);
260 if(m_matrix.isCompressed())
261 return m_matrix.valuePtr()[m_matrix.outerIndexPtr()[m_outerStart+1]-1];
263 return m_matrix.valuePtr()[m_matrix.outerIndexPtr()[m_outerStart]+m_matrix.innerNonZeroPtr()[m_outerStart]-1];
266 EIGEN_STRONG_INLINE
Index rows()
const {
return IsRowMajor ? m_outerSize.value() : m_matrix.rows(); }
267 EIGEN_STRONG_INLINE
Index cols()
const {
return IsRowMajor ? m_matrix.cols() : m_outerSize.value(); }
269 inline const SparseMatrixType& nestedExpression()
const {
return m_matrix; }
270 inline SparseMatrixType& nestedExpression() {
return m_matrix; }
271 Index startRow()
const {
return IsRowMajor ? m_outerStart : 0; }
272 Index startCol()
const {
return IsRowMajor ? 0 : m_outerStart; }
273 Index blockRows()
const {
return IsRowMajor ? m_outerSize.value() : m_matrix.rows(); }
274 Index blockCols()
const {
return IsRowMajor ? m_matrix.cols() : m_outerSize.value(); }
286 template<
typename _Scalar,
int _Options,
typename _StorageIndex,
int BlockRows,
int BlockCols>
291 typedef _StorageIndex StorageIndex;
299 : Base(xpr, startRow, startCol, blockRows, blockCols)
302 using Base::operator=;
305 template<
typename _Scalar,
int _Options,
typename _StorageIndex,
int BlockRows,
int BlockCols>
310 typedef _StorageIndex StorageIndex;
318 : Base(xpr, startRow, startCol, blockRows, blockCols)
321 using Base::operator=;
332 template<
typename Derived>
339 template<
typename Derived>
346 template<
typename Derived>
351 IsRowMajor ? outerStart : 0, IsRowMajor ? 0 : outerStart,
352 IsRowMajor ? outerSize : rows(), IsRowMajor ? cols() : outerSize);
359 template<
typename Derived>
364 IsRowMajor ? outerStart : 0, IsRowMajor ? 0 : outerStart,
365 IsRowMajor ? outerSize : rows(), IsRowMajor ? cols() : outerSize);
372 template<
typename XprType,
int BlockRows,
int BlockCols,
bool InnerPanel>
378 using Base::convert_index;
381 EIGEN_SPARSE_PUBLIC_INTERFACE(BlockType)
389 m_startRow( (BlockRows==1) && (BlockCols==XprType::ColsAtCompileTime) ? convert_index(i) : 0),
390 m_startCol( (BlockRows==XprType::RowsAtCompileTime) && (BlockCols==1) ? convert_index(i) : 0),
391 m_blockRows(BlockRows==1 ? 1 : xpr.rows()),
392 m_blockCols(BlockCols==1 ? 1 : xpr.cols())
398 : m_matrix(xpr), m_startRow(convert_index(startRow)), m_startCol(convert_index(startCol)), m_blockRows(convert_index(blockRows)), m_blockCols(convert_index(blockCols))
401 inline Index rows()
const {
return m_blockRows.value(); }
402 inline Index cols()
const {
return m_blockCols.value(); }
406 return m_matrix.coeffRef(row + m_startRow.value(), col + m_startCol.value());
409 inline const Scalar coeff(
Index row,
Index col)
const 411 return m_matrix.coeff(row + m_startRow.value(), col + m_startCol.value());
414 inline Scalar& coeffRef(
Index index)
416 return m_matrix.coeffRef(m_startRow.value() + (RowsAtCompileTime == 1 ? 0 : index),
417 m_startCol.value() + (RowsAtCompileTime == 1 ? index : 0));
420 inline const Scalar coeff(
Index index)
const 422 return m_matrix.coeff(m_startRow.value() + (RowsAtCompileTime == 1 ? 0 : index),
423 m_startCol.value() + (RowsAtCompileTime == 1 ? index : 0));
426 inline const XprType& nestedExpression()
const {
return m_matrix; }
427 inline XprType& nestedExpression() {
return m_matrix; }
428 Index startRow()
const {
return m_startRow.value(); }
429 Index startCol()
const {
return m_startCol.value(); }
430 Index blockRows()
const {
return m_blockRows.value(); }
431 Index blockCols()
const {
return m_blockCols.value(); }
435 friend struct internal::unary_evaluator<Block<XprType,BlockRows,BlockCols,InnerPanel>, internal::IteratorBased, Scalar >;
451 EIGEN_STATIC_ASSERT(
sizeof(
T)==0, THIS_SPARSE_BLOCK_SUBEXPRESSION_IS_READ_ONLY);
459 template<
typename ArgType,
int BlockRows,
int BlockCols,
bool InnerPanel>
461 :
public evaluator_base<Block<ArgType,BlockRows,BlockCols,InnerPanel> >
463 class InnerVectorInnerIterator;
464 class OuterVectorInnerIterator;
467 typedef typename XprType::StorageIndex StorageIndex;
468 typedef typename XprType::Scalar Scalar;
471 IsRowMajor = XprType::IsRowMajor,
473 OuterVector = (BlockCols==1 && ArgType::IsRowMajor)
476 (BlockRows==1 && !ArgType::IsRowMajor),
479 Flags = XprType::Flags
485 : m_argImpl(op.nestedExpression()), m_block(op)
488 inline Index nonZerosEstimate()
const {
489 Index nnz = m_block.nonZeros();
491 return m_argImpl.nonZerosEstimate() * m_block.size() / m_block.nestedExpression().size();
499 const XprType &m_block;
502 template<
typename ArgType,
int BlockRows,
int BlockCols,
bool InnerPanel>
504 :
public EvalIterator
506 enum { IsRowMajor = unary_evaluator::IsRowMajor };
512 : EvalIterator(aEval.m_argImpl, outer + (IsRowMajor ? aEval.m_block.startRow() : aEval.m_block.startCol())),
513 m_block(aEval.m_block),
514 m_end(IsRowMajor ? aEval.m_block.startCol()+aEval.m_block.blockCols() : aEval.m_block.startRow()+aEval.m_block.blockRows())
516 while( (EvalIterator::operator
bool()) && (EvalIterator::index() < (IsRowMajor ? m_block.startCol() : m_block.startRow())) )
517 EvalIterator::operator++();
520 inline StorageIndex index()
const {
return EvalIterator::index() - convert_index<StorageIndex>(IsRowMajor ? m_block.startCol() : m_block.startRow()); }
521 inline Index outer()
const {
return EvalIterator::outer() - (IsRowMajor ? m_block.startRow() : m_block.startCol()); }
522 inline Index row()
const {
return EvalIterator::row() - m_block.startRow(); }
523 inline Index col()
const {
return EvalIterator::col() - m_block.startCol(); }
525 inline operator bool()
const {
return EvalIterator::operator bool() && EvalIterator::index() < m_end; }
528 template<
typename ArgType,
int BlockRows,
int BlockCols,
bool InnerPanel>
531 enum { IsRowMajor = unary_evaluator::IsRowMajor };
541 m_outerPos( (IsRowMajor ? aEval.m_block.startCol() : aEval.m_block.startRow()) - 1),
542 m_innerIndex(IsRowMajor ? aEval.m_block.startRow() : aEval.m_block.startCol()),
544 m_end(IsRowMajor ? aEval.m_block.startCol()+aEval.m_block.blockCols() : aEval.m_block.startRow()+aEval.m_block.blockRows())
546 EIGEN_UNUSED_VARIABLE(outer);
547 eigen_assert(outer==0);
552 inline StorageIndex index()
const {
return convert_index<StorageIndex>(m_outerPos - (IsRowMajor ? m_eval.m_block.startCol() : m_eval.m_block.startRow())); }
553 inline Index outer()
const {
return 0; }
554 inline Index row()
const {
return IsRowMajor ? 0 : index(); }
555 inline Index col()
const {
return IsRowMajor ? index() : 0; }
557 inline Scalar value()
const {
return m_value; }
559 inline OuterVectorInnerIterator& operator++()
562 while(++m_outerPos<m_end)
564 EvalIterator it(m_eval.m_argImpl, m_outerPos);
566 while(it && it.index() < m_innerIndex) ++it;
567 if(it && it.index()==m_innerIndex)
569 m_value = it.value();
576 inline operator bool()
const {
return m_outerPos < m_end; }
579 template<
typename _Scalar,
int _Options,
typename _StorageIndex,
int BlockRows,
int BlockCols>
581 :
evaluator<SparseCompressedBase<Block<SparseMatrix<_Scalar, _Options, _StorageIndex>,BlockRows,BlockCols,true> > >
588 template<
typename _Scalar,
int _Options,
typename _StorageIndex,
int BlockRows,
int BlockCols>
590 :
evaluator<SparseCompressedBase<Block<const SparseMatrix<_Scalar, _Options, _StorageIndex>,BlockRows,BlockCols,true> > >
602 #endif // EIGEN_SPARSE_BLOCK_H A versatible sparse matrix representation.
Definition: SparseMatrix.h:92
Definition: CoreEvaluators.h:65
Namespace containing all symbols from the Eigen library.
Definition: bench_norm.cpp:85
Eigen::Index Index
The interface type of indices.
Definition: EigenBase.h:37
Definition: XprHelper.h:89
BlockImpl(XprType &xpr, Index i)
Column or Row constructor.
Definition: SparseBlock.h:387
Definition: Constants.h:529
BlockImpl(XprType &xpr, Index startRow, Index startCol, Index blockRows, Index blockCols)
Dynamic-size constructor.
Definition: SparseBlock.h:397
Definition: CoreEvaluators.h:109
Base class of any sparse matrices or sparse expressions.
Definition: ForwardDeclarations.h:281
EIGEN_DEFAULT_DENSE_INDEX_TYPE Index
The Index type as used for the API.
Definition: Meta.h:33
InnerVectorReturnType innerVector(Index outer)
Definition: SparseBlock.h:333
The type used to identify a general sparse storage.
Definition: Constants.h:494
A matrix or vector expression mapping an existing expression.
Definition: Ref.h:190
Definition: BandTriangularSolver.h:13
InnerVectorsReturnType innerVectors(Index outerStart, Index outerSize)
Definition: SparseBlock.h:348
Expression of a fixed-size or dynamic-size block.
Definition: Block.h:103
const int Dynamic
This value means that a positive quantity (e.g., a size) is not known at compile-time, and that instead the value is stored in some runtime variable.
Definition: Constants.h:21
Generic expression where a coefficient-wise unary operator is applied to an expression.
Definition: CwiseUnaryOp.h:55
Common base class for sparse [compressed]-{row|column}-storage format.
Definition: SparseCompressedBase.h:15
Definition: SparseBlock.h:96
Definition: ForwardDeclarations.h:17