使用MATIO库写入MATLAB结构体(struct)数据的示例程序

使用MATIO库写入MATLAB结构体(struct)数据的示例程序

MATIO是一个用于读写MATLAB数据文件(.mat)的开源C库。下面是一个完整的示例程序，展示如何使用MATIO库创建一个包含结构体数据的MAT文件。

示例程序

#include <stdio.h>
#include <stdlib.h>
#include <matio.h>

int main() {
    // 初始化mat文件
    mat_t *matfp = Mat_CreateVer("struct_example.mat", NULL, MAT_FT_MAT5);
    if (matfp == NULL) {
        fprintf(stderr, "Error creating MAT file\n");
        return EXIT_FAILURE;
    }

    // 创建结构体字段名称
    const char *fields[] = {"name", "value", "data"};
    size_t struct_dims[2] = {1, 1}; // 1x1结构体

    // 创建结构体数组
    matvar_t *struct_array = Mat_VarCreateStruct("my_struct", 2, struct_dims, fields, 3);
    if (struct_array == NULL) {
        fprintf(stderr, "Error creating struct variable\n");
        Mat_Close(matfp);
        return EXIT_FAILURE;
    }

    // 添加第一个字段: name (字符串)
    const char *name = "example struct";
    size_t name_dims[2] = {1, strlen(name)};
    matvar_t *name_var = Mat_VarCreate(NULL, MAT_C_CHAR, MAT_T_UINT8, 2, 
                                      name_dims, (void*)name, 0);
    if (name_var == NULL) {
        fprintf(stderr, "Error creating name variable\n");
        Mat_VarFree(struct_array);
        Mat_Close(matfp);
        return EXIT_FAILURE;
    }
    Mat_VarSetStructFieldByName(struct_array, "name", 0, name_var);

    // 添加第二个字段: value (双精度浮点数)
    double value = 3.14159;
    size_t value_dims[2] = {1, 1};
    matvar_t *value_var = Mat_VarCreate(NULL, MAT_C_DOUBLE, MAT_T_DOUBLE, 2, 
                                       value_dims, &value, 0);
    if (value_var == NULL) {
        fprintf(stderr, "Error creating value variable\n");
        Mat_VarFree(struct_array);
        Mat_Close(matfp);
        return EXIT_FAILURE;
    }
    Mat_VarSetStructFieldByName(struct_array, "value", 0, value_var);

    // 添加第三个字段: data (2x3矩阵)
    double data[] = {1.0, 2.0, 3.0, 4.0, 5.0, 6.0};
    size_t data_dims[2] = {2, 3};
    matvar_t *data_var = Mat_VarCreate(NULL, MAT_C_DOUBLE, MAT_T_DOUBLE, 2, 
                                      data_dims, data, 0);
    if (data_var == NULL) {
        fprintf(stderr, "Error creating data variable\n");
        Mat_VarFree(struct_array);
        Mat_Close(matfp);
        return EXIT_FAILURE;
    }
    Mat_VarSetStructFieldByName(struct_array, "data", 0, data_var);

    // 将结构体写入文件
    if (Mat_VarWrite(matfp, struct_array, MAT_COMPRESSION_NONE) != 0) {
        fprintf(stderr, "Error writing variable to file\n");
    }

    // 清理资源
    Mat_VarFree(struct_array);
    Mat_Close(matfp);

    printf("MAT file with struct created successfully!\n");
    return EXIT_SUCCESS;
}

编译说明

要编译此程序，你需要先安装MATIO库。在Linux系统上，可以使用以下命令安装：

sudo apt-get install libmatio-dev

然后使用gcc编译：

gcc -o mat_struct_example mat_struct_example.c -lmatio

程序说明

1. 程序创建了一个名为"struct_example.mat"的MAT文件
2. 文件中包含一个名为"my_struct"的结构体变量
3. 结构体包含三个字段：
   • "name": 字符串类型
   • "value": 双精度标量
   • "data": 2x3双精度矩阵

在MATLAB中验证

生成的.mat文件可以在MATLAB中加载并检查：

load('struct_example.mat');
disp(my_struct);
disp(my_struct.name);
disp(my_struct.value);
disp(my_struct.data);

注意事项

1. 确保在程序结束时释放所有分配的资源以避免内存泄漏
2. 检查每个MATIO函数的返回值以确保操作成功
3. 对于大型数据，考虑使用MAT_COMPRESSION_ZLIB进行压缩
4. 结构体可以嵌套，即结构体字段可以是另一个结构体

这个示例展示了MATIO库的基本用法，你可以根据需要扩展它来处理更复杂的数据结构。