python - 如何从 tvtk PolyData 实例中提取顶点数据？

Question

我正在尝试从 tvtk.PolyData 实例中提取顶点数据。例如:

from numpy import array
from tvtk.api import tvtk
from mayavi.sources.vtk_data_source import VTKDataSource
from mayavi.modules.surface import Surface
from mayavi import mlab

pointArr=array([[0,0,0],[1,0,0],[1,1,0],[0,1,0],[0,1,1],[1,1,1]],dtype="float64")
faceArr=array([[0,1,2,3],[2,3,4,5]])
faces = tvtk.PolyData()
faces.points=pointArr
print faces.points
faces.polys=faceArr
faces.point_data.scalars = pointArr[:,2]
faces.point_data.scalars.name = 'Height'
src = VTKDataSource(data = faces)
mlab.pipeline.surface(src, opacity=1.)
mlab.show()

这会打印

vtkPoints (0x9811398)
  Debug: Off
  Modified Time: 838816
  Reference Count: 2
  Registered Events: 
    Registered Observers:
      vtkObserver (0x98112b8)
        Event: 33
        EventName: ModifiedEvent
        Command: 0x9811688
        Priority: 0
        Tag: 1
  Data: 0x9811480
  Data Array Name: Points
  Number Of Points: 6
  Bounds: 
    Xmin,Xmax: (0, 1)
    Ymin,Ymax: (0, 1)
    Zmin,Zmax: (0, 1)

这里发生了什么？我期望看到 NumPy 数组:

[[ 0.  0.  0.]
 [ 1.  0.  0.]
 [ 1.  1.  0.]
 [ 0.  1.  0.]
 [ 0.  1.  1.]
 [ 1.  1.  1.]]

Answer 1

tvtk 的重点是处理 python 对象(包括 numpy 数组)和 VTK 对象保存类之间的转换。如果您研究 faces.points，您会发现它实际上是一个名为 Points 的 tvtk 类实例，但您可以使用轻松获取 numpy 表示

np.array( faces.points )

不同的 View 不应该复制内存，所以这一切都非常有效。

编辑:tvtk 自然也提供了基于事件的 API 和图形编辑元素