matrix

作者: 世界上的一道风 | 来源:发表于2019-07-11 08:29 被阅读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;
    }
    
    

    相关文章

      网友评论

        本文标题:matrix

        本文链接:https://www.haomeiwen.com/subject/kawlkctx.html