Function Vmtk_centerlines

Code/Source/PythonAPI/Vmtk_PyModule.cxx:374–502

");

static PyObject *
Vmtk_centerlines(PyObject* self, PyObject* args, PyObject* kwargs)
{
  //std::cout << "========== Vmtk_centerlines ==========" << std::endl;
  auto api = PyUtilApiFunction("OO!O!|O!O!", PyRunTimeErr, __func__);
  static char *keywords[] = {"surface", "inlet_ids", "outlet_ids", "split", "use_face_ids", NULL};
  PyObject* surfaceArg;
  PyObject* inletIdsArg;
  PyObject* outletIdsArg;
  PyObject* splitArg = nullptr;
  bool splitCenterlines = true;
  PyObject* useFaceIdsArg = nullptr;
  bool useFaceIds = false;

  if (!PyArg_ParseTupleAndKeywords(args, kwargs, api.format, keywords, &surfaceArg, &PyList_Type, &inletIdsArg, 
        &PyList_Type, &outletIdsArg, &PyBool_Type, &splitArg, &PyBool_Type, &useFaceIdsArg)) {
      return api.argsError();
  }

  // Check inlet IDs.
  //
  // [TODO:DaveP] Add this as a util function.
  //
  std::vector<int> sources;
  int numInletIds = PyList_Size(inletIdsArg);
  if (numInletIds == 0) {
    api.error("The 'inlet_ids' argument is empty.");
    return nullptr;
  }
  for (int i = 0; i < numInletIds; i++) {
      auto item = PyList_GetItem(inletIdsArg, i);
      if (!PyLong_Check(item)) {
          api.error("The 'inlet_ids' argument is not a list of integers.");
          return nullptr;
      }
      int id = PyLong_AsLong(item);
      //std::cout << "[Vmtk_centerlines]   ID: " << id << std::endl;
      sources.push_back(id);
  }

  // Get the target IDs.
  //
  int numOutletIds = PyList_Size(outletIdsArg);
  std::vector<int> targets;
  if (numOutletIds == 0) {
    api.error("The 'outlet_ids' argument is empty.");
    return nullptr;
  }
  for (int i = 0; i < numOutletIds; i++) {
      auto item = PyList_GetItem(outletIdsArg, i);
      if (!PyLong_Check(item)) {
          api.error("The 'outlet_ids' argument is not a list of integers.");
          return nullptr;
      }
      int id = PyLong_AsLong(item);
      targets.push_back(id);
  }