diagonal_matrix.h

浏览该文件的文档.

// The libMesh Finite Element Library.
// Copyright (C) 2002-2023 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_DIAGONAL_MATRIX_H
#define LIBMESH_DIAGONAL_MATRIX_H

#include "libmesh/id_types.h"
#include "libmesh/sparse_matrix.h"

#include <memory>

namespace libMesh
{
template <typename>
class NumericVector;
namespace Parallel
{
class Communicator;
}

template <typename T>
class DiagonalMatrix : public SparseMatrix<T>
{
public:
  explicit DiagonalMatrix(const Parallel::Communicator & comm);

  virtual ~DiagonalMatrix() = default;

  DiagonalMatrix(DiagonalMatrix &&) = default;
  DiagonalMatrix & operator=(DiagonalMatrix &&) = default;

  DiagonalMatrix & operator=(const NumericVector<T> & vec);

  DiagonalMatrix & operator=(NumericVector<T> && vec);


  virtual
  void init(const numeric_index_type m,
            const numeric_index_type n,
            const numeric_index_type m_l,
            const numeric_index_type n_l,
            const numeric_index_type nnz = 30,
            const numeric_index_type noz = 10,
            const numeric_index_type blocksize = 1) override;

  virtual void init(ParallelType type = PARALLEL) override;

  virtual void init(const NumericVector<T> & other, const bool fast = false);

  virtual void init(const DiagonalMatrix<T> & other, const bool fast = false);

  virtual void clear() override;

  virtual void zero() override;

  virtual std::unique_ptr<SparseMatrix<T>> zero_clone () const override;

  virtual std::unique_ptr<SparseMatrix<T>> clone () const override;

  virtual void close() override;

  virtual numeric_index_type m() const override;

  virtual numeric_index_type n() const override;

  virtual numeric_index_type row_start() const override;

  virtual numeric_index_type row_stop() const override;

  virtual void set(const numeric_index_type i, const numeric_index_type j, const T value) override;

  virtual void add(const numeric_index_type i, const numeric_index_type j, const T value) override;

  virtual
  void add_matrix(const DenseMatrix<T> & dm,
                  const std::vector<numeric_index_type> & rows,
                  const std::vector<numeric_index_type> & cols) override;

  virtual
  void add_matrix(const DenseMatrix<T> & dm,
                  const std::vector<numeric_index_type> & dof_indices) override;

  virtual void add(const T a, const SparseMatrix<T> & X) override;

  virtual T operator()(const numeric_index_type i, const numeric_index_type j) const override;

  virtual Real l1_norm() const override;

  virtual Real linfty_norm() const override;

  virtual bool closed() const override;

  virtual void print_personal(std::ostream & os = libMesh::out) const override;

  virtual void get_diagonal(NumericVector<T> & dest) const override;

  virtual void get_transpose(SparseMatrix<T> & dest) const override;

  virtual void zero_rows(std::vector<numeric_index_type> & rows, T val = 0) override;

  const NumericVector<T> & diagonal() const;

protected:
  std::unique_ptr<NumericVector<T>> _diagonal;
};
} // namespace libMesh

#endif // LIBMESH_DIAGONAL_MATRIX_H