twirls/CommonLib/matlab_io.cpp

/*********************************************************************************************
	Description: <EFBFBD>ṩ<EFBFBD><EFBFBD><EFBFBD><EFBFBD>matlab<EFBFBD><EFBFBD>д<EFBFBD><EFBFBD><EFBFBD>ݹ<EFBFBD><EFBFBD><EFBFBD>

	Copyright : All right reserved by ZheYuan.BJ

	Author : Zhang Zhonghai
	Date : 2023/09/18
***********************************************************************************************/

#include <string>
#include <mat.h>
#include <iostream>
#include <vector>
#include "matlab_io.h"
using namespace std;

/* <20><>ȡ<EFBFBD>ַ<EFBFBD><D6B7><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> */
bool ReadMtxString(const string& filePath, const string& mtxName, 
	vector<string>& vStr, int* pRowNum, int* pColNum) {
	
	MATFile* pMatFile = nullptr;
	mxArray* pMxArray = nullptr;
	mxArray* pCell = nullptr;
	int rowNum, colNum;
	char strBuf[STRING_BUF_SIZE];

	pMatFile = matOpen(filePath.c_str(), "r"); //<2F><><EFBFBD><EFBFBD>.mat<61>ļ<EFBFBD>
	if (pMatFile == nullptr) {
		cout << "filePath is error!" << endl;
		return false;
	}
	pMxArray = matGetVariable(pMatFile, mtxName.c_str()); //<2F><>ȡ.mat<61>ļ<EFBFBD><C4BC><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ΪmatrixName<6D>ľ<EFBFBD><C4BE><EFBFBD>
	rowNum = (int)mxGetM(pMxArray);
	colNum = (int)mxGetN(pMxArray);

	vStr.resize(rowNum * colNum);
	for (int i = 0; i < rowNum; ++i) {
		for (int j = 0; j < colNum; ++j) {
			pCell = mxGetCell(pMxArray, j * rowNum + i);
			if (mxGetString(pCell, strBuf, STRING_BUF_SIZE) != 0) {
				cout << "String is too large to fit in the buffer! " << i + 1 << '\t' << j + 1 << endl;
				return false;
			}
			vStr[i * colNum + j] = strBuf;
		}
	}
	*pRowNum = rowNum;
	*pColNum = colNum;

	return true;
}

/* <20><>mat<61>ļ<EFBFBD><C4BC>ж<EFBFBD>ȡ<EFBFBD><C8A1><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ƶľ<C6B5><C4BE><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ݣ<EFBFBD><DDA3><EFBFBD><EFBFBD><EFBFBD>ȡ<EFBFBD><C8A1><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ֵ */
T* ReadMtxDouble(const string& filePath, const string& mtxName, int* pRowNum, int* pColNum) {
	T* dst = nullptr;
	MATFile* pMatFile = nullptr;
	mxArray* pMxArray = nullptr;
	int rowNum, colNum;
	double* matData;
	pMatFile = matOpen(filePath.c_str(), "r"); //<2F><><EFBFBD><EFBFBD>.mat<61>ļ<EFBFBD>
	if (pMatFile == nullptr) {
		cout << "filePath is error!" << endl;
		return nullptr;
	}
	pMxArray = matGetVariable(pMatFile, mtxName.c_str()); //<2F><>ȡ.mat<61>ļ<EFBFBD><C4BC><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ΪmatrixName<6D>ľ<EFBFBD><C4BE><EFBFBD>
	rowNum = (int)mxGetM(pMxArray);
	colNum = (int)mxGetN(pMxArray);
	// cout << rowNum << " " << colNum << endl;
	matData = (double*)mxGetData(pMxArray); //<2F><>ȡָ<C8A1><D6B8>

	dst = new T[rowNum * colNum];
	for (int i = 0; i < rowNum; ++i) {
		for (int j = 0; j < colNum; ++j) {
			dst[i * colNum + j] = T(matData[j * rowNum + i]);
		}
	}

	mxDestroyArray(pMxArray); //<2F>ͷ<EFBFBD><CDB7>ڴ<EFBFBD>
	matClose(pMatFile); // <20>ر<EFBFBD><D8B1>ļ<EFBFBD>

	*pRowNum = rowNum;
	*pColNum = colNum;
	return dst;
}

/* <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD>д<EFBFBD><D0B4>mat<61>ļ<EFBFBD><C4BC>У<EFBFBD><D0A3>ø<EFBFBD><C3B8><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> */
bool SaveMtxDouble(T* src, MATFile* pMatFile, string matrixName, int rowNum, int colNum)
{
	//ת<>ô洢
	int datasize = colNum * rowNum;
	double* mtxData = new double[datasize];//<2F><><EFBFBD>洢<EFBFBD><E6B4A2><EFBFBD><EFBFBD>תΪdouble<6C><65>ʽ
	// memset(mtxData, 0, datasize * sizeof(double));
	if (pMatFile == nullptr)
	{
		cout << "mat file pointer is error!" << endl;
		return false;
	}
	for (int i = 0; i < rowNum; i++)
	{
		for (int j = 0; j < colNum; j++)
		{
			mtxData[j * rowNum + i] = double(src[i * colNum + j]);
			// *(mtxData + j * rowNum + i) = (double)src[i * colNum + j]; <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
		}
	}
	mxArray* pWriteArray = NULL;//matlab<61><62>ʽ<EFBFBD><CABD><EFBFBD><EFBFBD>
	//<2F><><EFBFBD><EFBFBD>һ<EFBFBD><D2BB>rowNum*colNum<75>ľ<EFBFBD><C4BE><EFBFBD>  
	pWriteArray = mxCreateDoubleMatrix(rowNum, colNum, mxREAL);
	//<2F><>data<74><61>ֵ<EFBFBD><D6B5><EFBFBD><EFBFBD>pWriteArrayָ<79><D6B8>
	memcpy((void*)(mxGetPr(pWriteArray)), (void*)mtxData, sizeof(double) * datasize);
	//<2F><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ΪmatrixName
	matPutVariable(pMatFile, matrixName.c_str(), pWriteArray);

	mxDestroyArray(pWriteArray);//release resource
	delete[]mtxData;//release resource

	return true;
}