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