From 1b9b005820149e117c6598d71d07780889f4c005 Mon Sep 17 00:00:00 2001
From: QwiglyDee <qwiglydee@gmail.com>
Date: Thu, 24 Dec 2020 21:04:44 +0700
Subject: [PATCH] decorations colour. close #1197

---
 .../blenderbim/bim/decoration.py              | 59 ++++++++++---------
 src/ifcblenderexport/blenderbim/bim/prop.py   |  3 +
 src/ifcblenderexport/blenderbim/bim/ui.py     |  4 +-
 3 files changed, 37 insertions(+), 29 deletions(-)

diff --git a/src/ifcblenderexport/blenderbim/bim/decoration.py b/src/ifcblenderexport/blenderbim/bim/decoration.py
index 5038c9b61d..41927d6a2f 100644
--- a/src/ifcblenderexport/blenderbim/bim/decoration.py
+++ b/src/ifcblenderexport/blenderbim/bim/decoration.py
@@ -19,6 +19,26 @@ class BaseDecorator():
     # base name of objects to decorate
     basename = "IfcAnnotation/Something"
 
+    DEF_GLSL = """
+        #define PI 3.141592653589793
+        #define CIRCLE_SEGS 24
+        // max points = SEGS (circle) + 2 (stem) + 3 (arrow)
+        #define MAX_POINTS 29
+        #define ARROW1_ANGLE PI / 12.0
+        #define ARROW2_ANGLE PI / 3.0
+        #define ARROW1_SIZE 16.0
+        #define ARROW2_SIZE 24.0
+        #define CIRCLE1_SIZE 8.0
+        #define CIRCLE2_SIZE 8.0
+        #define DASH1_SIZE 16.0
+        #define DASH2_SIZE 48.0
+        #define DASH_GAP 4.0
+        #define CROSS_SIZE 16.0
+        #define CALLOUT_GAP 8.0
+        #define CALLOUT_SIZE 64.0
+        #define COLOR vec4({color[0]}, {color[1]}, {color[2]}, {color[3]})
+    """
+
     VERT_GLSL = """
     uniform mat4 viewMatrix;
     in vec3 pos;
@@ -31,6 +51,7 @@ class BaseDecorator():
         type = topo;
     }
     """
+
     GEOM_GLSL = """
     layout(lines) in;
     layout(line_strip, max_vertices=2) out;
@@ -49,7 +70,7 @@ class BaseDecorator():
     out vec4 fragColor;
 
     void main() {
-        fragColor = vec4(1.0, 1.0, 1.0, 1.0);
+        fragColor = COLOR;
     }
     """
 
@@ -91,30 +112,13 @@ class BaseDecorator():
     }
     """
 
-    DEF_GLSL = """
-        #define PI 3.141592653589793
-        #define CIRCLE_SEGS 24
-        // max points = SEGS (circle) + 2 (stem) + 3 (arrow)
-        #define MAX_POINTS 29
-        #define ARROW1_ANGLE PI / 12.0
-        #define ARROW2_ANGLE PI / 3.0
-        #define ARROW1_SIZE 16.0
-        #define ARROW2_SIZE 24.0
-        #define CIRCLE1_SIZE 8.0
-        #define CIRCLE2_SIZE 8.0
-        #define DASH1_SIZE 16.0
-        #define DASH2_SIZE 48.0
-        #define DASH_GAP 4.0
-        #define CROSS_SIZE 16.0
-        #define CALLOUT_GAP 8.0
-        #define CALLOUT_SIZE 64.0
-    """
-
     # class var for single handler
     installed = None
 
     @classmethod
     def install(cls, *args, **kwargs):
+        if cls.installed:
+            cls.uninstall()
         handler = cls(*args, **kwargs)
         cls.installed = SpaceView3D.draw_handler_add(handler, (), 'WINDOW', 'POST_PIXEL')
 
@@ -134,13 +138,13 @@ class BaseDecorator():
         self.font_size = 16
         self.dpi = context.preferences.system.dpi
 
-    @classmethod
-    def create_shader(cls):
+    def create_shader(self):
+        defines = self.DEF_GLSL.format(color=self.props.decorations_colour)
         # NB: libcode param doesn't work
-        return GPUShader(vertexcode=cls.VERT_GLSL,
-                         fragcode=cls.FRAG_GLSL,
-                         geocode=cls.LIB_GLSL + cls.GEOM_GLSL,
-                         defines=cls.DEF_GLSL)
+        return GPUShader(vertexcode=self.VERT_GLSL,
+                         fragcode=self.FRAG_GLSL,
+                         geocode=self.LIB_GLSL + self.GEOM_GLSL,
+                         defines=defines)
 
     def __call__(self):
         if self.props.active_drawing_index is None or len(self.props.drawings) == 0:
@@ -280,6 +284,7 @@ class BaseDecorator():
         blf.position(self.font_id, pos.x, pos.y, 0)
 
         blf.rotation(self.font_id, ang)
+        blf.color(self.font_id, *self.props.decorations_colour)
         blf.draw(self.font_id, text)
         blf.disable(self.font_id, blf.ROTATION)
 
@@ -580,7 +585,7 @@ class HiddenDecorator(BaseDecorator):
         if (t > 0.5) {
             discard;
         } else {
-            fragColor = vec4(1.0, 1.0, 1.0, 1.0);
+            fragColor = COLOR;
         }
     }
     """
diff --git a/src/ifcblenderexport/blenderbim/bim/prop.py b/src/ifcblenderexport/blenderbim/bim/prop.py
index e0f8ecf986..017d494c19 100644
--- a/src/ifcblenderexport/blenderbim/bim/prop.py
+++ b/src/ifcblenderexport/blenderbim/bim/prop.py
@@ -723,6 +723,9 @@ class DocProperties(PropertyGroup):
     ifc_files: CollectionProperty(name="IFCs", type=StrProperty)
     drawing_styles: CollectionProperty(name="Drawing Styles", type=DrawingStyle)
     should_draw_decorations: BoolProperty(name="Should Draw Decorations", update=toggleDecorations)
+    decorations_colour: FloatVectorProperty(name="Decorations Colour", subtype="COLOR", default=(1, 0, 0, 1),
+                                            min=0.0, max=1.0, size=4,
+                                            update=toggleDecorations)
 
 
 class BIMCameraProperties(PropertyGroup):
diff --git a/src/ifcblenderexport/blenderbim/bim/ui.py b/src/ifcblenderexport/blenderbim/bim/ui.py
index c8ea3dc1ea..79bd7b63d7 100644
--- a/src/ifcblenderexport/blenderbim/bim/ui.py
+++ b/src/ifcblenderexport/blenderbim/bim/ui.py
@@ -2461,8 +2461,8 @@ class BIM_PT_annotation_utilities(Panel):
                 row.operator("bim.remove_drawing", icon="X", text="").index = props.active_drawing_index
             layout.template_list("BIM_UL_generic", "", props, "drawings", props, "active_drawing_index")
 
-        row = layout.row()
-        row.prop(props, "should_draw_decorations")
+        layout.prop(props, "should_draw_decorations")
+        layout.prop(props, "decorations_colour")
 
 
 class BIM_PT_qto_utilities(Panel):