mirror of
https://github.com/IfcOpenShell/IfcOpenShell.git
synced 2026-09-23 07:52:37 +00:00
Dion Moult
64 lines
2.1 KiB
Python
64 lines
2.1 KiB
Python
import bpy
|
|
from blenderbim.bim.prop import Attribute
|
|
from blenderbim.bim.module.pset.data import Data
|
|
import blenderbim.bim.schema
|
|
from bpy.types import PropertyGroup
|
|
from bpy.props import (
|
|
PointerProperty,
|
|
StringProperty,
|
|
EnumProperty,
|
|
BoolProperty,
|
|
IntProperty,
|
|
FloatProperty,
|
|
FloatVectorProperty,
|
|
CollectionProperty,
|
|
)
|
|
|
|
|
|
psetnames = {}
|
|
qtonames = {}
|
|
|
|
|
|
def getPsetNames(self, context):
|
|
global psetnames
|
|
obj = context.active_object
|
|
if "/" in obj.name:
|
|
ifc_class = obj.name.split("/")[0]
|
|
if ifc_class not in psetnames:
|
|
psets = blenderbim.bim.schema.ifc.psetqto.get_applicable(ifc_class, pset_only=True)
|
|
psetnames[ifc_class] = [(p.Name, p.Name, "") for p in psets]
|
|
assigned_names = [
|
|
Data.psets[p]["Name"] for p in Data.products[obj.BIMObjectProperties.ifc_definition_id]["psets"]
|
|
]
|
|
return [p for p in psetnames[ifc_class] if p[0] not in assigned_names]
|
|
return []
|
|
|
|
|
|
def getMaterialPsetNames(self, context):
|
|
global psetnames
|
|
ifc_class = "IfcMaterial"
|
|
if ifc_class not in psetnames:
|
|
psets = blenderbim.bim.schema.ifc.psetqto.get_applicable(ifc_class, pset_only=True)
|
|
psetnames[ifc_class] = [(p.Name, p.Name, "") for p in psets]
|
|
return psetnames[ifc_class]
|
|
|
|
|
|
def getQtoNames(self, context):
|
|
global qtonames
|
|
if "/" in context.active_object.name:
|
|
ifc_class = context.active_object.name.split("/")[0]
|
|
if ifc_class not in qtonames:
|
|
psets = blenderbim.bim.schema.ifc.psetqto.get_applicable(ifc_class, qto_only=True)
|
|
qtonames[ifc_class] = [(p.Name, p.Name, "") for p in psets]
|
|
return qtonames[ifc_class]
|
|
return []
|
|
|
|
|
|
class PsetProperties(PropertyGroup):
|
|
active_pset_id: IntProperty(name="Active Pset ID")
|
|
active_pset_name: StringProperty(name="Pset Name")
|
|
properties: CollectionProperty(name="Properties", type=Attribute)
|
|
pset_name: EnumProperty(items=getPsetNames, name="Pset Name")
|
|
qto_name: EnumProperty(items=getQtoNames, name="Qto Name")
|
|
material_pset_name: EnumProperty(items=getMaterialPsetNames, name="Pset Name")
|