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

New feature to have multiple section planes at any angle

Browse Source
This commit is contained in:
Dion Moult
2020-04-25 17:40:24 +10:00
parent b250472d60
commit 3671953472
1 changed files with 35 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/ifcblenderexport/blenderbim/bim/operator.py
+35 -6
View File
@@ -14,7 +14,7 @@ from . import bcf
from . import ifc from . import ifc
from bpy_extras.io_utils import ImportHelper from bpy_extras.io_utils import ImportHelper
from itertools import cycle from itertools import cycle
from mathutils import Vector, Matrix from mathutils import Vector, Matrix, Euler
from math import radians, atan, tan from math import radians, atan, tan
from pathlib import Path from pathlib import Path
@@ -1820,6 +1820,8 @@ class AddSectionPlane(bpy.types.Operator):
if not self.has_section_override_node(): if not self.has_section_override_node():
self.create_section_compare_node() self.create_section_compare_node()
self.create_section_override_node(obj) self.create_section_override_node(obj)
else:
self.append_obj_to_section_override_node(obj)
self.add_default_material_if_none_exists() self.add_default_material_if_none_exists()
self.override_materials() self.override_materials()
return {'FINISHED'} return {'FINISHED'}
@@ -1828,6 +1830,8 @@ class AddSectionPlane(bpy.types.Operator):
section = bpy.data.objects.new('Section', None) section = bpy.data.objects.new('Section', None)
section.empty_display_type = 'SINGLE_ARROW' section.empty_display_type = 'SINGLE_ARROW'
section.empty_display_size = 5 section.empty_display_size = 5
section.rotation_euler = Euler((radians(180.0), 0.0, 0.0), 'XYZ')
section.show_in_front = True
collection = bpy.data.collections.get('Sections') collection = bpy.data.collections.get('Sections')
if not collection: if not collection:
collection = bpy.data.collections.new('Sections') collection = bpy.data.collections.new('Sections')
@@ -1880,6 +1884,7 @@ class AddSectionPlane(bpy.types.Operator):
transparent = group.nodes.new(type='ShaderNodeBsdfTransparent') transparent = group.nodes.new(type='ShaderNodeBsdfTransparent')
section_mix = group.nodes.new(type='ShaderNodeMixShader') section_mix = group.nodes.new(type='ShaderNodeMixShader')
section_mix.name = 'Section Mix'
group.links.new(transparent.outputs['BSDF'], section_mix.inputs[1]) group.links.new(transparent.outputs['BSDF'], section_mix.inputs[1])
group.links.new(backfacing_mix.outputs['Shader'], section_mix.inputs[2]) group.links.new(backfacing_mix.outputs['Shader'], section_mix.inputs[2])
@@ -1888,12 +1893,33 @@ class AddSectionPlane(bpy.types.Operator):
cut_obj = group.nodes.new(type='ShaderNodeTexCoord') cut_obj = group.nodes.new(type='ShaderNodeTexCoord')
cut_obj.object = obj cut_obj.object = obj
section_and = group.nodes.new(type='ShaderNodeGroup') section_compare = group.nodes.new(type='ShaderNodeGroup')
section_and.node_tree = bpy.data.node_groups.get('Section Compare') section_compare.node_tree = bpy.data.node_groups.get('Section Compare')
section_compare.name = 'Last Section Compare'
value = group.nodes.new(type='ShaderNodeValue') value = group.nodes.new(type='ShaderNodeValue')
group.links.new(cut_obj.outputs['Object'], section_and.inputs[0]) value.name = 'Mock Section'
group.links.new(value.outputs['Value'], section_and.inputs[1]) group.links.new(cut_obj.outputs['Object'], section_compare.inputs[0])
group.links.new(section_and.outputs['Value'], section_mix.inputs['Fac']) group.links.new(value.outputs['Value'], section_compare.inputs[1])
group.links.new(section_compare.outputs['Value'], section_mix.inputs['Fac'])
def append_obj_to_section_override_node(self, obj):
group = bpy.data.node_groups.get('Section Override')
cut_obj = group.nodes.new(type='ShaderNodeTexCoord')
cut_obj.object = obj
section_compare = group.nodes.new(type='ShaderNodeGroup')
section_compare.node_tree = bpy.data.node_groups.get('Section Compare')
last_compare = group.nodes.get('Last Section Compare')
last_compare.name = 'Section Compare'
mock_section = group.nodes.get('Mock Section')
section_mix = group.nodes.get('Section Mix')
group.links.new(last_compare.outputs['Value'], section_compare.inputs[0])
group.links.new(mock_section.outputs['Value'], section_compare.inputs[1])
group.links.new(cut_obj.outputs['Object'], last_compare.inputs[1])
group.links.new(section_compare.outputs['Value'], section_mix.inputs['Fac'])
section_compare.name = 'Last Section Compare'
def add_default_material_if_none_exists(self): def add_default_material_if_none_exists(self):
material = bpy.data.materials.get('Section Override') material = bpy.data.materials.get('Section Override')
@@ -1910,11 +1936,14 @@ class AddSectionPlane(bpy.types.Operator):
override = bpy.data.node_groups.get('Section Override') override = bpy.data.node_groups.get('Section Override')
for material in bpy.data.materials: for material in bpy.data.materials:
material.use_nodes = True material.use_nodes = True
if material.node_tree.nodes.get('Section Override'):
continue
material.blend_method = 'HASHED' material.blend_method = 'HASHED'
material.shadow_method = 'HASHED' material.shadow_method = 'HASHED'
material_output = material.node_tree.nodes['Material Output'] material_output = material.node_tree.nodes['Material Output']
from_socket = material_output.inputs['Surface'].links[0].from_socket from_socket = material_output.inputs['Surface'].links[0].from_socket
section_override = material.node_tree.nodes.new(type='ShaderNodeGroup') section_override = material.node_tree.nodes.new(type='ShaderNodeGroup')
section_override.name = 'Section Override'
section_override.node_tree = override section_override.node_tree = override
material.node_tree.links.new(from_socket, section_override.inputs[0]) material.node_tree.links.new(from_socket, section_override.inputs[0])
material.node_tree.links.new(section_override.outputs[0], material_output.inputs['Surface']) material.node_tree.links.new(section_override.outputs[0], material_output.inputs['Surface'])