Rename ClassificationReferencesData -> ObjectClassificationsData

To make it more clear what element's classification data it's storing.
This commit is contained in:
Andrej730
2025-07-31 12:20:51 +05:00
parent 37a62e3004
commit 9cc5338602
3 changed files with 11 additions and 11 deletions
@@ -27,7 +27,7 @@ from bonsai.bim.ifc import IfcStore
def refresh(): def refresh():
ClassificationsData.is_loaded = False ClassificationsData.is_loaded = False
ClassificationReferencesData.is_loaded = False ObjectClassificationsData.is_loaded = False
MaterialClassificationsData.is_loaded = False MaterialClassificationsData.is_loaded = False
CostClassificationsData.is_loaded = False CostClassificationsData.is_loaded = False
@@ -92,7 +92,7 @@ class ReferencesData:
return [(str(e.id()), e.Name, "") for e in tool.Ifc.get().by_type("IfcClassification")] return [(str(e.id()), e.Name, "") for e in tool.Ifc.get().by_type("IfcClassification")]
class ClassificationReferencesData(ReferencesData): class ObjectClassificationsData(ReferencesData):
data: dict[str, Any] = {} data: dict[str, Any] = {}
is_loaded = False is_loaded = False
@@ -19,7 +19,7 @@
import bpy import bpy
import bonsai.tool as tool import bonsai.tool as tool
from bonsai.bim.prop import Attribute from bonsai.bim.prop import Attribute
from bonsai.bim.module.classification.data import ClassificationsData, ClassificationReferencesData from bonsai.bim.module.classification.data import ClassificationsData, ObjectClassificationsData
from bpy.types import PropertyGroup from bpy.types import PropertyGroup
from bpy.props import ( from bpy.props import (
PointerProperty, PointerProperty,
@@ -43,9 +43,9 @@ def get_available_classifications(
def get_classifications(self, context): def get_classifications(self, context):
if not ClassificationReferencesData.is_loaded: if not ObjectClassificationsData.is_loaded:
ClassificationReferencesData.load() ObjectClassificationsData.load()
return ClassificationReferencesData.data["classifications"] return ObjectClassificationsData.data["classifications"]
def get_classification_source( def get_classification_source(
@@ -24,7 +24,7 @@ import bonsai.bim.module.classification.prop as classification_prop
from bpy.types import Panel, UIList from bpy.types import Panel, UIList
from bonsai.bim.module.classification.data import ( from bonsai.bim.module.classification.data import (
ClassificationsData, ClassificationsData,
ClassificationReferencesData, ObjectClassificationsData,
MaterialClassificationsData, MaterialClassificationsData,
CostClassificationsData, CostClassificationsData,
) )
@@ -115,7 +115,7 @@ class ReferenceUI:
layout: bpy.types.UILayout layout: bpy.types.UILayout
data: type[ data: type[
Union[ Union[
ClassificationReferencesData, ObjectClassificationsData,
MaterialClassificationsData, MaterialClassificationsData,
CostClassificationsData, CostClassificationsData,
] ]
@@ -266,9 +266,9 @@ class BIM_PT_classification_references(Panel, ReferenceUI):
return bool(tool.Ifc.get_entity(context.active_object)) return bool(tool.Ifc.get_entity(context.active_object))
def draw(self, context): def draw(self, context):
if not ClassificationReferencesData.is_loaded: if not ObjectClassificationsData.is_loaded:
ClassificationReferencesData.load() ObjectClassificationsData.load()
self.data = ClassificationReferencesData self.data = ObjectClassificationsData
self.obj = context.active_object.name self.obj = context.active_object.name
self.obj_type = "Object" self.obj_type = "Object"
self.draw_ui(context) self.draw_ui(context)