---

matrix.hh

Go to the documentation of this file.

#ifndef MATRIX_HH
#define MATRIX_HH

#include <vector>
#include <iostream>
#include <math.h>
#include "dataref"
#include "mtrx_except.hh"

class Matrix;
class SubMatrix;
class MatMTimesMat;
class MatTimesMat;
class MatCrossMat;

class Matrix {
#include "matrix_friend_ops.hh"    // friend declarations of matrix operations
  friend class SubMatrix;
public:

  // Constructors
  Matrix();
  Matrix(const Matrix&);
  Matrix(int);
  Matrix(double);
  Matrix(int m, int n, double re = 0);
  Matrix(int m, int n, double *re, double *im = NULL);
  Matrix(int m, int n, double **v);
  Matrix(bool isc, bool isd);

  // Destructor
  ~Matrix();

  // Assignment
  Matrix& operator = (const Matrix&);

  // Type casts
  operator double() const;
  operator int() const;
  operator bool() const;

  // Indexing
  template <class T>
  inline SubMatrix operator () (const T&);
  template <class T>
  inline SubMatrix operator () (const T&) const;
  template <class T, class U>
  inline SubMatrix operator () (const T&, const U&);
  template <class T, class U>
  inline SubMatrix operator () (const T&, const U&) const;

  // Info
  inline void Size(int *dims) const;
  inline bool IsEmpty() const;
  inline bool IsDIN() const;
  inline bool IsColon() const;
  inline bool IsVector() const;
  inline bool IsScalar() const;
  inline bool IsReal() const;
  inline bool IsComplex() const;

  // Data extraction
  void ExtractData(double*) const;
  inline double* GetPr();
  inline const double* GetPr() const;
  inline double* GetPi();
  inline const double* GetPi() const;

  // Arithmetic plus assign
  Matrix& operator += (Matrix);
  inline Matrix& operator += (const MatMTimesMat&);
  inline Matrix& operator += (const MatTimesMat&);
  inline Matrix& operator += (const MatCrossMat&);
  Matrix& operator -= (Matrix);
  inline Matrix& operator *= (const Matrix&);
  inline Matrix& operator /= (const Matrix&);

  static Matrix DIN;

protected:
  
  class MatrixData {
  public:
    int m, n, l;
    double *re, *im;
    bool iscolon, isdin;

    MatrixData();
    MatrixData(const MatrixData&);
    MatrixData(int m, int n, double re = 0);
    MatrixData(int m, int n, double *re, double *im = NULL);
    MatrixData(bool isc, bool isd);
    ~MatrixData();

    MatrixData& operator = (const MatrixData&);

    inline bool IsEmpty() const;
    inline bool IsReal() const;
    inline bool IsComplex() const;

    inline double* GetPr();
    inline const double* GetPr() const;
    inline double* GetPi();
    inline const double* GetPi() const;

    inline double index(int) const;
    inline double& index(int);
    inline double index(int, int) const;
    inline double& index(int, int);
    inline int DoubleToSingleIndex(int i, int j) const;
  };

  typedef DataRef<MatrixData> MatrixRef;

  MatrixRef ref;
  static Matrix COLON;

  inline void Assign(const Matrix&);
};

class SubMatrix : public Matrix {
  friend class Matrix;
public:
  void operator = (const SubMatrix& rhs) {
    *base = subsasgn(*base,ind,jnd,Matrix(rhs));
  }
  void operator = (const Matrix& rhs) {
    *base = subsasgn(*base,ind,jnd,rhs);
  }
  ~SubMatrix() {}
private:
  SubMatrix(Matrix *b, const Matrix& i, const Matrix& j = Matrix::DIN)
    : Matrix(subsref(*b,i,j)), base(b), cbase(b), ind(i), jnd(j) {}
  SubMatrix(const Matrix *b, const Matrix& i, const Matrix& j = Matrix::DIN)
    : Matrix(subsref(*b,i,j)), base(NULL), cbase(b), ind(i), jnd(j) {}

  Matrix *base;
  const Matrix *cbase;
  const Matrix ind, jnd;
};

class MatMTimesMat {
public:
  MatMTimesMat(const Matrix& mm1, const Matrix& mm2) : m1(mm1), m2(mm2) {}
  operator Matrix() {return mtimes_eval(m1,m2);};
  const Matrix& m1;
  const Matrix& m2;
};

class MatTimesMat {
public:
  MatTimesMat(const Matrix& mm1, const Matrix& mm2) : m1(mm1), m2(mm2) {}
  operator Matrix() {return times_eval(m1,m2);};
  const Matrix& m1;
  const Matrix& m2;
};

class MatCrossMat {
public:
  MatCrossMat(const Matrix& mm1, const Matrix& mm2) : m1(mm1), m2(mm2) {}
  operator Matrix() {return cross_eval(m1,m2);};
  const Matrix& m1;
  const Matrix& m2;
};

#include "matrix_ops.hh" // declarations of non-friend matrix operations
#include "matrix_member_inline.hh" // inline definitions of matrix members
#include "matrix_friend_ops_inline.hh" //inline defintions of matrix friend ops
#include "matrixdata_inline.hh"

#endif

---