TA的每日心情 | 怒 2019-11-20 15:22 |
---|
签到天数: 2 天 [LV.1]初来乍到
|
EDA365欢迎您登录!
您需要 登录 才可以下载或查看,没有帐号?注册
x
' V, y( q9 I3 C+ G7 H8 k! T
用C编写mex程序
* N# M% S; [, z, W8 T 大家都知道,matlab是一种解释型的编程环境,也就是说,跟以前的basic一样,是读3 l; g1 c/ X. z& X. K# J( B! H
一句执行一句的。这样做可以很方便的实现编程过程中的交互,也免去了麻烦又耗时的
7 j+ u H& M* ~: }" r0 x# r. H3 F$ d编译过程。但凡事有一利必有一弊,matlab在执行时速度慢也就根源于此。在matlab里5 [* w1 u+ P; t0 Z: A% }
tic8 R! q/ h8 R, r6 p; {4 \
for i=1:100009 \) a7 u q( H$ W1 k
b(i)=a(10001-i);7 c' X1 N9 S- u+ q3 f4 M
end: |- L( e& C; m8 E' [- t
怎么样,是不是很慢?
! t( ^0 S2 l( y2 h 你的程序里如果再多几个这样的循环,运行速度就可想而知了。5 K, ~/ r& O+ S
上面程序的功能是将向量a里的数据逆序赋给向量b。下面的程序可以实现相同的功能
y' o% r% K) ^4 _( ` H6 e tic6 u) i1 ^% C; I* E) N2 T
b=a(10000:-1:1);
- |5 J, H( l8 K- S/ f9 X 为什么这个程序运行速度就这么快呢?这是因为matlab里的基础矩阵运算函数,像转
! _6 ~; I% t& M. o5 J置,复制等等,都是以二进制程序的形式存在的,运行起来速度当然比解释执行10000次6 Q8 y* p9 t* S5 B( b# t
所以编matlab程序时,应该尽量避免用循环语句,而使用等效的矩阵运算。虽然这样
' M0 Z! V; l, ?, y 但总是有的时候没法找到对应的矩阵运算来等效,或编出来的程序复杂得让人没法修
! F9 A1 K* A6 D 简单地说,mex程序就是根据一定的接口规范(mtlab提出的)编写的一个dll,matla
, t) A1 b6 {0 ]比如我编了一个mex函数,名字叫max2.dll,那么只要把这个dll所在的目录加到matlab
7 C+ |% m" f2 W) d/ Y的搜索路径里(用addpath),就可以像调用普通matlab函数一样来调用它了。因为把- w7 }' v: e c8 o8 L" B% |8 |
4 T8 Z- Q Z9 g循环体放到了二进制程序中,执行速度快得多。9 R1 n6 y1 G2 l) b2 B7 i! J4 L1 L
, ~, P0 `4 }3 j( ^' S
+ z8 n" K0 D: u! o4 ^
Mex文件既可以用c,也可以用fortran来编。因为我用的是c语言,所以下面的介绍都
( P4 e2 L2 N4 [. n1 Q! @
6 w' E3 P& l+ S& \" O/ c是用c语言编写mex文件的方法。如果你用的是fortran,请你自己去看Apiguide.pdf,里# [, V5 Q) ] D% w" Y
4 n- \. z4 o3 u) z$ L8 ?+ `* b! u; z [
面有详细说明。6 h/ b" {/ Q3 k! B/ |
& C1 \! l0 W5 `: B' i6 m& V# G前面说到通过把耗时长的函数用c语言实现,并编译成mex函数可以加快执行速度。这
5 f' b0 S' L, m2 i' m g Matlab5.1本身是不带c语言的编译器的,所以要求你的机器上已经安装有VC,BC或Wat7 ?' _* N. s$ }4 m5 @1 E1 @
com C中的一种。如果你在安装Matlab时已经设置过编译器,那么现在你应该就可以使用
3 h$ ^$ h o+ j2 n* U4 Fmex命令来编译c语言的程序了。如果当时没有选,只要在Matlab里键入 mex -setup6 \6 k' ^4 W, X
,就会出现一个DOS方式窗口,下面只要根据提示一步步设置就可以了。由于我用的是w7 I' n9 h0 w2 j+ ?
听说Matlab5.2已经内置了C语言的编译器,那么下面的这些可能就用不着了。可惜现
+ s8 V# x# j [3 X 需要注意的是,在设置编译器路径时,只能使用路径名称的8字符形式。比如我用的V
, g4 [ |, W: WC5装在路径 C:\PROGRAM FILES\DEVSTUDIO下,那在设置路径时就要写成:C:\PROGRA~1
. X: w& F5 u* x! f' M9 j7 M 这样设置完之后,mex就可以执行了。为了测试你的路径设置正确与否,把下面的程序
; B J; H: w. x% O: y存为hello.c。
1 K/ Q/ z! Q" h# j9 ]! A' Q. Y! t2 v: h( a/ d* p8 K( A$ [* |
) A* \4 z# H0 o#include "mex.h"! x4 ]# S2 w) k6 @' R3 }5 w! ]
: f+ O& M7 f* |! r: {6 Kvoid mexFunction(int nlhs, mxArray *plhs[], int nrhs, const mxArray *prhs[])) x" }' N, B$ P6 }: X4 g) z5 R
3 i9 l. B7 d0 ~1 Y) O, k7 c
4 ?2 U/ O1 c$ k, ?( c& s4 @3 a
{
4 v q: I- C! X% O8 x* _9 z0 y7 B5 J; s
mexPrintf("hello,world!\n");; x* P- @* z. d3 w
, |- S* c2 w+ v/ u/ j
}
* I; t% b0 O; B$ _% k6 n1 }: J0 f5 P! `' ]8 b' c4 }
5 F7 z& Q3 m, _9 H3 ?8 f2 ~( C8 D: ]
假设你把hello.c放在了C:\TEST\下,在Matlab里用CD C:\TEST\ 将当前目录改为C:\
0 ?% p0 F6 S3 [- {9 U3 k0 n% |
3 Y( X: X7 U5 @/ a( NTEST\(注意,仅将C:\TEST\加入搜索路径是没有用的)。现在敲:# K2 `/ [. A3 r- c, W
" @' ?( |4 a0 {$ N* d8 C, J
4 r; w5 q9 L) Y/ e' N mex hello.c
7 D/ Q+ W% e1 g9 _ K& d& o3 j, z' a' T# f* C1 R6 @
- }7 V8 i8 ?% |1 u4 [ r0 v
如果一切顺利,编译应该在出现编译器提示信息后正常退出。如果你已将C:\TEST\加
) I& k+ F& [+ X
) b( {7 H0 \3 B入了搜索路径,现在键入hello,程序会在屏幕上打出一行:+ p- c" }( ~, C; e& F! \, s0 {/ q
- m1 [2 D* M8 d/ z) H
/ f& z# f& b, Z- y R- T: i
hello,world!
. O( A0 f C' _5 D: r9 G& i1 H( \) o6 }7 R) [
* `1 r. D3 s6 l
看看C\TEST\目录下,你会发现多了一个文件:HELLO.DLL。( Q2 u' ?6 _+ A! t- r. Y# U2 i* D$ z
* T' I! p; l8 w7 u+ o) m/ T4 Y9 B
" O% j. T$ C$ h& \" q! z 这样,第一个mex函数就算完成了。怎么样,很简单吧。下一次,会对这个最简单的程" b3 [/ Y! P2 }; |. a [
1 X' Z7 \1 x) K, P/ X
序进行分析,并给它增加一些功能。
" Y' s4 V3 F2 i7 ?+ y7 z' K% B分析hello.c,可以看到程序的结构是十分简单的,整个程序由一个接口子过程
" Q O3 v# I) _/ g& EmexFunction构成。前面提到过,Matlab的mex函数有一定的接口规范,就是指这5 k2 C X. N$ s
nlhs:输出参数数目/ k1 c6 u2 n8 M/ R
plhs:指向输出参数的指针
" u( Y- |0 N/ h' I" y" A. vnrhs:输入参数数目
0 @' E' A. q2 D( m例如,使用 [a,b]=test(c,d,e) 调用mex函数test时,传给test的这四个参数分别是2,! _' P1 C; }3 z2 r4 X+ r8 D, g! f
plhs,3,prhs。其中:
, q2 u$ d& q+ V, q' o r yprhs[0]=c
4 U# j" B' _+ }% }: l1 O1 x( T
7 L5 I9 W% \* X& R. \- iprhs[1]=d5 }$ }8 D( G$ {* l2 P& U
2 s% [3 ?* z; D1 \3 |1 w6 G1 N7 L3 R8 y
prhs[2]=e- w" ^7 W8 z4 {0 T
I5 W8 Y2 N% g! Y- b4 d4 k) D- U
当函数返回时,将会把你放在plhs[0],plhs[1]里的地址赋给a和b,达到返回数据的目
$ r3 W5 \4 l6 `9 Q" s( \, J" \
% Y+ B- G( h/ k f# @ e) W: p的。% {- W$ m: |7 Y# n
) a6 x9 J) V* u
, |8 E5 _. m7 a6 j( ] 细心的你也许已经注意到,prhs和plhs都是指向类型mxArray类型数据的指针。# _/ U! \3 A! f
& J# D$ _1 H" p2 q h这个类型是在mex.h中定义的,事实上,在Matlab里大多数数据都是以这种类型存在。当
- w+ B7 V' Y- ^( L: y7 h0 G |( }* x/ {) O. v: V9 s, g& ]
然还有其他的数据类型,可以参考Apiguide.pdf里的介绍。
) c3 I7 n% g0 J( b. I* L' V+ l/ y/ e& h$ p2 J
# x8 ~/ t$ n6 w/ n, ^2 q$ @2 } 为了让大家能更直观地了解参数传递的过程,我们把hello.c改写一下,使它能根据输- e: M4 y; w7 s
7 ?/ S/ Y N& u& Q" [" y5 l) W入参数的变化给出不同的屏幕输出:
) |0 Q2 y( h4 Q6 G: f
6 \$ d. R- T4 z% X% k5 c
) V& D- u D' n+ A: F0 f//hello.c 2.0; ]8 Y0 y% n$ {' j& s# ^9 G+ y4 ]! V6 K3 c
7 e9 I* s S* u V: H) w+ `+ o
6 V' |1 o/ c2 K" o3 f9 M
#include "mex.h"/ ~( o4 m: k( o5 d2 n* @0 b" [
* h0 {+ m0 |9 \/ z+ E
void mexFunction(int nlhs, mxArray *plhs[], int nrhs, const mxArray *prhs[])
8 ?5 A8 ~4 H2 \/ C& d0 T# L# X4 D# o7 B1 B2 D
{# f- o( z+ ~8 e& p
7 g j8 A- C! U" S9 aint i;
! q2 J, ]: P$ n( j! J+ }9 d( T
" R, b$ R) _: Ui=mxGetScalar(prhs[0]);2 p3 q' W$ X) u
$ R, l+ A( ?( v; N7 E4 j3 S4 D4 Tif(i==1)1 s; Y( i+ }" l* t% L; {* [6 }& Y
' c c+ G9 x E! S5 ~( }: S mexPrintf("hello,world!\n");$ g7 S7 N% }6 ~, h. S( Z7 @
" f+ |0 Q Z& m; l; ?9 O
else9 B B6 W( M- K2 i2 R; _. K
F# I8 w6 m% x$ u+ c( T
mexPrintf("大家好!\n"); A& |* h6 v* }/ I# ^* i' V
) x( p. L* T- t3 H/ x; w}
6 z3 l7 M+ E. \+ w4 Z2 w
( z" y' L; P* W9 }4 {; z9 J, R4 M; U3 {: T, A
将这个程序编译通过后,执行hello(1),屏幕上会打出:- C, i+ x3 X2 W9 H. l: t
8 v2 X' n( f9 f" D: y. t
hello,world!6 [! ^* J' V0 M- I8 G, l3 Z4 [
: ?( t$ |, F9 T0 j2 [) v3 m 而hello(0)将会得到:
$ a: A# A+ d( m, x, ?3 \$ Z
" J1 H& ~' M& v/ c' ]8 r, ?大家好!
7 L$ L% K* b; J2 s3 }# M C- L+ w5 b0 q, ?) Q$ C8 @
. |/ Q: I0 r7 s% N6 T: U& D
现在,程序hello已经可以根据输入参数来给出相应的屏幕输出。在这个程序里,除了用% d. q% B, K* ]1 I2 K1 {" f* N
1 ?# r8 n' }; z- k5 v
到了屏幕输出函数mexPrintf(用法跟c里的printf函数几乎完全一样)外,还用到了一. d3 q! n# M" e! G+ q* s7 M( I7 d
) }# n0 U7 ?# \6 p M8 e6 j个函数:mxGetScalar,调用方式如下:
8 X" Y# r+ c6 A; T
5 R8 P0 @# @) i5 {* Yi=mxGetScalar(prhs[0]);/ e: b$ i7 `( g1 V# j; S
1 M( Q3 H) J" ~; p& O; c "Scalar"就是标量的意思。在Matlab里数据都是以数组的形式存在的,mxGetScalar的
& c2 N8 `$ \, d. h2 p" x9 `3 L
* q8 U/ i* ~" {5 ` q( L) F作用就是把通过prhs[0]传递进来的mxArray类型的指针指向的数据(标量)赋给C程序里9 F! i1 k" |" j+ e: r4 ?
2 |: u5 w. M, G5 \8 O
的变量。这个变量本来应该是double类型的,通过强制类型转换赋给了整形变量i。+ J8 b- x6 q7 t" G9 \. [' R/ ^
' V" A2 m# U: u2 n* H
I: a& V9 Z' y r0 T1 V. V. _; X 既然有标量,显然还应该有矢量,否则矩阵就没法传了。看下面的程序:; i; v4 U5 e/ x A
& ?! Y8 t' u7 {: a: {% I; X
0 ^+ S3 n/ s( u
//hello.c 2.15 R: g+ P6 [% r3 L5 _+ k( y
# [" U& O# R* h* _" D5 a. X+ h) H- ]1 y5 H" m0 S. `
#include "mex.h"
2 R5 K0 V( x# I
2 Y* o0 d2 F8 Z& t Zvoid mexFunction(int nlhs, mxArray *plhs[], n, U, v) ?' {
/ E, n+ y, H' [6 s+ A
int nrhs, const mxArray *prhs[])
6 Q4 ]$ |4 c+ s5 K: s0 N
2 l/ e1 \. w( x; c{
* W' o3 V' I, y5 N% ^
, H1 R- F: g# g# ]4 x1 ]int *i;" O3 P6 D8 W9 B$ t( n2 f
. q& @- A! L3 b7 ^% W, }i=mxGetPr(prhs[0]);( Z" r `$ N7 k3 g% J1 X
9 `- ^6 ~: y+ U5 M xif(i[0]==1)
2 r& C4 u( c! I$ S3 r ~# }. n! L$ G8 F# ^9 x( Z; P k" @
mexPrintf("hello,world!\n");
$ c* O$ w- V2 L
@, Q5 B8 J* i2 h3 gelse
2 E# }7 Z' |8 x7 Q
) M- j. Z8 l. z! p( M% P mexPrintf("大家好!\n");
: L z, [. i9 f) i I
& ]- V( G+ ~0 v$ \4 s& x8 Q# z}
! t$ J5 R* I# H( q( x3 E! p7 p5 |
9 }: p) m3 P7 I- |
- x) i ?9 W# y1 `5 M 这样,就通过mxGetPr函数从指向mxArray类型数据的prhs[0]获得了指向double类型的
/ A d2 |" R, M3 k$ r8 y' Q; j; v9 m0 F2 y+ o
指针。
* Q5 y3 a. l' t! c$ X$ z8 u5 w& k( E: ]
0 F) u3 J) r, u& E 但是,还有个问题,如果输入的不是单个的数据,而是向量或矩阵,那该怎么处理呢
: [6 x4 q0 N) x9 w/ O* x
2 k2 a9 w( d. _9 ^! h?通过mxGetPr只能得到指向这个矩阵的指针,如果我们不知道这个矩阵的确切大小,就
! C& a8 M0 o, a* d& J# j9 f
8 g2 ]4 c$ J/ ^没法对它进行计算。& J- I S" {5 |5 }
+ w7 M; O d5 a& M7 p! u% d; `. i
, s- w0 y7 |7 l) t/ @6 _ 为了解决这个问题,Matlab提供了两个函数mxGetM和mxGetN来获得传进来参数的行数
" e% m: U- ] y ]4 g8 u4 R7 P
2 W' d, |/ t& e6 j和列数。下面例程的功能很简单,就是获得输入的矩阵,把它在屏幕上显示出来:% ?. H* g5 k" e! g9 F$ l- _/ [
1 g+ Y% u; }3 D8 X! k2 V/ l2 l
//show.c 1.0" z. X. X+ v# b
4 L. J2 K! I& ]7 Q8 I: T" k; W
, V6 K/ w" F9 ^#include "mex.h"
; B- }# c6 L4 x+ r0 C; ~
- t3 v- s3 y$ x0 Y+ w ~* X% f* [void mexFunction(int nlhs, mxArray *plhs[], int nrhs, const mxArray *prhs[])
5 ~. M G6 f6 `4 F. t3 N
% I0 x. h* b/ P2 H8 o4 ~+ J5 D' K4 l4 B5 T, H1 I& u
{! B& J0 K) C7 |! p R
$ k' \5 |2 G+ v/ F" F6 E: C
double *data;
) M% N, B3 e b- H
- {% S3 ^8 `1 h' y$ Cint M,N;3 j1 h: q3 K, d& I. A
- A' ~5 g. T" \% N8 e' l4 f7 d# u
int i,j;* V7 R8 b% h7 @7 |, O6 ?& B
1 |$ c T1 Q; q3 } Q0 t
) |. w% I& A; x2 ?5 L# N; s; w
?5 m1 x: x6 w# h, `7 z- @( ~data=mxGetPr(prhs[0]); //获得指向矩阵的指针6 h5 [; X6 l; I8 H* L
, C0 I% e9 m0 s6 H( G7 sM=mxGetM(prhs[0]); //获得矩阵的行数
2 y% \1 M6 h+ u+ A: b& u* D( K8 [ I- ]4 r9 M+ s
N=mxGetN(prhs[0]); //获得矩阵的列数, p. N, g0 k7 B' H
- h3 k9 I* }, r7 x! H" n) H1 x* L
- E& v$ d# f! p: [9 V5 \% p
for(i=0;i<M;i++)9 \0 c3 z9 I/ ]* p4 H' S" X
b5 _$ j1 S* E4 O h, ?{
5 M: j, @9 v5 ^3 u6 J$ n' m4 T# P1 F/ d. J1 u1 F; I5 C
for(j=0;j<N;j++)
; Q* R2 f, u8 Z7 V+ I4 D7 F9 O8 ~
]% `# [+ K1 r1 J: m! L- ~ mexPrintf("%4.3f ",data[j*M+i]);
' M2 z. o8 _* B' g5 e9 o; q
" W5 F9 S3 s. p" ~0 Q# ~* Y: U; x8 P mexPrintf("\n");( H2 h, g+ W! v( g
$ y! @6 C7 g0 r1 j" Q& ^/ a4 [- ]}
# I+ p9 d& g' m! y" j! {, ^/ b' \8 v/ l/ B- G; N$ f2 k
}; n0 W# r; ?- R: o
V* |! h" S x3 X* |
. T I4 B6 l o* U6 x2 c: G 编译完成后,用下面的命令测试一下:
) O& s+ [7 L8 g u: l- @; L/ [# K- O4 _) y
0 m; _4 a' C* y% p0 ~6 N a=1:10;6 X' i/ f3 d9 H4 Q
4 {! g7 X8 C4 [4 s% M+ F7 R
b=[a;a+1];
" U' S) H. N$ W4 }/ ?1 n! C7 |: S! ] v9 n/ z4 z! r
show(a)# X$ G2 k I' W& c- ]0 E6 w
( C: @0 \: u0 k5 N$ D- Y0 L show(b)$ l$ {3 } R+ l) H/ J
5 R. _2 b. b/ w* h8 @ a3 v# h
V/ P4 d- q5 U, c* w+ { ]3 ?/ g6 p 需要注意的是,在Matlab里,矩阵第一行是从1开始的,而在C语言中,第一行的序数/ I8 I( R( e( J: J* Q8 d7 Z
% }4 z7 m. |5 u为零,Matlab里的矩阵元素b(i,j)在传递到C中的一维数组大data后对应于data[j*M+i]
, Z, f# Z; b: C/ u# ?" M2 K( M' L. y
。
4 C; G2 X4 T$ S, Z& S+ q! _# e* ~
输入数据是在函数调用之前已经在Matlab里申请了内存的,由于mex函数与Matlab共用同
( |% O* ?& X1 ?+ C3 ^8 n# R% l$ v一个地址空间,因而在prhs[]里传递指针就可以达到参数传递的目的。但是,输出参数
$ @: ^, Z9 \. ^8 V6 V却需要在mex函数内申请到内存空间,才能将指针放在plhs[]中传递出去。由于返回指针) \8 R1 P0 k% Y9 L
类型必须是mxArray,所以Matlab专门提供了一个函数:mxCreateDoubleMatrix来实现内3 [2 {4 I/ N4 ?
存的申请,函数原型如下: n% \* W. a2 a
mxArray *mxCreateDoubleMatrix(int m, int n, mxComplexity ComplexFlag)! }' F6 {2 @1 r" C+ o0 @) A: u2 J
m:待申请矩阵的行数
/ r% F+ ?7 b- I( h9 N4 xn:待申请矩阵的列数4 e. n) f( x, J/ }2 I
为矩阵申请内存后,得到的是mxArray类型的指针,就可以放在plhs[]里传递回去了。但
A9 M* W8 _0 B7 ^- X. [7 O是对这个新矩阵的处理,却要在函数内完成,这时就需要用到前面介绍的mxGetPr。使用
( @; \/ b+ ?9 u: O7 {mxGetPr获得指向这个矩阵中数据区的指针(double类型)后,就可以对这个矩阵进行各
; d5 \7 L3 z! N( |, T2 p种操作和运算了。下面的程序是在上面的show.c的基础上稍作改变得到的,功能是将输
. O4 V {' Z8 \0 C& w# `# l
& j3 g/ r/ E* A5 F8 j$ q//reverse.c 1.0& F8 s) e0 H4 \
: O9 p5 g. H& u0 Q0 m) e- |
; {( p: V- f& D6 y% a#include "mex.h"
! y' n7 d6 Q3 @5 g% @) G3 N/ q4 u# W% |* s) B. H2 `1 s
void mexFunction(int nlhs, mxArray *plhs[],
* m- f1 b# C1 D" q
' w* y8 e, b9 g9 W, ]) y int nrhs, const mxArray *prhs[]). j( W4 X; e$ ~; h& }! h
* E- S9 h! \3 X9 u: i
{
2 ?% u' b8 I8 C8 f8 ? t& {$ v$ B
, o/ F! x* v* ^4 J5 fdouble *inData;! p* Z2 L* c$ }: n0 g# `) w8 _1 Q
9 z' q0 X; N1 {0 F& H' \, `. J
double *outData;
; \/ t- [# ]' K+ |4 O1 @& `: k
4 s$ r+ | r' s9 W" iint M,N;. ~; a$ Q" l" T4 W, | n* t
1 i/ V! C& W' P' h
int i,j;; w/ Z" W% q/ ?# L- t
/ g, p, m+ _8 L- }% b$ N4 G; p- g& p9 W Y" Q* t" k, `2 B
5 M' q% t) p' s' _7 F; w
inData=mxGetPr(prhs[0]);- _" j2 i6 ?. E
+ z" f" T4 I; u F
M=mxGetM(prhs[0]);! G, X& Q- y: C
% j M6 O) U8 Q K! R- `* u
N=mxGetN(prhs[0]);: f$ W; c; H/ p! c$ f
. f0 s1 n( E$ @% V" B( J9 L, f$ L
w5 D x1 J# z( c$ Q" Vplhs[0]=mxCreateDoubleMatrix(M,N,mxREAL);
: w5 q. X0 D+ T6 L+ t, x( e
3 J" Z# k$ R# A6 E8 QoutData=mxGetPr(plhs[0]);
* c' {, h! |2 S( D3 {4 u: G# f
% Q% F! v7 n1 G0 ]
1 O+ y1 Z0 @; I9 `- C- Ifor(i=0;i<M;i++)
9 k7 v6 R% r1 }) \1 |: G5 V
2 o8 }$ H. I3 |& e/ K* z; C for(j=0;j<N;j++)
; I4 f* A& k4 c* V* n! h+ ? t: X9 T& b
outData[j*M+i]=inData[(N-1-j)*M+i];
. |4 l3 q/ ?. D4 ?9 r5 b: f( N1 N: q4 a: x7 e! S/ s. J
}) y M. u2 `4 V% e# @& r, Q
3 m& L" `7 S+ W- B' F6 X* U2 d( B6 A' }
当然,Matlab里使用到的并不是只有double类型这一种矩阵,还有字符串类型、稀疏矩
4 ?' X" g# }. K3 n6 p% T! A: {) e$ v+ X( k
阵、结构类型矩阵等等,并提供了相应的处理函数。本文用到编制mex程序中最经常遇到
9 D8 f1 z* w3 m% l3 P
# l) y" A) N" Y! F7 n的一些函数,其余的详细情况清参考Apiref.pdf。- W1 @4 T: N$ Z: v- a# c
通过前面两部分的介绍,大家对参数的输入和输出方法应该有了基本的了解。具备了这 P9 X# ]9 x t8 Z+ k) X- z8 L
些知识,就能够满足一般的编程需要了。但这些程序还有些小的缺陷,以前面介绍的re- E' k7 S! L/ m0 w* T
由于前面的例程中没有对输入、输出参数的数目及类型进行检查,导致程序的容错性很' U& B/ L$ q6 S3 y4 e
#include "mex.h"7 s+ t% S v. g2 i* N0 n0 Y: T- J
void mexFunction(int nlhs, mxArray *plhs[],1 n8 l# i/ Z s1 g1 J3 V
int nrhs, const mxArray *prhs[])& \& [! r# ~, ~6 v
{$ @/ D$ b- E& K* W( g8 e( l! N
double *inData;- C G* s! h0 Y( M& z; M, \) m$ ]
double *outData;' L0 J& g8 m2 U, E
int M,N;
: A, r4 o' Z6 q- o+ @# k$ r, L1 G//异常处理, ]. `5 n) Q3 H& D* K; `
if(nrhs!=1)
8 p: m: S7 e8 U$ H' A/ n8 Z7 P& Q5 {
mexErrMsgTxt("USAGE: b=reverse(a)\n");5 r7 I; q) D/ P9 Z1 w
5 h, `* |9 Z e K
if(!mxIsDouble(prhs[0]))# M$ {! K4 L' X2 @
* s( R @7 j6 [
mexErrMsgTxt("the Input Matrix must be double!\n");9 P! d. V C. T. A$ `4 ]6 I
% V5 }# Q% f" A0 y* |+ x3 c
, f: q6 U7 V0 T: UinData=mxGetPr(prhs[0]);
; f% _6 z3 J& L# x& s, G; r; A
* a, X7 z% G# O6 |M=mxGetM(prhs[0]);4 x1 s1 D( J; c( [ m3 d3 v
& q9 T3 \7 F7 W" _5 X4 j: f* GN=mxGetN(prhs[0]);5 c- h8 Z: O3 j! e# U8 ?8 Q
8 D5 I1 r1 W) y8 x( v
$ { V [) s4 ~7 @plhs[0]=mxCreateDoubleMatrix(M,N,mxREAL);
& R y; c8 e3 o4 u
( t7 |8 c* }3 p( T5 u6 VoutData=mxGetPr(plhs[0]); [: X. J( k& w6 \
9 n" [$ W v2 e0 G% i" M. ?1 [6 Z" ?) U6 N0 E+ ^
for(i=0;i<M;i++)- M" a9 z9 V6 o7 v1 U
6 O/ b. z" S: `3 I0 i6 ^9 b2 ^; I for(j=0;j<N;j++)
y2 a% I$ L" F% T- {3 A; q9 ?
/ @5 y* E6 K8 z* v outData[j*M+i]=inData[(N-1-j)*M+i];
3 s% T5 t+ H5 A0 I# @) G8 {( c( O8 m/ e' o
}
, J3 }7 M0 R/ E3 _# O) S- j; w& i! p2 P0 \- h: E
1 J( A/ L4 ]3 L; r在上面的异常处理中,使用了两个新的函数:mexErrMsgTxt和mxIsDouble。MexErrMsgT3 A9 B% S _+ K
, ?7 ?3 H4 g" Fxt在给出出错提示的同时退出当前程序的运行。MxIsDouble则用于判断mxArray中的数据
8 q% x) U/ m* M# t' N
$ o2 v* B4 u. V* t( T是否double类型。当然Matlab还提供了许多用于判断其他数据类型的函数,这里不加详; n' ]$ d- `: h$ @, J
% G/ U) n4 Q1 f3 {
述。
% Z) R; ]2 E+ c+ L
; s2 a9 e/ U% t$ ^3 S5 n, f7 n; j/ E; P5 c$ d1 Q
需要说明的是,Matlab提供的API中,函数前缀有mex-和mx-两种。带mx-前缀的大多是对
, {. }5 J* s) m: L5 v/ z1 T- T, u( g/ `# E7 l/ S
mxArray数据进行操作的函数,如mxIsDouble,mxCreateDoubleMatrix等等。而带mx前缀* ]$ A2 v: M( J+ V( b
" N6 a9 V; f: n# c K2 n* b- Q4 u
的则大多是与Matlab环境进行交互的函数,如mexPrintf,mxErrMsgTxt等等。了解了这
5 a6 i0 W* [4 a( U4 M2 J
2 ^( C2 X3 ~+ E) O一点,对在Apiref.pdf中查找所需的函数很有帮助。
' c* r# Y3 L$ l. F4 U& a
& i+ z! b6 m$ S0 y: u) j
7 M3 `$ O; d- W! ?+ y/ w Z# Z至此为止,使用C编写mex函数的基本过程已经介绍完了。下面会在介绍几个非常有用的
# |/ _& v4 Z5 L2 J$ T3 v6 ?1 W8 i+ k* W; z' X; r* o$ r4 B L
函数调用。如果有足够的时间,也许还会有一个更复杂一些的例程。
# g+ c0 j. f( M& g
' ?, A' D1 G% h/ `- U$ t4 F* p" q; C我们之所以使用Matlab,很重要的考虑是Matlab提供了相当丰富的矩阵运算函数和各) [6 J( }4 u* B) Y0 w q
种toolbox。在编制mex函数时,有时我们也会遇到一些操作,在Matlab下,只需要一个
8 M) u$ R. Q: M% m) E7 F. w为了在mex函数里调用Matlab命令,我们就需要用到一个函数mexCallMATLAB,原型如下:* T0 T; X: L; f" y) v. E
int mexCallMATLAB(int nlhs, mxArray *plhs[], int nrhs, mxArray *prhs[],
, u& [, f" D2 k. D; S const char *command_name);/ V1 b* \% N; @6 l3 \$ R
有了前面的基础,使用这个函数就显得十分容易了。下面给出一个例程,功能是将输入8 C5 p ]( x" i% C* [! e5 L
#include "mex.h"2 h' Q$ Q6 j5 g4 C! g. k; J
void mexFunction(int nlhs, mxArray *plhs[],
! N* Q: ^3 m# f3 _; }5 t) g int nrhs, const mxArray *prhs[])8 ^! g0 x! Q* p* \) i9 J
{7 M, E- ^/ [5 e
2 T; E/ @7 E3 `1 u' y
double *inData;
7 R8 _$ ~7 E8 {5 ~" x+ v
8 o2 a: Q9 a5 nmxArray *IN[1];+ N" l: g9 P8 k5 a* f
7 g9 I; i! R- s' i vmxArray *OUT[1];3 M) G# ?( N% ^, }
5 l* ~/ Q$ F+ tdouble *outData;0 D8 {; i0 f5 x: V, k6 m) Y1 {
5 X9 W- K( }) f, J- [$ y
int M,N;4 A4 R2 B5 k T8 R- R
8 M. k/ z; a9 P7 c4 @int i,j;
; l4 c# a/ b* D
( z0 e G2 `" P1 [; U, z7 }% q, D3 N3 [; x: `0 i) \
//异常处理
( b1 W# P- N! ?" M, O7 B: n# s/ _) T: a0 J
if(nrhs!=1)
# I* u8 s$ A7 e) X* K; N4 I
% }+ P' t3 m I. f. y6 B s mexErrMsgTxt("USAGE: b=rot(a)\n");
6 n/ R7 I" J1 u0 b2 n$ x1 G1 N) c
- p" D. _9 ~! T/ |if(!mxIsDouble(prhs[0]))
& [& H, ^- \. n/ @- [1 G9 e$ p
5 T0 G: ^& M* Y8 f5 k! C mexErrMsgTxt("the Input Matrix must be double!\n");1 K0 E( |2 x* O, E- V
% l7 m1 D$ w/ J/ T
8 ^' F' f: [% j8 {//计算转置
) D5 G! C% O8 Q3 z4 `/ c. Q# g- }8 B: J, F, l
if(mexCallMATLAB(1,OUT,1,prhs,"'"))# \ r3 p! X8 J2 l, ~* u: U* J, ?* p4 N
b; h0 F# F# T0 Q4 L mexErrMsgTxt("Error when compute!\n");
7 o) ?0 [! _" g' o) _% M1 x0 T
/ N0 X9 F) _: R( ^' Y" f! W- i4 k$ f; j( e, t6 W
//根据输入参数数目决定是否显示
) ]. C. a8 ?9 h" _7 m: r( e
! Y1 ?3 \4 W% d4 R. z$ Qif(nlhs==0)% U" g6 Q" M( h$ T0 W
. s. O5 f: o7 Y) l9 _
mexCallMATLAB(0,IN,1,OUT,"disp");
0 g( I0 l) p$ b
?. K |+ H$ U# Y" eelse" Y/ X- K; I* e
! Q7 b0 g, m4 [5 w+ m. M* T1 f plhs[0]=OUT[0];2 r8 Q" A9 X. D
8 Y2 |6 G# [4 Q
} |
|