From bf340f12d9b8393fd70d6e05c5e507eb1e04e084 Mon Sep 17 00:00:00 2001
From: Josh Holtrop <jholtrop+git@gmail.com>
Date: Mon, 3 Oct 2011 17:55:28 -0400
Subject: [PATCH] add draw() to Mode; draw in-progress Connects

---
 SketchWidget.py | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/SketchWidget.py b/SketchWidget.py
index b1888c5..9fe047e 100644
--- a/SketchWidget.py
+++ b/SketchWidget.py
@@ -52,6 +52,7 @@ class SketchWidget(object):
         self.axis_color = (1.0, 0.0, 0.0, 1.0)
         self.hover_color = (1.0, 1.0, 1.0, 1.0)
         self.constraint_color = (0.8, 1.0, 0.0, 1.0)
+        self.constraint_ip_color = (0.1, 1.0, 0.5, 1.0)
         self.snap_angle = 10
         self.hv_snap_dist = 10
         self.snap_dist = 6
@@ -156,6 +157,8 @@ class SketchWidget(object):
             s, p = self.hover_snap_ptref.shape, self.hover_snap_ptref.ptNum
             self.drawConnect(Connect(s, p, s, p))
 
+        self.mode.draw(self)
+
         if gldrawable.is_double_buffered():
             gldrawable.swap_buffers()
         else:
@@ -513,6 +516,8 @@ class Mode(object):
         pass
     def get_cursor(self):
         return 'arrow'
+    def draw(self, sw):
+        pass
 
 class SelectMode(Mode):
     pass
@@ -661,6 +666,12 @@ class ConnectMode(Mode):
         else:
             sw.window.set_mode('')
 
+    def draw(self, sw):
+        if self.first_ptref is not None:
+            glColor(*sw.constraint_ip_color)
+            s, p = self.first_ptref.shape, self.first_ptref.ptNum
+            sw.drawConstraint(Connect(s, p, s, p))
+
 class HorizontalMode(ConnectMode):
     def do_left_click(self, sw, x, y):
         if sw.hover_snap_ptref is not None: