Matrix.h

#ifndef __MATRIX_H__
#define __MATRIX_H__

#include <iostream>
using namespace std;

template <typename T> class Matrix {
private:
  int cols, rows;
  T **mat;

public:
  Matrix(int cols, int rows) : cols(cols), rows(rows), mat(NULL) {
    mat = new T *[rows];
    for (int i = 0; i < rows; i++) {
      mat[i] = new T[cols];
    }
  }

  ~Matrix() {
    for (int i = 0; i < this->rows; i++) {
      delete[] mat[i];
    }
    delete[] mat;
  }

  int getRows() { return this->rows; };

  int getCols() { return this->cols; };

  T &elem(int r, int c) { return mat[r][c]; };

  void print() { cout << "Mat" << rows << "x" << cols; };

  T **getMartix() { return mat; };
};

#endif