在不关闭 MATLAB 的情况下停止 mex 函数 (C)

Question

我想在检测到 NaN 后停止执行 C 代码并使用 mexWarnMsgTxt 将错误消息发送到 MATLAB。

C 代码通过 mex 文件从 MATLAB 执行。我尝试使用 abort() 和 exit() 这确实杀死了 c 程序，但也杀死了 MATLAB（我猜是因为它是这里的调用进程。）

#include "mex.h"
#include "matrix.h"


for (int i = 0; i <= 5; i++)
       {
           if (mxIsFinite(out[i])) {

           }
           else if (mxIsInf(out[i])) {
               char *err_msg = malloc(max_len_err_msg);
               snprintf(err_msg, max_len_err_msg, "Inf detected in file %s at line %d", __FILE__, __LINE__);
               mexWarnMsgTxt(err_msg);
               abort();
               //free(err_msg);
               //abort(1);
               /* NOTE: Test for NaN is here for illustration only.  If a double
                * is not finite and is not infinity, then it is a NaN */

           }
           else if (mxIsNaN(out[i])) {
               char *err_msg = malloc(max_len_err_msg);
               snprintf(err_msg, max_len_err_msg, "NaN detected in file %s at line %d", __FILE__, __LINE__);
               mexWarnMsgTxt(err_msg);
               abort();
               //free(err_msg);

           }
       }

我只希望我的 mexFunction 停止但不终止 Matlab。

Answer 1

mex 函数是普通的 C 函数，所以要提前离开函数，只需使用 return。

如果您分配的资源需要手动清理，C 中已建立的习惯用法是使用goto cleanup;（这是goto 的少数（如果不是唯一的话）可接受和普遍接受的用法之一):

void mexFunction(
    int nlhs, mxArray *plhs[],
    int nrhs, const mxArray *prhs[]
) {
    for (int i = 0; i <= 5; i++) {
        if (mxIsInf(out[i])) {
            char *err_msg = malloc(max_len_err_msg);
            snprintf(err_msg, max_len_err_msg, "Inf detected in file %s at line %d", __FILE__, __LINE__);
            mexWarnMsgTxt(err_msg);
            free(err_msg);
            goto cleanup;
        } else if (mxIsNaN(out[i])) {
            char *err_msg = malloc(max_len_err_msg);
            snprintf(err_msg, max_len_err_msg, "NaN detected in file %s at line %d", __FILE__, __LINE__);
            mexWarnMsgTxt(err_msg);
            free(err_msg);
            goto cleanup;
        }
        …
    }

cleanup:

    // Perform cleanup here.
}

（请注意，在此代码中，err_msg 清理是在其自身范围内执行的，而不是全局清理。）

但在不进行清理的情况下，goto 语句是不必要的，可以用return 代替。

Answer 2

我当前的解决方案是在 C 中定义一个全局变量 abort_flag，如果发生错误，则将其设置为 1，并基于此中断我的所有循环并从函数返回。 有点“手动”但有效：

int  abort_flag = 0;
// to use in other file insert into header: extern int abort_flag;

// in the NaN case (see above)
abort_flag = 1;

// in the loops
if (abort_flag==1) { break; };

// in the functions
if (abort_flag==1) { return; };