SquareMatrix2D.h

// =======================================================================
// @(#)SquareMatrix2D.h        1.3 11/20/98
// 
// Class:       TSquareMatrix2D
//
// Purpose:     store a square matrix (ansatz = test space) in 2d
//
// Author:      Gunar Matthies
//
// History:     10.08.1998 start implementation
//
// =======================================================================

#ifndef __SQUAREMATRIX2D__
#define __SQUAREMATRIX2D__

#include <SquareMatrix.h>
#include <SquareStructure2D.h>

class TSquareMatrix2D : public TSquareMatrix
{
  protected:
    TSquareStructure2D *structure;

  public:
    TSquareMatrix2D(TSquareStructure2D *squarestructure);

//     /** generate an empty nxn matrix */
//     explicit TSquareMatrix2D(int n);
    
    void SetStructure(TSquareStructure2D *squarestructure);
    
    ~TSquareMatrix2D();

    TFESpace2D *GetFESpace() const
    { return structure->GetFESpace(); }

    TSquareStructure2D *GetMatrixStructure() const
    { return structure; }
    TSquareStructure2D *GetStructure() const
    { return structure; }

    
    TSquareMatrix2D& operator*=(double alpha);
    TSquareMatrix2D& operator+=(TSquareMatrix2D& rhsMat);
    TSquareMatrix2D& operator+=(TSquareMatrix2D* rhsMat)
    { *this += *rhsMat; return *this; }
    
    TSquareMatrix2D& operator=(const TSquareMatrix2D& rhs);

    friend TSquareMatrix2D& operator+(const TSquareMatrix2D & A, 
                                      const TSquareMatrix2D & B);
    
    friend TSquareMatrix2D& operator*(const TSquareMatrix2D & A,
                                      const double alpha);
    friend TSquareMatrix2D& operator*(const double alpha,
                                      const TSquareMatrix2D & A);

    friend double* operator*(const TSquareMatrix2D & A, const double* x);
};

#endif