在 C MexFunction 中使用 Besselk 函数

Using Besselk function in C MexFunction

我想用C mexFunction实现Matern相关函数,需要计算第二类修正Besselk函数。

在MATLAB中,可以使用函数besselk。但是,在任何 C 库中都没有这样的等价物(我说的对吗?)。我知道 boost 库(一个 C++ 库)提供了对第二类修改后的 Besselk 函数的访问,请参阅 https://en.cppreference.com/w/cpp/experimental/special_math,但我无法在我的 [=24] 上使用 MATLAB 2018a 在我的 C mexFunction 中使用它=] 以及 linux 系统。顺便说一句,我不想​​在 C mex 代码中调用 MATLAB 函数 besselk。

如有任何建议,我们将不胜感激。下面是一个使用 C mex 代码构建母体相关函数的最小示例。

#include "mex.h"
#include "matrix.h"
#include <math.h>
//#include <boost/math/special_functions.hpp>
double matern(double h, double nu)
{
    double tau = sqrt(2.0*nu) * h;
    double c = 0.0;
    if(h>0.0){
      c = (pow(tau, nu) * pow(2.0, 1.0-nu) / tgamma(nu)) * boost::math::cyl_bessel_k(nu, tau);
    }else{
      c = 1.0;
    }
    return c;
  }
void mexFunction(int nlhs, mxArray *plhs[], int nrhs, const mxArray *prhs[])
{
    if(nrhs!=2){
        mexErrMsgTxt("Two Inputs are required.");
    }
    double h = mxGetScalar(prhs[0]);
    double nu = mxGetScalar(prhs[1]);
    double corr = matern(h, nu);
    mexPrint("matern(h=%g, nu=%g) = %g", h, nu, corr);
    plhs[0] = mxCreateDoubleMatrix(1, 1, mxREAL);
    mxSetPr(plhs[0], &corr);
         return;
}

如果我将上面的 C mex 代码翻译成 C++ 代码文件,我可以在 mac 上成功获得使用 g++ 编译器编译的 C++ 代码。

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

#include <boost/math/special_functions.hpp>

double matern(const double h, const double nu)
{

  double tau = sqrt(2.0*nu) * h;

  double c = 0.0;
  if(h>0.0){
    c = (pow(tau, nu) * pow(2.0, 1.0-nu) / tgamma(nu)) * boost::math::cyl_bessel_k(nu, tau);
  }else{
    c = 1.0;
  }


  return c;
}

int main(){
    double nu=0.5, h=1.0;
    double corr = matern(h, nu);
    printf("corr=%lf\n", corr);

    return 0;
}

再次强调我的问题,我不需要C++代码,相反我想成功地制作我的C mex代码运行。

这不是真的Mex-specific。

您可以在 C 程序中使用 Boost 特殊数学函数;它们被设计用于 C 或 C++。

为此,您需要添加

#include <boost/math/tr1.hpp>

在 C++ 中,此文件中的声明被插入到命名空间 booth::math 但显然您不能在 C 代码中使用 C++ 命名空间。这些函数本身是用 C linkage 编译的,所以它们是全局名称,可以简单地按原样使用。

请注意,Boost 提供了 C99 和 TR1 特殊数学函数(因此它可以与早于 C99 的标准 C 库一起使用,如果您仍然生活在过去),但它们存在于不同的库中,如在Boost documentation中解释。即使您有 C99 标准库,在包含 Boost header 之后,您也需要将 Boost 库用于 C99 函数。因此,您可能需要在构建命令中添加两个 link 选项:

-lboost_math_c99 -lboost_math_tr1

如果您需要 floatlong double 版本,您将需要其他库。有关详细信息,请参阅上述 link 到 Boost 文档。