From dad0a5c0ec5006236cb6e7ea9beb3cce35a3f4cf Mon Sep 17 00:00:00 2001
From: Oliver Sander <sander@dune-project.org>
Date: Thu, 6 Jan 2005 18:11:07 +0000
Subject: [PATCH] This is a helper class which stores a set of matrix entries,
i.e. a set of (i,j) integer pairs. Pairs can be added to the matrix in any
order. A MatrixIndexSet can be used to initialize a BCRSMatrix via the
exportIdx method.
MatrixIndexSets are meant to help with the sometimes unwieldy
BCRSMatrix class. If that class' handling gets easier, (or
if somebody explain to me how BCRSMatrix are really supposed
to be used), MatrixIndexSets may very well disappear again.
[[Imported from SVN: r134]]
---
istl/matrixindexset.hh | 132 +++++++++++++++++++++++++++++++++++++++++
1 file changed, 132 insertions(+)
create mode 100644 istl/matrixindexset.hh
diff --git a/istl/matrixindexset.hh b/istl/matrixindexset.hh
new file mode 100644
index 000000000..5fa813a56
--- /dev/null
+++ b/istl/matrixindexset.hh
@@ -0,0 +1,132 @@
+// -*- tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*-
+// vi: set et ts=4 sw=2 sts=2:
+#ifndef DUNE_MATRIX_INDEX_SET_HH
+#define DUNE_MATRIX_INDEX_SET_HH
+
+namespace Dune {
+
+
+ /** \brief Stores the nonzero entries in a sparse matrix */
+ class MatrixIndexSet
+ {
+
+ public:
+
+ /** \brief Constructor setting the matrix size */
+ MatrixIndexSet(int rows, int cols) : rows_(rows), cols_(cols) {
+ indices.resize(rows_);
+ }
+
+ /** \brief Add an index to the index set */
+ void add(int i, int j) {
+
+ if (!containsSorted(indices[i], j))
+ insertSorted(indices[i], j);
+
+ }
+
+ /** \brief Return the number of entries */
+ int size() const {
+ int entries = 0;
+ for (int i=0; i<rows_; i++)
+ entries += indices[i].size();
+
+ return entries;
+ }
+
+ /** \brief Return the number of rows */
+ int rows() const {return rows_;}
+
+
+ /** \brief Return the number of entries in a given row */
+ int rowsize(int row) const {return indices[row].size();}
+
+ /** \brief Import all nonzero entries of a sparse matrix into the index set
+ * \tparam MatrixType Needs to be BCRSMatrix<...>
+ */
+ template <class MatrixType>
+ int import(const MatrixType& m, int rowOffset=0, int colOffset=0) {
+
+ typedef typename MatrixType::row_type RowType;
+ typedef typename RowType::ConstIterator ColumnIterator;
+
+ for (int rowIdx=0; rowIdx<m.N(); rowIdx++) {
+
+ const RowType& row = m[rowIdx];
+
+ ColumnIterator cIt = row.begin();
+ ColumnIterator cEndIt = row.end();
+
+ for(; cIt!=cEndIt; ++cIt)
+ add(rowIdx+rowOffset, cIt.index()+colOffset);
+
+ }
+
+ }
+
+ /** \brief Initializes a BCRSMatrix with the indices contained
+ in this MatrixIndexSet
+ \tparam MatrixType Needs to be BCRSMatrix<...>
+ */
+ template <class MatrixType>
+ void exportIdx(MatrixType& m) const {
+
+ MatrixType matrix(rows_, cols_, MatrixType::random);
+
+ for (int i=0; i<rows_; i++) {
+ matrix.setrowsize(i, indices[i].size());
+ }
+
+ matrix.endrowsizes();
+
+ for (int i=0; i<rows_; i++) {
+ for (int j=0; j<indices[i].size(); j++) {
+ //std::cout << "setting index " << i <<", " << rowSizes[i][j] << std::endl;
+ matrix.addindex(i, indices[i][j]);
+ }
+ }
+
+ matrix.endindices();
+
+ m = matrix;
+ }
+
+ private:
+
+ /** \todo Doesn't use the fact that the array is sorted! */
+ bool containsSorted(const std::vector<int>& nb, int idx) {
+
+ for (int i=0; i<nb.size(); i++)
+ if (nb[i]==idx)
+ return true;
+
+ return false;
+ }
+
+ void insertSorted(std::vector<int>& nb, int idx) {
+
+ int i;
+ // Find correct slot for insertion
+ for (i=0; i<nb.size(); i++)
+ if (nb[i] >= idx)
+ break;
+
+ // Insert
+ nb.push_back(0);
+ for (int j=nb.size()-1; j>i; j--)
+ nb[j] = nb[j-1];
+
+ nb[i] = idx;
+
+ }
+
+ Dune::Array<std::vector<int> > indices;
+
+ int rows_, cols_;
+
+ };
+
+
+} // end namespace Dune
+
+#endif
--
GitLab