OpenNest/OpenNest.Core/Drawing.cs

using System;
using System.Collections.Generic;
using System.Drawing;
using System.Linq;
using System.Threading;
using OpenNest.Bending;
using OpenNest.CNC;
using OpenNest.Converters;
using OpenNest.Geometry;

namespace OpenNest
{
    public class Drawing
    {
        private static int nextId;
        private static int nextColorIndex;
        private Program program;

        public static Color[] PartColors = new Color[]
        {
            Color.FromArgb(205, 92, 92), // Indian Red
            Color.FromArgb(148, 103, 189), // Medium Purple
            Color.FromArgb(75, 180, 175), // Teal
            Color.FromArgb(210, 190, 75), // Goldenrod
            Color.FromArgb(190, 85, 175), // Orchid
            Color.FromArgb(185, 115, 85), // Sienna
            Color.FromArgb(120, 100, 190), // Slate Blue
            Color.FromArgb(200, 100, 140), // Rose
            Color.FromArgb(80, 175, 155), // Sea Green
            Color.FromArgb(195, 160, 85), // Dark Khaki
            Color.FromArgb(175, 95, 160), // Plum
            Color.FromArgb(215, 130, 130), // Light Coral
        };

        public static bool UseGoldenAngleColors { get; set; }

        public static Color GetPartColor(int index)
        {
            ArgumentOutOfRangeException.ThrowIfNegative(index);
            return UseGoldenAngleColors
                ? PartColorPalette.GoldenAngle(index)
                : PartColors[index % PartColors.Length];
        }

        public static Color GetNextColor() => GetPartColor(unchecked(nextColorIndex++) & int.MaxValue);

        public Drawing()
            : this(string.Empty, new Program()) { }

        public Drawing(string name)
            : this(name, new Program()) { }

        public Drawing(string name, Program pgm)
        {
            Id = Interlocked.Increment(ref nextId);
            Name = name;
            Material = new Material();
            Constraints = new NestConstraints();
            Source = new SourceInfo();
            Program = pgm;
        }

        public int Id { get; }

        public string Name { get; set; }

        public string Customer { get; set; }

        public int Priority { get; set; }

        public DwgQty Quantity;

        public Material Material { get; set; }

        public Program Program
        {
            get { return program; }
            set
            {
                program = value;
                UpdateArea();
                RecomputeCanonicalAngle();
            }
        }

        /// <summary>
        /// Recomputes and stores the canonical angle from the current Program.
        /// Callers that mutate Program in place (rather than reassigning it) must invoke this explicitly.
        /// Cut-off drawings are left with Angle=0.
        /// </summary>
        public void RecomputeCanonicalAngle()
        {
            if (Source == null)
                Source = new SourceInfo();

            if (program == null || IsCutOff)
            {
                Source.Angle = 0.0;
                return;
            }

            Source.Angle = CanonicalAngle.Compute(this);
        }

        public Color Color { get; set; }

        public bool IsCutOff { get; set; }

        /// <summary>
        /// Cut this drawing's parts before every other part when automatic sequencing runs,
        /// e.g. a test coupon checked before the rest of the sheet is cut.
        /// </summary>
        public bool CutFirst { get; set; }

        public NestConstraints Constraints { get; set; }

        public SourceInfo Source { get; set; }

        public List<Bend> Bends { get; set; } = new List<Bend>();

        /// <summary>
        /// The library revision this drawing was copied from, or null when it did not come
        /// from a drawing library. Editing the drawing in a nest keeps the link as provenance.
        /// </summary>
        public DrawingLibraryLink Library { get; set; }

        /// <summary>
        /// Contours this drawing says must be cut a fixed way, in the source frame. At most one
        /// pin per contour is used; <see cref="CNC.CuttingPlanning.CutPinResolver"/> maps them to
        /// a placement. Pins do not follow geometry into new drawings (split pieces).
        /// </summary>
        public List<CNC.CuttingStrategy.ContourCutPin> CutPins { get; set; } = new List<CNC.CuttingStrategy.ContourCutPin>();

        /// <summary>
        /// Scrap pieces tabbed to the part, in the source frame. <see cref="Program"/> already has
        /// them applied; they are kept so an editor can change them from <see cref="SourceEntities"/>.
        /// </summary>
        public List<Breakoffs.Breakoff> Breakoffs { get; set; } = new List<Breakoffs.Breakoff>();

        /// <summary>
        /// Holes whose slugs stay attached by tabs, in the source frame. <see cref="Program"/> already
        /// has them applied (cut pieces plus <see cref="CNC.LayerType.Tab"/> spans); they are kept so
        /// an editor can change them from <see cref="SourceEntities"/>.
        /// </summary>
        public List<TabbedCutouts.TabbedCutout> TabbedCutouts { get; set; } = new List<TabbedCutouts.TabbedCutout>();

        /// <summary>
        /// Complete set of source entities with stable GUIDs.
        /// Null when the drawing was created from G-code or an older nest file.
        /// </summary>
        public List<Entity> SourceEntities { get; set; }

        /// <summary>
        /// IDs of entities in <see cref="SourceEntities"/> that are suppressed (hidden).
        /// Suppressed entities are excluded from the active Program but preserved for re-enabling.
        /// </summary>
        public HashSet<Guid> SuppressedEntityIds { get; set; } = new HashSet<Guid>();

        /// <summary>
        /// Role overrides for entities in <see cref="SourceEntities"/>; entities not listed keep the
        /// role their layer implies (see <see cref="Converters.EntityRoles"/>). <see cref="Program"/> already has
        /// them applied; display-only entities are not in it and come from
        /// <see cref="Converters.EntityRoles.DisplayGeometry"/>.
        /// </summary>
        public Dictionary<Guid, EntityRole> EntityRoles { get; set; } = new Dictionary<Guid, EntityRole>();

        public double Area { get; protected set; }

        public void UpdateArea()
        {
            var geometry = ConvertProgram
                .ToGeometry(Program)
                .Where(entity => SpecialLayers.IsMaterial(entity.Layer));
            var shapes = ShapeBuilder.GetShapes(geometry);

            if (shapes.Count == 0)
                return;

            var areas = new double[shapes.Count];

            for (int i = 0; i < shapes.Count; i++)
            {
                var shape = shapes[i];
                areas[i] = shape.Area();
            }

            int largestAreaIndex = 0;

            for (int i = 1; i < areas.Length; i++)
            {
                var area = areas[i];

                if (area > areas[largestAreaIndex])
                    largestAreaIndex = i;
            }

            var outerArea = areas[largestAreaIndex];

            Area = outerArea - (areas.Sum() - outerArea);
        }

        public override bool Equals(object obj)
        {
            if (obj is Drawing == false)
                return false;

            var dwg = (Drawing)obj;

            return Name == dwg.Name;
        }

        public override int GetHashCode()
        {
            return Name.GetHashCode();
        }
    }

    public class SourceInfo
    {
        /// <summary>
        /// Path to the source file.
        /// </summary>
        public string Path { get; set; }

        /// <summary>
        /// Offset distances to the original location.
        /// </summary>
        public Vector Offset { get; set; }

        /// <summary>
        /// Rotation (radians) that maps the source program geometry to its canonical
        /// (MBR-axis-aligned) frame. Populated automatically by the <see cref="Drawing.Program"/>
        /// setter via <see cref="CanonicalAngle.Compute"/>. A value of 0 means the drawing is
        /// already canonical or <see cref="Drawing.IsCutOff"/> is true. Callers that mutate
        /// <see cref="Drawing.Program"/> in place must invoke
        /// <see cref="Drawing.RecomputeCanonicalAngle"/> to refresh.
        /// </summary>
        public double Angle { get; set; }
    }
}