get/set animation node state

Author: tny-cesilco

MCPForUnity/Editor/Tools/Animation/ControllerCreate.cs

@@ -422,6 +422,105 @@ public static object AssignToGameObject(JObject @params)
             };
         }
 
+        // Reads node graph positions for every state in all layers (recurses into
+        // sub-state-machines). Returns [{ name, x, y, layer }] so a caller can analyze
+        // the current layout before sending back a revised one.
+        public static object GetStatePositions(JObject @params)
+        {
+            var controller = LoadController(@params);
+            if (controller == null)
+                return ControllerNotFoundError(@params);
+
+            var nodes = new List<object>();
+            for (int li = 0; li < controller.layers.Length; li++)
+                CollectPositions(controller.layers[li].stateMachine, li, nodes);
+
+            return new
+            {
+                success = true,
+                message = $"Read {nodes.Count} state position(s).",
+                data = new { count = nodes.Count, nodes }
+            };
+        }
+
+        private static void CollectPositions(AnimatorStateMachine sm, int layer, List<object> outList)
+        {
+            var children = sm.states;
+            for (int i = 0; i < children.Length; i++)
+                outList.Add(new
+                {
+                    name = children[i].state.name,
+                    x = children[i].position.x,
+                    y = children[i].position.y,
+                    layer
+                });
+            foreach (var sub in sm.stateMachines)
+                CollectPositions(sub.stateMachine, layer, outList);
+        }
+
+        // Sets node graph positions from a 'positions' array of { name, x, y }.
+        // Matches states by name across all layers (recurses into sub-state-machines)
+        // and reassigns stateMachine.states so the edits persist on the asset.
+        public static object SetStatePositions(JObject @params)
+        {
+            var controller = LoadController(@params);
+            if (controller == null)
+                return ControllerNotFoundError(@params);
+
+            if (!(@params["positions"] is JArray positions) || positions.Count == 0)
+                return new { success = false, message = "'positions' array is required: [{ name, x, y }, ...]" };
+
+            var want = new Dictionary<string, Vector2>();
+            foreach (var token in positions)
+            {
+                string name = token["name"]?.ToString();
+                if (string.IsNullOrEmpty(name))
+                    continue;
+                float x = token["x"]?.ToObject<float>() ?? 0f;
+                float y = token["y"]?.ToObject<float>() ?? 0f;
+                want[name] = new Vector2(x, y);
+            }
+            if (want.Count == 0)
+                return new { success = false, message = "No valid entries in 'positions' (each needs a 'name')." };
+
+            var matched = new HashSet<string>();
+            for (int li = 0; li < controller.layers.Length; li++)
+                ApplyPositions(controller.layers[li].stateMachine, want, matched);
+
+            EditorUtility.SetDirty(controller);
+            AssetDatabase.SaveAssets();
+
+            var unmatched = want.Keys.Where(k => !matched.Contains(k)).ToList();
+            return new
+            {
+                success = true,
+                message = $"Positioned {matched.Count} state(s); {unmatched.Count} name(s) unmatched.",
+                data = new
+                {
+                    matched = matched.Count,
+                    requested = want.Count,
+                    unmatched
+                }
+            };
+        }
+
+        private static void ApplyPositions(AnimatorStateMachine sm, Dictionary<string, Vector2> want, HashSet<string> matched)
+        {
+            var children = sm.states;
+            for (int i = 0; i < children.Length; i++)
+            {
+                if (want.TryGetValue(children[i].state.name, out var p))
+                {
+                    children[i].position = new Vector3(p.x, p.y, 0f);
+                    matched.Add(children[i].state.name);
+                }
+            }
+            sm.states = children; // reassign so position edits persist
+
+            foreach (var sub in sm.stateMachines)
+                ApplyPositions(sub.stateMachine, want, matched);
+        }
+
         private static AnimatorController LoadController(JObject @params)
         {
             string controllerPath = @params["controllerPath"]?.ToString();

MCPForUnity/Editor/Tools/Animation/ManageAnimation.cs

@@ -217,6 +217,8 @@ private static object HandleControllerAction(JObject @params, string action)
             {
                 case "create": return ControllerCreate.Create(@params);
                 case "add_state": return ControllerCreate.AddState(@params);
+                case "set_state_positions": return ControllerCreate.SetStatePositions(@params);
+                case "get_state_positions": return ControllerCreate.GetStatePositions(@params);
                 case "add_transition": return ControllerCreate.AddTransition(@params);
                 case "add_parameter": return ControllerCreate.AddParameter(@params);
                 case "get_info": return ControllerCreate.GetInfo(@params);
@@ -228,7 +230,7 @@ private static object HandleControllerAction(JObject @params, string action)
                 case "create_blend_tree_2d": return ControllerBlendTrees.CreateBlendTree2D(@params);
                 case "add_blend_tree_child": return ControllerBlendTrees.AddBlendTreeChild(@params);
                 default:
-                    return new { success = false, message = $"Unknown controller action: {action}. Valid: create, add_state, add_transition, add_parameter, get_info, assign, add_layer, remove_layer, set_layer_weight, create_blend_tree_1d, create_blend_tree_2d, add_blend_tree_child" };
+                    return new { success = false, message = $"Unknown controller action: {action}. Valid: create, add_state, set_state_positions, get_state_positions, add_transition, add_parameter, get_info, assign, add_layer, remove_layer, set_layer_weight, create_blend_tree_1d, create_blend_tree_2d, add_blend_tree_child" };
             }
         }
 

Server/src/services/tools/manage_animation.py

@@ -15,7 +15,9 @@
 ]
 
 CONTROLLER_ACTIONS = [
-    "controller_create", "controller_add_state", "controller_add_transition",
+    "controller_create", "controller_add_state",
+    "controller_set_state_positions", "controller_get_state_positions",
+    "controller_add_transition",
     "controller_add_parameter", "controller_get_info", "controller_assign",
     "controller_add_layer", "controller_remove_layer", "controller_set_layer_weight",
     "controller_create_blend_tree_1d", "controller_create_blend_tree_2d", "controller_add_blend_tree_child",