Glassvision/IfcOpenShell
1
0
Fork 0
mirror of https://github.com/IfcOpenShell/IfcOpenShell.git synced 2026-08-10 09:48:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

Custom snap symbols #5325.

Browse Source
This commit is contained in:
Bruno Perdigão
2024-09-13 17:34:08 -03:00
parent 96f3ee2f3a
commit a2cf95038a
1 changed files with 50 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/bonsai/bonsai/bim/module/model/decorator.py
+50 -5
View File
@@ -27,6 +27,7 @@ import math
from bpy_extras import view3d_utils
from mathutils import Vector, Matrix
from gpu_extras.batch import batch_for_shader
from gpu_extras.presets import draw_circle_2d
from typing import Union
from bonsai.bim.module.drawing.helper import format_distance
@@ -317,6 +318,7 @@ class PolylineDecorator:
if cls.is_installed:
cls.uninstall()
handler = cls()
cls.handlers.append(SpaceView3D.draw_handler_add(handler.draw_snap_point, (context,), "WINDOW", "POST_PIXEL"))
cls.handlers.append(SpaceView3D.draw_handler_add(handler.draw_measurements, (context,), "WINDOW", "POST_PIXEL"))
cls.handlers.append(SpaceView3D.draw_handler_add(handler.draw_input_ui, (context,), "WINDOW", "POST_PIXEL"))
cls.handlers.append(SpaceView3D.draw_handler_add(handler.draw_product_preview, (context,), "WINDOW", "POST_VIEW"))
@@ -485,6 +487,49 @@ class PolylineDecorator:
self.draw_text_background(context, coords_angle, text_dim)
blf.draw(self.font_id, text)
def draw_snap_point(self, context):
self.line_shader = gpu.shader.from_builtin("POLYLINE_UNIFORM_COLOR")
self.line_shader.bind() # required to be able to change uniforms of the shader
self.line_shader.uniform_float("viewportSize", (context.region.width, context.region.height))
self.shader = gpu.shader.from_builtin("UNIFORM_COLOR")
self.line_shader.uniform_float("lineWidth", 1.0)
theme = context.preferences.themes.items()[0][1]
decorator_color_object_active = (*theme.view_3d.object_active, 1) #unwrap color values and adds alpha=1
region = context.region
rv3d = region.data
snap_prop = context.scene.BIMPolylineProperties.snap_mouse_point[0]
mouse_point = Vector((snap_prop.x, snap_prop.y, snap_prop.z))
coords = view3d_utils.location_3d_to_region_2d(region, rv3d, mouse_point)
padding = 6
verts = []
if snap_prop.snap_type in ["Edge", "Vertex"]:
p1 = (coords[0] - padding, coords[1] + padding)
p2 = (coords[0] + padding, coords[1] + padding)
p3 = (coords[0] + padding, coords[1] - padding)
p4 = (coords[0] - padding, coords[1] - padding)
verts = [p1, p2, p3, p4]
if snap_prop.snap_type == "Edge":
edges = [[0, 1], [1, 3], [3, 2], [2, 0]]
else:
edges = [[0, 1], [1, 2], [2, 3], [3, 0]]
if snap_prop.snap_type == "Edge Center":
p1 = (coords[0], coords[1] + padding)
p2 = (coords[0] + padding, coords[1] - padding)
p3 = (coords[0] - padding, coords[1] - padding)
verts = [p1, p2, p3]
edges = [[0, 1], [1, 2], [2, 0]]
if snap_prop.snap_type == "Face":
draw_circle_2d(coords, decorator_color_object_active, padding)
return
self.draw_batch("LINES", verts, decorator_color_object_active, edges)
def __call__(self, context):
self.addon_prefs = tool.Blender.get_addon_preferences()
@@ -495,7 +540,7 @@ class PolylineDecorator:
decorator_color_unselected = self.addon_prefs.decorator_color_unselected
decorator_color_background = self.addon_prefs.decorator_color_background
theme = context.preferences.themes.items()[0][1]
decorator_color_vertex_selected = (*theme.view_3d.vertex_select, 1) #unwrap color values and adds alpha=1
decorator_color_object_active = (*theme.view_3d.object_active, 1) #unwrap color values and adds alpha=1
decorator_color_x_axis = (*theme.user_interface.axis_x, 1)
decorator_color_y_axis = (*theme.user_interface.axis_y, 1)
decorator_color_z_axis = (*theme.user_interface.axis_z, 1)
@@ -577,13 +622,11 @@ class PolylineDecorator:
# self.draw_batch("TRIS", polyline_points, (0, 1, 0, 0.1), edges)
# Mouse points
if snap_prop.snap_type in ["Face", "Plane"]:
if snap_prop.snap_type in ["Plane"]:
self.draw_batch("POINTS", mouse_point, decorator_color)
else:
self.draw_batch("POINTS", mouse_point, decorator_color_vertex_selected)
if ref_point:
self.draw_batch("POINTS", ref_point, decorator_color_vertex_selected)
self.draw_batch("POINTS", ref_point, decorator_color_object_active)
# Line between last polyline point and mouse
edges = [[0, 1]]
@@ -594,7 +637,9 @@ class PolylineDecorator:
self.draw_batch("LINES", [polyline_points[-1]] + mouse_point, decorator_color, edges)
# Draw polyline with selected points
print("EDGES",polyline_edges)
self.line_shader.uniform_float("lineWidth", 2.0)
print("VS", polyline_points)
self.draw_batch("POINTS", polyline_points, decorator_color_special)
if len(polyline_points) > 1:
self.draw_batch("LINES", polyline_points, decorator_color_special, polyline_edges)