mirror of
https://github.com/IfcOpenShell/IfcOpenShell.git
synced 2026-09-16 21:42:19 +00:00
fixed create_entity and minor changes to bmesh_to_ifc
This commit is contained in:
martinaCodes
@@ -1,115 +1,173 @@
|
||||
import bpy
|
||||
|
||||
# IfcSverchok - IFC Sverchok extension
|
||||
# Copyright (C) 2020, 2021 Dion Moult <dion@thinkmoult.com>
|
||||
#
|
||||
# This file is part of IfcSverchok.
|
||||
#
|
||||
# IfcSverchok is free software: you can redistribute it and/or modify
|
||||
# it under the terms of the GNU General Public License as published by
|
||||
# the Free Software Foundation, either version 3 of the License, or
|
||||
# (at your option) any later version.
|
||||
#
|
||||
# IfcSverchok is distributed in the hope that it will be useful,
|
||||
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
# GNU General Public License for more details.
|
||||
#
|
||||
# You should have received a copy of the GNU General Public License
|
||||
# along with IfcSverchok. If not, see <http://www.gnu.org/licenses/>.
|
||||
# from helper import SayHello
|
||||
import ifcopenshell
|
||||
import ifcsverchok.helper
|
||||
from ifcsverchok.ifcstore import SvIfcStore
|
||||
|
||||
# import ifcsverchok.ifc_store
|
||||
# from bpy.props import StringProperty
|
||||
# from sverchok.node_tree import SverchCustomTreeNode
|
||||
# from sverchok.data_structure import updateNode
|
||||
|
||||
import bpy
|
||||
import ifcopenshell
|
||||
import ifcsverchok.helper
|
||||
from bpy.props import StringProperty
|
||||
|
||||
from bpy.props import StringProperty, EnumProperty, IntProperty, BoolProperty, PointerProperty
|
||||
from sverchok.node_tree import SverchCustomTreeNode
|
||||
from sverchok.data_structure import updateNode
|
||||
from sverchok.data_structure import updateNode, flatten_data, repeat_last_for_length
|
||||
|
||||
input_map = {}
|
||||
|
||||
|
||||
class SvIfcCreateEntity(bpy.types.Node, SverchCustomTreeNode, ifcsverchok.helper.SvIfcCore):
|
||||
bl_idname = "SvIfcCreateEntity"
|
||||
bl_label = "IFC Create Entity"
|
||||
file: StringProperty(name="file", update=updateNode)
|
||||
ifc_class: StringProperty(name="ifc_class", update=updateNode)
|
||||
current_ifc_class: StringProperty(name="current_ifc_class")
|
||||
node_dict = {}
|
||||
|
||||
is_scene_dependent = True # if True and is_interactive then the node will be updated upon scene changes
|
||||
|
||||
def refresh_node(self, context):
|
||||
if self.refresh_local:
|
||||
self.process()
|
||||
self.refresh_local = False
|
||||
|
||||
refresh_local: BoolProperty(name="Update Node", description="Update Node", update=refresh_node)
|
||||
|
||||
Names: StringProperty(name="Names", default="", update=updateNode)
|
||||
Descriptions: StringProperty(name="Descriptions", default="", update=updateNode)
|
||||
IfcClass: StringProperty(name="IfcClass", update=updateNode)
|
||||
Representations: StringProperty(name="Representations", default="", update=updateNode)
|
||||
Properties: StringProperty(name="properties", update=updateNode)
|
||||
|
||||
def sv_init(self, context):
|
||||
self.inputs.new("SvStringsSocket", "file").prop_name = "file"
|
||||
self.inputs.new("SvStringsSocket", "ifc_class").prop_name = "ifc_class"
|
||||
self.outputs.new("SvStringsSocket", "file")
|
||||
self.outputs.new("SvStringsSocket", "entity")
|
||||
self.inputs.new("SvStringsSocket", "Names").prop_name = "Names"
|
||||
self.inputs.new("SvStringsSocket", "Descriptions").prop_name = "Descriptions"
|
||||
self.inputs.new("SvStringsSocket", "IfcClass").prop_name = "IfcClass"
|
||||
self.inputs.new("SvStringsSocket", "Representations").prop_name = "Representations"
|
||||
self.inputs.new("SvStringsSocket", "Properties").prop_name = "Properties"
|
||||
self.outputs.new("SvStringsSocket", "Entities")
|
||||
self.outputs.new("SvStringsSocket", "file") # only for testing
|
||||
self.node_dict[hash(self)] = {}
|
||||
|
||||
def process(self):
|
||||
self.sv_input_names = ["file", "ifc_class"]
|
||||
ifc_class = self.inputs["ifc_class"].sv_get()[0][0]
|
||||
if ifc_class:
|
||||
file = self.inputs["file"].sv_get()[0][0]
|
||||
if file:
|
||||
schema_name = file.wrapped_data.schema
|
||||
else:
|
||||
schema_name = "IFC4"
|
||||
self.schema = ifcopenshell.ifcopenshell_wrapper.schema_by_name(schema_name)
|
||||
self.entity_schema = self.schema.declaration_by_name(ifc_class)
|
||||
print("#"*20, "\n running create_entity3 PROCESS()... \n", "#"*20,)
|
||||
print("#"*20, "\n hash(self):", hash(self), "\n", "#"*20,)
|
||||
|
||||
if ifc_class != self.current_ifc_class:
|
||||
self.generate_inputs(ifc_class)
|
||||
try:
|
||||
for i in range(0, len(self.inputs)):
|
||||
input = self.inputs[i].sv_get()
|
||||
except:
|
||||
# This occurs when a blender save file is reloaded
|
||||
self.generate_inputs(ifc_class)
|
||||
|
||||
for i in range(0, self.entity_schema.attribute_count()):
|
||||
self.sv_input_names.append(self.entity_schema.attribute_by_index(i).name())
|
||||
super().process()
|
||||
self.names = self.inputs["Names"].sv_get()
|
||||
self.descriptions = self.inputs["Descriptions"].sv_get()
|
||||
self.ifc_class = self.inputs["IfcClass"].sv_get()[0][0]
|
||||
self.representations = self.inputs["Representations"].sv_get()
|
||||
self.properties = self.inputs["Properties"].sv_get()
|
||||
|
||||
def generate_inputs(self, ifc_class):
|
||||
while len(self.inputs) > 2:
|
||||
self.inputs.remove(self.inputs[-1])
|
||||
for i in range(0, self.entity_schema.attribute_count()):
|
||||
name = self.entity_schema.attribute_by_index(i).name()
|
||||
setattr(SvIfcCreateEntity, name, StringProperty(name=name))
|
||||
self.inputs.new("SvStringsSocket", name).prop_name = name
|
||||
self.current_ifc_class = ifc_class
|
||||
|
||||
def process_ifc(self, file, ifc_class, *attributes):
|
||||
entity = file.create_entity(ifc_class)
|
||||
for i in range(0, len(entity)):
|
||||
self.sv_input_names = [i.name for i in self.inputs]
|
||||
|
||||
if hash(self) not in self.node_dict:
|
||||
self.node_dict[hash(self)] = {} #happens if node is already on canvas when blender loads
|
||||
if not self.node_dict[hash(self)]:
|
||||
self.node_dict[hash(self)].update(dict.fromkeys(self.sv_input_names, 0))
|
||||
print("node_dict: ", self.node_dict)
|
||||
if not self.inputs["IfcClass"].sv_get()[0][0]:
|
||||
raise Exception("Insert IfcClass")
|
||||
return
|
||||
|
||||
edit = False
|
||||
for i in range(len(self.inputs)):
|
||||
input = self.inputs[self.sv_input_names[i]].sv_get(deepcopy=False)
|
||||
# print("input: ", input)
|
||||
# print("self.node_dict[hash(self)][self.inputs[i].name]: ", self.node_dict[hash(self)][self.inputs[i].name])
|
||||
if isinstance(self.node_dict[hash(self)][self.inputs[i].name], list) and input != self.node_dict[hash(self)][self.inputs[i].name]:
|
||||
edit = True
|
||||
self.node_dict[hash(self)][self.inputs[i].name] = input
|
||||
|
||||
if self.refresh_local:
|
||||
edit = True
|
||||
|
||||
|
||||
self.names = repeat_last_for_length(self.names, len(self.representations), deepcopy=False)
|
||||
self.descriptions = repeat_last_for_length(self.descriptions, len(self.representations), deepcopy=False)
|
||||
# print("REPRESENTATION: ", self.representations)
|
||||
# print("IfcClass: ",self.ifc_class)
|
||||
print("Names: ",self.names)
|
||||
print("Descriptions: ",self.descriptions)
|
||||
|
||||
|
||||
|
||||
self.file = SvIfcStore.get_file()
|
||||
|
||||
if self.node_id not in SvIfcStore.id_map:
|
||||
enities = self.create()
|
||||
else:
|
||||
if edit is True:
|
||||
enities = self.edit()
|
||||
else:
|
||||
enities = self.get_existing_element()
|
||||
|
||||
print("Entities: ", enities)
|
||||
print("SvIfcStore.id_map: ", SvIfcStore.id_map)
|
||||
|
||||
self.outputs["Entities"].sv_set(enities)
|
||||
self.outputs["file"].sv_set([[self.file]])
|
||||
|
||||
def create(self):
|
||||
# print("#"*20, "\n running create entity CREATE()... \n")
|
||||
results = []
|
||||
for i, name in enumerate(self.names):
|
||||
print(type(self.names[i][0]))
|
||||
try:
|
||||
value = attributes[i]
|
||||
if isinstance(value, str) and value == "":
|
||||
value = None
|
||||
except:
|
||||
value = None
|
||||
if value is None and entity.is_a("IfcRoot") and i == 0:
|
||||
value = ifcopenshell.guid.new()
|
||||
if value is not None and entity.attribute_name(i) == "Representation":
|
||||
value = file.createIfcProductDefinitionShape(None, None, Representations = [value])
|
||||
#product_shape = ifcopenshell.api.run("geometry.assign_representation", file, product = , representation = value)
|
||||
if value is not None:
|
||||
value = self.cast_value(self.entity_schema.attribute_by_index(i), value)
|
||||
try:
|
||||
entity[i] = value
|
||||
except:
|
||||
raise Exception(
|
||||
"The IFC class {} requires a valid value for {} - you provided {}".format(
|
||||
ifc_class, entity.attribute_name(i), value
|
||||
)
|
||||
)
|
||||
self.entity = ifcopenshell.api.run("root.create_entity", self.file, ifc_class=self.ifc_class, name=self.names[i][0], description=self.descriptions[i][0])
|
||||
try:
|
||||
ifcopenshell.api.run("geometry.assign_representation", self.file, product=self.entity, representation=self.representations[i][0])
|
||||
except IndexError:
|
||||
pass
|
||||
results.append([self.entity])
|
||||
SvIfcStore.id_map.setdefault(self.node_id, []).append(self.entity.id())
|
||||
except:
|
||||
raise Exception("Something went wrong. Cannot create entity.")
|
||||
return results
|
||||
|
||||
def edit(self):
|
||||
results = []
|
||||
id_map_copy = SvIfcStore.id_map[self.node_id].copy()
|
||||
for i, step_id in enumerate(id_map_copy):
|
||||
self.entity = self.file.by_id(step_id)
|
||||
self.entity.Name = self.names[i][0]
|
||||
self.entity.Description = self.descriptions[i][0]
|
||||
|
||||
if self.representations[i][0] and not self.file.by_type("IFCPRODUCTDEFINITIONSHAPE"):
|
||||
ifcopenshell.api.run("geometry.assign_representation", self.file, product=self.entity, representation=self.representations[i][0])
|
||||
elif self.representations[i][0]:
|
||||
self.entity.Representation = self.representations[i][0]
|
||||
else:
|
||||
pass
|
||||
|
||||
self.outputs["file"].sv_set([[file]])
|
||||
self.outputs["entity"].sv_set([[entity]])
|
||||
if self.entity.is_a() != self.ifc_class:
|
||||
SvIfcStore.id_map[self.node_id].remove(step_id)
|
||||
self.entity = ifcopenshell.util.schema.reassign_class(self.file, self.entity, self.ifc_class)
|
||||
SvIfcStore.id_map.setdefault(self.node_id, []).append(self.entity.id())
|
||||
results.append([self.entity])
|
||||
|
||||
def cast_value(self, attribute, value):
|
||||
data_type = self.get_attribute_data_type(attribute.type_of_attribute())
|
||||
if data_type == "integer":
|
||||
value = int(value)
|
||||
return value
|
||||
return results
|
||||
|
||||
def get_attribute_data_type(self, data_type):
|
||||
if hasattr(data_type, "declared_type"):
|
||||
return self.get_attribute_data_type(data_type.declared_type())
|
||||
return data_type
|
||||
def get_existing_element(self):
|
||||
# print("#"*20, "\n running create entity get_existing_element()... \n", "#"*20,)
|
||||
results = []
|
||||
for i, step_id in enumerate(SvIfcStore.id_map[self.node_id]):
|
||||
self.entity = self.file.by_id(step_id)
|
||||
results.append([self.entity])
|
||||
return results
|
||||
|
||||
|
||||
def sv_free(self):
|
||||
|
||||
try:
|
||||
print('DELETING')
|
||||
del SvIfcStore.id_map[self.node_id]
|
||||
del self.node_dict[hash(self)]
|
||||
print('Node was deleted')
|
||||
except KeyError or AttributeError:
|
||||
pass
|
||||
|
||||
|
||||
def register():
|
||||
|
||||
Reference in New Issue
Block a user