Function GetPointsFromList

Code/Source/PythonAPI/Math_PyModule.cxx:76–110

------------------- GetPointsFromList ------------------- Create an array of points from a Python list. [TODO:DaveP] should really use exceptions here.

// [TODO:DaveP] should really use exceptions here.
//
static double **
GetPointsFromList(PyUtilApiFunction& api, PyObject* listArg, int dim, const std::string& argName)
{
  // First check that the listArg is a list.
  if (!PyList_Check(listArg)){
      api.error("The " + argName + " argument is not a list.");
      return nullptr;
  }

  int listSize = PyList_Size(listArg);
  if (listSize == 0) {
      api.error("The " + argName + " argument is empty.");
      return nullptr;
  }

  // Convert the Python list to a double array.
  //
  double **points = cvMath().createArray(listSize, dim);
  for (int i = 0; i < listSize; i++) {
      PyObject *temp = PyList_GetItem(listArg,i);
      if (temp == nullptr) {
          api.error("The " + std::to_string(i) + "th element of the " + argName + "argument is not defined.");
          return nullptr;
      }
      if (PyList_Size(temp) != dim) {
          api.error("The " + std::to_string(i) + "th element of the " + argName + " argument list != "+std::to_string(dim)+".");
          return nullptr;
      }
      for (int j = 0; j < dim; j++) {
          points[i][j] = PyFloat_AsDouble(PyList_GetItem(temp,j));
      }
  }

  return points;
}

//////////////////////////////////////////////////////
//          M o d u l e  F u n c t i o n s          //