1
我正在写一个程序,涉及到寻找二次和三次多项式的实根,并且由于MATLAB中的roots
函数用于具有一般度和计算量大的多项式,我选择了GSL Library这样的函数,我想MEX它并在MATLAB中使用它。下面的代码:如何访问传递给MEX函数的矩阵的成员?
/*
* mx_solve_quadratic.cpp
*
* Solves for real roots of the standard quadratic equation
*
* The calling syntax is:
*
* rootsMatrix = mx_solve_quadratic(coefficientsMatrix)
*
* This is a MEX file for MATLAB.
*/
#include <config.h>
#include <math.h>
#include "mex.h"
#include "matrix.h"
int gsl_poly_solve_quadratic (double , double , double , double *, double *)
/* The gateway function */
void mexFunction(int nlhs, mxArray *plhs[], int nrhs, const mxArray *prhs[])
{
double a; /* coefficient for x^2 */
double b; /* coefficient for x */
double c; /* coefficient for 1 */
double *x0 /* pointer to the smaller root */
double *x1 /* pointer to the bigger root */
mxGetData(prhs[0])
}
int gsl_poly_solve_quadratic (double a, double b, double c, double *x0, double *x1)
{
if (a == 0) /* Handle linear case */
{
if (b == 0)
{
return 0;
}
else
{
*x0 = -c/b;
return 1;
};
}
{
double disc = b * b - 4 * a * c;
if (disc > 0)
{
if (b == 0)
{
double r = sqrt (-c/a);
*x0 = -r;
*x1 = r;
}
else
{
double sgnb = (b > 0 ? 1 : -1);
double temp = -0.5 * (b + sgnb * sqrt (disc));
double r1 = temp/a ;
double r2 = c/temp ;
if (r1 < r2)
{
*x0 = r1 ;
*x1 = r2 ;
}
else
{
*x0 = r2 ;
*x1 = r1 ;
}
}
return 2;
}
else if (disc == 0)
{
*x0 = -0.5 * b/a ;
*x1 = -0.5 * b/a ;
return 2 ;
}
else
{
return 0;
}
}
}
因为单一类型的矩阵传递给MEX功能,prhs
是mxArray只用一个构件和与3个成员的矩阵。
我得到这个矩阵由mxGetData(prhs[0])
,但我不知道如何访问矩阵内的成员,并得到a
,b
,c
?