CMatrix类的实现
一. 头文件定义
#ifndef CMATRIX_H
#define CMATRIX_H
#include <iostream>
using namespace std;
class CMatrix{
public:
CMatrix();
CMatrix(int nRow=0,int nCOL=0,double *pData=NULL);
CMatrix(const CMatrix& m);
CMatrix(const char *strPath);
~CMatrix();
bool Create(int nRow,int nCol,double *pData=NULL);
void Set(int nRow,int nCol,double dVale);
void Release();
friend istream & operator>>(istream &is,CMatrix & m);
friend ostream & operator<<(ostream &os,const CMatrix &m);
CMatrix& operator=(const CMatrix& m);
CMatrix& operator+=(const CMatrix& m);
double & operator[](int nIndex);
double & operator()(int nRow,int nCol);
bool operator ==(const CMatrix& m);
bool operator !=(const CMatrix& m);
operator double();
private:
int m_nRow;
int m_nCol;
double *m_pData;
};
CMatrix operator+(const CMatrix& m1,const CMatrix& m2);
inline void CMatrix::Set(int nRow,int nCol,double dVal){
m_pData[nRow*m_nCol+nCol]=dVal;
}
#endif
二. CMatrix.cpp
#include "CMatrix.h"
#include <fstream>
#include <assert.h>
#include <string.h>
CMatrix::CMatrix():m_nRow(0),m_nCol(0),m_pData(0){
}
CMatrix::CMatrix(int nRow,int nCol,double *pData):m_pData(0){
Create(nRow,nCol,pData);
}
CMatrix::CMatrix(const CMatrix &m):m_pData(0){
*this = m;
}
CMatrix::CMatrix(const char * strPath){
m_pData = 0;
m_nRow=m_nCol=0;
ifstream cin (strPath);
cin>>*this;
}
CMatrix::~CMatrix(){
Release();
}
bool CMatrix::Create(int nRow,int nCol,double *pData){
Release();
m_pData=new double[nRow*nCol];
m_nRow=nRow;
m_nCol=nCol;
if(pData){
memcpy(m_pData,pData,nRow*nCol*sizeof(double));
}
}
void CMatrix::Release(){
if(m_pData){
delete []m_pData;
m_pData=NULL;
}
m_nRow = m_nCol = 0;
}
CMatrix& CMatrix::operator=(const CMatrix &m){
if(this!=&m){
Create(m.m_nRow,m.m_nCol,m.m_pData);
}
return *this;
}
CMatrix& CMatrix::operator+=(const CMatrix &m){
assert(m_nRow==m.m_nRow && m_nCol == m.m_nCol);
for(int i=0;i<m_nRow*m_nCol;i++){
m_pData[i]+=m.m_pData[i];
}
return *this;
}
CMatrix operator+(const CMatrix&m1,const CMatrix&m2){
CMatrix m3(m1);
m3 +=m2;
return m3;
}
double & CMatrix::operator[](int nIndex){
assert(nIndex<m_nRow*m_nCol);
return m_pData[nIndex];
}
double &CMatrix::operator()(int nRow,int nCol){
assert(nRow*m_nCol+nCol<m_nRow*m_nCol);
return m_pData[nRow*m_nCol+nCol];
}
bool CMatrix::operator == (const CMatrix &m){
if(!(m_nRow==m.m_nRow && m_nCol==m.m_nCol)){
return false;
}
for(int i=0;i<m_nRow*m_nCol;i++){
if(m_pData[i]!=m.m_pData[i]){
return false;
}
}
return true;
}
bool CMatrix::operator !=(const CMatrix& m){
return !((*this)==m);
}
CMatrix::operator double(){
double dS=0;
for(int i=0;i<m_nRow*m_nCol;i++){
dS+=m_pData[i];
}
return dS;
}
istream & operator>>(istream & is,CMatrix&m){
is>>m.m_nRow>>m.m_nCol;
m.Create(m.m_nRow,m.m_nCol);
for(int i=0;i<m.m_nRow*m.m_nCol;i++){
is>>m.m_pData[i];
}
return is;
}
ostream & operator<<(ostream& os, const CMatrix&m){
os<<m.m_nRow<<""<<m.m_nCol<<endl;
double * pData = m.m_pData;
for(int i=0;i<m.m_nRow;i++){
for(int j=0;j<m.m_nCol;j++){
os<<*pData++<<"";
}
os<<endl;
}
return os;
}
构造函数
1.不带参数的构造函数(初始化)
CMatrix::CMatrix():m_nRow(0),m_nCol(0),m_pData(0){
}
2.带行、列及数据指针等参数的构造函数,并且参数带默认值
CMatrix::CMatrix(int nRow, int nCol, double *pData):m_pData(0){
Create(nRow,nCol,pData);
}
3.带文件路径参数的构造函数
CMatrix::CMatrix(const char * strPath):m_nRow(0),m_nCol(0),m_pData(0){
ifstream cin(strPath);
cin>>*this;
}
4.拷贝构造函数
CMatrix::CMatrix(const CMatrix& m):m_pData(0){
*this = m;
}
析构函数
1.调用Release()
CMatrix::~CMatrix(){
Release();
}
2.将内存释放,并将行列设置为0
bool CMatrix::Create(int nRow, int nCol, double *pData){
Release();
m_pData = new double[nRow*nCol];
m_nRow = nRow;
m_nCol = nCol;
if(pData != NULL){
memcpy(m_pData, pData, nRow*nCol*sizeof(double));
}
}
void CMatrix::Release(){
if(m_pData != NULL){
delete []m_pData;
m_pData = NULL;
}
m_nRow = m_nCol = 0;
}
运算符重载
1.算术运算符重载:+, -, +=, -=
CMatrix operator+(const CMatrix& m1, const CMatrix& m2){
CMatrix m3(m1);
m3 += m2;
return m3;
}
CMatrix operator-(const CMatrix& m1, const CMatrix& m2){
CMatrix m3(m1);
m3 -= m2;
return m3;
}
CMatrix& CMatrix::operator+=(const CMatrix& m){
assert(m_nRow==m.m_nRow && m_nCol==m.m_nCol);
for (int i=0;i<m_nRow*m_nCol;i++){
m_pData[i]+=m.m_pData[i];
}
return *this;
}
CMatrix& CMatrix::operator-=(const CMatrix& m){
assert(m_nRow==m.m_nRow && m_nCol==m.m_nCol);
for (int i=0;i<m_nRow*m_nCol;i++){
m_pData[i]-=m.m_pData[i];
}
return *this;
}
2.关系运算符重载:==,!=
bool CMatrix::operator == (const CMatrix& m){
if(!(m_nRow==m.m_nRow && m_nCol==m.m_nCol)){
return false;
}
for(int i=0;i<m_nRow*m_nCol;i++)
{
if(m_pData[i]!=m.m_pData[i]){
return false;
}
}
return true;
}
bool CMatrix::operator !=(const CMatrix& m){
return !((*this)==m);
}
3.下标操作符:[], ()
double & CMatrix::operator[](int nIndex){
assert(nIndex<m_nRow*m_nCol);
return m_pData[nIndex];
}
double & CMatrix::operator()(int nRow, int nCol){
assert(nRow * m_nCol + nCol < m_nRow * m_nCol);
return m_pData[nRow * m_nCol + nCol];
}
4.强制类型转换: double
CMatrix::operator double(){
double dS=0;
for(int i=0;i<m_nRow*m_nCol;i++){
dS+=m_pData[i];
}
return dS;
}
5.赋值运算符:=,尤其注意当m1=m1特殊情况的处理
CMatrix& CMatrix::operator=(const CMatrix& m){
if(this!=&m){
Create(m.m_nRow, m.m_nCol, m.m_pData);
}
return *this;
}
输入和输出函数
1.输入和输出运输符:<<, >>
istream & operator>>(istream& is, CMatrix & m){
is>>m.m_nRow>>m.m_nCol;
m.Create(m.m_nRow, m.m_nCol);
for(int i=0;i<m.m_nRow*m.m_nCol;i++){
is>>m.m_pData[i];
}
return is;
}
ostream & operator<<(ostream& os, const CMatrix &m){
os<<"size:["<<m.m_nRow<<","<<m.m_nCol<<']'<<endl;
double * pData = m.m_pData;
for(int i=0;i<m.m_nRow;i++){
for(int j=0;j<m.m_nCol;j++){
os<<*pData++<<" ";
}
os<<endl;
}
return os;
}
三. 主函数main.cpp
#include<iostream>
#include "CComplex.h"
#include <stdio.h>
#include "CMatrix.h"
using namespace std;
int main(int argc,char** argv){
double pData[10] = {2, 3, 4, 5};
CMatrix m1, m2(2, 5, pData), m3("d:\\1.txt"), m4(m2);
cin >> m1;
m2.Set(1, 3, 10);
cout << m1 << m2 << m3 << m4;
m4 = m3;
m4[2] = m4 + 1;
if(m4 == m3){
cout << "Error !" << endl;
}
m4 += m3;
cout << "sum of m4 = " << (double)m4 << endl;
return 0;
}
ps:用vscode写c++要配置的东西好多好麻烦