Question

我有一个接受双矩阵作为输入的mex函数，但我只是意识到这个函数用于的代码，也可以有单精度矩阵。是否可以允许函数接受？

如果没有，解决这个问题的另一种方法是什么？

Answer 1

简单的解决方案是将MATLAB中的输入转换为一致的类型（可能是双重），但如果您希望MEX函数处理多种类型，则这是一种方法。

使用mxIsSingle和mxIsDouble（或mxIsClass）检查输入类型并进行相应处理。您可能在if中有一个mexFunction语句，用于设置输入和输出，然后调用模板函数。请参阅下面的示例，该示例使用C ++标准库函数模板std::min<T>对数组中的所有值进行阈值处理，而无需任何数据转换。

<强> flexibleFunction.cpp

#include "mex.h"
#include <algorithm> // std::min

template <typename T>
void threshArrayLT(T *out, const T *arr, mwSize n, T c)
{  // you allocate out
    for (mwSize i = 0; i < n; ++i)
        out[i] = std::min<T>(arr[i], c);
}

void mexFunction(int nlhs, mxArray *plhs[], int nrhs, const mxArray *prhs[])
{
    if (nlhs > 1 || nrhs != 2)
        mexErrMsgTxt("Syntax:\n\tH = flexibleFunction(arr,c)");

    if (!mxIsDouble(prhs[0]) && !mxIsSingle(prhs[0]))
        mexErrMsgTxt("Array must be double or single.");

    if ((mxIsDouble(prhs[0]) && !mxIsDouble(prhs[1])) || 
            (mxIsSingle(prhs[0]) && !mxIsSingle(prhs[1])))
        mexErrMsgTxt("Arguments must have same type.");

    const mwSize* dims = mxGetDimensions(prhs[0]);
    int ndims = static_cast<int>(mxGetNumberOfDimensions(prhs[0]));
    size_t numel = mxGetNumberOfElements(prhs[0]);

    if (mxIsDouble(prhs[0])) {
        double *arr = mxGetPr(prhs[0]);
        double c = mxGetScalar(prhs[1]);
        plhs[0] = mxCreateNumericArray(ndims,dims,mxDOUBLE_CLASS,mxREAL);
        threshArrayLT(mxGetPr(plhs[0]),arr,numel,c);
        // In reality, if it's this simple, use std::transform with lambda or bind:
        //std::transform(arr, arr+numel, mxGetPr(plhs[0]), 
        //    [&](double s){ return std::min(s,c); });
    } else if (mxIsSingle(prhs[0])) {
        float *arr = (float*)mxGetData(prhs[0]);
        float c = static_cast<float>(mxGetScalar(prhs[1]));
        plhs[0] = mxCreateNumericArray(ndims,dims,mxSINGLE_CLASS,mxREAL);
        threshArrayLT((float*)mxGetData(plhs[0]),arr,numel,c);
    }
}

您还可以在C ++中使用函数重载（相同名称，不同的参数类型）。

示例

>> v = rand(1,8); c = 0.5; >> whos v c Name Size Bytes Class Attributes c 1x1 8 double v 1x8 64 double >> flexibleFunction(v,c) ans = 0.2760 0.5000 0.5000 0.1626 0.1190 0.4984 0.5000 0.3404 >> flexibleFunction(single(v),single(c)) ans = 0.2760 0.5000 0.5000 0.1626 0.1190 0.4984 0.5000 0.3404

MATLAB Mex函数可以接受单个和双精度吗？

1 个答案: