def set_polydata_vertices(polydata, vertices):
"""Set polydata vertices with a numpy array (ndarrays Nx3 int).
Parameters
----------
polydata : vtkPolyData
vertices : vertices, represented as 2D ndarrays (Nx3)
"""
vtk_points = Points()
vtk_points.SetData(numpy_support.numpy_to_vtk(vertices, deep=True))
polydata.SetPoints(vtk_points)
return polydata
def numpy_to_vtk_points(points):
"""Convert Numpy points array to a vtk points array.
Parameters
----------
points : ndarray
Returns
-------
vtk_points : vtkPoints()
"""
vtk_points = Points()
vtk_points.SetData(
numpy_support.numpy_to_vtk(np.asarray(points), deep=True))
return vtk_points
