Function Math_fft

Code/Source/PythonAPI/Math_PyModule.cxx:131–174

");

static PyObject *
Math_fft(PyObject *self, PyObject *args)
{
  auto api = PyUtilApiFunction("Oii", PyRunTimeErr, __func__);
  PyObject *pointsArg;
  int nterms = 0;
  int numInterpPoints = 0;

  if (!PyArg_ParseTuple(args,api.format, &pointsArg, &nterms, &numInterpPoints)) {
      return api.argsError();
  }

  // [TODO:DaveP] check that other arguments are valid.

  // Get an array of points from the pointsArgs list.
  int dim = 2;
  auto pts = GetPointsFromList(api, pointsArg, dim, "points");
  if (pts == nullptr) {
      return nullptr;
  }
  int nlistpts = PyList_Size(pointsArg);

  // Perform the fft operation.
  auto mathObj = cvMath();
  double **terms = NULL;
  if (mathObj.FFT(pts, nlistpts, numInterpPoints, nterms, &terms) == SV_ERROR) {
       mathObj.deleteArray(pts,nlistpts,dim);
       api.error("Error calculating the fft.");
       return nullptr;
  }

  // Create result list.
  PyObject *pylist = PyList_New(nterms);
  for (int i = 0; i < nterms; i++) {
      PyObject* rr = PyList_New(dim);
      PyList_SetItem(rr,0,PyFloat_FromDouble(terms[i][0]));
      PyList_SetItem(rr,1,PyFloat_FromDouble(terms[i][1]));
      PyList_SET_ITEM(pylist, i, rr);
  }

  mathObj.deleteArray(pts,nlistpts,dim);
  mathObj.deleteArray(terms,nterms,dim);
  return pylist;
}

//--------------------
// Math_inverse_fft