Multiple mlab scene models example

Example showing a dialog with multiple embedded scenes.

When using several embedded scenes with mlab, you should be very careful always to pass the scene you want to use for plotting to the mlab function used, elsewhere it uses the current scene. In this example, failing to do so would result in only one scene being used, the last one created.

The trick is to use the ‘mayavi_scene’ attribute of the MlabSceneModel, and pass it as a keyword argument to the mlab functions.

For more examples on embedding mlab scenes in dialog, see also: the examples Mlab interactive dialog example, and Lorenz ui example, as well as the section of the user manual Embedding a Mayavi scene in a Traits dialog.

Source code: multiple_mlab_scene_models.py

import numpy as np

from enthought.traits.api import HasTraits, Instance, Button, \
    on_trait_change
from enthought.traits.ui.api import View, Item, HSplit, Group

from enthought.mayavi import mlab
from enthought.mayavi.tools.mlab_scene_model import MlabSceneModel
from enthought.tvtk.pyface.scene_editor import SceneEditor

class MyDialog(HasTraits):

    scene1 = Instance(MlabSceneModel, ())
    scene2 = Instance(MlabSceneModel, ())

    button1 = Button('Redraw')
    button2 = Button('Redraw')

    @on_trait_change('button1')
    def redraw_scene1(self):
        self.redraw_scene(self.scene1)

    @on_trait_change('button2')
    def redraw_scene2(self):
        self.redraw_scene(self.scene2)

    def redraw_scene(self, scene):
        # Notice how each mlab 
        mlab.clf(figure=scene.mayavi_scene)
        x, y, z, s = np.random.random((4, 100))
        mlab.points3d(x, y, z, s, figure=scene.mayavi_scene)

    # The layout of the dialog created
    view = View(HSplit(
                  Group(
                       Item('scene1',
                            editor=SceneEditor(), height=250, 
                            width=300),
                       'button1', 
                       show_labels=False,
                  ),
                  Group(
                       Item('scene2',
                            editor=SceneEditor(), height=250, 
                            width=300, show_label=False),
                       'button2', 
                       show_labels=False,
                  ),
                ),
                resizable=True,
                )

m = MyDialog()
m.configure_traits()
                                

Previous topic

Wx embedding example

Next topic

Multiple engines example

This Page