RectilinearGridSource从tvtk.RectilinearGrid()

Question

我试图从tvtk.RectilinearGridSource对象构建tvtk.RectilinearGrid，以便将其添加到mayavi.engine中。

我以前经常这样做：

import mayavi.mlab as mlab

r = tvtk.RectilinearGrid()
r.point_data.scalars = data.ravel()
r.point_data.scalars.name = 'Temperature'

d = mlab.pipline.add_dataset(r)

但相反，我宁愿这样称呼它：

from mayavi.api import Engine
e = Engine()
e.start()
s = e.new_scene()
src = tvtk.RectilinearGridSource()

然后将src链接到r，即，使用前面定义的RectilinearGrid。

有办法这样做吗？

Answer 1

我找到了一个答案：

r = tvtk.RectilinearGrid()
r.point_data.scalars = data.ravel()
r.point_data.scalars.name = 'Temperature'

from mayavi.sources.vtk_data_source import VTKDataSource
src = VTKDataSource(data=r)

e.add_source(d)