matrix_block.hpp

Go to the documentation of this file.

// This file is a part of NiHu, a C++ BEM template library.
//
// Copyright (C) 2012-2014  Peter Fiala <fiala@hit.bme.hu>
// Copyright (C) 2012-2014  Peter Rucz <rucz@hit.bme.hu>
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
#ifndef MATRIX_BLOCK_HPP_INCLUDED
#define MATRIX_BLOCK_HPP_INCLUDED
 
namespace NiHu
{
 
template <class Matrix, class RowIndex, class ColIndex = RowIndex>
class matrix_block
{
public:
    matrix_block(Matrix &m, RowIndex const &rows, ColIndex const &cols)
        : m_matrix(m), m_rows(rows), m_cols(cols)
    {
    }
 
    template <class SubMatrix>
    void operator +=(SubMatrix const &rhs) const
    {
        for (int i = 0; i < m_rows.size(); ++i)
            for (int j = 0; j < m_cols.size(); ++j)
                m_matrix(m_rows(i), m_cols(j)) += rhs(i, j);
    }
 
protected:
    Matrix &m_matrix;       
    RowIndex const &m_rows; 
    ColIndex const &m_cols; 
};
 
 
template <class Matrix, class RowIndex, class ColIndex>
matrix_block<Matrix, RowIndex, ColIndex>
    block(Matrix &matrix, RowIndex const &rows, ColIndex const &cols)
{
    return matrix_block<Matrix, RowIndex, ColIndex>(matrix, rows, cols);
}
 
} // end of namespace NiHu
 
#endif /* MATRIX_BLOCK_HPP_INCLUDED */