chore: import upstream snapshot with attribution
This commit is contained in:
wehub-resource-sync
@@ -0,0 +1,437 @@
|
||||
using System;
|
||||
using System.Collections.Generic;
|
||||
using Newtonsoft.Json.Linq;
|
||||
using UnityEditor;
|
||||
using UnityEngine;
|
||||
using MCPForUnity.Editor.Helpers;
|
||||
using MCPForUnity.Runtime.Helpers;
|
||||
|
||||
namespace MCPForUnity.Editor.Tools.Physics
|
||||
{
|
||||
internal static class PhysicsRigidbodyOps
|
||||
{
|
||||
private static readonly HashSet<string> Valid3DKeys = new HashSet<string>
|
||||
{
|
||||
"mass", "drag", "lineardamping", "angulardrag", "angulardamping",
|
||||
"usegravity", "iskinematic", "interpolation", "collisiondetectionmode", "constraints"
|
||||
};
|
||||
|
||||
private static readonly HashSet<string> Valid2DKeys = new HashSet<string>
|
||||
{
|
||||
"mass", "gravityscale", "drag", "lineardamping", "angulardrag", "angulardamping",
|
||||
"bodytype", "simulated", "collisiondetectionmode", "constraints"
|
||||
};
|
||||
|
||||
public static object GetRigidbody(JObject @params)
|
||||
{
|
||||
var p = new ToolParams(@params);
|
||||
|
||||
var targetToken = p.GetRaw("target");
|
||||
if (targetToken == null || string.IsNullOrEmpty(targetToken.ToString()))
|
||||
return new ErrorResponse("'target' parameter is required.");
|
||||
|
||||
string searchMethod = p.Get("search_method");
|
||||
string dimensionParam = p.Get("dimension")?.ToLowerInvariant();
|
||||
|
||||
var go = GameObjectLookup.FindByTarget(targetToken, searchMethod ?? "by_name");
|
||||
if (go == null)
|
||||
return new ErrorResponse($"GameObject not found: '{targetToken}'.");
|
||||
|
||||
bool has3D = go.GetComponent<Rigidbody>() != null;
|
||||
bool has2D = go.GetComponent<Rigidbody2D>() != null;
|
||||
|
||||
if (!has3D && !has2D)
|
||||
return new ErrorResponse($"No Rigidbody or Rigidbody2D found on '{go.name}'.");
|
||||
|
||||
bool is2D;
|
||||
if (dimensionParam == "2d")
|
||||
is2D = true;
|
||||
else if (dimensionParam == "3d")
|
||||
is2D = false;
|
||||
else
|
||||
is2D = has2D && !has3D;
|
||||
|
||||
if (is2D)
|
||||
return GetRigidbody2D(go);
|
||||
|
||||
return GetRigidbody3D(go);
|
||||
}
|
||||
|
||||
private static object GetRigidbody3D(GameObject go)
|
||||
{
|
||||
var rb = go.GetComponent<Rigidbody>();
|
||||
if (rb == null)
|
||||
return new ErrorResponse($"No Rigidbody found on '{go.name}'.");
|
||||
|
||||
var data = new Dictionary<string, object>
|
||||
{
|
||||
["target"] = go.name,
|
||||
["instanceID"] = go.GetInstanceIDCompat(),
|
||||
["dimension"] = "3d",
|
||||
["mass"] = rb.mass,
|
||||
#if UNITY_6000_0_OR_NEWER
|
||||
["linearDamping"] = rb.linearDamping,
|
||||
["angularDamping"] = rb.angularDamping,
|
||||
#else
|
||||
["linearDamping"] = rb.drag,
|
||||
["angularDamping"] = rb.angularDrag,
|
||||
#endif
|
||||
["useGravity"] = rb.useGravity,
|
||||
["isKinematic"] = rb.isKinematic,
|
||||
["position"] = new[] { rb.position.x, rb.position.y, rb.position.z },
|
||||
["rotation"] = new[] { rb.rotation.x, rb.rotation.y, rb.rotation.z, rb.rotation.w },
|
||||
#if UNITY_6000_0_OR_NEWER
|
||||
["velocity"] = new[] { rb.linearVelocity.x, rb.linearVelocity.y, rb.linearVelocity.z },
|
||||
#else
|
||||
["velocity"] = new[] { rb.velocity.x, rb.velocity.y, rb.velocity.z },
|
||||
#endif
|
||||
["angularVelocity"] = new[] { rb.angularVelocity.x, rb.angularVelocity.y, rb.angularVelocity.z },
|
||||
["interpolation"] = rb.interpolation.ToString(),
|
||||
["collisionDetectionMode"] = rb.collisionDetectionMode.ToString(),
|
||||
["constraints"] = rb.constraints.ToString(),
|
||||
["isSleeping"] = rb.IsSleeping(),
|
||||
["centerOfMass"] = new[] { rb.centerOfMass.x, rb.centerOfMass.y, rb.centerOfMass.z },
|
||||
["maxAngularVelocity"] = rb.maxAngularVelocity,
|
||||
#if UNITY_6000_0_OR_NEWER
|
||||
["maxLinearVelocity"] = rb.maxLinearVelocity,
|
||||
#endif
|
||||
};
|
||||
|
||||
return new
|
||||
{
|
||||
success = true,
|
||||
message = $"Retrieved Rigidbody state for '{go.name}'.",
|
||||
data
|
||||
};
|
||||
}
|
||||
|
||||
private static object GetRigidbody2D(GameObject go)
|
||||
{
|
||||
var rb2d = go.GetComponent<Rigidbody2D>();
|
||||
if (rb2d == null)
|
||||
return new ErrorResponse($"No Rigidbody2D found on '{go.name}'.");
|
||||
|
||||
var data = new Dictionary<string, object>
|
||||
{
|
||||
["target"] = go.name,
|
||||
["instanceID"] = go.GetInstanceIDCompat(),
|
||||
["dimension"] = "2d",
|
||||
["mass"] = rb2d.mass,
|
||||
["gravityScale"] = rb2d.gravityScale,
|
||||
#if UNITY_6000_0_OR_NEWER
|
||||
["linearDamping"] = rb2d.linearDamping,
|
||||
["angularDamping"] = rb2d.angularDamping,
|
||||
#else
|
||||
["linearDamping"] = rb2d.drag,
|
||||
["angularDamping"] = rb2d.angularDrag,
|
||||
#endif
|
||||
["bodyType"] = rb2d.bodyType.ToString(),
|
||||
["simulated"] = rb2d.simulated,
|
||||
["position"] = new[] { rb2d.position.x, rb2d.position.y },
|
||||
["rotation"] = rb2d.rotation,
|
||||
#if UNITY_6000_0_OR_NEWER
|
||||
["velocity"] = new[] { rb2d.linearVelocity.x, rb2d.linearVelocity.y },
|
||||
#else
|
||||
["velocity"] = new[] { rb2d.velocity.x, rb2d.velocity.y },
|
||||
#endif
|
||||
["angularVelocity"] = rb2d.angularVelocity,
|
||||
["collisionDetectionMode"] = rb2d.collisionDetectionMode.ToString(),
|
||||
["constraints"] = rb2d.constraints.ToString(),
|
||||
["isSleeping"] = rb2d.IsSleeping(),
|
||||
["isAwake"] = rb2d.IsAwake(),
|
||||
["centerOfMass"] = new[] { rb2d.centerOfMass.x, rb2d.centerOfMass.y },
|
||||
};
|
||||
|
||||
return new
|
||||
{
|
||||
success = true,
|
||||
message = $"Retrieved Rigidbody2D state for '{go.name}'.",
|
||||
data
|
||||
};
|
||||
}
|
||||
|
||||
public static object ConfigureRigidbody(JObject @params)
|
||||
{
|
||||
var p = new ToolParams(@params);
|
||||
|
||||
var targetToken = p.GetRaw("target");
|
||||
if (targetToken == null || string.IsNullOrEmpty(targetToken.ToString()))
|
||||
return new ErrorResponse("'target' parameter is required.");
|
||||
|
||||
var properties = p.GetRaw("properties") as JObject;
|
||||
if (properties == null || properties.Count == 0)
|
||||
return new ErrorResponse("'properties' parameter is required and must be a non-empty object.");
|
||||
|
||||
string searchMethod = p.Get("search_method");
|
||||
string dimensionParam = p.Get("dimension")?.ToLowerInvariant();
|
||||
|
||||
var go = GameObjectLookup.FindByTarget(targetToken, searchMethod ?? "by_name");
|
||||
if (go == null)
|
||||
return new ErrorResponse($"GameObject not found: '{targetToken}'.");
|
||||
|
||||
bool has3D = go.GetComponent<Rigidbody>() != null;
|
||||
bool has2D = go.GetComponent<Rigidbody2D>() != null;
|
||||
bool is2D;
|
||||
|
||||
if (dimensionParam == "2d")
|
||||
is2D = true;
|
||||
else if (dimensionParam == "3d")
|
||||
is2D = false;
|
||||
else
|
||||
is2D = has2D && !has3D;
|
||||
|
||||
if (is2D)
|
||||
return ConfigureRigidbody2D(go, properties);
|
||||
|
||||
return ConfigureRigidbody3D(go, properties);
|
||||
}
|
||||
|
||||
private static object ConfigureRigidbody3D(GameObject go, JObject properties)
|
||||
{
|
||||
var rb = go.GetComponent<Rigidbody>();
|
||||
if (rb == null)
|
||||
return new ErrorResponse($"No Rigidbody found on '{go.name}'.");
|
||||
|
||||
// Validate all keys before applying any changes
|
||||
var unknown = new List<string>();
|
||||
foreach (var prop in properties.Properties())
|
||||
{
|
||||
string key = prop.Name.ToLowerInvariant().Replace("_", "");
|
||||
if (!Valid3DKeys.Contains(key))
|
||||
unknown.Add(prop.Name);
|
||||
}
|
||||
if (unknown.Count > 0)
|
||||
return new ErrorResponse(
|
||||
$"Unknown Rigidbody property(ies): {string.Join(", ", unknown)}.");
|
||||
|
||||
Undo.RecordObject(rb, "Configure Rigidbody");
|
||||
|
||||
var changed = new List<string>();
|
||||
|
||||
foreach (var prop in properties.Properties())
|
||||
{
|
||||
string key = prop.Name.ToLowerInvariant().Replace("_", "");
|
||||
switch (key)
|
||||
{
|
||||
case "mass":
|
||||
rb.mass = prop.Value.Value<float>();
|
||||
changed.Add("mass");
|
||||
break;
|
||||
case "drag":
|
||||
case "lineardamping":
|
||||
#if UNITY_6000_0_OR_NEWER
|
||||
rb.linearDamping = prop.Value.Value<float>();
|
||||
#else
|
||||
rb.drag = prop.Value.Value<float>();
|
||||
#endif
|
||||
changed.Add("linearDamping");
|
||||
break;
|
||||
case "angulardrag":
|
||||
case "angulardamping":
|
||||
#if UNITY_6000_0_OR_NEWER
|
||||
rb.angularDamping = prop.Value.Value<float>();
|
||||
#else
|
||||
rb.angularDrag = prop.Value.Value<float>();
|
||||
#endif
|
||||
changed.Add("angularDamping");
|
||||
break;
|
||||
case "usegravity":
|
||||
rb.useGravity = prop.Value.Value<bool>();
|
||||
changed.Add("useGravity");
|
||||
break;
|
||||
case "iskinematic":
|
||||
rb.isKinematic = prop.Value.Value<bool>();
|
||||
changed.Add("isKinematic");
|
||||
break;
|
||||
case "interpolation":
|
||||
{
|
||||
string val = prop.Value.ToString();
|
||||
if (Enum.TryParse<RigidbodyInterpolation>(val, true, out var interp))
|
||||
{
|
||||
rb.interpolation = interp;
|
||||
changed.Add("interpolation");
|
||||
}
|
||||
else
|
||||
{
|
||||
return new ErrorResponse(
|
||||
$"Invalid interpolation value: '{val}'. Valid: None, Interpolate, Extrapolate.");
|
||||
}
|
||||
break;
|
||||
}
|
||||
case "collisiondetectionmode":
|
||||
{
|
||||
string val = prop.Value.ToString();
|
||||
if (Enum.TryParse<CollisionDetectionMode>(val, true, out var mode))
|
||||
{
|
||||
rb.collisionDetectionMode = mode;
|
||||
changed.Add("collisionDetectionMode");
|
||||
}
|
||||
else
|
||||
{
|
||||
return new ErrorResponse(
|
||||
$"Invalid collisionDetectionMode: '{val}'. Valid: Discrete, Continuous, ContinuousDynamic, ContinuousSpeculative.");
|
||||
}
|
||||
break;
|
||||
}
|
||||
case "constraints":
|
||||
{
|
||||
var token = prop.Value;
|
||||
if (token.Type == JTokenType.Integer)
|
||||
{
|
||||
rb.constraints = (RigidbodyConstraints)token.Value<int>();
|
||||
changed.Add("constraints");
|
||||
}
|
||||
else
|
||||
{
|
||||
string val = token.ToString();
|
||||
if (Enum.TryParse<RigidbodyConstraints>(val, true, out var c))
|
||||
{
|
||||
rb.constraints = c;
|
||||
changed.Add("constraints");
|
||||
}
|
||||
else
|
||||
{
|
||||
return new ErrorResponse(
|
||||
$"Invalid constraints value: '{val}'. Use enum name (e.g. 'FreezePositionX, FreezeRotationY') or int flags.");
|
||||
}
|
||||
}
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
EditorUtility.SetDirty(rb);
|
||||
|
||||
return new
|
||||
{
|
||||
success = true,
|
||||
message = $"Configured Rigidbody on '{go.name}': {string.Join(", ", changed)}.",
|
||||
data = new { target = go.name, dimension = "3d", changed }
|
||||
};
|
||||
}
|
||||
|
||||
private static object ConfigureRigidbody2D(GameObject go, JObject properties)
|
||||
{
|
||||
var rb2d = go.GetComponent<Rigidbody2D>();
|
||||
if (rb2d == null)
|
||||
return new ErrorResponse($"No Rigidbody2D found on '{go.name}'.");
|
||||
|
||||
// Validate all keys before applying any changes
|
||||
var unknown = new List<string>();
|
||||
foreach (var prop in properties.Properties())
|
||||
{
|
||||
string key = prop.Name.ToLowerInvariant().Replace("_", "");
|
||||
if (!Valid2DKeys.Contains(key))
|
||||
unknown.Add(prop.Name);
|
||||
}
|
||||
if (unknown.Count > 0)
|
||||
return new ErrorResponse(
|
||||
$"Unknown Rigidbody2D property(ies): {string.Join(", ", unknown)}.");
|
||||
|
||||
Undo.RecordObject(rb2d, "Configure Rigidbody2D");
|
||||
|
||||
var changed = new List<string>();
|
||||
|
||||
foreach (var prop in properties.Properties())
|
||||
{
|
||||
string key = prop.Name.ToLowerInvariant().Replace("_", "");
|
||||
switch (key)
|
||||
{
|
||||
case "mass":
|
||||
rb2d.mass = prop.Value.Value<float>();
|
||||
changed.Add("mass");
|
||||
break;
|
||||
case "gravityscale":
|
||||
rb2d.gravityScale = prop.Value.Value<float>();
|
||||
changed.Add("gravityScale");
|
||||
break;
|
||||
case "drag":
|
||||
case "lineardamping":
|
||||
#if UNITY_6000_0_OR_NEWER
|
||||
rb2d.linearDamping = prop.Value.Value<float>();
|
||||
#else
|
||||
rb2d.drag = prop.Value.Value<float>();
|
||||
#endif
|
||||
changed.Add("linearDamping");
|
||||
break;
|
||||
case "angulardrag":
|
||||
case "angulardamping":
|
||||
#if UNITY_6000_0_OR_NEWER
|
||||
rb2d.angularDamping = prop.Value.Value<float>();
|
||||
#else
|
||||
rb2d.angularDrag = prop.Value.Value<float>();
|
||||
#endif
|
||||
changed.Add("angularDamping");
|
||||
break;
|
||||
case "bodytype":
|
||||
{
|
||||
string val = prop.Value.ToString();
|
||||
if (Enum.TryParse<RigidbodyType2D>(val, true, out var bt))
|
||||
{
|
||||
rb2d.bodyType = bt;
|
||||
changed.Add("bodyType");
|
||||
}
|
||||
else
|
||||
{
|
||||
return new ErrorResponse(
|
||||
$"Invalid bodyType: '{val}'. Valid: Dynamic, Kinematic, Static.");
|
||||
}
|
||||
break;
|
||||
}
|
||||
case "simulated":
|
||||
rb2d.simulated = prop.Value.Value<bool>();
|
||||
changed.Add("simulated");
|
||||
break;
|
||||
case "collisiondetectionmode":
|
||||
{
|
||||
string val = prop.Value.ToString();
|
||||
if (Enum.TryParse<CollisionDetectionMode2D>(val, true, out var mode))
|
||||
{
|
||||
rb2d.collisionDetectionMode = mode;
|
||||
changed.Add("collisionDetectionMode");
|
||||
}
|
||||
else
|
||||
{
|
||||
return new ErrorResponse(
|
||||
$"Invalid collisionDetectionMode: '{val}'. Valid: Discrete, Continuous.");
|
||||
}
|
||||
break;
|
||||
}
|
||||
case "constraints":
|
||||
{
|
||||
var token = prop.Value;
|
||||
if (token.Type == JTokenType.Integer)
|
||||
{
|
||||
rb2d.constraints = (RigidbodyConstraints2D)token.Value<int>();
|
||||
changed.Add("constraints");
|
||||
}
|
||||
else
|
||||
{
|
||||
string val = token.ToString();
|
||||
if (Enum.TryParse<RigidbodyConstraints2D>(val, true, out var c))
|
||||
{
|
||||
rb2d.constraints = c;
|
||||
changed.Add("constraints");
|
||||
}
|
||||
else
|
||||
{
|
||||
return new ErrorResponse(
|
||||
$"Invalid constraints value: '{val}'. Use enum name (e.g. 'FreezePositionX, FreezeRotation') or int flags.");
|
||||
}
|
||||
}
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
EditorUtility.SetDirty(rb2d);
|
||||
|
||||
return new
|
||||
{
|
||||
success = true,
|
||||
message = $"Configured Rigidbody2D on '{go.name}': {string.Join(", ", changed)}.",
|
||||
data = new { target = go.name, dimension = "2d", changed }
|
||||
};
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user