2016-08-28 51 views
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

回答

0

简单地

x0=mxGetPr(prhs[0]); 

mxGetPr返回double指针。

然后你就可以通过

a=x0[0]; 
b=x0[1]; 
c=x0[2]; 

访问的x0成员如果你的数据类型是不同的double,您可以使用

type variable = (type *) mxGetData(prhs[0]); 
相关问题