#ifndef CLASS_TRIMATRIX
#define CLASS_TRIMATRIX

#include "Field.h"

class TriMatrix {
 public:
  // Constructors
  TriMatrix(int N);
  TriMatrix(const TriMatrix&);
  ~TriMatrix();

  TriMatrix& operator= (const TriMatrix &);
  double& operator()(int i, int j);
  
  TriMatrix operator+(const TriMatrix &) const;
  TriMatrix operator-(const TriMatrix &) const;
  TriMatrix operator*(const double)      const;
  Field     operator*(Field &)           const;
  Field     operator/(const Field &)     const;
  Field     triSolve (const Field &)     const;
  
  void      print    ();

  static TriMatrix id(int N);
  
 private:
  int N; // Size of TriMatrix
  double *mDiag;
  double *mUpper;
  double *mLower;
};

#endif