python - 如何更改 Mayavi 中线图的管半径？

Question

我正在尝试动态修改 Mayavi2 中 3D 线图的管半径。例如

from traits.api import HasTraits, Float, Instance, on_trait_change
from traitsui.api import View, Item, Group
from mayavi.core.api import PipelineBase
from mayavi.core.ui.api import MayaviScene, SceneEditor, MlabSceneModel
import numpy

def curve():
    n_mer, n_long = 6, 11
    pi = numpy.pi
    dphi = pi / 1000.0
    phi = numpy.arange(0.0, 2 * pi + 0.5 * dphi, dphi)
    mu = phi * n_mer
    x = numpy.cos(mu) * (1 + numpy.cos(n_long * mu / n_mer) * 0.5)
    y = numpy.sin(mu) * (1 + numpy.cos(n_long * mu / n_mer) * 0.5)
    z = numpy.sin(n_long * mu / n_mer) * 0.5
    t = numpy.sin(mu)

    return x, y, z, t

class MyModel(HasTraits):
    radius = Float(0.025)
    scene = Instance(MlabSceneModel, ())
    plot = Instance(PipelineBase)

    @on_trait_change('radius,scene.activated')
    def update_plot(self):
        x, y, z, t = curve()
        if self.plot is None:
            self.plot = self.scene.mlab.plot3d(x, y, z, t,
                                tube_radius=self.radius, colormap='Spectral')
        else:
            print self.radius
            self.plot.mlab_source.set(tube_radius=self.radius)
            self.scene.mlab.draw()

    view = View(Item('scene', editor=SceneEditor(scene_class=MayaviScene),
                     height=250, width=300, show_label=False),
                Group(
                        'radius',
                     ),
                resizable=True,
                )

my_model = MyModel()
my_model.configure_traits()

这给出:

但是，当我更改半径时，视线图不会发生任何变化。

Answer 1

不要使用trait.set或trait.reset来设置mayavi或vtk属性。事实上，您使用的 MLineSource 没有这样的属性，但即使有也可能无法工作。

手动查找控制此功能的 mayavi 属性通常很有用，以查看它分配给哪个 mayavi 对象。在本例中，通过 mayavi 管道 GUI 显示它位于管过滤器上。

mlab.plot3d 是一个辅助函数，它尝试为您做所有事情，并且不维护引用所使用的​​过滤器。但总的来说，如果您自己构建管道，最好保留对 Mayavi 管道中每个步骤的引用。这样您就可以轻松访问控制它的 mayavi 对象。

如果您不自己构建管道，您始终可以通过手动导航父对象和子对象树来在管道中访问它。在这种情况下，您可以像这样访问它:

@on_trait_change('radius,scene.activated')
    def update_plot(self):
        x, y, z, t = curve()
        if self.plot is None:
            self.plot = self.scene.mlab.plot3d(x, y, z, t,
                                tube_radius=self.radius, colormap='Spectral')
        else:
            self.plot.parent.parent.filter.radius = self.radius