Function Vmtk_cap

Code/Source/PythonAPI/Vmtk_PyModule.cxx:194–255

");

static PyObject *
Vmtk_cap(PyObject* self, PyObject* args,  PyObject* kwargs)
{
  auto api = PyUtilApiFunction("O|O!", PyRunTimeErr, __func__);
  static char *keywords[] = {"surface", "use_center", NULL};
  PyObject* surfaceArg;
  PyObject* useCenterArg = nullptr;

  if (!PyArg_ParseTupleAndKeywords(args, kwargs, api.format, keywords, &surfaceArg, &PyBool_Type, &useCenterArg)) {
      return api.argsError();
  }

  // Set cap type.
  //
  // This determines whether to cap regularly or cap with a point each hole center.
  //     0 - regular,
  //     1 - point in center
  //
  bool radialFill = false;
  if ((useCenterArg != nullptr) && PyObject_IsTrue(useCenterArg)) {
    radialFill = true;
  }

  // Get the vtkPolyData object from the Python object.
  //
  auto surfPolydata = GetVtkPolyData(api, surfaceArg);
  if (surfPolydata == nullptr) {
      return nullptr;
  }
  cvPolyData cvSurfPolydata(surfPolydata);

  // Perform cap operation.
  //
  std::vector<int> centerIDs;
  cvPolyData *cappedSurface = NULL;
  int numIds, *ids;
  if (sys_geom_cap(&cvSurfPolydata, radialFill, centerIDs, &cappedSurface) != SV_OK) {
    api.error("Error capping model.");
    return nullptr;
  }

/*
  // [TODO:DaveP] what are the IDs that were not found?
  if (numIds == 0) {
      api.error("No cap IDs were found.");
      return nullptr;
  }

  // Build return list of IDs.
  //
  PyObject* pyList = PyList_New(numIds);
  for (int i = 0; i < numIds; i++) {
      PyObject* pint = Py_BuildValue("i", ids[i]);
      PyList_SetItem(pyList, i, pint);
  }

  delete [] ids;
  return pyList;