將C/C++三維陣列轉換為MATLAB mat檔案
阿新 • • 發佈:2019-01-31
測試的三維陣列X,Y,Z維度數分別為4,3,2
需要注意的是C/C++三維陣列按行優先儲存,MATLAB按列優先儲存
// 新增依賴的標頭檔案
#include <mat.h>
#include <matrix.h>
// 新增依賴庫
#pragma comment(lib, "libmat.lib")
#pragma comment(lib, "libmx.lib")
#pragma comment(lib, "libmex.lib")
int main() {
// 填充測試陣列
double _3d_array[2][3][4];
int sn = 1;
for (int z = 0; z < 2; ++z) {
for (int y = 0; y < 3; ++y) {
for (int x = 0; x < 4; ++x) {
_3d_array[z][y][x] = sn++;
}
}
}
// 建立mat檔案
MATFile *mat_file = matOpen("d:\\test.mat", "w");
// 建立MATLAB三維陣列。在傳遞指定維度大小的陣列時,按 Y,X,Z的順序設定。即矩陣為Y行,X列,共Z個矩陣
size_t dims[3] = { 3, 4, 2 };
mxArray *mat_array = mxCreateNumericArray(3, dims, mxDOUBLE_CLASS, mxREAL);
double * dst = (double *)(mxGetPr(pMxArray));
double * d = dst;
for (int z = 0; z < 2; ++z) {
// MATLAB按列優先儲存
for (int x = 0; x < 4; ++x) {
for (int y = 0; y < 3; ++y) {
(*d++) = src[z * 3 * 4 + y * 4 + x];
}
}
}
// 儲存三維陣列
matPutVariable(mat_file, "test_value", mat_array);
// 釋放mat檔案物件
matClose(mat_file);
return 0;
}
需要注意的是C/C++三維陣列按行優先儲存,MATLAB按列優先儲存
// 新增依賴的標頭檔案
#include <mat.h>
#include <matrix.h>
// 新增依賴庫
#pragma comment(lib, "libmat.lib")
#pragma comment(lib, "libmx.lib")
#pragma comment(lib, "libmex.lib")
int main() {
// 填充測試陣列
double _3d_array[2][3][4];
int sn = 1;
for (int z = 0; z < 2; ++z) {
for (int y = 0; y < 3; ++y) {
for (int x = 0; x < 4; ++x) {
_3d_array[z][y][x] = sn++;
}
}
}
// 建立mat檔案
MATFile *mat_file = matOpen("d:\\test.mat", "w");
// 建立MATLAB三維陣列。在傳遞指定維度大小的陣列時,按 Y,X,Z的順序設定。即矩陣為Y行,X列,共Z個矩陣
size_t dims[3] = { 3, 4, 2 };
mxArray *mat_array = mxCreateNumericArray(3, dims, mxDOUBLE_CLASS, mxREAL);
double * dst = (double *)(mxGetPr(pMxArray));
double * d = dst;
for (int z = 0; z < 2; ++z) {
// MATLAB按列優先儲存
for (int x = 0; x < 4; ++x) {
for (int y = 0; y < 3; ++y) {
(*d++) = src[z * 3 * 4 + y * 4 + x];
}
}
}
// 儲存三維陣列
matPutVariable(mat_file, "test_value", mat_array);
// 釋放mat檔案物件
matClose(mat_file);
return 0;
}