python - Mayavi:自定义工具栏

Question

有没有办法自定义mayavi场景的默认工具栏？我想删除一些按钮，因为我不需要它们(例如保存按钮)。在这里您可以看到我正在谈论哪个工具栏:

该代码只是示例代码:

import os
os.environ['ETS_TOOLKIT'] = 'qt4'
from pyface.qt import QtGui, QtCore
from traits.api import HasTraits, Instance, on_trait_change
from traitsui.api import View, Item
from mayavi.core.ui.api import MayaviScene, MlabSceneModel, SceneEditor
from tvtk.pyface.api import DecoratedScene
from pyface.api import ImageResource
from pyface.action.api import Action

class MyCustomScene(DecoratedScene):

    def _actions_default(self):
        actions = [ 
                Action(
                    image = ImageResource("path to image",
                        search_path = [self._get_image_path()],
                        ),
                    tooltip = "blabla",
                    on_perform = self._save_snapshot,
                    )
                ]
        actions.extend(DecoratedScene._actions_default(self))
        return actions

#The actual visualization
class Visualization(HasTraits):
    scene = Instance(MlabSceneModel, ())

    @on_trait_change('scene.activated')
    def update_plot(self):
        # We can do normal mlab calls on the embedded scene.
        self.scene.mlab.test_points3d()

    # the layout of the dialog screated
    view = View(Item('scene', editor=SceneEditor(scene_class=MyCustomScene),
                     height=250, width=300, show_label=False),
                resizable=True # We need this to resize with the parent widget
                )
################################################################################
# The QWidget containing the visualization, this is pure PyQt4 code.
class MayaviQWidget(QtGui.QWidget):
    def __init__(self, parent=None):
        QtGui.QWidget.__init__(self, parent)
        layout = QtGui.QVBoxLayout(self)
        layout.setContentsMargins(0,0,0,0)
        layout.setSpacing(0)
        self.visualization = Visualization()

        # The edit_traits call will generate the widget to embed.
        self.ui = self.visualization.edit_traits(parent=self,
                                                 kind='subpanel').control
        layout.addWidget(self.ui)
        self.ui.setParent(self)



if __name__ == "__main__":
    # Don't create a new QApplication, it would unhook the Events
    # set by Traits on the existing QApplication. Simply use the
    # '.instance()' method to retrieve the existing one.
    app = QtGui.QApplication.instance()
    container = QtGui.QWidget()
    container.setWindowTitle("Embedding Mayavi in a PyQt4 Application")
    # define a "complex" layout to test the behaviour
    layout = QtGui.QGridLayout(container)

    # put some stuff around mayavi
    label_list = []
    for i in range(3):
        for j in range(3):
            if (i==1) and (j==1):continue
            label = QtGui.QLabel(container)
            label.setText("Your QWidget at (%d, %d)" % (i,j))
            label.setAlignment(QtCore.Qt.AlignHCenter|QtCore.Qt.AlignVCenter)
            layout.addWidget(label, i, j)
            label_list.append(label)
    mayavi_widget = MayaviQWidget(container)

    layout.addWidget(mayavi_widget, 1, 1)
    container.show()
    window = QtGui.QMainWindow()
    window.setCentralWidget(container)
    window.show()

    # Start the main event loop.
    app.exec_()

我认为它隐藏在 MayaviScene 中的某个地方。也许有必要创建一个具有新场景或类似内容的新类？

Answer 1

您应该检查_actions_default MayaviScene和 DecoratedScene了解如何创建您自己的一个。第二个展示了如何从头开始创建工具栏，而第一个展示了工具栏代码如何与其他组件交互。

class MyCustomScene(DecoratedScene):
    # …
    def _actions_default(self):
        actions = [ 
                # add icons here
                # … 
                ]
        return actions