From f65ecafa4abf15c64e47192b8bbe3ee95f5e46b1 Mon Sep 17 00:00:00 2001
From: Greg <gregzzmail@gmail.com>
Date: Fri, 19 Aug 2016 16:05:11 +0200
Subject: [PATCH] Fix T49090 - NW broke some color wheel drawing

Thanks to @Severin for help on IRC :)
---
 node_wrangler.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/node_wrangler.py b/node_wrangler.py
index 977c7aa69..7e3e179ae 100644
--- a/node_wrangler.py
+++ b/node_wrangler.py
@@ -742,6 +742,9 @@ def store_mouse_cursor(context, event):
 
 
 def draw_line(x1, y1, x2, y2, size, colour=[1.0, 1.0, 1.0, 0.7]):
+    shademodel_state = bgl.Buffer(bgl.GL_INT, 1)
+    bgl.glGetIntegerv(bgl.GL_SHADE_MODEL, shademodel_state)
+
     bgl.glEnable(bgl.GL_BLEND)
     bgl.glLineWidth(size * dpifac())
     bgl.glShadeModel(bgl.GL_SMOOTH)
@@ -756,7 +759,8 @@ def draw_line(x1, y1, x2, y2, size, colour=[1.0, 1.0, 1.0, 0.7]):
     except:
         pass
     bgl.glEnd()
-    bgl.glShadeModel(bgl.GL_FLAT)
+
+    bgl.glShadeModel(shademodel_state[0])
     bgl.glDisable(bgl.GL_LINE_SMOOTH)
 
 
-- 
GitLab