头文件:
class X_MATH_EXPORT XMatrix { public: XMatrix() = delete; XMatrix(size_t row, size_t column, XDecimal val); XMatrix(size_t row, size_t column, const XDecimal vals[]); XMatrix(size_t row, size_t column, std::initializer_list<XDecimal> vals); XMatrix(const XMatrix& m); virtual ~XMatrix(); /** 获取行数 */ size_t getRowCount() const; /** 获取列数 */ size_t getColumnCount() const; /** 获取元素值 */ XDecimal getValue(size_t row, size_t column) const; /** 设置元素值 */ void setValue(size_t row, size_t column, XDecimal val); /** 矩阵转置 */ virtual XMatrix& transpose(); /** 是否为方阵 */ virtual bool isSquareMatrix() const; /** 是否为行矩阵 */ virtual bool isRowMatrix() const; /** 是否为列矩阵 */ virtual bool isColumnMatrix() const; /** 是否为零矩阵 */ virtual bool isZeroMatrix() const; /** 重载运算符 XMatrix == otherXMatrix */ bool operator==(const XMatrix& m) const; /** 重载运算符 XMatrix != otherXMatrix */ bool operator!=(const XMatrix& m) const; /** 重载运算符 XMatrix += otherXMatrix */ XMatrix& operator+=(const XMatrix& m); /** 重载运算符 XMatrix -= otherXMatrix */ XMatrix& operator-=(const XMatrix& m); /** 重载运算符 XMatrix *= XDecimal */ XMatrix& operator*=(XDecimal val); /** 重载运算符 XMatrix /= XDecimal */ XMatrix& operator/=(XDecimal val); // XDecimal*& operator[](size_t row); /** 重载运算符 XMatrix + otherXMatrix */ XMatrix operator+(const XMatrix& m) const; /** 重载运算符 XMatrix - otherXMatrix */ XMatrix operator-(const XMatrix& m) const; /** 重载运算符 XMatrix * otherXMatrix */ XMatrix operator*(const XMatrix& m) const; /** 重载运算符 XMatrix * XDecimal */ XMatrix operator*(XDecimal val) const; /** 重载运算符 XMatrix / XDecimal */ XMatrix operator/(XDecimal val) const; /** 重载运算符 -XMatrix */ friend X_MATH_EXPORT XMatrix operator-(const XMatrix& m); /** 重载运算符 XDecimal * XMatrix */ friend X_MATH_EXPORT XMatrix operator*(XDecimal factor, const XMatrix& m); /** 重载运算符 ostream << XMatrix */ friend X_MATH_EXPORT std::ostream& operator<<(std::ostream& outStream, const XMatrix& m); protected: XMatrix(size_t row, size_t column); XMatrix& swapRow(size_t row1, size_t row2); XMatrix& swapColumn(size_t col1, size_t col2); XMatrix& addTimesRow(size_t targetRow, size_t sourceRow, XDecimal times = 1.0); XMatrix& addTimesColumn(size_t targetCol, size_t sourceCol, XDecimal times = 1.0); protected: /** * @brief 矩阵的数值(行主序) */ XDecimal* _vals{ nullptr }; size_t _row{ 0 }; size_t _column{ 0 }; };
源文件:
XMatrix::XMatrix(size_t row, size_t column, XDecimal val) : _row(row) , _column(column) { XAssert(row != 0 && column != 0); size_t count = row * column; _vals = new XDecimal[count]; // 手动填充数组,避免编译器默认处理不同的情况 for(size_t i = 0; i < count; ++i) { _vals[i] = val; } } XMatrix::XMatrix(size_t row, size_t column, const XDecimal* vals) : _row(row) , _column(column) { XAssert(row != 0 && column != 0); size_t count = row * column; _vals = new XDecimal[count]; for(size_t i = 0; i < count; ++i) { _vals[i] = vals[i]; } } XMatrix::XMatrix(size_t row, size_t column, std::initializer_list<XDecimal> vals) : _row(row) , _column(column) { size_t count = row * column; XAssert(vals.size() == count); _vals = new XDecimal[count]; size_t index = 0; for(auto it = vals.begin(); index < count, it != vals.end(); ++it) { _vals[index] = *it; ++index; } } XMatrix::XMatrix(const XMatrix& m) : _row(m._row) , _column(m._column) { if(m._vals == nullptr) return; size_t count = _row * _column; _vals = new XDecimal[count]; for(size_t i = 0; i < count; ++i) { _vals[i] = m._vals[i]; } } XMatrix::~XMatrix() { if(_vals != nullptr) { delete[] _vals; _vals = nullptr; } } size_t XMatrix::getRowCount() const { return _row; } size_t XMatrix::getColumnCount() const { return _column; } XDecimal XMatrix::getValue(size_t row, size_t column) const { XAssert(row < _row && column < _column); return _vals[_column * row + column]; } void XMatrix::setValue(size_t row, size_t column, XDecimal val) { XAssert(row < _row && column < _column); _vals[_column * row + column] = val; } bool XMatrix::operator==(const XMatrix& m) const { if(_row != m._row || _column != m._column) return false; size_t count = _row * _column; for(size_t i = 0; i < count; ++i) { if(!X::isEqual(_vals[i], m._vals[i])) return false; } return true; } bool XMatrix::operator!=(const XMatrix& m) const { return !(*this == m); } XMatrix& XMatrix::operator+=(const XMatrix& m) { XAssert(_row == m._row && _column == m._column); size_t count = _row * _column; for(size_t i = 0; i < count; ++i) { _vals[i] += m._vals[i]; } return *this; } XMatrix& XMatrix::operator-=(const XMatrix& m) { XAssert(_row == m._row && _column == m._column); size_t count = _row * _column; for(size_t i = 0; i < count; ++i) { _vals[i] -= m._vals[i]; } return *this; } XMatrix& XMatrix::operator*=(XDecimal val) { size_t count = _row * _column; for(size_t i = 0; i < count; ++i) { _vals[i] *= val; } return *this; } XMatrix& XMatrix::operator/=(XDecimal val) { size_t count = _row * _column; for(size_t i = 0; i < count; ++i) { _vals[i] /= val; } return *this; } XMatrix XMatrix::operator+(const XMatrix& m) const { XAssert(_row == m._row && _column == m._column); XMatrix mat(_row, _column); size_t count = _row * _column; for(size_t i = 0; i < count; ++i) { mat._vals[i] = _vals[i] + m._vals[i]; } return mat; } XMatrix XMatrix::operator-(const XMatrix& m) const { XAssert(_row == m._row && _column == m._column); XMatrix mat(_row, _column); size_t count = _row * _column; for(size_t i = 0; i < count; ++i) { mat._vals[i] = _vals[i] - m._vals[i]; } return mat; } XMatrix XMatrix::operator*(const XMatrix& m) const { XAssert(_column == m._row); XMatrix mat(_row, m._column); XDecimal tempVal = 0.0; for(size_t row = 0; row < mat._row; ++row) { for(size_t col = 0; col < mat._column; ++col) { tempVal = 0.0; for(int i = 0; i < _column; ++i) { tempVal += _vals[_column * row + i] * m._vals[m._column * i + col]; } mat._vals[mat._column * row + col] = tempVal; } } return mat; } XMatrix XMatrix::operator*(XDecimal val) const { XMatrix mat(_row, _column); size_t count = _row * _column; for(size_t i = 0; i < count; ++i) { mat._vals[i] = _vals[i] * val; } return mat; } XMatrix XMatrix::operator/(XDecimal val) const { XMatrix mat(_row, _column); size_t count = _row * _column; for(size_t i = 0; i < count; ++i) { mat._vals[i] = _vals[i] / val; } return mat; } XMatrix operator-(const XMatrix& m) { XMatrix mat(m._row, m._column); size_t count = m._row * m._column; for(size_t i = 0; i < count; ++i) { mat._vals[i] = -m._vals[i]; } return mat; } XMatrix operator*(XDecimal val, const XMatrix& m) { return m * val; } std::ostream& operator<<(std::ostream& outStream, const XMatrix& m) { char paddedNumber[25]; outStream << "{"; for(int row = 0; row < m._row; ++row) { outStream << "["; for(int col = 0; col < m._column; ++col) { sprintf(paddedNumber, "% -4.8f", m._vals[row * m._column + col]); outStream << paddedNumber; if(col < m._column) { outStream << " "; } } outStream << "] "; } outStream << "}"; return outStream; } XMatrix::XMatrix(size_t row, size_t column) : _row(row) , _column(column) { XAssert(row != 0 && column != 0); _vals = new XDecimal[row * column]; } XMatrix& XMatrix::swapRow(size_t row1, size_t row2) { XAssert(row1 < _row && row2 < _row); // 如果自己和自己交换则直接返回 if(row1 == row2) return *this; XDecimal tempVal = 0.0; for(size_t col = 0; col < _column; ++col) { tempVal = _vals[_column * row1 + col]; _vals[_column * row1 + col] = _vals[_column * row2 + col]; this->_vals[_column * row2 + col] = tempVal; } return *this; } XMatrix& XMatrix::swapColumn(size_t col1, size_t col2) { XAssert(col1 < _row && col2 < _row); // 如果自己和自己交换则直接返回 if(col1 == col2) return *this; XDecimal tempVal = 0.0; for(size_t row = 0; row < _column; ++row) { tempVal = _vals[_column * row + col1]; _vals[_column * row + col1] = _vals[_column * row + col2]; this->_vals[_column * row + col2] = tempVal; } return *this; } XMatrix& XMatrix::addTimesRow(size_t targetRow, size_t sourceRow, XDecimal times) { XAssert(targetRow < _row && sourceRow < _row); if(!X::isZero(times)) { for(size_t col = 0; col < _column; ++col) { _vals[_column * targetRow + col] += _vals[_column * sourceRow + col] * times; } } return *this; } XMatrix& XMatrix::addTimesColumn(size_t targetCol, size_t sourceCol, XDecimal times) { XAssert(targetCol < _row && sourceCol < _row); if(!X::isZero(times)) { for(size_t row = 0; row < _column; ++row) { _vals[_column * row + targetCol] += _vals[_column * row + sourceCol] * times; } } return *this; } XMatrix& XMatrix::transpose() { auto* vals = new XDecimal[_column * _row]; for(int row = 0; row < _row; ++row) { for(int col = 0; col < _column; ++col) { vals[col * _row + row] = _vals[row * _column + col]; } } delete[] _vals; _vals = vals; auto row = _row; _row = _column; _column = row; return *this; } bool XMatrix::isSquareMatrix() const { return _row == _column; } bool XMatrix::isRowMatrix() const { return _row == 1; } bool XMatrix::isColumnMatrix() const { return _column == 1; } bool XMatrix::isZeroMatrix() const { for(int row = 0; row < _row; ++row) { for(int col = 0; col < _column; ++col) { if(!X::isZero(_vals[row * _column + col])) { return false; } } } return true; }
标签:const,column,矩阵,XMatrix,vals,row,size From: https://www.cnblogs.com/chanyuantiandao/p/18306977