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()

这给出了：

enter image description here

然而，当我改变半径时，视线图没有任何反应。

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

如何改变Mayavi线图的管半径？

1 个答案: