io.hh

Go to the documentation of this file.

// -*- tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*-
// vi: set et ts=4 sw=2 sts=2:
#ifndef DUNE_ISTL_IO_HH
#define DUNE_ISTL_IO_HH
 
#include <cmath>
#include <complex>
#include <limits>
#include <ios>
#include <iomanip>
#include <fstream>
#include <string>
 
#include "matrixutils.hh"
#include "istlexception.hh"
#include <dune/common/fvector.hh>
#include <dune/common/fmatrix.hh>
#include <dune/common/dynmatrix.hh>
#include <dune/common/diagonalmatrix.hh>
#include <dune/common/unused.hh>
 
#include <dune/istl/matrix.hh>
#include <dune/istl/scaledidmatrix.hh>
#include "bcrsmatrix.hh"
 
 
namespace Dune {
 
  //
  // pretty printing of vectors
  //
 
  template<class V>
  void recursive_printvector (std::ostream& s, const V& v, std::string rowtext,
                              int& counter, int columns, int width,
                              int precision)
  {
    for (typename V::ConstIterator i=v.begin(); i!=v.end(); ++i)
      recursive_printvector(s,*i,rowtext,counter,columns,width,precision);
  }
 
  template<class K, int n>
  void recursive_printvector (std::ostream& s, const FieldVector<K,n>& v,
                              std::string rowtext, int& counter, int columns,
                              int width, int precision)
  {
    DUNE_UNUSED_PARAMETER(precision);
    // we now can print n numbers
    for (int i=0; i<n; i++)
    {
      if (counter%columns==0)
      {
        s << rowtext; // start a new row
        s << " ";     // space in front of each entry
        s.width(4);   // set width for counter
        s << counter; // number of first entry in a line
      }
      s << " ";         // space in front of each entry
      s.width(width);   // set width for each entry anew
      s << v[i];        // yeah, the number !
      counter++;        // increment the counter
      if (counter%columns==0)
        s << std::endl; // start a new line
    }
  }
 
 
  template<class V>
  void printvector (std::ostream& s, const V& v, std::string title,
                    std::string rowtext, int columns=1, int width=10,
                    int precision=2)
  {
    // count the numbers printed to make columns
    int counter=0;
 
    // remember old flags
    std::ios_base::fmtflags oldflags = s.flags();
 
    // set the output format
    s.setf(std::ios_base::scientific, std::ios_base::floatfield);
    int oldprec = s.precision();
    s.precision(precision);
 
    // print title
    s << title << " [blocks=" << v.N() << ",dimension=" << v.dim() << "]"
      << std::endl;
 
    // print data from all blocks
    recursive_printvector(s,v,rowtext,counter,columns,width,precision);
 
    // check if new line is required
    if (counter%columns!=0)
      s << std::endl;
 
    // reset the output format
    s.flags(oldflags);
    s.precision(oldprec);
  }
 
 
  //
  // pretty printing of matrices
  //
 
  inline void fill_row (std::ostream& s, int m, int width, int precision)
  {
    DUNE_UNUSED_PARAMETER(precision);
    for (int j=0; j<m; j++)
    {
      s << " ";         // space in front of each entry
      s.width(width);   // set width for each entry anew
      s << ".";         // yeah, the number !
    }
  }
 
  template<class M>
  void print_row (std::ostream& s, const M& A, typename M::size_type I,
                  typename M::size_type J, typename M::size_type therow,
                  int width, int precision)
  {
    typename M::size_type i0=I;
    for (typename M::size_type i=0; i<A.N(); i++)
    {
      if (therow>=i0 && therow<i0+MatrixDimension<M>::rowdim(A,i))
      {
        // the row is in this block row !
        typename M::size_type j0=J;
        for (typename M::size_type j=0; j<A.M(); j++)
        {
          // find this block
          typename M::ConstColIterator it = A[i].find(j);
 
          // print row or filler
          if (it!=A[i].end())
            print_row(s,*it,i0,j0,therow,width,precision);
          else
            fill_row(s,MatrixDimension<M>::coldim(A,j),width,precision);
 
          // advance columns
          j0 += MatrixDimension<M>::coldim(A,j);
        }
      }
      // advance rows
      i0 += MatrixDimension<M>::rowdim(A,i);
    }
  }
 
  template<class K, int n, int m>
  void print_row (std::ostream& s, const FieldMatrix<K,n,m>& A,
                  typename FieldMatrix<K,n,m>::size_type I,
                  typename FieldMatrix<K,n,m>::size_type J,
                  typename FieldMatrix<K,n,m>::size_type therow, int width,
                  int precision)
  {
    DUNE_UNUSED_PARAMETER(J);
    DUNE_UNUSED_PARAMETER(precision);
 
    typedef typename FieldMatrix<K,n,m>::size_type size_type;
 
    for (size_type i=0; i<n; i++)
      if (I+i==therow)
        for (int j=0; j<m; j++)
        {
          s << " ";         // space in front of each entry
          s.width(width);   // set width for each entry anew
          s << A[i][j];     // yeah, the number !
        }
  }
 
  template<class K>
  void print_row (std::ostream& s, const FieldMatrix<K,1,1>& A,
                  typename FieldMatrix<K,1,1>::size_type I,
                  typename FieldMatrix<K,1,1>::size_type J,
                  typename FieldMatrix<K,1,1>::size_type therow,
                  int width, int precision)
  {
    DUNE_UNUSED_PARAMETER(J);
    DUNE_UNUSED_PARAMETER(precision);
 
    if (I==therow)
    {
      s << " ";         // space in front of each entry
      s.width(width);   // set width for each entry anew
      s << static_cast<K>(A);         // yeah, the number !
    }
  }
 
  template<class M>
  void printmatrix (std::ostream& s, const M& A, std::string title,
                    std::string rowtext, int width=10, int precision=2)
  {
 
    // remember old flags
    std::ios_base::fmtflags oldflags = s.flags();
 
    // set the output format
    s.setf(std::ios_base::scientific, std::ios_base::floatfield);
    int oldprec = s.precision();
    s.precision(precision);
 
    // print title
    s << title
      << " [n=" << A.N()
      << ",m=" << A.M()
      << ",rowdim=" << MatrixDimension<M>::rowdim(A)
      << ",coldim=" << MatrixDimension<M>::coldim(A)
      << "]" << std::endl;
 
    // print all rows
    for (typename M::size_type i=0; i<MatrixDimension<M>::rowdim(A); i++)
    {
      s << rowtext;  // start a new row
      s << " ";      // space in front of each entry
      s.width(4);    // set width for counter
      s << i;        // number of first entry in a line
      print_row(s,A,0,0,i,width,precision); // generic print
      s << std::endl; // start a new line
    }
 
    // reset the output format
    s.flags(oldflags);
    s.precision(oldprec);
  }
 
  template<class B, int n, int m, class A>
  void printSparseMatrix(std::ostream& s,
                         const BCRSMatrix<FieldMatrix<B,n,m>,A>& mat,
                         std::string title, std::string rowtext,
                         int width=3, int precision=2)
  {
    typedef BCRSMatrix<FieldMatrix<B,n,m>,A> Matrix;
    // remember old flags
    std::ios_base::fmtflags oldflags = s.flags();
    // set the output format
    s.setf(std::ios_base::scientific, std::ios_base::floatfield);
    int oldprec = s.precision();
    s.precision(precision);
    // print title
    s << title
      << " [n=" << mat.N()
      << ",m=" << mat.M()
      << ",rowdim=" << MatrixDimension<Matrix>::rowdim(mat)
      << ",coldim=" << MatrixDimension<Matrix>::coldim(mat)
      << "]" << std::endl;
 
    typedef typename Matrix::ConstRowIterator Row;
 
    for(Row row=mat.begin(); row != mat.end(); ++row) {
      int skipcols=0;
      bool reachedEnd=false;
 
      while(!reachedEnd) {
        for(int innerrow=0; innerrow<n; ++innerrow) {
          int count=0;
          typedef typename Matrix::ConstColIterator Col;
          Col col=row->begin();
          for(; col != row->end(); ++col,++count) {
            if(count<skipcols)
              continue;
            if(count>=skipcols+width)
              break;
            if(innerrow==0) {
              if(count==skipcols) {
                s << rowtext;           // start a new row
                s << " ";               // space in front of each entry
                s.width(4);             // set width for counter
                s << row.index()<<": "; // number of first entry in a line
              }
              s.width(4);
              s<<col.index()<<": |";
            } else {
              if(count==skipcols) {
                for(typename std::string::size_type i=0; i < rowtext.length(); i++)
                  s<<" ";
                s<<"       ";
              }
              s<<"      |";
            }
            for(int innercol=0; innercol < m; ++innercol) {
              s.width(9);
              s<<(*col)[innerrow][innercol]<<" ";
            }
 
            s<<"|";
          }
          if(innerrow==n-1 && col==row->end())
            reachedEnd = true;
          else
            s << std::endl;
        }
        skipcols += width;
        s << std::endl;
      }
      s << std::endl;
    }
 
    // reset the output format
    s.flags(oldflags);
    s.precision(oldprec);
  }
 
  namespace
  {
    template<typename T>
    struct MatlabPODWriter
    {
      static std::ostream& write(const T& t,  std::ostream& s)
      {
        s << t;
        return s;
      }
    };
    template<typename T>
    struct MatlabPODWriter<std::complex<T> >
    {
      static std::ostream& write(const std::complex<T>& t,  std::ostream& s)
      {
        s << t.real() << " " << t.imag();
        return s;
      }
    };
  } // anonymous namespace
 
  template <class FieldType, int dim>
  void writeMatrixToMatlabHelper(const ScaledIdentityMatrix<FieldType,dim>& matrix, int rowOffset, int colOffset, std::ostream& s)
  {
    for (int i=0; i<dim; i++)
    {
      //+1 for Matlab numbering
      s << rowOffset + i + 1 << " " << colOffset + i + 1 << " ";
      MatlabPODWriter<FieldType>::write(matrix.scalar(), s)<< std::endl;
    }
  }
 
  template <class FieldType, int dim>
  void writeMatrixToMatlabHelper(const DiagonalMatrix<FieldType,dim>& matrix, int rowOffset, int colOffset, std::ostream& s)
  {
    for (int i=0; i<dim; i++)
    {
      //+1 for Matlab numbering
      s << rowOffset + i + 1 << " " << colOffset + i + 1 << " ";
      MatlabPODWriter<FieldType>::write(matrix.diagonal(i), s)<< std::endl;
    }
  }
 
  template <class FieldType, int rows, int cols>
  void writeMatrixToMatlabHelper
    (   const FieldMatrix<FieldType,rows,cols>& matrix, int rowOffset,
    int colOffset, std::ostream& s)
  {
    for (int i=0; i<rows; i++)
      for (int j=0; j<cols; j++) {
        //+1 for Matlab numbering
        s << rowOffset + i + 1 << " " << colOffset + j + 1 << " ";
        MatlabPODWriter<FieldType>::write(matrix[i][j], s)<< std::endl;
      }
  }
 
  template <class FieldType>
  void writeMatrixToMatlabHelper(const DynamicMatrix<FieldType>& matrix, int rowOffset,
                                 int colOffset, std::ostream& s)
  {
    for (int i=0; i<matrix.N(); i++)
      for (int j=0; j<matrix.M(); j++) {
        //+1 for Matlab numbering
        s << rowOffset + i + 1 << " " << colOffset + j + 1 << " ";
        MatlabPODWriter<FieldType>::write(matrix[i][j], s)<< std::endl;
      }
  }
 
  template <class MatrixType>
  void writeMatrixToMatlabHelper(const MatrixType& matrix,
                                 int externalRowOffset, int externalColOffset,
                                 std::ostream& s)
  {
    // Precompute the accumulated sizes of the columns
    std::vector<typename MatrixType::size_type> colOffset(matrix.M());
    if (colOffset.size() > 0)
      colOffset[0] = 0;
 
    for (typename MatrixType::size_type i=0; i<matrix.M()-1; i++)
      colOffset[i+1] = colOffset[i] +
                       MatrixDimension<MatrixType>::coldim(matrix,i);
 
    typename MatrixType::size_type rowOffset = 0;
 
    // Loop over all matrix rows
    for (typename MatrixType::size_type rowIdx=0; rowIdx<matrix.N(); rowIdx++)
    {
 
      const typename MatrixType::row_type& row = matrix[rowIdx];
 
      typename MatrixType::row_type::ConstIterator cIt   = row.begin();
      typename MatrixType::row_type::ConstIterator cEndIt = row.end();
 
      // Loop over all columns in this row
      for (; cIt!=cEndIt; ++cIt)
        writeMatrixToMatlabHelper(*cIt,
                                  externalRowOffset+rowOffset,
                                  externalColOffset + colOffset[cIt.index()],
                                  s);
 
      rowOffset += MatrixDimension<MatrixType>::rowdim(matrix, rowIdx);
    }
 
  }
 
  template <class MatrixType>
  void writeMatrixToMatlab(const MatrixType& matrix,
                           const std::string& filename, int outputPrecision = 18)
  {
    std::ofstream outStream(filename.c_str());
    int oldPrecision = outStream.precision();
    outStream.precision(outputPrecision);
 
    writeMatrixToMatlabHelper(matrix, 0, 0, outStream);
    outStream.precision(oldPrecision);
  }
 
  // Recursively print all the blocks
  template<class V>
  void writeVectorToMatlabHelper (const V& v, std::ostream& stream)
  {
    for (const auto& entry : v)
      writeVectorToMatlabHelper(entry, stream);
  }
 
  // Recursively print all the blocks -- specialization for FieldVector
  template<class K, int n>
  void writeVectorToMatlabHelper (const FieldVector<K,n>& v, std::ostream& s)
  {
    for (const auto& entry : v)
    {
      s << entry << std::endl;
    }
  }
 
  // Recursively print all the blocks -- specialization for std::vector
  template<class K>
  void writeVectorToMatlabHelper (const std::vector<K>& v, std::ostream& s)
  {
    for (const auto& entry : v)
    {
      s << entry << std::endl;
    }
  }
 
  // Recursively print all the blocks -- specialization for std::array
  template<class K, std::size_t n>
  void writeVectorToMatlabHelper (const std::array<K,n>& v, std::ostream& s)
  {
    for (const auto& entry : v)
    {
      s << entry << std::endl;
    }
  }
 
  template <class VectorType>
  void writeVectorToMatlab(const VectorType& vector,
                           const std::string& filename, int outputPrecision = 18)
  {
    std::ofstream outStream(filename.c_str());
    int oldPrecision = outStream.precision();
    outStream.precision(outputPrecision);
 
    writeVectorToMatlabHelper(vector, outStream);
    outStream.precision(oldPrecision);
  }
 
} // namespace Dune
 
#endif