2025-12-01

2026-03-17 14:58:51 -06:00
parent 183e865f8b
commit 4b82b57113
6846 changed files with 954887 additions and 162606 deletions
@@ -0,0 +1,76 @@
+# SPDX-FileCopyrightText: 2025 Blender Studio Tools Authors
+#
+# SPDX-License-Identifier: GPL-3.0-or-later
+
+from bpy.types import Modifier, Object, PropertyGroup
+from bpy.props import StringProperty, CollectionProperty, IntProperty, PointerProperty
+from .operators import geomod_get_identifier
+
+
+class GeoNodeShapeKey(PropertyGroup):
+    name: StringProperty(
+        description="Name of the modifier, storage object, etc", override={'LIBRARY_OVERRIDABLE'}
+    )
+
+    # On overridden objects, this stores the local object for sculpting.
+    # Used for deletion and back and forth switching.
+    storage_object: PointerProperty(
+        name="Storage Object", type=Object, override={'LIBRARY_OVERRIDABLE'}
+    )
+
+    @property
+    def ob_name(self) -> str:
+        return self.id_data.name + "." + self.name
+
+    @property
+    def modifier(self) -> Modifier:
+        for m in self.id_data.modifiers:
+            if m.type == 'NODES':
+                identifier = geomod_get_identifier(m, 'Sculpt')
+                if not identifier:
+                    continue
+                sculpt_ob = m[identifier]
+                if not sculpt_ob:
+                    continue
+                if sculpt_ob == self.storage_object:
+                    return m
+
+    @property
+    def other_affected_objects(self) -> list[Object]:
+        if not self.storage_object:
+            return []
+
+        ret = []
+        for target in self.storage_object.geonode_shapekey_targets:
+            if target.obj in [None, self.id_data]:
+                continue
+            ret.append(target.obj)
+
+        return ret
+
+    @property
+    def index(self):
+        obj = self.id_data
+        for i, gnsk in enumerate(obj.geonode_shapekeys):
+            if gnsk == self:
+                return i
+
+
+class GNSK_TargetObject(PropertyGroup):
+    obj: PointerProperty(name="Target Object", type=Object)
+
+
+registry = [GeoNodeShapeKey, GNSK_TargetObject]
+
+
+def register():
+    Object.geonode_shapekeys = CollectionProperty(
+        type=GeoNodeShapeKey, override={'LIBRARY_OVERRIDABLE', 'USE_INSERTION'}
+    )
+    Object.geonode_shapekey_index = IntProperty(
+        options={'LIBRARY_EDITABLE'}, override={'LIBRARY_OVERRIDABLE'}
+    )
+
+    # On local objects for sculpting, this stores the overridden object.
+    # Used for swapping back and forth between the two objects.
+    Object.geonode_shapekey_targets = CollectionProperty(type=GNSK_TargetObject)