Matlab C++ mex函数实现输入矩阵转块对角矩阵报错求解

• 原逻辑仅将输入矩阵的单个元素映射到输出对角线位置，没有把输入的每一行完整元素作为连续块，放到输出矩阵对应行的专属块区域
• Matlab的mxArray采用列优先存储，原索引计算不符合该存储规则，导致元素位置错误

#include <matrix.h>
#include <mex.h>

void mexFunction(int nlhs, mxArray *plhs[], int nrhs, const mxArray *prhs[]) {
        const mwSize *dims;
        double *a, *b; 
        int rows, cols;

        // 输入校验（可选，提升健壮性）
        if (nrhs != 1 || !mxIsDouble(prhs[0]) || mxIsComplex(prhs[0])) {
            mexErrMsgIdAndTxt("block_diag:invalidInput", "仅支持输入非复数的双精度矩阵");
        }
        if (nlhs > 1) {
            mexErrMsgIdAndTxt("block_diag:invalidOutput", "仅支持1个输出参数");
        }

        // 获取输入矩阵维度
        dims = mxGetDimensions(prhs[0]);
        rows = (int) dims[0];
        cols = (int) dims[1];

        // 创建输出矩阵：rows行，rows*cols列
        plhs[0] = mxCreateDoubleMatrix(rows, rows * cols, mxREAL);
        
        // 获取输入、输出数组的指针
        a = mxGetPr(prhs[0]);
        b = mxGetPr(plhs[0]);

        // 生成块对角矩阵
        for (int i = 0; i < rows; i++) {
            // 当前行对应的块的起始列偏移
            int block_col_start = i * cols;
            for (int j = 0; j < cols; j++) {
                // 按列优先规则计算输入、输出元素的索引
                int input_idx = j * rows + i;
                int output_col = block_col_start + j;
                int output_idx = output_col * rows + i;
                b[output_idx] = a[input_idx];
            }
        }
}

• 删除了冗余的mxDuplicateArray调用，无需额外复制输入矩阵，直接读取输入指针即可，减少内存开销
• 调整循环逻辑：按输入矩阵的行遍历，每一行对应输出矩阵的一行，同时对应输出矩阵中一个长度为cols的连续块
• 修正索引计算：完全适配Matlab列优先的存储规则，输入、输出元素的位置计算符合mxArray的内存布局
• 新增可选的输入输出校验逻辑，避免非法输入导致程序崩溃

内容的提问来源于stack exchange，提问作者Joshua Cochrane