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