feat(mcp): add input tools — load_nest, import_dxf, create_drawing
Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
213
OpenNest.Mcp/Tools/InputTools.cs
Normal file
213
OpenNest.Mcp/Tools/InputTools.cs
Normal file
@@ -0,0 +1,213 @@
|
||||
using System.Collections.Generic;
|
||||
using System.ComponentModel;
|
||||
using System.IO;
|
||||
using System.Text;
|
||||
using ModelContextProtocol.Server;
|
||||
using OpenNest.Converters;
|
||||
using OpenNest.Geometry;
|
||||
using OpenNest.IO;
|
||||
using CncProgram = OpenNest.CNC.Program;
|
||||
|
||||
namespace OpenNest.Mcp.Tools
|
||||
{
|
||||
public class InputTools
|
||||
{
|
||||
private readonly NestSession _session;
|
||||
|
||||
public InputTools(NestSession session)
|
||||
{
|
||||
_session = session;
|
||||
}
|
||||
|
||||
[McpServerTool(Name = "load_nest")]
|
||||
[Description("Load a .nest zip file into the session. Returns a summary of plates, parts, and drawings.")]
|
||||
public string LoadNest([Description("Absolute path to the .nest file")] string path)
|
||||
{
|
||||
if (!File.Exists(path))
|
||||
return $"Error: file not found: {path}";
|
||||
|
||||
var reader = new NestReader(path);
|
||||
var nest = reader.Read();
|
||||
_session.Nest = nest;
|
||||
|
||||
var sb = new StringBuilder();
|
||||
sb.AppendLine($"Loaded nest: {nest.Name}");
|
||||
sb.AppendLine($"Units: {nest.Units}");
|
||||
sb.AppendLine($"Plates: {nest.Plates.Count}");
|
||||
|
||||
for (var i = 0; i < nest.Plates.Count; i++)
|
||||
{
|
||||
var plate = nest.Plates[i];
|
||||
var work = plate.WorkArea();
|
||||
sb.AppendLine($" Plate {i}: {plate.Size.Width:F1} x {plate.Size.Height:F1}, " +
|
||||
$"parts={plate.Parts.Count}, " +
|
||||
$"utilization={plate.Utilization():P1}, " +
|
||||
$"work area={work.Width:F1} x {work.Height:F1}");
|
||||
}
|
||||
|
||||
sb.AppendLine($"Drawings: {nest.Drawings.Count}");
|
||||
|
||||
foreach (var dwg in nest.Drawings)
|
||||
{
|
||||
var bbox = dwg.Program.BoundingBox();
|
||||
sb.AppendLine($" {dwg.Name}: bbox={bbox.Width:F2} x {bbox.Height:F2}, " +
|
||||
$"required={dwg.Quantity.Required}, nested={dwg.Quantity.Nested}");
|
||||
}
|
||||
|
||||
return sb.ToString();
|
||||
}
|
||||
|
||||
[McpServerTool(Name = "import_dxf")]
|
||||
[Description("Import a DXF file as a new drawing. Returns drawing name and bounding box.")]
|
||||
public string ImportDxf(
|
||||
[Description("Absolute path to the DXF file")] string path,
|
||||
[Description("Name for the drawing (defaults to filename without extension)")] string name = null)
|
||||
{
|
||||
if (!File.Exists(path))
|
||||
return $"Error: file not found: {path}";
|
||||
|
||||
var importer = new DxfImporter();
|
||||
|
||||
if (!importer.GetGeometry(path, out var geometry))
|
||||
return "Error: failed to read DXF file";
|
||||
|
||||
if (geometry.Count == 0)
|
||||
return "Error: no geometry found in DXF file";
|
||||
|
||||
var pgm = ConvertGeometry.ToProgram(geometry);
|
||||
|
||||
if (pgm == null)
|
||||
return "Error: failed to convert geometry to program";
|
||||
|
||||
var drawingName = name ?? Path.GetFileNameWithoutExtension(path);
|
||||
var drawing = new Drawing(drawingName, pgm);
|
||||
_session.Drawings.Add(drawing);
|
||||
|
||||
var bbox = pgm.BoundingBox();
|
||||
return $"Imported drawing '{drawingName}': bbox={bbox.Width:F2} x {bbox.Height:F2}";
|
||||
}
|
||||
|
||||
[McpServerTool(Name = "create_drawing")]
|
||||
[Description("Create a drawing from a built-in shape or G-code string. Shape can be: rectangle, circle, l_shape, t_shape, gcode.")]
|
||||
public string CreateDrawing(
|
||||
[Description("Name for the drawing")] string name,
|
||||
[Description("Shape type: rectangle, circle, l_shape, t_shape, gcode")] string shape,
|
||||
[Description("Width of the shape (not used for circle or gcode)")] double width = 10,
|
||||
[Description("Height of the shape (not used for circle or gcode)")] double height = 10,
|
||||
[Description("Radius for circle shape")] double radius = 5,
|
||||
[Description("G-code string (only used when shape is 'gcode')")] string gcode = null)
|
||||
{
|
||||
CncProgram pgm;
|
||||
|
||||
switch (shape.ToLower())
|
||||
{
|
||||
case "rectangle":
|
||||
pgm = CreateRectangle(width, height);
|
||||
break;
|
||||
|
||||
case "circle":
|
||||
pgm = CreateCircle(radius);
|
||||
break;
|
||||
|
||||
case "l_shape":
|
||||
pgm = CreateLShape(width, height);
|
||||
break;
|
||||
|
||||
case "t_shape":
|
||||
pgm = CreateTShape(width, height);
|
||||
break;
|
||||
|
||||
case "gcode":
|
||||
if (string.IsNullOrWhiteSpace(gcode))
|
||||
return "Error: gcode parameter is required when shape is 'gcode'";
|
||||
pgm = ParseGcode(gcode);
|
||||
if (pgm == null)
|
||||
return "Error: failed to parse G-code";
|
||||
break;
|
||||
|
||||
default:
|
||||
return $"Error: unknown shape '{shape}'. Use: rectangle, circle, l_shape, t_shape, gcode";
|
||||
}
|
||||
|
||||
var drawing = new Drawing(name, pgm);
|
||||
_session.Drawings.Add(drawing);
|
||||
|
||||
var bbox = pgm.BoundingBox();
|
||||
return $"Created drawing '{name}': bbox={bbox.Width:F2} x {bbox.Height:F2}";
|
||||
}
|
||||
|
||||
private static CncProgram CreateRectangle(double width, double height)
|
||||
{
|
||||
var entities = new List<Entity>
|
||||
{
|
||||
new Line(0, 0, width, 0),
|
||||
new Line(width, 0, width, height),
|
||||
new Line(width, height, 0, height),
|
||||
new Line(0, height, 0, 0)
|
||||
};
|
||||
|
||||
return ConvertGeometry.ToProgram(entities);
|
||||
}
|
||||
|
||||
private static CncProgram CreateCircle(double radius)
|
||||
{
|
||||
var entities = new List<Entity>
|
||||
{
|
||||
new Circle(0, 0, radius)
|
||||
};
|
||||
|
||||
return ConvertGeometry.ToProgram(entities);
|
||||
}
|
||||
|
||||
private static CncProgram CreateLShape(double width, double height)
|
||||
{
|
||||
var hw = width / 2;
|
||||
var hh = height / 2;
|
||||
|
||||
var entities = new List<Entity>
|
||||
{
|
||||
new Line(0, 0, width, 0),
|
||||
new Line(width, 0, width, hh),
|
||||
new Line(width, hh, hw, hh),
|
||||
new Line(hw, hh, hw, height),
|
||||
new Line(hw, height, 0, height),
|
||||
new Line(0, height, 0, 0)
|
||||
};
|
||||
|
||||
return ConvertGeometry.ToProgram(entities);
|
||||
}
|
||||
|
||||
private static CncProgram CreateTShape(double width, double height)
|
||||
{
|
||||
var stemWidth = width / 3;
|
||||
var topHeight = height / 3;
|
||||
var stemLeft = (width - stemWidth) / 2;
|
||||
var stemRight = stemLeft + stemWidth;
|
||||
var stemBottom = 0.0;
|
||||
var stemTop = height - topHeight;
|
||||
|
||||
var entities = new List<Entity>
|
||||
{
|
||||
new Line(stemLeft, stemBottom, stemRight, stemBottom),
|
||||
new Line(stemRight, stemBottom, stemRight, stemTop),
|
||||
new Line(stemRight, stemTop, width, stemTop),
|
||||
new Line(width, stemTop, width, height),
|
||||
new Line(width, height, 0, height),
|
||||
new Line(0, height, 0, stemTop),
|
||||
new Line(0, stemTop, stemLeft, stemTop),
|
||||
new Line(stemLeft, stemTop, stemLeft, stemBottom)
|
||||
};
|
||||
|
||||
return ConvertGeometry.ToProgram(entities);
|
||||
}
|
||||
|
||||
private static CncProgram ParseGcode(string gcode)
|
||||
{
|
||||
using var stream = new MemoryStream(Encoding.UTF8.GetBytes(gcode));
|
||||
var reader = new ProgramReader(stream);
|
||||
var pgm = reader.Read();
|
||||
reader.Close();
|
||||
return pgm;
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user