PHXH
/
unity-mcp
51
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #15 from justinpbarnett/bug/material-saving 

material saving fix
main
Justin P Barnett 2025-03-19 18:12:00 -04:00 committed by GitHub
parent 9edb03d1b3 c2be2bfa34
commit 03c12d0f3c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 80 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

68
Editor/Commands/MaterialCommandHandler.cs
View File

@ -2,6 +2,8 @@ using UnityEngine;
using Newtonsoft.Json.Linq;
using UnityEngine.Rendering.Universal;
using UnityEngine.Rendering;
using UnityEditor;
using System.IO;
namespace MCPServer.Editor.Commands
{
@ -22,26 +24,72 @@ namespace MCPServer.Editor.Commands
// Check if URP is being used
bool isURP = GraphicsSettings.currentRenderPipeline is UniversalRenderPipelineAsset;
// Create material with appropriate shader based on render pipeline
Material material;
if (isURP)
Material material = null;
string materialName = (string)@params["material_name"];
bool createIfMissing = (bool)(@params["create_if_missing"] ?? true);
string materialPath = null;
// If material name is specified, try to find or create it
if (!string.IsNullOrEmpty(materialName))
{
material = new Material(Shader.Find("Universal Render Pipeline/Lit"));
// Ensure Materials folder exists
const string materialsFolder = "Assets/Materials";
if (!Directory.Exists(materialsFolder))
{
Directory.CreateDirectory(materialsFolder);
}
materialPath = $"{materialsFolder}/{materialName}.mat";
material = AssetDatabase.LoadAssetAtPath<Material>(materialPath);
if (material == null && createIfMissing)
{
// Create new material with appropriate shader
material = new Material(isURP ? Shader.Find("Universal Render Pipeline/Lit") : Shader.Find("Standard"));
material.name = materialName;
// Save the material asset
AssetDatabase.CreateAsset(material, materialPath);
AssetDatabase.SaveAssets();
}
else if (material == null)
{
throw new System.Exception($"Material '{materialName}' not found and create_if_missing is false.");
}
}
else
{
material = new Material(Shader.Find("Standard"));
// Create a temporary material if no name specified
material = new Material(isURP ? Shader.Find("Universal Render Pipeline/Lit") : Shader.Find("Standard"));
}
if (@params.ContainsKey("material_name")) material.name = (string)@params["material_name"];
// Apply color if specified
if (@params.ContainsKey("color"))
{
var colorArray = (JArray)@params["color"] ?? throw new System.Exception("Invalid color parameter.");
if (colorArray.Count != 3) throw new System.Exception("Color must be an array of 3 floats [r, g, b].");
material.color = new Color((float)colorArray[0], (float)colorArray[1], (float)colorArray[2]);
var colorArray = (JArray)@params["color"];
if (colorArray.Count < 3 || colorArray.Count > 4)
throw new System.Exception("Color must be an array of 3 (RGB) or 4 (RGBA) floats.");
Color color = new Color(
(float)colorArray[0],
(float)colorArray[1],
(float)colorArray[2],
colorArray.Count > 3 ? (float)colorArray[3] : 1.0f
);
material.color = color;
// If this is a saved material, make sure to save the color change
if (!string.IsNullOrEmpty(materialPath))
{
EditorUtility.SetDirty(material);
AssetDatabase.SaveAssets();
}
}
// Apply the material to the renderer
renderer.material = material;
return new { material_name = material.name };
return new { material_name = material.name, path = materialPath };
}
}
}

31
Python/tools/material_tools.py
View File

@ -1,5 +1,5 @@
from mcp.server.fastmcp import FastMCP, Context
from typing import List
from typing import List, Optional
from unity_connection import get_unity_connection
def register_material_tools(mcp: FastMCP):
@ -9,18 +9,22 @@ def register_material_tools(mcp: FastMCP):
def set_material(
ctx: Context,
object_name: str,
material_name: str = None,
color: List[float] = None,
material_name: Optional[str] = None,
color: Optional[List[float]] = None,
create_if_missing: bool = True
) -> str:
"""
Apply or create a material for a game object.
Apply or create a material for a game object. If material_name is provided,
the material will be saved as a shared asset in the Materials folder.
Args:
object_name: Target game object.
material_name: Optional material name.
color: Optional [R, G, B] values (0.0-1.0).
material_name: Optional material name. If provided, creates/uses a shared material asset.
color: Optional [R, G, B] or [R, G, B, A] values (0.0-1.0).
create_if_missing: Whether to create the material if it doesn't exist (default: True).
Returns:
str: Status message indicating success or failure.
"""
try:
unity = get_unity_connection()
@ -63,14 +67,23 @@ def register_material_tools(mcp: FastMCP):
return f"Error: Color {channel} value must be in the range 0.0-1.0, but got {value}."
# Set up parameters for the command
params = {"object_name": object_name}
params = {
"object_name": object_name,
"create_if_missing": create_if_missing
}
if material_name:
params["material_name"] = material_name
params["create_if_missing"] = create_if_missing
if color:
params["color"] = color
result = unity.send_command("SET_MATERIAL", params)
return f"Applied material to {object_name}: {result.get('material_name', 'unknown')}"
material_name = result.get("material_name", "unknown")
material_path = result.get("path")
if material_path:
return f"Applied shared material '{material_name}' to {object_name} (saved at {material_path})"
else:
return f"Applied instance material '{material_name}' to {object_name}"
except Exception as e:
return f"Error setting material: {str(e)}"