DiagMatrix.h

Go to the documentation of this file.

// $Id$
//==============================================================================
//==============================================================================
 
#ifndef _DIAG_MATRIX_H
#define _DIAG_MATRIX_H
 
#include "SystemMatrix.h"
 
 
class DiagMatrix : public SystemMatrix
{
public:
  DiagMatrix(size_t m = 0) : myMat(m) {}
  DiagMatrix(const DiagMatrix& A) : SystemMatrix(A), myMat(A.myMat) {}
  DiagMatrix(const RealArray& data, size_t nrows = 0);
  virtual ~DiagMatrix() {}
 
  virtual LinAlg::MatrixType getType() const { return LinAlg::DIAG; }
 
  virtual SystemMatrix* copy() const { return new DiagMatrix(*this); }
 
  bool redim(size_t r) { return myMat.resize(r,utl::RETAIN); }
 
  virtual size_t dim(int) const { return myMat.size(); }
 
  Vector& getMat() { return myMat; }
  Real& operator()(size_t r) { return myMat(r); }
  const Real& operator()(size_t r) const { return myMat(r); }
 
  virtual void dump(std::ostream& os, LinAlg::StorageFormat format,
                    const char* label);
 
  virtual void initAssembly(const SAM& sam, char);
 
  virtual void init() { myMat.fill(Real(0)); }
 
  virtual bool assemble(const Matrix& eM, const SAM& sam, int e);
  virtual bool assemble(const Matrix& eM, const SAM& sam,
                        SystemVector& B, int e);
  virtual bool assemble(const Matrix& eM, const SAM& sam,
                        SystemVector& B, const IntVec& meq);
 
  virtual bool assemble(const Matrix& eM, const IntVec& meq);
 
  bool assembleStruct(int val, const SAM& sam, const IntVec& meq);
 
  virtual void mult(Real alpha) { myMat *= alpha; }
 
  virtual bool add(const SystemMatrix& B, Real alpha);
 
  virtual bool add(Real sigma, int ieq);
 
  virtual bool multiply(const SystemVector& B, SystemVector& C) const;
 
  using SystemMatrix::solve;
  virtual bool solve(SystemVector& B, Real*);
 
  virtual Real Linfnorm() const { return myMat.normInf(); }
 
protected:
  virtual std::ostream& write(std::ostream& os) const { return os << myMat; }
 
private:
  Vector myMat; 
};
 
#endif