def __init__(self, node, *args, **kw):
self.node = node
self.style = style._make_style()
self.text_widget = TextEdit(self.style, self._get_text, self._set_text, [all_printable], {'\r':'\n'})
self.node.obs.add_observer(self, '_node_')
Widget.__init__(self, *args, **kw)
self._size = MovingValue(Point)
self._pos = MovingValue(Point)
from Shapes.Ellipse import Ellipse
self.shape = Ellipse(pygame.Rect(0,0,1,1))
self.obs_loc = Observable()
self._register_keys()
self.reset_max_text_size()
self.mouse_area = MouseArea(self.in_bounds)
class NodeWidget(Widget):
bg_color=(10,10,100)
activated_bg_color=(40,40,160)
fg_color=(70,70,150)
activated_fg_color=(100,100,250)
key_stop_edit=Key(0, pygame.K_ESCAPE)
def __init__(self, node, *args, **kw):
self.node = node
self.style = style._make_style()
self.text_widget = TextEdit(self.style, self._get_text, self._set_text, [all_printable], {'\r':'\n'})
self.node.obs.add_observer(self, '_node_')
Widget.__init__(self, *args, **kw)
self._size = MovingValue(Point)
self._pos = MovingValue(Point)
from Shapes.Ellipse import Ellipse
self.shape = Ellipse(pygame.Rect(0,0,1,1))
self.obs_loc = Observable()
self._register_keys()
self.reset_max_text_size()
self.mouse_area = MouseArea(self.in_bounds)
def __getstate__(self):
d= self.__dict__.copy()
del d['focus_keymap']
del d['keymap']
return d
def __setstate__(self, d):
for k,v in d.iteritems():
self.__dict__[k] = v
self.focus_keymap = Keymap()
self.keymap = Keymap()
self._register_keys()
def _register_keys(self):
r = self.focus_keymap.register_key
r(Key(pygame.KMOD_CTRL, pygame.K_RIGHT), keydown_noarg(self._move_right))
r(Key(pygame.KMOD_CTRL, pygame.K_LEFT), keydown_noarg(self._move_left))
r(Key(pygame.KMOD_CTRL, pygame.K_UP), keydown_noarg(self._move_up))
r(Key(pygame.KMOD_CTRL, pygame.K_DOWN), keydown_noarg(self._move_down))
r(Key(0, pygame.K_RETURN), keydown_noarg(self._edit_value))
def in_bounds(self, p):
return self.rect().collidepoint(tuple(p))
def _get_text(self):
if self.node.value is None:
return ''
return str(self.node.value)
def _set_text(self, text):
self.node.value = text
self._update_text_size()
def reset_max_text_size(self, value=None):
# returns the calculated actual size
if value is None:
value = FONT_SIZE_ABSOLUTE_MAX
self.max_font_size = value
return self.calc_text_size()
def calc_text_size(self):
min_font_size = 2
ratio = 0.7
yratio = 1
size = self.style.font_size
self.text_widget.update()
while ((self.text_widget.size[0] < self._size.final[0]*ratio
or self.text_widget.size[1] < self._size.final[1]*yratio)
and self.style.font_size < min(self.max_font_size, FONT_SIZE_ABSOLUTE_MAX)):
self.style.font_size = min(self.style.font_size + 1, FONT_SIZE_ABSOLUTE_MAX)
self._update_text_size()
while ((self.text_widget.size[0] > self._size.final[0]*ratio
or self.text_widget.size[1] > self._size.final[1]*yratio)
and self.style.font_size > min_font_size or self.style.font_size > self.max_font_size):
self.style.font_size -= 1
self._update_text_size()
return self.style.font_size
def _update_text_size(self, new_size=None):
if new_size is not None:
self.style.font_size = new_size
self.text_widget.set_style(self.style)
self.text_widget.update(force=True)
def get_size(self):
return tuple(self._size.current)
def set_size(self, p):
p = Point(p)
self._size.final = p
self._update_text_size()
self.obs_loc.notify.size_set(p)
size = property(get_size, set_size)
def get_pos(self):
return tuple(self._pos.current)
def set_pos(self, p):
p = Point(p)
self._pos.final = p
self.obs_loc.notify.pos_set(p)
pos = property(get_pos, set_pos)
def rect(self):
s = self.size
return pygame.Rect(self.pos[0],self.pos[1],s[0],s[1])
def final_rect(self):
s = self._size.final
p = self._pos.final
return pygame.Rect(p[0],p[1],s[0],s[1])
def _node_connect(self, e):
pass
def _node_disconnect(self, e):
pass
def update(self):
self.text_widget.update()
self._update_text_size()
self._size.update()
self._pos.update()
self.shape.rect = self.rect()
def _draw(self, surface, pos_offset):
self.shape.paint(pos_offset, surface, self.fg_color, self.bg_color)
self.text_widget.draw(surface, tuple(Point(pos_offset) + Point(self.rect().center) - Point(self.text_widget.size)/2))
def _move_right(self):
'''move right'''
self.pos = Point(self.pos) + Point((5,0))
def _move_left(self):
'''move left'''
self.pos = Point(self.pos) + Point((-5,0))
def _move_up(self):
'''move up'''
self.pos = Point(self.pos) + Point((0,-5))
def _move_down(self):
'''move down'''
self.pos = Point(self.pos) + Point((0,5))
def _edit_value(self):
'''Edit value'''
self.keymap.set_next_keymap(self.text_widget.focus_keymap)
self.text_widget._start_editing()
self.keymap.register_key(self.key_stop_edit, keydown_noarg(self._stop_edit_value))
def _stop_edit_value(self):
'''Stop editing value'''
self.text_widget._stop_editing()
self.keymap.set_next_keymap(self.focus_keymap)
self.keymap.unregister_key(self.key_stop_edit)