C++步步为营

matrix

2019-07-11  本文已影响0人  世界上的一道风

Matrix.h文件:

//
// Created by zc on 2019-07-09.
//

#ifndef OBJECTS_MATRIX_H
#define OBJECTS_MATRIX_H

#endif //OBJECTS_62MATRIX_H
#include<fstream>
#include<iostream>

using namespace std;


template <typename elemType>
class Matrix
{

public:
    Matrix(int rows, int columns);
    Matrix(const Matrix&);
    ~Matrix(){delete [] _matrix;};
  friend 
  Matrix<elemType>
  operator+(const Matrix<elemType> &m1, const Matrix<elemType> &m2)
  {
      Matrix<elemType> result(m1);
      result += m2;
      return result;
  }


  friend 
  Matrix<elemType>
  operator*(const Matrix<elemType> &m1, const Matrix<elemType> &m2)
  {
      Matrix<elemType> result(m1.rows(), m2.cols());
      for (int ix=0; ix<m1.rows(); ix++){
          for (int jx=0; jx<m1.cols(); jx++){
              result(ix, jx) = 0;
              // m1矩阵的列数等于m2矩阵的行数
              for( int kx=0; kx<m1.cols(); kx++)
                  result(ix, jx) += m1(ix,kx) * m2(kx, jx);
          }
    }
    return result;
  }
    Matrix& operator=(const Matrix&);
    void operator+=(const Matrix&);


    // 同一个()操作,对应不同的elemType,一个可以改变元素,一个不能。
    elemType& operator()(int row, int column)
    {return _matrix[row * cols() + column];}
    const elemType& operator()(int row, int column) const
    {return _matrix[row * cols() + column];}

    int rows() const{return _rows;}
    int cols() const{return _cols;}

    bool same_size(const Matrix &m) const
    { return rows() == m.rows() && cols() == m.cols();}

    std::ostream& print(std::ostream&) const;


protected:
    int _rows;
    int _cols;
    elemType *_matrix;
};

template <typename elemType>
inline std::ostream&
operator<<(std::ostream& os, const Matrix<elemType> &m)
{
    return m.print(os);
}

Matrix.cpp文件:

//
// Created by 郑楚 on 2019-07-09.
//

#include "Matrix.h"
#include<fstream>
using namespace std;


template <typename elemType>
Matrix<elemType>::Matrix(int rows, int cols): _rows(rows), _cols(cols)
{
   int size = _rows * _cols;
   _matrix = new elemType[size];
   for (int ix=0; ix<size; ++ix)
       _matrix[ix] = elemType(); //不同类型的初始化构造函数被调用
}

template <typename elemType>
Matrix<elemType>::Matrix(const Matrix &rhs) {
    _rows = rhs._rows; _cols = rhs._cols;
    int mat_size = _rows * _cols;
    _matrix = new elemType[mat_size];
    for (int ix=0; ix<mat_size; ++ix)
        _matrix[ix] = rhs._matrix[ix];

}


//deepcopy
template <typename elemType>
Matrix<elemType>& Matrix<elemType>::operator=(const Matrix &rhs)
{
    if (this != &rhs){
        _rows = rhs._rows; _cols = rhs._cols;
        int mat_size = _rows * _cols;
        delete [] _matrix;
        _matrix = new elemType[mat_size];
        for (int ix=0; ix<mat_size; ++ix)
            _matrix[ix] = rhs._matrix[ix];
    }
    return *this;
}



template <typename elemType>
void Matrix<elemType>::operator+=(const Matrix<elemType> &m)
{
    int matrix_size = cols() * rows();
    for (int ix=0; ix<matrix_size; ix++)
        (* (_matrix + ix)) += (*( m._matrix + ix));
}

template <typename elemType>
std::ostream& Matrix<elemType>::print( std::ostream &os) const{
    int col = cols();
    int matrix_size = col * rows();
    for (int ix=0; ix<matrix_size; ++ix){
        if (ix % col == 0) os << std::endl;
        os << ( *(_matrix + ix)) << ' ';
    }
    os << std::endl;
    return os;
}

int main()
{
    ofstream log("log.txt");
    if (!log)
    {cerr << "can't open log file!\n"; return -1;}
    Matrix<float> identity(4, 4);
    log << "identity: " << identity << endl;
    float ar[16] = {1.0, 0., 0., 0.,
                    0., 1.0, 0., 0.,
                    0., 0., 1., 0.,
                    0., 0., 0., 1.};
    for (int i=0, k=0; i<4; ++i)
        for(int j=0; j<4; j++)
            identity(i, j) = ar[k++];

    log << "identity after set: " << identity << endl;

    Matrix<float> m(identity);
    log << "m: merberwise initialized: " << m << endl;

    Matrix<float> m2(8, 12);
    log << "m2: 8x12: " << m2 << endl;

    m2 = m;
    log << "m2 after memberwise assigned to m: " << m2 << endl;

    float ar2[16] = {1.3, 0.4, 2.6, 8.2, 6.2, 1.7, 1.3, 8.3,
                     4.2, 7.4, 2.7, 1.9, 6.3, 8.1, 5.6, 6.6};

    Matrix<float> m3(4, 4);
    for (int i=0, k=0; i<4; ++i)
        for(int j=0; j<4; j++)
            m3(i, j) = ar2[k++];

    log << "m3: assigned random values: " << m3 << endl;

    Matrix<float> m4 = m3 * identity; log << m4 << endl;
    Matrix<float> m5 = m3 + m4; log << m5 << endl;

    m3 += m4; log << m3 << endl;
    return 0;
}

上一篇下一篇

猜你喜欢

热点阅读