mirror of
https://github.com/IfcOpenShell/IfcOpenShell.git
synced 2026-08-10 01:41:57 +00:00
Dion Moult
28 lines
1.2 KiB
Python
28 lines
1.2 KiB
Python
class Usecase:
|
|
def __init__(self, file, **settings):
|
|
self.file = file
|
|
self.settings = {"style": None, "ifc_class": "IfcSurfaceStyleRendering", "attributes": {}}
|
|
for key, value in settings.items():
|
|
self.settings[key] = value
|
|
|
|
def execute(self):
|
|
for key, value in self.settings["attributes"].items():
|
|
if key == "SurfaceColour" and value:
|
|
self.settings["attributes"][key] = self.create_colour_rgb(value)
|
|
if key == "DiffuseColour" and isinstance(value, dict):
|
|
self.settings["attributes"][key] = self.create_colour_rgb(value)
|
|
|
|
style_item = self.file.create_entity(self.settings["ifc_class"], **self.settings["attributes"])
|
|
styles = list(self.settings["style"].Styles or [])
|
|
|
|
duplicate_items = [s for s in styles if s.is_a(self.settings["ifc_class"])]
|
|
for duplicate_item in duplicate_items:
|
|
self.file.remove(duplicate_item)
|
|
|
|
styles.append(style_item)
|
|
self.settings["style"].Styles = styles
|
|
return style_item
|
|
|
|
def create_colour_rgb(self, value):
|
|
return self.file.createIfcColourRgb(value["Name"], value["Red"], value["Green"], value["Blue"])
|