矩阵计算C语言源代码.doc

/#include matrix.h#define EX_TRUE (0)#define EX_FALSE (1)#define EX_MALLOC (2)typedef struct tagMatrixunsigned char ucRow;unsigned char ucCol;unsigned short usRev;float *pfDataAdd;Matrix_s;extern int InputRowColInfo(unsigned char *pucRow, unsigned char *pucCol);extern int CreateMatrixMem(void *pvData);extern int InputMatrixData(void *pvData);extern int PrintMatrixData(void *pvData);extern int PrintInverseMatrixData(void *pvData);extern int Show_matrix(Matrix_s *pstMatrixA, Matrix_s *pstMatrixB);/#include operate_matrix.h#define EX_MATRIX_INIT (1)#define EX_MATRIX_UNINIT (0)typedef enumEX_INPUT_MATRIX = 1,EX_ADD_MATRIX,EX_SUBTRACT_MATRIX,EX_MULTIPLY_MATRIX,EX_INVERSE_MATRIX,EX_QUIT_MATRIX,EX_BOTTOM_MATRIXMatrix_opcode_en;extern int OperateMatrix(int iOpCode, Matrix_s *pstMatrixA, Matrix_s *pstMatrix);extern int ShowMatrixMenu();extern int CheckMatrix(int iOpCode, Matrix_s *pstMatrixA, Matrix_s *pstMatrixB);extern int AddMatrix(int iOpCode, Matrix_s *pstMatrixA, Matrix_s *pstMatrixB, Matrix_s *pstMatrixC);extern int SubtractMatrix(int iOpCode, Matrix_s *pstMatrixA, Matrix_s *pstMatrixB, Matrix_s *pstMatrixC);extern int MultiplyMatrix(int iOpCode, Matrix_s *pstMatrixA, Matrix_s *pstMatrixB, Matrix_s *pstMatrixC);extern int CreateResultMatrix(unsigned char ucRow, unsigned char ucCol, void *pvData);extern int InverseMatrix(float a, int n);/*Guanlin Luo ID:10529749*/#include #include #include #include #include matrix.h/*We have already done this header.*/#include operate_matrix.h/*We have already done this header.*/char g_cFlag = EX_MATRIX_UNINIT;int main ()int iInputOpCode;int iRet = EX_FALSE;Matrix_s stMatrixA, stMatrixB;memset(&stMatrixA, 0, sizeof(stMatrixA);memset(&stMatrixB, 0, sizeof(stMatrixB);while(1)/* show menu */(void)ShowMatrixMenu();/* operete matrix */scanf(%d, &iInputOpCode);iRet = OperateMatrix(iInputOpCode, &stMatrixA, &stMatrixB);if (EX_TRUE = iRet)if (EX_QUIT_MATRIX = iInputOpCode)break;if (NULL != stMatrixA.pfDataAdd)free(stMatrixA.pfDataAdd);if (NULL != stMatrixB.pfDataAdd)free(stMatrixB.pfDataAdd);return EX_TRUE;int ShowMatrixMenu()/*choices for users.*/printf(1 - Enter matrices A & Brn);printf(2 - Add matricesrn);printf(3 - Subtract matricesrn);printf(4 - Multiply matricesrn);printf(5 - Inverse of Arn);printf(6 - Quit programrn);printf(Option: );return EX_TRUE;int OperateMatrix(int iOpCode, Matrix_s *pstMatrixA, Matrix_s *pstMatrixB)int iRet = EX_FALSE;Matrix_s stMatrixC;float *pfTmpBuf = NULL;memset(&stMatrixC, 0, sizeof(Matrix_s);switch (iOpCode)case EX_INPUT_MATRIX:iRet = Show_matrix(pstMatrixA, pstMatrixB);if (EX_FALSE = iRet)printf(err input matrix info!n);return iRet;else/* at first, we need input matrix data */g_cFlag = EX_MATRIX_INIT;break;case EX_ADD_MATRIX:/*Addition part.*/iRet = AddMatrix(EX_ADD_MATRIX, pstMatrixA, pstMatrixB, &stMatrixC);if (EX_TRUE != iRet)/*For checking.*/printf(err multiply matrix operation!n);if (EX_MALLOC = iRet);elsefree(stMatrixC.pfDataAdd);return iRet;printf(rnA + B =: rn);iRet = PrintMatrixData(&stMatrixC);if (iRet = EX_FALSE)printf(err print Matrix result!rn);free(stMatrixC.pfDataAdd);return iRet;free(stMatrixC.pfDataAdd);break;case EX_SUBTRACT_MATRIX:/*Subtract two matrices*/iRet = SubtractMatrix(EX_SUBTRACT_MATRIX, pstMatrixA, pstMatrixB, &stMatrixC);if (EX_TRUE != iRet)/*For checking.*/printf(err subtract matrix operation!rn);if (EX_MALLOC = iRet);elsefree(stMatrixC.pfDataAdd);return iRet;printf(rnA - B =: rn);iRet = PrintMatrixData(&stMatrixC);if (iRet = EX_FALSE)printf(err print Matrix result!rn);free(stMatrixC.pfDataAdd);return iRet;free(stMatrixC.pfDataAdd);break;case EX_MULTIPLY_MATRIX:/*multiplication part.*/iRet = MultiplyMatrix(EX_MULTIPLY_MATRIX, pstMatrixA, pstMatrixB, &stMatrixC);if (EX_TRUE != iRet)/*For checking.*/printf(rnerr multiply matrix operation!rn);if (EX_MALLOC = iRet);elsefree(stMatrixC.pfDataAdd);return iRet;printf(rnA * B =: rn);iRet = PrintMatrixData(&stMatrixC);if (EX_TRUE != iRet)printf(err print Matrix result!rn);free(stMatrixC.pfDataAdd);return iRet;free(stMatrixC.pfDataAdd);break;case EX_INVERSE_MATRIX:/*inverse.*/if (EX_MATRIX_UNINIT = g_cFlag)/*For checking the input.*/printf(please choose 1 operation at first.rn);return EX_FALSE;pfTmpBuf = (float *)malloc(pstMatrixA-ucRow * pstMatrixA-ucCol * sizeof(float);if (pfTmpBuf = NULL)/*For checking.*/printf(err malloc memory.rn);return EX_MALLOC;elsememset(pfTmpBuf, 0, (pstMatrixA-ucRow * pstMatrixA-ucCol * sizeof(float);memcpy(pfTmpBuf, pstMatrixA-pfDataAdd, (pstMatrixA-ucRow * pstMatrixA-ucCol * sizeof(float);stMatrixC.ucCol = pstMatrixA-ucCol;stMatrixC.ucRow = pstMatrixA-ucRow;stMatrixC.pfDataAdd = pfTmpBuf;iRet = InverseMatrix(stMatrixC.pfDataAdd, stMatrixC.ucRow);if (EX_TRUE != iRet)/*For checking.*/printf(rnerr inverse matrix operation!rn);if (EX_MALLOC = iRet);elsefree(stMatrixC.pfDataAdd);return iRet;printf(rnInverse of A =: rn);iRet = PrintInverseMatrixData(&stMatrixC);if (EX_TRUE != iRet)/*For checking.*/printf(err print Matrix result!rn);free(stMatrixC.pfDataAdd);return iRet;free(stMatrixC.pfDataAdd);break;case EX_QUIT_MATRIX:/*users chose to end this program.*/printf(will quit matrix menu!rn);break;default:/*For checking.*/printf(err operate code!rn);return EX_FALSE;return EX_TRUE;int CheckMatrix(int iOpCode, Matrix_s *pstMatrixA, Matrix_s *pstMatrixB)if (pstMatrixA = NULL) | (pstMatrixB = NULL)printf(matrix para is null.rn);return EX_FALSE;if (EX_MATRIX_UNINIT = g_cFlag)printf(please choose 1 operation at first.rn);return EX_FALSE;switch (iOpCode)case EX_ADD_MATRIX: case EX_SUBTRACT_MATRIX:if (pstMatrixA-ucCol = pstMatrixB-ucCol)& (pstMatrixA-ucRow = pstMatrixB-ucRow);else/* ADD / Subtract operation */printf(rnAdd/Subtract operation is invalid!rn);return EX_FALSE;break;case EX_INVERSE_MATRIX:if (2 = pstMatrixA-ucRow)& (2 = pstMatrixA-ucCol);else/* Inverse operation */printf(rnMultiply/Inverse operation is invalid!rn);return EX_FALSE;break;case EX_MULTIPLY_MATRIX:if (pstMatrixA-ucRow = pstMatrixB-ucCol)& (pstMatrixA-ucCol = pstMatrixB-ucRow);else/* Multiply operation */printf(rnMultiply operation is invalid!rn);return EX_FALSE;break;default:printf(err operate code!rn);return EX_FALSE;return EX_TRUE;int AddMatrix(int iOpCode, Matrix_s *pstMatrixA, Matrix_s *pstMatrixB, Matrix_s *pstMatrixC)int iRet = EX_FALSE;unsigned char ucRow;unsigned char ucCol;unsigned char i, j;float fTmpData = 0.0;if (pstMatrixA = NULL) | (pstMatrixB = NULL)printf(matrix para is null.rn);return EX_FALSE;iRet = CheckMatrix(iOpCode, pstMatrixA, pstMatrixB);if (EX_FALSE = iRet)return iRet;elseucRow = pstMatrixA-ucRow;ucCol = pstMatrixA-ucCol;/* create result matrix C */iRet = CreateResultMatrix(ucRow, ucCol, pstMatrixC);if (iRet != EX_TRUE)return iRet;for (i = 0; i ucRow; i+)for (j = 0; j pfDataAdd + i * ucCol + j) + *(pstMatrixB-pfDataAdd + i * ucCol + j);*(pstMatrixC-pfDataAdd + i * ucCol + j) = fTmpData;return EX_TRUE;int SubtractMatrix(int iOpCode, Matrix_s *pstMatrixA, Matrix_s *pstMatrixB, Matrix_s *pstMatrixC)int iRet = EX_FALSE;unsigned char ucRow;unsigned char ucCol;unsigned char i, j;float fTmpData = 0.0;if (pstMatrixA = NULL) | (pstMatrixB = NULL)printf(matrix para is null.rn);return EX_FALSE;iRet = CheckMatrix(iOpCode, pstMatrixA, pstMatrixB);if (EX_FALSE = iRet)return iRet;elseucRow = pstMatrixA-ucRow;ucCol = pstMatrixA-ucCol;/* create result matrix C */iRet = CreateResultMatrix(ucRow, ucCol, pstMatrixC);if (iRet != EX_TRUE)return iRet;for (i = 0; i ucRow; i+)for (j = 0; j pfDataAdd + i * ucCol + j) - *(pstMatrixB-pfDataAdd + i * ucCol + j);*(pstMatrixC-pfDataAdd + i * ucCol + j) = fTmpData;return EX_TRUE;int MultiplyMatrix(int iOpCode, Matrix_s *pstMatrixA, Matrix_s *pstMatrixB, Matrix_s *pstMatrixC)int iRet = EX_FALSE;unsigned char ucRow;unsigned char ucCol;unsigned char i, j, k;float fTmpData = 0.0;if (pstMatrixA = NULL) | (pstMatrixB = NULL)printf(matrix para is null.rn);return EX_FALSE;iRet = CheckMatrix(iOpCode, pstMatrixA, pstMatrixB);if (EX_FALSE = iRet)return iRet;elseucRow = pstMatrixA-ucRow;ucCol = pstMatrixB-ucCol;/* create result matrix C */iRet = CreateResultMatrix(ucRow, ucCol, pstMatrixC);if (iRet != EX_TRUE)return iRet;for (i = 0; i ucRow; i+)for (j = 0; j ucCol; j+)for (k = 0; k ucCol; k+)fTmpData += (float)(*(pstMatrixA-pfDataAdd + i * ucCol + k) * *(pstMatrixB-pfDataAdd + j + k * ucCol);*(pstMatrixC-pfDataAdd + i * ucCol + j) = fTmpData;fTmpData = 0;return EX_TRUE;int CreateResultMatrix(unsigned char ucRow, unsigned char ucCol, void *pvData)Matrix_s *pstTmpData = NULL;int iRet = EX_FALSE;if (pvData = NULL)printf(para is null.rn);return EX_FALSE;elsepstTmpData = (Matrix_s *)pvData;/* create result matrix C */memset(pstTmpData, 0, sizeof(Matrix_s);pstTmpData-ucRow = ucRow;pstTmpData-ucCol = ucCol;iRet = CreateMatrixMem(pstTmpData);if (iRet != EX_TRUE)printf(err create result Matrix C!rn);return i