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():
ClassificationsData.is_loaded = False
ClassificationReferencesData.is_loaded = False
ObjectClassificationsData.is_loaded = False
MaterialClassificationsData.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")]
class ClassificationReferencesData(ReferencesData):
class ObjectClassificationsData(ReferencesData):
data: dict[str, Any] = {}
is_loaded = False
@@ -19,7 +19,7 @@
import bpy
import bonsai.tool as tool
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.props import (
PointerProperty,
@@ -43,9 +43,9 @@ def get_available_classifications(
def get_classifications(self, context):
if not ClassificationReferencesData.is_loaded:
ClassificationReferencesData.load()
return ClassificationReferencesData.data["classifications"]
if not ObjectClassificationsData.is_loaded:
ObjectClassificationsData.load()
return ObjectClassificationsData.data["classifications"]
def get_classification_source(
@@ -24,7 +24,7 @@ import bonsai.bim.module.classification.prop as classification_prop
from bpy.types import Panel, UIList
from bonsai.bim.module.classification.data import (
ClassificationsData,
ClassificationReferencesData,
ObjectClassificationsData,
MaterialClassificationsData,
CostClassificationsData,
)
@@ -115,7 +115,7 @@ class ReferenceUI:
layout: bpy.types.UILayout
data: type[
Union[
ClassificationReferencesData,
ObjectClassificationsData,
MaterialClassificationsData,
CostClassificationsData,
]
@@ -266,9 +266,9 @@ class BIM_PT_classification_references(Panel, ReferenceUI):
return bool(tool.Ifc.get_entity(context.active_object))
def draw(self, context):
if not ClassificationReferencesData.is_loaded:
ClassificationReferencesData.load()
self.data = ClassificationReferencesData
if not ObjectClassificationsData.is_loaded:
ObjectClassificationsData.load()
self.data = ObjectClassificationsData
self.obj = context.active_object.name
self.obj_type = "Object"
self.draw_ui(context)