feat: improving the user experience

bioxelnodes/bioxelutils/container.py

@@ -6,12 +6,12 @@ from mathutils import Matrix, Vector
 
 
 from .layer import Layer, layer_to_obj, obj_to_layer
-from ..nodes import custom_nodes
-from .utils import (get_container_layer_objs,
-                    get_layer_prop_value,
-                    get_nodes_by_type,
-                    move_node_to_node)
-
+from .common import (get_container_layer_objs,
+                     get_layer_prop_value,
+                     get_nodes_by_type,
+                     move_node_to_node)
+from .node import add_node_to_graph
+from ..utils import get_use_link
 
 NODE_TYPE = {
     "label": "BioxelNodes_MaskByLabel",
@@ -82,13 +82,17 @@ def add_layers(layers: list[Layer],
                cache_dir: str):
 
     node_group = container_obj.modifiers[0].node_group
-    output_node = get_nodes_by_type(node_group,
-                                    'NodeGroupOutput')[0]
+    try:
+        output_node = get_nodes_by_type(node_group,
+                                        'NodeGroupOutput')[0]
+    except:
+        output_node = node_group.nodes.new("NodeGroupOutput")
 
     for i, layer in enumerate(layers):
         layer_obj = layer_to_obj(layer, container_obj, cache_dir)
-        fetch_node = custom_nodes.add_node(node_group,
-                                           "BioxelNodes_FetchLayer")
+        fetch_node = add_node_to_graph("FetchLayer",
+                                       node_group,
+                                       get_use_link())
         fetch_node.label = get_layer_prop_value(layer_obj, "name")
         fetch_node.inputs[0].default_value = layer_obj
 
@@ -104,6 +108,7 @@ def add_layers(layers: list[Layer],
 
 def container_to_obj(container: Container,
                      scene_scale: float,
+                     step_size: float,
                      cache_dir: str):
     # Wrapper a Container
 
@@ -134,11 +139,26 @@ def container_to_obj(container: Container,
     container_obj.name = container.name
     container_obj.data.name = container.name
     container_obj.show_in_front = True
+
+    container_obj.lock_location[0] = True
+    container_obj.lock_location[1] = True
+    container_obj.lock_location[2] = True
+
+    container_obj.lock_rotation[0] = True
+    container_obj.lock_rotation[1] = True
+    container_obj.lock_rotation[2] = True
+
+    container_obj.lock_scale[0] = True
+    container_obj.lock_scale[1] = True
+    container_obj.lock_scale[2] = True
+
     container_obj['bioxel_container'] = True
+    container_obj["scene_scale"] = scene_scale
+    container_obj["step_size"] = step_size
 
     modifier = container_obj.modifiers.new("GeometryNodes", 'NODES')
     node_group = bpy.data.node_groups.new('GeometryNodes', 'GeometryNodeTree')
-    node_group.interface.new_socket(name="Component",
+    node_group.interface.new_socket(name="Output",
                                     in_out="OUTPUT",
                                     socket_type="NodeSocketGeometry")
     modifier.node_group = node_group