作者lihgong (人生,是一句引用句)
看板MATLAB
标题[心得] 用 Matlab 写 MEX 函数加速 vol.4
时间Thu May 17 09:13:45 2007
最後一篇, 这篇会提出一个简单的 MEX file 的 Framework
废话不多说, 直接看 example, 程式码才是最好的说明书
这个 example 实际上是 vol.2 和 vol.3 的合体
如果前面的程式都看懂了, 这个程式应该不会有任何困难
建议把 code 复制到 Matlab 的 Editor 来看, 比较不会伤眼睛 :p
范例程式接受一个 2-D 的 input, 把每个元素 +1 以後输出
output = input + 1;
在程式码里, 我把程式区分了很多块
以一个标准的 MEX 档来说, 大概会有这几个部分
* 取得输入参数的资料 (eg. dimension)
* 配置输出参数 (Output Allocation)
* 根据输入, 计算输出 (Data Processing)
把下面的程式看懂, 碰到要写 MEX 的时候
直接套下面这个范例, 应该可以省不少时间 :)
#include "mex.h"
#include <math.h>
#include <stdio.h>
// Program framework for input, output and processing
// usage:
// input = zeros(2, 10)
// input(1, :) = 1:10
// input(2, :) = 10:-1:1
// mex test3.c
// a = test3(input)
// note: type all commands above in Matlab Command Window
void mexFunction(
int nlhs, mxArray *plhs[],
int nrhs, const mxArray *prhs[])
{
int i, j, k;
// input
int input_dim_x;
int input_dim_y;
double *in; // pointer to process content of the input
// output
int output_dimension[2]; // note: you should specify output dimension
int output_dim_x;
int output_dim_y;
double *out;
/* -------------------------------- */
/* NECESSARY (input processing) */
in = mxGetPr(prhs[0]); // get data pointer
// mx: Matrix
// P: Pointer
// r: real
// mxGetPr() has a counterprt mxGetPi();
input_dim_x = mxGetM(prhs[0]);
input_dim_y = mxGetN(prhs[0]);
// print some message about input data
printf("nrhs: %d\n", nrhs);
printf("mxGetM(prhs[0]): %d\n", mxGetM(prhs[0]));
printf("mxGetN(prhs[0]): %d\n", mxGetN(prhs[0]));
// print input content
for(i=0; i<input_dim_x; i++) // x
for(j=0; j<input_dim_y; j++) // y
// notice: data type is "float", you shall use "%f" insted of "%d"
printf("%f\n", in[i + j*input_dim_x]);
/* -------------------------------- */
/* NECESSARY (output allocation) */
// specify output matrix's dimension
output_dim_x = input_dim_x;
output_dim_y = input_dim_y;
output_dimension[0] = output_dim_x; // mxCreateNumericArray() required
output_dimension[1] = output_dim_y;
// Allocate the output matrix
plhs[0] = mxCreateNumericArray(
2,
output_dimension,
mxDOUBLE_CLASS,
mxREAL);
// Get output matrix's data pointer
out = mxGetPr(plhs[0]); // just point at the start
/* -------------------------------- */
/* NECESSARY (data processing) */
// Copy and add 1
for(i=0; i<output_dim_x; i++)
for(j=0; j<output_dim_y; j++)
{
int temp = i + j*input_dim_x;
out[temp] = in[temp]*2;
}
}
--
※ 发信站: 批踢踢实业坊(ptt.cc)
◆ From: 140.113.128.237
1F:推 kaishi:推荐这篇文章 我也在 140.113.128.XXX 哈 05/17 11:24
2F:推 Evanny:范例似乎是将输入乘上两倍 而不是加一喔 考我有没有认真看 05/17 23:19
3F:→ Evanny:真巧妙 刚好过了一年 05/17 23:20