#ifndef MT3Q6_H
#define MT3Q6_H
///////////////////////////////////////////////////////////////
// MT3Q6.h - Matrix                                          //
//                                                           //
// Jim Fawcett, CSE687 - Object Oriented Design, Spring 2012 //
///////////////////////////////////////////////////////////////

#include <vector>
#include <iostream>

template <typename T>
class Matrix
{
public:
  typedef std::vector<T> row;  // row wrapper class would let us handle 
                               // index error by throwing exception
  Matrix(size_t M, size_t N);
  //////////////////////////////// don't need these ///
  // Matrix(const Matrix<t>& m);
  // Matrix<T>& operator=(const Matrix<T>& m);
  //
  row& operator[](size_t i);
  row operator[](size_t i) const;
  size_t rows() const;
  size_t cols() const;
private:
  std::vector<row> _matrix;
  size_t _M, _N;
};

template <typename T>
size_t Matrix<T>::rows() const { return _M; }

template <typename T>
size_t Matrix<T>::cols() const { return _N; }

template<typename T>
Matrix<T>::Matrix(size_t M, size_t N) : _M(M), _N(N)
{
  for(size_t i=0; i<_M; ++i)
  {
    row r;
    for(size_t j=0; j<_N; ++j)
    {
      r.push_back(T());
    }
    _matrix.push_back(r);
  }
}

template<typename T>
typename Matrix<T>::row& Matrix<T>::operator[](size_t i)
{
  if(i<0 || _M<=i)
    throw std::exception("row index error");
  return _matrix[i];
}

template<typename T>
typename Matrix<T>::row Matrix<T>::operator[](size_t i) const
{
  if(i<0 || _M<=i)
    throw std::exception("row index error");  
  return _matrix[i];
}

template <typename T>
void Display(const Matrix<T>& m)
{
  for(size_t i=0; i<m.rows(); ++i)
  {
    std::cout << "\n  ";
    for(size_t j=0; j<m.cols(); ++j)
      std::cout << m[i][j] << " ";
  }
  std::cout << "\n";
}

#endif