matrix.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_MATRIX_HH
#define DUNE_ISTL_MATRIX_HH
 
#include <memory>
#include <cmath>
 
#include <dune/common/ftraits.hh>
 
#include <dune/istl/istlexception.hh>
#include <dune/istl/bvector.hh>
 
namespace Dune {
 
namespace MatrixImp
{
  template<class B, class A=std::allocator<B> >
  class DenseMatrixBase : public Imp::block_vector_unmanaged<B,A>
                              // this derivation gives us all the blas level 1 and norms
                              // on the large array. However, access operators have to be
                              // overwritten.
  {
  public:
 
    //===== type definitions and constants
 
    typedef typename B::field_type field_type;
 
    typedef A allocator_type;
 
    typedef typename A::size_type size_type;
 
    typedef BlockVector<B,A> value_type;
 
    typedef BlockVector<B,A> block_type;
 
    // just a shorthand
    typedef Imp::BlockVectorWindow<B,A> window_type;
 
    typedef window_type reference;
 
    typedef const window_type const_reference;
 
 
    //===== constructors and such
 
    DenseMatrixBase () : Imp::block_vector_unmanaged<B,A>()
    {
      // nothing is known ...
      rows_ = 0;
      columns_ = 0;
    }
 
    DenseMatrixBase (size_type rows, size_type columns) : Imp::block_vector_unmanaged<B,A>()
    {
      // and we can allocate the big array in the base class
      this->n = rows*columns;
      columns_ = columns;
      if (this->n>0)
      {
        this->p = allocator_.allocate(this->n);
        new (this->p)B[this->n];
      }
      else
      {
        this->n = 0;
        this->p = 0;
      }
 
      // we can allocate the windows now
      rows_ = rows;
    }
 
    DenseMatrixBase (const DenseMatrixBase& a)
    {
      // allocate the big array in the base class
      this->n = a.n;
      columns_ = a.columns_;
      if (this->n>0)
      {
        // allocate and construct objects
        this->p = allocator_.allocate(this->n);
        new (this->p)B[this->n];
 
        // copy data
        for (size_type i=0; i<this->n; i++)
          this->p[i]=a.p[i];
      }
      else
      {
        this->n = 0;
        this->p = nullptr;
      }
 
      // we can allocate the windows now
      rows_ = a.rows_;
    }
 
    ~DenseMatrixBase ()
    {
      if (this->n>0) {
        size_type i=this->n;
        while (i)
          this->p[--i].~B();
        allocator_.deallocate(this->p,this->n);
      }
    }
 
    void resize (size_type rows, size_type columns)
    {
      // deconstruct objects and deallocate memory if necessary
      if (this->n>0) {
        size_type i=this->n;
        while (i)
          this->p[--i].~B();
        allocator_.deallocate(this->p,this->n);
      }
 
      // and we can allocate the big array in the base class
      this->n = rows*columns;
      if (this->n>0)
      {
        this->p = allocator_.allocate(this->n);
        new (this->p)B[this->n];
      }
      else
      {
        this->n = 0;
        this->p = nullptr;
      }
 
      // we can allocate the windows now
      rows_ = rows;
      columns_ = columns;
    }
 
    DenseMatrixBase& operator= (const DenseMatrixBase& a)
    {
      if (&a!=this)     // check if this and a are different objects
      {
        columns_ = a.columns_;
        // reallocate arrays if necessary
        // Note: still the block sizes may vary !
        if (this->n!=a.n || rows_!=a.rows_)
        {
          // deconstruct objects and deallocate memory if necessary
          if (this->n>0) {
            size_type i=this->n;
            while (i)
              this->p[--i].~B();
            allocator_.deallocate(this->p,this->n);
          }
 
          // allocate the big array in the base class
          this->n = a.n;
          if (this->n>0)
          {
            // allocate and construct objects
            this->p = allocator_.allocate(this->n);
            new (this->p)B[this->n];
          }
          else
          {
            this->n = 0;
            this->p = nullptr;
          }
 
          // Copy number of rows
          rows_ = a.rows_;
        }
 
        // and copy the data
        for (size_type i=0; i<this->n; i++)
          this->p[i]=a.p[i];
      }
 
      return *this;
    }
 
 
    //===== assignment from scalar
 
    DenseMatrixBase& operator= (const field_type& k)
    {
      (static_cast<Imp::block_vector_unmanaged<B,A>&>(*this)) = k;
      return *this;
    }
 
 
    //===== access to components
    // has to be overwritten from base class because it must
    // return access to the windows
 
    reference operator[] (size_type i)
    {
#ifdef DUNE_ISTL_WITH_CHECKING
      if (i>=rows_) DUNE_THROW(ISTLError,"index out of range");
#endif
      return window_type(this->p + i*columns_, columns_);
    }
 
    const_reference operator[] (size_type i) const
    {
#ifdef DUNE_ISTL_WITH_CHECKING
      if (i<0 || i>=rows_) DUNE_THROW(ISTLError,"index out of range");
#endif
      return window_type(this->p + i*columns_, columns_);
    }
 
    // forward declaration
    class ConstIterator;
 
    class Iterator
    {
    public:
      Iterator ()
      : window_(nullptr,0)
      {
        i = 0;
      }
 
      Iterator (Iterator& other) = default;
      Iterator (Iterator&& other) = default;
 
      Iterator (B* data, size_type columns, size_type _i)
      : i(_i),
        window_(data + _i*columns, columns)
      {}
 
      Iterator& operator=(Iterator&& other)
      {
        i = other.i;
        // Do NOT use window_.operator=, because that copies the window content, not just the window!
        window_.set(other.window_.getsize(),other.window_.getptr());
        return *this;
      }
 
      Iterator& operator=(Iterator& other)
      {
        i = other.i;
        // Do NOT use window_.operator=, because that copies the window content, not just the window!
        window_.set(other.window_.getsize(),other.window_.getptr());
        return *this;
      }
 
      Iterator& operator++()
      {
        ++i;
        window_.setptr(window_.getptr()+window_.getsize());
        return *this;
      }
 
      Iterator& operator--()
      {
        --i;
        window_.setptr(window_.getptr()-window_.getsize());
        return *this;
      }
 
      bool operator== (const Iterator& it) const
      {
        return window_.getptr() == it.window_.getptr();
      }
 
      bool operator!= (const Iterator& it) const
      {
        return window_.getptr() != it.window_.getptr();
      }
 
      bool operator== (const ConstIterator& it) const
      {
        return window_.getptr() == it.window_.getptr();
      }
 
      bool operator!= (const ConstIterator& it) const
      {
        return window_.getptr() != it.window_.getptr();
      }
 
      window_type& operator* () const
      {
        return window_;
      }
 
      window_type* operator-> () const
      {
        return &window_;
      }
 
      // return index corresponding to pointer
      size_type index () const
      {
        return i;
      }
 
      friend class ConstIterator;
 
    private:
      size_type i;
      mutable window_type window_;
    };
 
    Iterator begin ()
    {
      return Iterator(this->p, columns_, 0);
    }
 
    Iterator end ()
    {
      return Iterator(this->p, columns_, rows_);
    }
 
    Iterator beforeEnd ()
    {
      return Iterator(this->p, columns_, rows_-1);
    }
 
    Iterator beforeBegin () const
    {
      return Iterator(this->p, columns_, -1);
    }
 
    Iterator find (size_type i)
    {
      return Iterator(this->p, columns_, std::min(i,rows_));
    }
 
    ConstIterator find (size_type i) const
    {
      return ConstIterator(this->p, columns_, std::min(i,rows_));
    }
 
    class ConstIterator
    {
    public:
      ConstIterator ()
      : window_(nullptr,0)
      {
        i = 0;
      }
 
      ConstIterator (const B* data, size_type columns, size_type _i)
      : i(_i),
        window_(const_cast<B*>(data + _i * columns), columns)
      {}
 
      ConstIterator (const Iterator& it)
      : i(it.i), window_(it.window_.getptr(),it.window_.getsize())
      {}
 
      ConstIterator& operator=(Iterator&& other)
      {
        i = other.i;
        // Do NOT use window_.operator=, because that copies the window content, not just the window!
        window_.set(other.window_.getsize(),other.window_.getptr());
        return *this;
      }
 
      ConstIterator& operator=(Iterator& other)
      {
        i = other.i;
        // Do NOT use window_.operator=, because that copies the window content, not just the window!
        window_.set(other.window_.getsize(),other.window_.getptr());
        return *this;
      }
 
      ConstIterator& operator++()
      {
        ++i;
        window_.setptr(window_.getptr()+window_.getsize());
        return *this;
      }
 
      ConstIterator& operator--()
      {
        --i;
        window_.setptr(window_.getptr()-window_.getsize());
        return *this;
      }
 
      bool operator== (const ConstIterator& it) const
      {
        return window_.getptr() == it.window_.getptr();
      }
 
      bool operator!= (const ConstIterator& it) const
      {
        return window_.getptr() != it.window_.getptr();
      }
 
      bool operator== (const Iterator& it) const
      {
        return window_.getptr() == it.window_.getptr();
      }
 
      bool operator!= (const Iterator& it) const
      {
        return window_.getptr() != it.window_.getptr();
      }
 
      const window_type& operator* () const
      {
        return window_;
      }
 
      const window_type* operator-> () const
      {
        return &window_;
      }
 
      // return index corresponding to pointer
      size_type index () const
      {
        return i;
      }
 
      friend class Iterator;
 
    private:
      size_type i;
      mutable window_type window_;
    };
 
    using iterator = Iterator;
 
    using const_iterator = ConstIterator;
 
    ConstIterator begin () const
    {
      return ConstIterator(this->p, columns_, 0);
    }
 
    ConstIterator end () const
    {
      return ConstIterator(this->p, columns_, rows_);
    }
 
    ConstIterator beforeEnd() const
    {
      return ConstIterator(this->p, columns_, rows_-1);
    }
 
    ConstIterator rend () const
    {
      return ConstIterator(this->p, columns_, -1);
    }
 
    //===== sizes
 
    size_type N () const
    {
      return rows_;
    }
 
 
  private:
    size_type rows_;            // number of matrix rows
    size_type columns_;           // number of matrix columns
 
    A allocator_;
  };
 
}  // namespace MatrixImp
 
  template<class T, class A=std::allocator<T> >
  class Matrix
  {
  public:
 
    typedef typename T::field_type field_type;
 
    typedef T block_type;
 
    typedef A allocator_type;
 
    typedef typename MatrixImp::DenseMatrixBase<T,A>::window_type row_type;
 
    typedef typename A::size_type size_type;
 
    typedef typename MatrixImp::DenseMatrixBase<T,A>::Iterator RowIterator;
 
    typedef typename row_type::iterator ColIterator;
 
    typedef typename MatrixImp::DenseMatrixBase<T,A>::ConstIterator ConstRowIterator;
 
    typedef typename row_type::const_iterator ConstColIterator;
 
    enum {
      blocklevel = T::blocklevel+1
    };
 
    Matrix() : data_(0,0), cols_(0)
    {}
 
    Matrix(size_type rows, size_type cols) : data_(rows,cols), cols_(cols)
    {}
 
    void setSize(size_type rows, size_type cols) {
      data_.resize(rows,cols);
      cols_ = cols;
    }
 
    RowIterator begin()
    {
      return data_.begin();
    }
 
    RowIterator end()
    {
      return data_.end();
    }
 
    RowIterator beforeEnd ()
    {
      return data_.beforeEnd();
    }
 
    RowIterator beforeBegin ()
    {
      return data_.beforeBegin();
    }
 
    ConstRowIterator begin() const
    {
      return data_.begin();
    }
 
    ConstRowIterator end() const
    {
      return data_.end();
    }
 
    ConstRowIterator beforeEnd() const
    {
      return data_.beforeEnd();
    }
 
    ConstRowIterator beforeBegin () const
    {
      return data_.beforeBegin();
    }
 
    Matrix& operator= (const field_type& t)
    {
      data_ = t;
      return *this;
    }
 
    row_type operator[](size_type row) {
#ifdef DUNE_ISTL_WITH_CHECKING
      if (row<0)
        DUNE_THROW(ISTLError, "Can't access negative rows!");
      if (row>=N())
        DUNE_THROW(ISTLError, "Row index out of range!");
#endif
      return data_[row];
    }
 
    const row_type operator[](size_type row) const {
#ifdef DUNE_ISTL_WITH_CHECKING
      if (row<0)
        DUNE_THROW(ISTLError, "Can't access negative rows!");
      if (row>=N())
        DUNE_THROW(ISTLError, "Row index out of range!");
#endif
      return data_[row];
    }
 
    size_type N() const {
      return data_.N();
    }
 
    size_type M() const {
      return cols_;
    }
 
    Matrix<T>& operator*=(const field_type& scalar) {
      data_ *= scalar;
      return (*this);
    }
 
    Matrix<T>& operator/=(const field_type& scalar) {
      data_ /= scalar;
      return (*this);
    }
 
    Matrix& operator+= (const Matrix& b) {
#ifdef DUNE_ISTL_WITH_CHECKING
      if(N()!=b.N() || M() != b.M())
        DUNE_THROW(RangeError, "Matrix sizes do not match!");
#endif
      data_ += b.data_;
      return (*this);
    }
 
    Matrix& operator-= (const Matrix& b) {
#ifdef DUNE_ISTL_WITH_CHECKING
      if(N()!=b.N() || M() != b.M())
        DUNE_THROW(RangeError, "Matrix sizes do not match!");
#endif
      data_ -= b.data_;
      return (*this);
    }
 
    Matrix transpose() const {
      Matrix out(M(), N());
      for (size_type i=0; i<N(); i++)
        for (size_type j=0; j<M(); j++)
          out[j][i] = (*this)[i][j];
 
      return out;
    }
 
    friend Matrix<T> operator*(const Matrix<T>& m1, const Matrix<T>& m2) {
      Matrix<T> out(m1.N(), m2.M());
      out = 0;
 
      for (size_type i=0; i<out.N(); i++ ) {
        for ( size_type j=0; j<out.M(); j++ )
          for (size_type k=0; k<m1.M(); k++)
            out[i][j] += m1[i][k]*m2[k][j];
      }
 
      return out;
    }
 
    template <class X, class Y>
    friend Y operator*(const Matrix<T>& m, const X& vec) {
#ifdef DUNE_ISTL_WITH_CHECKING
      if (m.M()!=vec.size())
        DUNE_THROW(ISTLError, "Vector size doesn't match the number of matrix columns!");
#endif
      Y out(m.N());
      out = 0;
 
      for (size_type i=0; i<out.size(); i++ ) {
        for ( size_type j=0; j<vec.size(); j++ )
          out[i] += m[i][j]*vec[j];
      }
 
      return out;
    }
 
    template <class X, class Y>
    void mv(const X& x, Y& y) const
    {
#ifdef DUNE_ISTL_WITH_CHECKING
      if (x.N()!=M()) DUNE_THROW(ISTLError,"vector/matrix size mismatch!");
      if (y.N()!=N()) DUNE_THROW(ISTLError,"vector/matrix size mismatch!");
#endif
 
      for (size_type i=0; i<data_.N(); i++) {
        y[i]=0;
        for (size_type j=0; j<cols_; j++)
          (*this)[i][j].umv(x[j], y[i]);
 
      }
 
    }
 
    template<class X, class Y>
    void mtv (const X& x, Y& y) const
    {
#ifdef DUNE_ISTL_WITH_CHECKING
      if (x.N()!=N()) DUNE_THROW(ISTLError,"index out of range");
      if (y.N()!=M()) DUNE_THROW(ISTLError,"index out of range");
#endif
 
      for(size_type i=0; i<y.N(); ++i)
        y[i]=0;
      umtv(x,y);
    }
 
    template <class X, class Y>
    void umv(const X& x, Y& y) const
    {
#ifdef DUNE_ISTL_WITH_CHECKING
      if (x.N()!=M()) DUNE_THROW(ISTLError,"vector/matrix size mismatch!");
      if (y.N()!=N()) DUNE_THROW(ISTLError,"vector/matrix size mismatch!");
#endif
 
      for (size_type i=0; i<data_.N(); i++) {
 
        for (size_type j=0; j<cols_; j++)
          (*this)[i][j].umv(x[j], y[i]);
 
      }
 
    }
 
    template<class X, class Y>
    void mmv (const X& x, Y& y) const
    {
#ifdef DUNE_ISTL_WITH_CHECKING
      if (x.N()!=M()) DUNE_THROW(ISTLError,"vector/matrix size mismatch!");
      if (y.N()!=N()) DUNE_THROW(ISTLError,"vector/matrix size mismatch!");
#endif
      ConstRowIterator endi=end();
      for (ConstRowIterator i=begin(); i!=endi; ++i)
      {
        ConstColIterator endj = (*i).end();
        for (ConstColIterator j=(*i).begin(); j!=endj; ++j)
          (*j).mmv(x[j.index()],y[i.index()]);
      }
    }
 
    template <class X, class Y>
    void usmv(const field_type& alpha, const X& x, Y& y) const
    {
#ifdef DUNE_ISTL_WITH_CHECKING
      if (x.N()!=M()) DUNE_THROW(ISTLError,"vector/matrix size mismatch!");
      if (y.N()!=N()) DUNE_THROW(ISTLError,"vector/matrix size mismatch!");
#endif
 
      for (size_type i=0; i<data_.N(); i++) {
 
        for (size_type j=0; j<cols_; j++)
          (*this)[i][j].usmv(alpha, x[j], y[i]);
 
      }
 
    }
 
    template<class X, class Y>
    void umtv (const X& x, Y& y) const
    {
#ifdef DUNE_ISTL_WITH_CHECKING
      if (x.N()!=N()) DUNE_THROW(ISTLError,"vector/matrix size mismatch!");
      if (y.N()!=M()) DUNE_THROW(ISTLError,"vector/matrix size mismatch!");
#endif
      ConstRowIterator endi=end();
      for (ConstRowIterator i=begin(); i!=endi; ++i)
      {
        ConstColIterator endj = (*i).end();
        for (ConstColIterator j=(*i).begin(); j!=endj; ++j)
          (*j).umtv(x[i.index()],y[j.index()]);
      }
    }
 
    template<class X, class Y>
    void mmtv (const X& x, Y& y) const
    {
#ifdef DUNE_ISTL_WITH_CHECKING
      if (x.N()!=N()) DUNE_THROW(ISTLError,"vector/matrix size mismatch!");
      if (y.N()!=M()) DUNE_THROW(ISTLError,"vector/matrix size mismatch!");
#endif
      ConstRowIterator endi=end();
      for (ConstRowIterator i=begin(); i!=endi; ++i)
      {
        ConstColIterator endj = (*i).end();
        for (ConstColIterator j=(*i).begin(); j!=endj; ++j)
          (*j).mmtv(x[i.index()],y[j.index()]);
      }
    }
 
    template<class X, class Y>
    void usmtv (const field_type& alpha, const X& x, Y& y) const
    {
#ifdef DUNE_ISTL_WITH_CHECKING
      if (x.N()!=N()) DUNE_THROW(ISTLError,"vector/matrix size mismatch!");
      if (y.N()!=M()) DUNE_THROW(ISTLError,"vector/matrix size mismatch!");
#endif
      ConstRowIterator endi=end();
      for (ConstRowIterator i=begin(); i!=endi; ++i)
      {
        ConstColIterator endj = (*i).end();
        for (ConstColIterator j=(*i).begin(); j!=endj; ++j)
          (*j).usmtv(alpha,x[i.index()],y[j.index()]);
      }
    }
 
    template<class X, class Y>
    void umhv (const X& x, Y& y) const
    {
#ifdef DUNE_ISTL_WITH_CHECKING
      if (x.N()!=N()) DUNE_THROW(ISTLError,"vector/matrix size mismatch!");
      if (y.N()!=M()) DUNE_THROW(ISTLError,"vector/matrix size mismatch!");
#endif
      ConstRowIterator endi=end();
      for (ConstRowIterator i=begin(); i!=endi; ++i)
      {
        ConstColIterator endj = (*i).end();
        for (ConstColIterator j=(*i).begin(); j!=endj; ++j)
          (*j).umhv(x[i.index()],y[j.index()]);
      }
    }
 
    template<class X, class Y>
    void mmhv (const X& x, Y& y) const
    {
#ifdef DUNE_ISTL_WITH_CHECKING
      if (x.N()!=N()) DUNE_THROW(ISTLError,"vector/matrix size mismatch!");
      if (y.N()!=M()) DUNE_THROW(ISTLError,"vector/matrix size mismatch!");
#endif
      ConstRowIterator endi=end();
      for (ConstRowIterator i=begin(); i!=endi; ++i)
      {
        ConstColIterator endj = (*i).end();
        for (ConstColIterator j=(*i).begin(); j!=endj; ++j)
          (*j).mmhv(x[i.index()],y[j.index()]);
      }
    }
 
    template<class X, class Y>
    void usmhv (const field_type& alpha, const X& x, Y& y) const
    {
#ifdef DUNE_ISTL_WITH_CHECKING
      if (x.N()!=N()) DUNE_THROW(ISTLError,"vector/matrix size mismatch!");
      if (y.N()!=M()) DUNE_THROW(ISTLError,"vector/matrix size mismatch!");
#endif
      ConstRowIterator endi=end();
      for (ConstRowIterator i=begin(); i!=endi; ++i)
      {
        ConstColIterator endj = (*i).end();
        for (ConstColIterator j=(*i).begin(); j!=endj; ++j)
          (*j).usmhv(alpha,x[i.index()],y[j.index()]);
      }
    }
 
    //===== norms
 
    typename FieldTraits<field_type>::real_type frobenius_norm () const
    {
      return std::sqrt(frobenius_norm2());
    }
 
    typename FieldTraits<field_type>::real_type frobenius_norm2 () const
    {
      double sum=0;
      for (size_type i=0; i<N(); ++i)
        for (size_type j=0; j<M(); ++j)
          sum += data_[i][j].frobenius_norm2();
      return sum;
    }
 
    template <typename ft = field_type,
              typename std::enable_if<!has_nan<ft>::value, int>::type = 0>
    typename FieldTraits<ft>::real_type infinity_norm() const {
      using real_type = typename FieldTraits<ft>::real_type;
      using std::max;
 
      real_type norm = 0;
      for (auto const &x : *this) {
        real_type sum = 0;
        for (auto const &y : x)
          sum += y.infinity_norm();
        norm = max(sum, norm);
      }
      return norm;
    }
 
    template <typename ft = field_type,
              typename std::enable_if<!has_nan<ft>::value, int>::type = 0>
    typename FieldTraits<ft>::real_type infinity_norm_real() const {
      using real_type = typename FieldTraits<ft>::real_type;
      using std::max;
 
      real_type norm = 0;
      for (auto const &x : *this) {
        real_type sum = 0;
        for (auto const &y : x)
          sum += y.infinity_norm_real();
        norm = max(sum, norm);
      }
      return norm;
    }
 
    template <typename ft = field_type,
              typename std::enable_if<has_nan<ft>::value, int>::type = 0>
    typename FieldTraits<ft>::real_type infinity_norm() const {
      using real_type = typename FieldTraits<ft>::real_type;
      using std::max;
 
      real_type norm = 0;
      real_type isNaN = 1;
      for (auto const &x : *this) {
        real_type sum = 0;
        for (auto const &y : x)
          sum += y.infinity_norm();
        norm = max(sum, norm);
        isNaN += sum;
      }
      isNaN /= isNaN;
      return norm * isNaN;
    }
 
    template <typename ft = field_type,
              typename std::enable_if<has_nan<ft>::value, int>::type = 0>
    typename FieldTraits<ft>::real_type infinity_norm_real() const {
      using real_type = typename FieldTraits<ft>::real_type;
      using std::max;
 
      real_type norm = 0;
      real_type isNaN = 1;
      for (auto const &x : *this) {
        real_type sum = 0;
        for (auto const &y : x)
          sum += y.infinity_norm_real();
        norm = max(sum, norm);
        isNaN += sum;
      }
      isNaN /= isNaN;
      return norm * isNaN;
    }
 
    //===== query
 
    bool exists (size_type i, size_type j) const
    {
#ifdef DUNE_ISTL_WITH_CHECKING
      if (i<0 || i>=N()) DUNE_THROW(ISTLError,"row index out of range");
      if (j<0 || i>=M()) DUNE_THROW(ISTLError,"column index out of range");
#else
      DUNE_UNUSED_PARAMETER(i);  DUNE_UNUSED_PARAMETER(j);
#endif
      return true;
    }
 
  protected:
 
    MatrixImp::DenseMatrixBase<T,A> data_;
 
    size_type cols_;
  };
 
} // end namespace Dune
 
#endif