# HG changeset patch
# User Neil Muller <neil@mcp.home.test>
# Date 1358288892 -7200
# Node ID 69a0349dce8e7381c8d13181745c015ac859a3af
# Parent  f5e381e819a1f86bc6cb48233f111d5191203d6d
Fix rect_drawer to call prepare

diff -r f5e381e819a1 -r 69a0349dce8e pyntnclick/tools/rect_drawer.py
--- 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