Fixed output for flexible matrix

Former-commit-id: e2c9913b6e

src/storage/FlexibleSparseMatrix.cpp

@@ -63,6 +63,11 @@ namespace storm {
             return getRow(rowGroupIndices[rowGroup] + offset);
         }
         
+        template<typename ValueType>
+        std::vector<typename FlexibleSparseMatrix<ValueType>::index_type> const& FlexibleSparseMatrix<ValueType>::getRowGroupIndices() const {
+            return rowGroupIndices;
+        }
+        
         template<typename ValueType>
         typename FlexibleSparseMatrix<ValueType>::index_type FlexibleSparseMatrix<ValueType>::getRowCount() const {
             return this->data.size();
@@ -165,48 +170,63 @@ namespace storm {
             return false;
         }
         
+        template<typename ValueType>
+        std::ostream& FlexibleSparseMatrix<ValueType>::printRow(std::ostream& out, index_type const& rowIndex) const {
+            index_type columnIndex = 0;
+            row_type row = this->getRow(rowIndex);
+            for (index_type column = 0; column < this->getColumnCount(); ++column) {
+                if (columnIndex < row.size() && row[columnIndex].getColumn() == column) {
+                    // Insert entry
+                    out << row[columnIndex].getValue() << "\t";
+                    ++columnIndex;
+                } else {
+                    // Insert zero
+                    out << "0\t";
+                }
+            }
+            return out;
+        }
+        
+
         template<typename ValueType>
         std::ostream& operator<<(std::ostream& out, FlexibleSparseMatrix<ValueType> const& matrix) {
+            typedef typename FlexibleSparseMatrix<ValueType>::index_type FlexibleIndex;
+            
             // Print column numbers in header.
             out << "\t\t";
-            for (typename FlexibleSparseMatrix<ValueType>::index_type i = 0; i < matrix.getColumnCount(); ++i) {
+            for (FlexibleIndex i = 0; i < matrix.getColumnCount(); ++i) {
                 out << i << "\t";
             }
             out << std::endl;
             
-            typename FlexibleSparseMatrix<ValueType>::index_type endIndex = matrix.hasNontrivialRowGrouping() ? matrix.getRowGroupCount() : matrix.getRowCount();
-            // Iterate over all rows.
-            for (typename SparseMatrix<ValueType>::index_type i = 0; i < endIndex; ++i) {
             if (matrix.hasNontrivialRowGrouping()) {
-                    out << "\t---- group " << i << "/" << (matrix.getRowGroupCount() - 1) << " ---- " << std::endl;
-                }
-                typename FlexibleSparseMatrix<ValueType>::index_type startRow = matrix.hasNontrivialRowGrouping() ? matrix.rowGroupIndices[i] : i;
-                typename FlexibleSparseMatrix<ValueType>::index_type endRow = matrix.hasNontrivialRowGrouping() ? matrix.rowGroupIndices[i + 1] : i+1;
-                for (typename FlexibleSparseMatrix<ValueType>::index_type rowIndex = startRow; rowIndex < endRow; ++rowIndex) {
+                // Iterate over all row groups
+                FlexibleIndex rowGroupCount = matrix.getRowGroupCount();
+                for (FlexibleIndex rowGroup = 0; rowGroup < rowGroupCount; ++rowGroup) {
+                    out << "\t---- group " << rowGroup << "/" << (rowGroupCount - 1) << " ---- " << std::endl;
+                    FlexibleIndex endRow = rowGroup+1 < rowGroupCount ? matrix.rowGroupIndices[rowGroup + 1] : matrix.getRowCount();
+                    // Iterate over all rows.
+                    for (FlexibleIndex row = matrix.rowGroupIndices[rowGroup]; row < endRow; ++row) {
                         // Print the actual row.
-                    out << i << "\t(\t";
-                    typename FlexibleSparseMatrix<ValueType>::row_type row = matrix.getRow(rowIndex);
-                    typename FlexibleSparseMatrix<ValueType>::index_type columnIndex = 0;
-                    for (auto const& entry : row) {
-                        //Insert zero between entries.
-                        while (columnIndex < entry.getColumn()) {
-                            out << "0\t";
-                            ++columnIndex;
+                        out << rowGroup << "\t(\t";
+                        matrix.printRow(out, row);
+                        out << "\t)\t" << rowGroup << std::endl;
                     }
-                        ++columnIndex;
-                        out << entry.getValue() << "\t";
                 }
-                    //Insert remaining zeros.
-                    while (columnIndex++ <= matrix.getColumnCount()) {
-                        out << "0\t";
-                    }
-                    out << "\t)\t" << i << std::endl;
+
+            } else {
+                // Iterate over all rows
+                for (FlexibleIndex row = 0; row < matrix.getRowCount(); ++row) {
+                    // Print the actual row.
+                    out << row << "\t(\t";
+                    matrix.printRow(out, row);
+                    out << "\t)\t" << row << std::endl;
                 }
             }
             
             // Print column numbers in footer.
             out << "\t\t";
-            for (typename FlexibleSparseMatrix<ValueType>::index_type i = 0; i < matrix.getColumnCount(); ++i) {
+            for (FlexibleIndex i = 0; i < matrix.getColumnCount(); ++i) {
                 out << i << "\t";
             }
             out << std::endl;
@@ -215,8 +235,11 @@ namespace storm {
 
         // Explicitly instantiate the matrix.
         template class FlexibleSparseMatrix<double>;
+        template std::ostream& operator<<(std::ostream& out, FlexibleSparseMatrix<double> const& matrix);
+        
 #ifdef STORM_HAVE_CARL
         template class FlexibleSparseMatrix<storm::RationalFunction>;
+        template std::ostream& operator<<(std::ostream& out, FlexibleSparseMatrix<storm::RationalFunction> const& matrix);
 #endif
 
     } // namespace storage

src/storage/FlexibleSparseMatrix.h

@@ -86,6 +86,13 @@ namespace storm {
              */
             row_type const& getRow(index_type rowGroup, index_type entryInGroup) const;
             
+            /*!
+             * Returns the grouping of rows of this matrix.
+             *
+             * @return The grouping of rows of this matrix.
+             */
+            std::vector<index_type> const& getRowGroupIndices() const;
+
             /*!
              * Returns the number of rows of the matrix.
              *
@@ -163,6 +170,14 @@ namespace storm {
              */
             bool rowHasDiagonalElement(storm::storage::sparse::state_type state);
 
+            /*!
+             * Print row.
+             * @param out Output stream.
+             * @param rowIndex Index of row to print.
+             * @return Output with printed row.
+             */
+            std::ostream& printRow(std::ostream& out, index_type const& rowIndex) const;
+            
             template<typename TPrime>
             friend std::ostream& operator<<(std::ostream& out, FlexibleSparseMatrix<TPrime> const& matrix);