comp_wfac.cc

来自「linear time－frequency toolbox」· CC 代码 · 共 39 行

#include <octave/oct.h>

extern "C"
{
  void wfac(double *g, const int L, const int R,
	    const int a, const int M, double *gf);

  int gcd(const int a, const int b,int *r, int *s);

}

DEFUN_DLD (comp_wfac, args, ,
  "Computes window factorization.\n\
  Usage: c=comp_wfac(g,a,M);\n\
  Yeah.")
{

  ComplexMatrix g = args(0).complex_matrix_value();
  const int a = args(1).int_value();
  const int M = args(2).int_value();
  const int L = g.rows();
  const int R = g.columns();

  const int b = L/M;

  int h_a, h_m;

  const int c=gcd(a, M,&h_a, &h_m);
  const int p=a/c;
  const int q=M/c;
  const int d=b/p;

  ComplexMatrix gf(p*q*R,c*d);
  
  wfac((double*)g.data(),L,R,a,M,(double*)gf.data());

  return octave_value (gf);
}