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

black format

Browse Source
This commit is contained in:
Bruno Perdigão
2024-08-05 17:27:04 -03:00
committed by Bruno Perdigão
parent bc0efd7389
commit 67b01dcd0f
1 changed files with 68 additions and 82 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/blenderbim/blenderbim/bim/module/model/wall.py
+45 -59
View File
@@ -293,24 +293,21 @@ class DrawPolylineWall(bpy.types.Operator):
@classmethod @classmethod
def poll(cls, context): def poll(cls, context):
return context.space_data.type == 'VIEW_3D' return context.space_data.type == "VIEW_3D"
def __init__(self): def __init__(self):
self.mousemove_count = 0 self.mousemove_count = 0
self.action_count = 0 self.action_count = 0
self.visible_objs = None self.visible_objs = None
self.number_options = {'0', '1', '2', '3', self.number_options = {"0", "1", "2", "3", "4", "5", "6", "7", "8", "9", ".", "-"}
'4', '5', '6', '7', '8', '9', '.', '-'}
self.number_input = [] self.number_input = []
self.number_output = '' self.number_output = ""
self.number_is_negative = False self.number_is_negative = False
self.is_input_on = False self.is_input_on = False
self.input_options = {'X', 'Y', 'D', 'A'} self.input_options = {"X", "Y", "D", "A"}
self.input_type = '' self.input_type = ""
self.input_value_xy = [None, None] self.input_value_xy = [None, None]
def get_visible_objects(self, context): def get_visible_objects(self, context):
depsgraph = context.evaluated_depsgraph_get() depsgraph = context.evaluated_depsgraph_get()
all_objs = [] all_objs = []
@@ -339,16 +336,14 @@ class DrawPolylineWall(bpy.types.Operator):
region = context.region region = context.region
borders = (region.width, region.height) borders = (region.width, region.height)
for i, axis in enumerate(zip(*transposed_bbox)): for i, axis in enumerate(zip(*transposed_bbox)):
if all(ax < 0 or ax > borders[i] for ax in axis): # Filter only objects in viewport if all(ax < 0 or ax > borders[i] for ax in axis): # Filter only objects in viewport
return (obj, None) return (obj, None)
min_point = min(axis) min_point = min(axis)
max_point = max(axis) max_point = max(axis)
bbox_2d.extend([min_point, max_point]) bbox_2d.extend([min_point, max_point])
return (obj, bbox_2d) return (obj, bbox_2d)
def snaping_movement(self, context, event): def snaping_movement(self, context, event):
scene = context.scene scene = context.scene
region = context.region region = context.region
@@ -357,17 +352,21 @@ class DrawPolylineWall(bpy.types.Operator):
offset = 5 offset = 5
mouse_offset = ( mouse_offset = (
(-offset, offset), (0, offset), (offset, offset), (-offset, offset),
(-offset, 0), (0, 0), (offset, 0), (0, offset),
(-offset, -offset), (0, -offset), (offset, -offset) (offset, offset),
(-offset, 0),
(0, 0),
(offset, 0),
(-offset, -offset),
(0, -offset),
(offset, -offset),
) )
# Plane to intersect. Default Container # Plane to intersect. Default Container
plane_origin = Vector((0, 0, 0)) plane_origin = Vector((0, 0, 0))
plane_normal = Vector((0, 0, 1)) plane_normal = Vector((0, 0, 1))
def get_viewport_ray_data(): def get_viewport_ray_data():
view_vector = view3d_utils.region_2d_to_vector_3d(region, rv3d, self.mouse_pos) view_vector = view3d_utils.region_2d_to_vector_3d(region, rv3d, self.mouse_pos)
ray_origin = view3d_utils.region_2d_to_origin_3d(region, rv3d, self.mouse_pos) ray_origin = view3d_utils.region_2d_to_origin_3d(region, rv3d, self.mouse_pos)
@@ -376,7 +375,6 @@ class DrawPolylineWall(bpy.types.Operator):
return ray_origin, ray_target, ray_direction return ray_origin, ray_target, ray_direction
def get_object_ray_data(obj_matrix): def get_object_ray_data(obj_matrix):
ray_origin, ray_target, ray_direction = get_viewport_ray_data() ray_origin, ray_target, ray_direction = get_viewport_ray_data()
matrix_inv = obj_matrix.inverted() matrix_inv = obj_matrix.inverted()
@@ -386,7 +384,6 @@ class DrawPolylineWall(bpy.types.Operator):
return ray_origin_obj, ray_target_obj, ray_direction_obj return ray_origin_obj, ray_target_obj, ray_direction_obj
def in_view_2d_bounding_box(obj, bbox): def in_view_2d_bounding_box(obj, bbox):
x, y = self.mouse_pos x, y = self.mouse_pos
xmin, xmax, ymin, ymax = bbox xmin, xmax, ymin, ymax = bbox
@@ -404,23 +401,19 @@ class DrawPolylineWall(bpy.types.Operator):
else: else:
return None, None, None return None, None, None
def ray_cast_to_plane(context, plane_origin, plane_normal): def ray_cast_to_plane(context, plane_origin, plane_normal):
intersection = Vector ((0,0,0,)) intersection = Vector((0, 0, 0))
try: try:
loc = view3d_utils.region_2d_to_location_3d( loc = view3d_utils.region_2d_to_location_3d(region, rv3d, self.mouse_pos, ray_direction)
region, rv3d, self.mouse_pos, ray_direction
)
intersection = mathutils.geometry.intersect_line_plane(ray_target, loc, plane_origin, plane_normal) intersection = mathutils.geometry.intersect_line_plane(ray_target, loc, plane_origin, plane_normal)
except: except:
intersection = Vector((0,0,0)) intersection = Vector((0, 0, 0))
if intersection == None: if intersection == None:
intersection = Vector((0,0,0)) intersection = Vector((0, 0, 0))
return intersection return intersection
def cast_rays_and_get_best_object(): def cast_rays_and_get_best_object():
best_length_squared = 1.0 best_length_squared = 1.0
best_obj = None best_obj = None
@@ -431,11 +424,10 @@ class DrawPolylineWall(bpy.types.Operator):
objs_to_raycast = [] objs_to_raycast = []
for obj, bbox_2d in self.objs_2d_bbox: for obj, bbox_2d in self.objs_2d_bbox:
if obj.type == 'MESH' and bbox_2d: if obj.type == "MESH" and bbox_2d:
if in_view_2d_bounding_box(obj, bbox_2d): if in_view_2d_bounding_box(obj, bbox_2d):
objs_to_raycast.append(obj) objs_to_raycast.append(obj)
for obj in objs_to_raycast: for obj in objs_to_raycast:
hit, normal, face_index = obj_ray_cast(obj) hit, normal, face_index = obj_ray_cast(obj)
if hit is None: if hit is None:
@@ -456,22 +448,18 @@ class DrawPolylineWall(bpy.types.Operator):
best_hit = hit_world best_hit = hit_world
best_face_index = face_index best_face_index = face_index
if best_obj is not None: if best_obj is not None:
return best_obj, best_hit, best_face_index return best_obj, best_hit, best_face_index
else: else:
return None, None, None return None, None, None
snap_threshold = 0.3 snap_threshold = 0.3
ray_origin, ray_target, ray_direction = get_viewport_ray_data() ray_origin, ray_target, ray_direction = get_viewport_ray_data()
obj, hit, face_index = cast_rays_and_get_best_object() obj, hit, face_index = cast_rays_and_get_best_object()
intersection = ray_cast_to_plane(context, plane_origin, plane_normal) intersection = ray_cast_to_plane(context, plane_origin, plane_normal)
rot_intersection = tool.Snaping.snap_on_axis(intersection) rot_intersection = tool.Snaping.snap_on_axis(intersection)
if obj is not None: if obj is not None:
original_obj = obj.original original_obj = obj.original
@@ -495,13 +483,14 @@ class DrawPolylineWall(bpy.types.Operator):
else: else:
tool.Snaping.update_snaping_point(intersection, "Plane") tool.Snaping.update_snaping_point(intersection, "Plane")
def modal(self, context, event): def modal(self, context, event):
if event.type == 'MOUSEMOVE': if event.type == "MOUSEMOVE":
self.mousemove_count += 1 self.mousemove_count += 1
self.is_input_on = False self.is_input_on = False
self.input_type = None
WallPolylineDecorator.set_input_panel(self.input_panel, self.input_type)
tool.Blender.update_viewport()
else: else:
self.mousemove_count = 0 self.mousemove_count = 0
@@ -515,13 +504,13 @@ class DrawPolylineWall(bpy.types.Operator):
WallPolylineDecorator.set_mouse_position(context, event) WallPolylineDecorator.set_mouse_position(context, event)
self.input_panel = WallPolylineDecorator.calculate_distance_and_angle(context, self.is_input_on) self.input_panel = WallPolylineDecorator.calculate_distance_and_angle(context, self.is_input_on)
tool.Blender.update_viewport() tool.Blender.update_viewport()
return {'RUNNING_MODAL'} return {"RUNNING_MODAL"}
if event.value == 'RELEASE' and event.type == 'LEFTMOUSE': if event.value == "RELEASE" and event.type == "LEFTMOUSE":
tool.Snaping.insert_polyline_point() tool.Snaping.insert_polyline_point()
if not self.is_input_on: if not self.is_input_on:
if event.value == 'RELEASE' and event.type == 'BACK_SPACE': if event.value == "RELEASE" and event.type == "BACK_SPACE":
tool.Snaping.remove_last_polyline_point() tool.Snaping.remove_last_polyline_point()
tool.Blender.update_viewport() tool.Blender.update_viewport()
@@ -542,40 +531,37 @@ class DrawPolylineWall(bpy.types.Operator):
self.input_type = None self.input_type = None
if self.input_type: if self.input_type:
if (event.ascii in self.number_options) or (event.value == 'RELEASE' and event.type == 'BACK_SPACE'): if (event.ascii in self.number_options) or (event.value == "RELEASE" and event.type == "BACK_SPACE"):
if event.ascii == '-': if event.ascii == "-":
if self.number_is_negative: if self.number_is_negative:
self.number_is_negative = False self.number_is_negative = False
else: else:
self.number_is_negative = True self.number_is_negative = True
else: else:
if event.ascii == '.' and self.number_output.count('.') == 1: if event.ascii == "." and self.number_output.count(".") == 1:
pass pass
elif event.type == 'BACK_SPACE': elif event.type == "BACK_SPACE":
self.number_input = self.number_input[:-1] self.number_input = self.number_input[:-1]
self.number_output = ''.join(self.number_input) self.number_output = "".join(self.number_input)
else: else:
self.number_input.append(event.ascii) self.number_input.append(event.ascii)
self.number_output = ''.join(self.number_input) self.number_output = "".join(self.number_input)
if self.number_is_negative: if self.number_is_negative:
self.number_output = '-' + self.number_output self.number_output = "-" + self.number_output
if not self.number_is_negative and self.number_output.startswith('-'): if not self.number_is_negative and self.number_output.startswith("-"):
self.number_output = self.number_output[1:] self.number_output = self.number_output[1:]
self.input_panel[self.input_type] = self.number_output self.input_panel[self.input_type] = self.number_output
WallPolylineDecorator.set_input_panel(self.input_panel, self.input_type) WallPolylineDecorator.set_input_panel(self.input_panel, self.input_type)
if self.input_type in {'X', 'Y'}: if self.input_type in {"X", "Y"}:
self.input_panel = WallPolylineDecorator.calculate_distance_and_angle(context, self.is_input_on) self.input_panel = WallPolylineDecorator.calculate_distance_and_angle(context, self.is_input_on)
elif self.input_type in {'D', 'A'}: elif self.input_type in {"D", "A"}:
self.input_panel = WallPolylineDecorator.calculate_x_and_y(context, self.is_input_on) self.input_panel = WallPolylineDecorator.calculate_x_and_y(context, self.is_input_on)
tool.Blender.update_viewport() tool.Blender.update_viewport()
if event.value == "PRESS" and event.type in {"RET", "NUMPAD_ENTER"}:
if event.value == 'PRESS' and event.type in {'RET', 'NUMPAD_ENTER'}: tool.Snaping.insert_polyline_point(float(self.input_panel["X"]), float(self.input_panel["Y"]))
tool.Snaping.insert_polyline_point(float(self.input_panel['X']), float(self.input_panel['Y']))
self.is_input_on = False self.is_input_on = False
self.input_type = None self.input_type = None
WallPolylineDecorator.set_input_panel(self.input_panel, self.input_type) WallPolylineDecorator.set_input_panel(self.input_panel, self.input_type)
@@ -595,19 +581,19 @@ class DrawPolylineWall(bpy.types.Operator):
def invoke(self, context, event): def invoke(self, context, event):
if context.space_data.type == 'VIEW_3D': if context.space_data.type == "VIEW_3D":
WallPolylineDecorator.install(context) WallPolylineDecorator.install(context)
self.input_panel = {'X': '', 'Y': '', 'D': '', 'A': ''} self.input_panel = {"X": "", "Y": "", "D": "", "A": ""}
WallPolylineDecorator.set_input_panel(self.input_panel, self.input_type) WallPolylineDecorator.set_input_panel(self.input_panel, self.input_type)
self.objs_2d_bbox = [] self.objs_2d_bbox = []
self.visible_objs = self.get_visible_objects(context) self.visible_objs = self.get_visible_objects(context)
for obj in self.visible_objs: # TODO check if matrix is needed here for obj in self.visible_objs: # TODO check if matrix is needed here
self.objs_2d_bbox.append(self.get_objects_2d_bounding_boxes(context, obj)) self.objs_2d_bbox.append(self.get_objects_2d_bounding_boxes(context, obj))
context.window_manager.modal_handler_add(self) context.window_manager.modal_handler_add(self)
return {'RUNNING_MODAL'} return {"RUNNING_MODAL"}
else: else:
self.report({'WARNING'}, "Active space must be a View3d") self.report({"WARNING"}, "Active space must be a View3d")
return {'CANCELLED'} return {"CANCELLED"}
class DumbWallAligner: class DumbWallAligner: