New node to get IFC attribute

This commit is contained in:
Dion Moult
2020-10-13 20:30:00 +11:00
parent 56209fa0ec
commit 88e20582dd
2 changed files with 43 additions and 1 deletions
+2 -1
View File
@@ -5,7 +5,7 @@ bl_info = {
"blender": (2, 90, 0),
"location": "Node Editor",
"category": "Node",
"description": "An extension to Sverchok to generate IFC data",
"description": "An extension to Sverchok to work with IFC data",
"warning": "",
}
@@ -36,6 +36,7 @@ def nodes_index():
("ifc.remove", "SvIfcRemove"),
("ifc.generate_guid", "SvIfcGenerateGuid"),
("ifc.get_property", "SvIfcGetProperty"),
("ifc.get_attribute", "SvIfcGetAttribute"),
("ifc.select_blender_objects", "SvIfcSelectBlenderObjects"),
])]
@@ -0,0 +1,41 @@
import bpy
import ifcopenshell
import ifcopenshell.util.element
import ifcsverchok.helper
from bpy.props import StringProperty
from sverchok.node_tree import SverchCustomTreeNode
from sverchok.data_structure import updateNode
class SvIfcGetAttribute(bpy.types.Node, SverchCustomTreeNode, ifcsverchok.helper.SvIfcCore):
bl_idname = 'SvIfcGetAttribute'
bl_label = 'IFC Get Attribute'
entity: StringProperty(name='entity', update=updateNode)
attribute_name: StringProperty(name='attribute_name', update=updateNode)
def sv_init(self, context):
self.inputs.new('SvStringsSocket', 'entity').prop_name = 'entity'
self.inputs.new('SvStringsSocket', 'attribute_name').prop_name = 'attribute_name'
self.outputs.new('SvStringsSocket', 'value')
def process(self):
self.sv_input_names = ['entity', 'attribute_name']
self.value_out = []
super().process()
self.outputs['value'].sv_set(self.value_out)
def process_ifc(self, entity, attribute_name):
try:
if attribute_name.isnumeric():
self.value_out.append(entity[int(attribute_name)])
else:
self.value_out.append(entity.get_info()[attribute_name])
except:
pass
def register():
bpy.utils.register_class(SvIfcGetAttribute)
def unregister():
bpy.utils.unregister_class(SvIfcGetAttribute)