fdct_usfft_mex.cpp

来自「This directory includes matlab interface」· C++ 代码 · 共 40 行

/*
  Copyright (C) 2004 Caltech
  Written by Lexing Ying
*/

#include "mex.h"
#include "matrix.h"

#include "fdct_usfft.hpp"

#include "mexaux.hpp"

using namespace std;
using namespace fdct_usfft_ns;

//digital curvelet transform
extern void _main();

void mexFunction(int nlhs, mxArray* plhs[], int nrhs, const mxArray* prhs[])
{
  if(nrhs!=6)
	 mexErrMsgTxt("6 inputs required");
  if(nlhs!=1)
	 mexErrMsgTxt("1 outputs required");
  
  int m; mex2cpp(prhs[0], m);
  int n; mex2cpp(prhs[1], n);
  int nbscales; mex2cpp(prhs[2], nbscales);
  int nbangles_coarse; mex2cpp(prhs[3], nbangles_coarse);
  int allcurvelets; mex2cpp(prhs[4], allcurvelets);
  CpxOffMat x; mex2cpp(prhs[5], x);
  
  vector< vector<CpxOffMat> > c;  //vector<int> extra;
  fdct_usfft(m, n, nbscales, nbangles_coarse, allcurvelets, x, c);
  
  cpp2mex(c, plhs[0]);
  
  return;
}