changeset 737:69a0349dce8e pyntnclick

Fix rect_drawer to call prepare
author Neil Muller <neil@mcp.home.test>
date Wed, 16 Jan 2013 00:28:12 +0200
parents f5e381e819a1
children b37c188ddeba
files pyntnclick/tools/rect_drawer.py
diffstat 1 files changed, 6 insertions(+), 0 deletions(-) [+]
line wrap: on
line diff
--- a/pyntnclick/tools/rect_drawer.py	Wed Jan 16 00:27:50 2013 +0200
+++ b/pyntnclick/tools/rect_drawer.py	Wed Jan 16 00:28:12 2013 +0200
@@ -54,6 +54,7 @@
         self._palette.cur_selection = self
 
     def draw(self, surface):
+        self.do_prepare()
         surface.fill(pygame.color.Color(0, 0, 0), self.rect)
         if self.selected:
             surface.fill(self.sel_colour, self._button_rect)
@@ -274,6 +275,7 @@
         self.clear_display = True
 
     def draw(self, surface):
+        self.do_prepare()
         if self.clear_display:
             surface.fill(pygame.color.Color(0, 0, 0))
             self.clear_display = False
@@ -521,6 +523,9 @@
         if self._parent.paused:
             return False
         if self.mode == 'draw':
+            if self.start_pos is None:
+                # We've come here not via a drawing situation, so bail
+                return False
             rect = pygame.rect.Rect(self.start_pos[0], self.start_pos[1],
                     self.end_pos[0] - self.start_pos[0],
                     self.end_pos[1] - self.start_pos[1])
@@ -563,6 +568,7 @@
                 fontsize=15, color=pygame.color.Color(128, 0, 255))
 
     def draw(self, surface):
+        self.do_prepare()
         text = 'Mode : %s' % self.app_image.mode
         if self.text != text:
             self.text = text