coupling_matrix.h

Go to the documentation of this file.

// The libMesh Finite Element Library.
// Copyright (C) 2002-2014 Benjamin S. Kirk, John W. Peterson, Roy H. Stogner

// This library is free software; you can redistribute it and/or
// modify it under the terms of the GNU Lesser General Public
// License as published by the Free Software Foundation; either
// version 2.1 of the License, or (at your option) any later version.

// This library 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
// Lesser General Public License for more details.

// You should have received a copy of the GNU Lesser General Public
// License along with this library; if not, write to the Free Software
// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA



#ifndef LIBMESH_COUPLING_MATRIX_H
#define LIBMESH_COUPLING_MATRIX_H

// Local Includes
#include "libmesh/libmesh_common.h"

// C++ includes
#include <vector>

namespace libMesh
{




class CouplingMatrix
{
public:

  explicit
  CouplingMatrix (const unsigned int n=0);

  unsigned char operator() (const unsigned int i,
                            const unsigned int j) const;

  unsigned char & operator() (const unsigned int i,
                              const unsigned int j);

  unsigned int size() const;

  void resize(const unsigned int n);

  void clear();

  bool empty() const;

private:

  std::vector<unsigned char> _values;

  unsigned int _size;
};






//--------------------------------------------------
// CouplingMatrix inline methods
inline
CouplingMatrix::CouplingMatrix (const unsigned int n) :
  _values(), _size(n)
{
  this->resize(n);
}



inline
unsigned char CouplingMatrix::operator() (const unsigned int i,
                                          const unsigned int j) const
{
  libmesh_assert_less (i, _size);
  libmesh_assert_less (j, _size);

  return _values[i*_size + j];
}



inline
unsigned char & CouplingMatrix::operator() (const unsigned int i,
                                            const unsigned int j)
{
  libmesh_assert_less (i, _size);
  libmesh_assert_less (j, _size);

  return _values[i*_size + j];
}



inline
unsigned int CouplingMatrix::size() const
{
  return _size;
}



inline
void CouplingMatrix::resize(const unsigned int n)
{
  _size = n;

  _values.resize(_size*_size);

  for (unsigned int i=0; i<_values.size(); i++)
    _values[i] = 0;
}



inline
void CouplingMatrix::clear()
{
  _size = 0;

  _values.clear();
}



inline
bool CouplingMatrix::empty() const
{
  return (_size == 0);
}


} // namespace libMesh


#endif // LIBMESH_COUPLING_MATRIX_H