source: git/factory/templates/ftmpl_matrix.h @ dce1e3

/* emacs edit mode for this file is -*- C++ -*- */
/* $Id$ */

#ifndef INCL_MATRIX_H
#define INCL_MATRIX_H

#include <factoryconf.h>

#ifndef NOSTREAMIO
#ifdef HAVE_IOSTREAM
#include <iostream>
#define OSTREAM std::ostream
#elif defined(HAVE_IOSTREAM_H)
#include <iostream.h>
#define OSTREAM ostream
#endif
#endif /* NOSTREAMIO */

template <class T>
class SubMatrix;

template <class T>
class Matrix;

#ifndef NOSTREAMIO
template <class T>
OSTREAM& operator<< (OSTREAM &, const Matrix<T> &);
#endif

template <class T>
class Matrix
{
private:
    int NR, NC;
    T ** elems;
#ifndef NOSTREAMIO
    void printrow ( OSTREAM & s, int i ) const;
#endif /* NOSTREAMIO */
    typedef T* T_ptr;
public:
    Matrix() : NR(0), NC(0), elems(0) {}
    Matrix( int nr, int nc );
    Matrix( const Matrix<T>& M );
    ~Matrix();
    Matrix<T>& operator= ( const Matrix<T>& M );
    int rows() const { return NR; }
    int columns() const { return NC; }
    SubMatrix<T> operator[] ( int i );
    const SubMatrix<T> operator[] ( int i ) const;
    T& operator() ( int row, int col );
    T operator() ( int row, int col ) const;
    SubMatrix<T> operator() ( int rmin, int rmax, int cmin, int cmax );
    const SubMatrix<T> operator() ( int rmin, int rmax, int cmin, int cmax ) const;
    void swapRow( int i, int j );
    void swapColumn( int i, int j );
#ifndef NOSTREAMIO
    void print( OSTREAM& s ) const;
    friend OSTREAM & operator<< <T>( OSTREAM & s, const Matrix<T>& M );
#endif /* NOSTREAMIO */
    friend class SubMatrix<T>;
};
    template <class T>
    Matrix<T> operator+ ( const Matrix<T>& lhs, const Matrix<T>& rhs );
    template <class T>
    Matrix<T> operator- ( const Matrix<T>& lhs, const Matrix<T>& rhs );
    template <class T>
    Matrix<T> operator* ( const Matrix<T>& lhs, const Matrix<T>& rhs );
    template <class T>
    Matrix<T> operator* ( const Matrix<T>& lhs, const T& rhs );
    template <class T>
    Matrix<T> operator* ( const T& lhs, const Matrix<T>& rhs );

template <class T>
class SubMatrix
{
private:
    int r_min, r_max, c_min, c_max;
    Matrix<T>& M;
    // we do not provide a default ctor, so nobody can declare an empty SubMatrix
    SubMatrix( int rmin, int rmax, int cmin, int cmax, const Matrix<T> & m );
public:
    SubMatrix( const SubMatrix<T> & S );
    SubMatrix<T>& operator= ( const SubMatrix<T>& S );
    SubMatrix<T>& operator= ( const Matrix<T>& S );
    operator Matrix<T>() const;
    T operator[] ( int i ) const;
    T& operator[] ( int i );
    friend class Matrix<T>;
};

#ifndef NOSTREAMIO
template <class T>
OSTREAM & operator<< ( OSTREAM & s, const Matrix<T>& M );
#endif /* NOSTREAMIO */

#endif /* ! INCL_MATRIX_H */