OpenNest/OpenNest.Tests/CutOffs/CutOffTests.cs

using System.Linq;
using OpenNest.CNC;
using OpenNest.Geometry;

namespace OpenNest.Tests.CutOffs;

public class CutOffTests
{
    [Fact]
    public void Drawing_IsCutOff_DefaultsFalse()
    {
        var drawing = new Drawing("test", new Program());
        Assert.False(drawing.IsCutOff);
    }

    [Fact]
    public void Plate_CutOffPart_DoesNotIncrementQuantity()
    {
        var drawing = new Drawing("cutoff", new Program()) { IsCutOff = true };
        var plate = new Plate(100, 100);
        plate.Parts.Add(new Part(drawing));
        Assert.Equal(0, drawing.Quantity.Nested);
    }

    [Fact]
    public void Plate_Utilization_ExcludesCutOffParts()
    {
        var pgm = new Program();
        pgm.Codes.Add(new RapidMove(new Vector(0, 0)));
        pgm.Codes.Add(new LinearMove(new Vector(10, 0)));
        pgm.Codes.Add(new LinearMove(new Vector(10, 10)));
        pgm.Codes.Add(new LinearMove(new Vector(0, 10)));
        pgm.Codes.Add(new LinearMove(new Vector(0, 0)));
        var realDrawing = new Drawing("real", pgm);
        var cutoffDrawing = new Drawing("cutoff", new Program()) { IsCutOff = true };

        var plate = new Plate(100, 100);
        plate.Parts.Add(new Part(realDrawing));
        plate.Parts.Add(new Part(cutoffDrawing));

        var utilization = plate.Utilization();
        var expected = realDrawing.Area / plate.Area();
        Assert.Equal(expected, utilization, 5);
    }

    [Fact]
    public void Plate_HasOverlappingParts_SkipsCutOffParts()
    {
        var pgm = new Program();
        pgm.Codes.Add(new RapidMove(new Vector(0, 0)));
        pgm.Codes.Add(new LinearMove(new Vector(10, 0)));
        pgm.Codes.Add(new LinearMove(new Vector(10, 10)));
        pgm.Codes.Add(new LinearMove(new Vector(0, 10)));
        pgm.Codes.Add(new LinearMove(new Vector(0, 0)));

        var realDrawing = new Drawing("real", pgm);
        var cutoffDrawing = new Drawing("cutoff", pgm) { IsCutOff = true };

        var plate = new Plate(100, 100);
        plate.Parts.Add(new Part(realDrawing));
        plate.Parts.Add(new Part(cutoffDrawing));

        var hasOverlap = plate.HasOverlappingParts(out var pts);
        Assert.False(hasOverlap);
    }

    [Fact]
    public void CutOff_VerticalCut_GeneratesFullLineOnEmptyPlate()
    {
        var plate = new Plate(100, 50);
        var settings = new CutOffSettings();
        var cutoff = new CutOff(new Vector(25, 20), CutOffAxis.Vertical);

        cutoff.Regenerate(plate, settings);

        Assert.NotNull(cutoff.Drawing);
        Assert.True(cutoff.Drawing.IsCutOff);
        Assert.True(cutoff.Drawing.Program.Codes.Count > 0);
    }

    [Fact]
    public void CutOff_HorizontalCut_GeneratesFullLineOnEmptyPlate()
    {
        var plate = new Plate(100, 50);
        var settings = new CutOffSettings();
        var cutoff = new CutOff(new Vector(25, 20), CutOffAxis.Horizontal);

        cutoff.Regenerate(plate, settings);

        var codes = cutoff.Drawing.Program.Codes;
        Assert.Equal(2, codes.Count);
    }

    [Fact]
    public void CutOff_VerticalCut_TrimsAroundPart()
    {
        // Create a 10x10 part at the origin, then move it to (20,20)
        // so the bounding box is Box(20,20,10,10) and doesn't span the origin.
        var pgm = new Program();
        pgm.Codes.Add(new RapidMove(new Vector(0, 0)));
        pgm.Codes.Add(new LinearMove(new Vector(10, 0)));
        pgm.Codes.Add(new LinearMove(new Vector(10, 10)));
        pgm.Codes.Add(new LinearMove(new Vector(0, 10)));
        pgm.Codes.Add(new LinearMove(new Vector(0, 0)));
        var drawing = new Drawing("sq", pgm);

        var plate = new Plate(50, 50);
        var part = Part.CreateAtOrigin(drawing);
        part.Location = new Vector(20, 20);
        plate.Parts.Add(part);

        // Vertical cut at X=25 runs along Y from 0 to 50.
        // Part BB at (20,20,10,10) with clearance 1 → exclusion X=[19,31], Y=[19,31].
        // X=25 is within [19,31] so exclusion applies: skip Y=[19,31].
        // Segments: (0, 19) and (31, 50) → 2 segments → 4 codes.
        var settings = new CutOffSettings { PartClearance = 1.0 };
        var cutoff = new CutOff(new Vector(25, 10), CutOffAxis.Vertical);
        cutoff.Regenerate(plate, settings);

        var codes = cutoff.Drawing.Program.Codes;
        Assert.Equal(4, codes.Count);
    }

    [Theory]
    [InlineData(0.3, 19.0)] // Narrower than twice the clearance: the slot closes up.
    [InlineData(4.0, 24.0)] // Wide slot: the cut runs in to 1 short of the slot's end.
    public void CutOff_UpASlot_KeepsClearanceFromPart(double slotWidth, double firstEnd)
    {
        // 10x10 part at (20,20) with a 5-deep slot up from the bottom edge, centered
        // on the cut line.
        var h = slotWidth / 2;
        var pgm = new Program();
        pgm.Codes.Add(new RapidMove(new Vector(0, 0)));
        pgm.Codes.Add(new LinearMove(new Vector(5 - h, 0)));
        pgm.Codes.Add(new LinearMove(new Vector(5 - h, 5)));
        pgm.Codes.Add(new LinearMove(new Vector(5 + h, 5)));
        pgm.Codes.Add(new LinearMove(new Vector(5 + h, 0)));
        pgm.Codes.Add(new LinearMove(new Vector(10, 0)));
        pgm.Codes.Add(new LinearMove(new Vector(10, 10)));
        pgm.Codes.Add(new LinearMove(new Vector(0, 10)));
        pgm.Codes.Add(new LinearMove(new Vector(0, 0)));

        var plate = new Plate(50, 50);
        var part = Part.CreateAtOrigin(new Drawing("slot", pgm));
        part.Location = new Vector(20, 20);
        plate.Parts.Add(part);

        var settings = new CutOffSettings { PartClearance = 1.0 };
        var cutoff = new CutOff(new Vector(25, 10), CutOffAxis.Vertical);
        cutoff.Regenerate(plate, settings, Plate.BuildPerimeterCache(plate));

        var ys = cutoff
            .Drawing.Program.Codes.OfType<Motion>()
            .Select(m => m.EndPoint.Y)
            .OrderBy(y => y)
            .ToList();

        Assert.Equal(4, ys.Count);
        Assert.Equal(0, ys[0], 6);
        // A closed slot leaves a shallow dent at its mouth: the cut stops 1 from the
        // mouth corners, at 20 - sqrt(1 - 0.15^2) = 19.011.
        Assert.InRange(ys[1], firstEnd - 0.02, firstEnd + 0.02);
        Assert.InRange(ys[2], 30.999, 31.02);
        Assert.Equal(50, ys[3], 6);
    }

    [Fact]
    public void CutOff_ShortSegment_FilteredByMinLength()
    {
        var pgm = new Program();
        pgm.Codes.Add(new RapidMove(new Vector(20, 0.02)));
        pgm.Codes.Add(new LinearMove(new Vector(30, 0.02)));
        pgm.Codes.Add(new LinearMove(new Vector(30, 10)));
        pgm.Codes.Add(new LinearMove(new Vector(20, 10)));
        pgm.Codes.Add(new LinearMove(new Vector(20, 0.02)));
        var drawing = new Drawing("sq", pgm);

        var plate = new Plate(50, 50);
        plate.Parts.Add(new Part(drawing));

        var settings = new CutOffSettings { PartClearance = 0.0, MinSegmentLength = 0.05 };
        var cutoff = new CutOff(new Vector(25, 10), CutOffAxis.Vertical);
        cutoff.Regenerate(plate, settings);

        var rapidCount = cutoff.Drawing.Program.Codes.Count(c => c is RapidMove);
        var lineCount = cutoff.Drawing.Program.Codes.Count(c => c is LinearMove);
        Assert.Equal(rapidCount, lineCount);
    }

    [Fact]
    public void CutOff_Overtravel_ExtendsFarEnd()
    {
        var plate = new Plate(100, 50);
        var settings = new CutOffSettings { Overtravel = 2.0 };
        var cutoff = new CutOff(new Vector(25, 10), CutOffAxis.Vertical);
        cutoff.Regenerate(plate, settings);

        // Plate(100, 50) = Width=100, Length=50. Vertical cut runs along Y (Width axis).
        // BoundingBox Y extent = Size.Width = 100. With 2" overtravel = 102.
        // Default AwayFromOrigin: RapidMove to near end (0), LinearMove to far end (102).
        var linearMoves = cutoff.Drawing.Program.Codes.OfType<LinearMove>().ToList();
        Assert.Single(linearMoves);
        Assert.Equal(102.0, linearMoves[0].EndPoint.Y, 5);
    }

    [Fact]
    public void CutOff_StartLimit_TruncatesNearEnd()
    {
        var plate = new Plate(100, 50);
        var settings = new CutOffSettings();
        var cutoff = new CutOff(new Vector(25, 10), CutOffAxis.Vertical) { StartLimit = 20.0 };
        cutoff.Regenerate(plate, settings);

        // AwayFromOrigin: RapidMove to near end (StartLimit=20), LinearMove to far end (100).
        var rapidMoves = cutoff.Drawing.Program.Codes.OfType<RapidMove>().ToList();
        Assert.Single(rapidMoves);
        Assert.Equal(20.0, rapidMoves[0].EndPoint.Y, 5);
    }

    [Fact]
    public void CutOff_EndLimit_TruncatesFarEnd()
    {
        var plate = new Plate(100, 50);
        var settings = new CutOffSettings();
        var cutoff = new CutOff(new Vector(25, 10), CutOffAxis.Vertical) { EndLimit = 80.0 };
        cutoff.Regenerate(plate, settings);

        // AwayFromOrigin: RapidMove to near end (0), LinearMove to far end (EndLimit=80).
        var linearMoves = cutoff.Drawing.Program.Codes.OfType<LinearMove>().ToList();
        Assert.Single(linearMoves);
        Assert.Equal(80.0, linearMoves[0].EndPoint.Y, 5);
    }

    [Fact]
    public void CutOff_BothLimits_LShapedCornerCut()
    {
        var plate = new Plate(60, 120);
        var settings = new CutOffSettings { PartClearance = 0 };

        var hCut = new CutOff(new Vector(85, 30), CutOffAxis.Horizontal) { EndLimit = 85.0 };
        hCut.Regenerate(plate, settings);

        var vCut = new CutOff(new Vector(85, 30), CutOffAxis.Vertical) { StartLimit = 30.0 };
        vCut.Regenerate(plate, settings);

        Assert.True(hCut.Drawing.Program.Codes.Count > 0);
        Assert.True(vCut.Drawing.Program.Codes.Count > 0);
    }

    [Fact]
    public void Plate_RegenerateCutOffs_MaterializesParts()
    {
        var plate = new Plate(100, 50);
        var cutoff = new CutOff(new Vector(25, 10), CutOffAxis.Vertical);
        plate.CutOffs.Add(cutoff);

        plate.RegenerateCutOffs(new CutOffSettings());

        Assert.Single(plate.Parts);
        Assert.True(plate.Parts[0].BaseDrawing.IsCutOff);
    }

    [Fact]
    public void Plate_RegenerateCutOffs_ReplacesOldParts()
    {
        var plate = new Plate(100, 50);
        var cutoff = new CutOff(new Vector(25, 10), CutOffAxis.Vertical);
        plate.CutOffs.Add(cutoff);

        var settings = new CutOffSettings();
        plate.RegenerateCutOffs(settings);
        plate.RegenerateCutOffs(settings);

        Assert.Single(plate.Parts);
    }

    [Fact]
    public void Plate_RegenerateCutOffs_DoesNotAffectRegularParts()
    {
        var pgm = new OpenNest.CNC.Program();
        pgm.Codes.Add(new OpenNest.CNC.RapidMove(new Vector(0, 0)));
        pgm.Codes.Add(new OpenNest.CNC.LinearMove(new Vector(5, 5)));
        var drawing = new Drawing("real", pgm);

        var plate = new Plate(100, 50);
        plate.Parts.Add(new Part(drawing));

        var cutoff = new CutOff(new Vector(25, 10), CutOffAxis.Vertical);
        plate.CutOffs.Add(cutoff);

        plate.RegenerateCutOffs(new CutOffSettings());

        Assert.Equal(2, plate.Parts.Count);
        Assert.False(plate.Parts[0].BaseDrawing.IsCutOff);
        Assert.True(plate.Parts[1].BaseDrawing.IsCutOff);
    }
}