如何在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没有这样的属性，但即使有，它也可能无法工作。在

通常情况下，AVI指定给AVI的属性是有用的。在本例中，通过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