如何从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 )

不同的视图不应该复制内存，所以这一切都是相当高效的。

补充一下：tvtk自然也提供了基于事件的API和图形编辑元素。