1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
|
"""A very silly example of using a scene editor. More complex examples are
available in mayavi. """
# Author: Prabhu Ramachandran <prabhu [at] aero.iitb.ac.in>
# Copyright (c) 2007, Enthought, Inc.
# License: BSD Style.
from enthought.traits.api import HasTraits, Enum, Instance, Any
from enthought.traits.ui.api import View, Item
from enthought.pyface.tvtk.scene_model import SceneModel
from enthought.pyface.tvtk.scene_editor import SceneEditor
from enthought.pyface.tvtk import actors
from enthought.tvtk.api import tvtk
######################################################################
class ActorViewer(HasTraits):
# A simple trait to change the actors/widgets.
actor_type = Enum('cone', 'sphere', 'plane_widget', 'box_widget')
# The scene model.
scene = Instance(SceneModel, ())
_current_actor = Any
######################
view = View(Item(name='actor_type'),
Item(name='scene',
editor=SceneEditor(),
show_label=False,
resizable=True,
height=500,
width=500)
)
def __init__(self, **traits):
super(ActorViewer, self).__init__(**traits)
self._actor_type_changed(self.actor_type)
####################################
# Private traits.
def _actor_type_changed(self, value):
scene = self.scene
if self._current_actor is not None:
scene.remove_actors(self._current_actor)
if value == 'cone':
a = actors.cone_actor()
scene.add_actors(a)
elif value == 'sphere':
a = actors.sphere_actor()
scene.add_actors(a)
elif value == 'plane_widget':
a = tvtk.PlaneWidget()
scene.add_actors(a)
elif value == 'box_widget':
a = tvtk.BoxWidget()
scene.add_actors(a)
self._current_actor = a
if __name__ == '__main__':
a = ActorViewer()
a.configure_traits()
|
