1. 程式人生 > >將C/C++三維陣列轉換為MATLAB mat檔案

將C/C++三維陣列轉換為MATLAB mat檔案

測試的三維陣列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;
}