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