1
""" The base class for moveable shapes. """
4
# Standard library imports.
7
# Enthought library imports.
8
from enthought.enable.api import ColorTrait, Component, Pointer
9
from enthought.kiva import Font, MODERN
10
from enthought.traits.api import Float, Property, Str, Tuple
13
class Shape(Component):
14
""" The base class for moveable shapes. """
16
#### 'Component' interface ################################################
18
# The background color of this component.
19
bgcolor = 'transparent'
21
#### 'Shape' interface ####################################################
23
# The coordinates of the center of the shape.
24
center = Property(Tuple)
27
fill_color = ColorTrait
29
# The pointer for the 'normal' event state.
30
normal_pointer = Pointer('arrow')
32
# The pointer for the 'moving' event state.
33
moving_pointer = Pointer('hand')
36
text_color = ColorTrait
38
# The text displayed in the shape.
41
#### 'Private' interface ##################################################
43
# The difference between the location of a mouse-click and the component's
48
###########################################################################
49
# 'Interactor' interface
50
###########################################################################
52
def normal_key_pressed(self, event):
53
""" Event handler. """
55
print 'normal_key_pressed', event.character
59
def normal_left_down(self, event):
60
""" Event handler. """
62
if self.is_in(event.x, event.y):
63
self.event_state = 'moving'
65
event.window.mouse_owner = self
66
event.window.set_pointer(self.moving_pointer)
68
self._offset_x = event.x - self.x
69
self._offset_y = event.y - self.y
71
# move this shape to the top of the z order. The components are
72
# drawn in order, so the last one will be drawn on top
73
siblings = self.container.components
80
def moving_mouse_move(self, event):
81
""" Event handler. """
83
top = event.y + self._offset_y
84
bottom = event.y - self._offset_y
85
left = event.x - self._offset_x
86
right = event.x + self._offset_x
88
# Keep the shape fully in the container
92
elif top > self.container.height:
93
bottom = self.container.height - self.height
97
elif right > self.container.width:
98
left = self.container.width - self.width
100
self.position = [left, bottom]
101
self.request_redraw()
105
def moving_left_up(self, event):
106
""" Event handler. """
108
self.event_state = 'normal'
110
event.window.set_pointer(self.normal_pointer)
111
event.window.mouse_owner = None
113
self.request_redraw()
117
def moving_mouse_leave(self, event):
118
""" Event handler. """
120
self.moving_left_up(event)
124
###########################################################################
126
###########################################################################
128
def _get_center(self):
129
""" Property getter. """
132
ox, oy = self.position
139
def _text_default(self):
140
""" Trait initializer. """
142
return type(self).__name__
144
###########################################################################
145
# Protected 'Shape' interface
146
###########################################################################
148
def _distance_between(self, (x1, y1), (x2, y2)):
149
""" Return the distance between two points. """
151
return math.sqrt(pow(x1 - x2, 2) + pow(y1 - y2, 2))
153
def _draw_text(self, gc):
154
""" Draw the shape's text. """
156
if len(self.text) > 0:
157
gc.set_fill_color(self._get_text_color(self.event_state))
159
gc.set_font(Font(family=MODERN, size=16))
160
tx, ty, tw, th = gc.get_text_extent(self.text)
164
gc.set_text_position(x+(dx-tw)/2, y+(dy-th)/2)
166
gc.show_text(self.text)
170
def _get_fill_color(self, event_state):
171
""" Return the fill color based on the event state. """
173
if event_state == 'normal':
174
fill_color = self.fill_color_
177
r, g, b, a = self.fill_color_
178
fill_color = (r, g, b, 0.5)
182
def _get_text_color(self, event_state):
183
""" Return the text color based on the event state. """
185
if event_state == 'normal':
186
text_color = self.text_color_
189
r, g, b, a = self.text_color_
190
text_color = (r, g, b, 0.5)
194
#### EOF ######################################################################
added
removed
enthought/enable/primitives/shape.py
