OpenNest/OpenNest.Posts.Cincinnati/CincinnatiPostProcessor.cs

using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Reflection;
using System.Text;
using System.Text.Json;
using System.Text.Json.Serialization;
using OpenNest.CNC;

namespace OpenNest.Posts.Cincinnati
{
    public sealed class CincinnatiPostProcessor : IPostProcessor
    {
        private static readonly JsonSerializerOptions JsonOptions = new()
        {
            WriteIndented = true,
            Converters = { new JsonStringEnumConverter() }
        };

        public string Name => "Cincinnati CL-707";
        public string Author => "OpenNest";
        public string Description => "Cincinnati CL-707/CL-800/CL-900/CL-940/CLX family";

        public CincinnatiPostConfig Config { get; }

        public CincinnatiPostProcessor()
        {
            var configPath = GetConfigPath();
            if (File.Exists(configPath))
            {
                var json = File.ReadAllText(configPath);
                Config = JsonSerializer.Deserialize<CincinnatiPostConfig>(json, JsonOptions);
            }
            else
            {
                Config = new CincinnatiPostConfig();
                SaveConfig();
            }
        }

        public CincinnatiPostProcessor(CincinnatiPostConfig config)
        {
            Config = config;
        }

        public void SaveConfig()
        {
            var configPath = GetConfigPath();
            var json = JsonSerializer.Serialize(Config, JsonOptions);
            File.WriteAllText(configPath, json);
        }

        private static string GetConfigPath()
        {
            var assemblyPath = typeof(CincinnatiPostProcessor).Assembly.Location;
            var dir = Path.GetDirectoryName(assemblyPath);
            var name = Path.GetFileNameWithoutExtension(assemblyPath);
            return Path.Combine(dir, name + ".json");
        }

        public void Post(Nest nest, Stream outputStream)
        {
            // 1. Create variable manager and register standard variables
            var vars = CreateVariableManager();

            // 2. Filter to non-empty plates
            var plates = nest.Plates
                .Where(p => p.Parts.Count > 0)
                .ToList();

            // 3. Build part sub-program registry (if enabled)
            Dictionary<(int, long), int> partSubprograms = null;
            List<(int subNum, string name, Program program)> subprogramEntries = null;

            if (Config.UsePartSubprograms)
            {
                partSubprograms = new Dictionary<(int, long), int>();
                subprogramEntries = new List<(int, string, Program)>();
                var nextSubNum = Config.PartSubprogramStart;

                foreach (var plate in plates)
                {
                    foreach (var part in plate.Parts)
                    {
                        if (part.BaseDrawing.IsCutOff) continue;
                        var key = CincinnatiPartSubprogramWriter.SubprogramKey(part);
                        if (!partSubprograms.ContainsKey(key))
                        {
                            var subNum = nextSubNum++;
                            partSubprograms[key] = subNum;

                            // Create normalized program at origin
                            var pgm = part.Program.Clone() as Program;
                            var bbox = pgm.BoundingBox();
                            pgm.Offset(-bbox.Location.X, -bbox.Location.Y);

                            subprogramEntries.Add((subNum, part.BaseDrawing.Name, pgm));
                        }
                    }
                }
            }

            // 4. Create writers
            var preamble = new CincinnatiPreambleWriter(Config);
            var sheetWriter = new CincinnatiSheetWriter(Config, vars);

            // 5. Build material description from first plate
            var material = plates.FirstOrDefault()?.Material;
            var materialDesc = material != null
                ? $"{material.Name}{(string.IsNullOrEmpty(material.Grade) ? "" : $", {material.Grade}")}"
                : "";

            // 6. Write to stream
            using var writer = new StreamWriter(outputStream, Encoding.UTF8, 1024, leaveOpen: true);

            // Main program
            preamble.WriteMainProgram(writer, nest.Name ?? "NEST", materialDesc, plates.Count);

            // Variable declaration subprogram
            preamble.WriteVariableDeclaration(writer, vars);

            // Sheet subprograms
            for (var i = 0; i < plates.Count; i++)
            {
                var sheetIndex = i + 1;
                var subNumber = Config.SheetSubprogramStart + i;
                sheetWriter.Write(writer, plates[i], nest.Name ?? "NEST", sheetIndex, subNumber,
                    partSubprograms);
            }

            // Part sub-programs (if enabled)
            if (subprogramEntries != null)
            {
                var partSubWriter = new CincinnatiPartSubprogramWriter(Config);
                var firstPlate = plates.FirstOrDefault();
                var sheetDiagonal = firstPlate != null
                    ? System.Math.Sqrt(firstPlate.Size.Width * firstPlate.Size.Width
                        + firstPlate.Size.Length * firstPlate.Size.Length)
                    : 100.0;

                foreach (var (subNum, name, pgm) in subprogramEntries)
                {
                    partSubWriter.Write(writer, pgm, name, subNum,
                        Config.DefaultLibraryFile ?? "", sheetDiagonal);
                }
            }

            writer.Flush();
        }

        public void Post(Nest nest, string outputFile)
        {
            using var fs = new FileStream(outputFile, FileMode.Create, FileAccess.Write);
            Post(nest, fs);
        }

        private ProgramVariableManager CreateVariableManager()
        {
            var vars = new ProgramVariableManager();
            vars.GetOrCreate("ProcessFeedrate", 148);  // Set by G89, no expression
            vars.GetOrCreate("LeadInFeedrate", 126, $"[#148*{Config.LeadInFeedratePercent}]");
            vars.GetOrCreate("LeadInArcLine2Feedrate", 127, $"[#148*{Config.LeadInArcLine2FeedratePercent}]");
            vars.GetOrCreate("CircleFeedrate", 128, Config.CircleFeedrateMultiplier.ToString("0.#"));
            return vars;
        }
    }
}