SparkClient/Model/Services/AlgorithmServer.cs

using System.Data;
using System.IO;
using System.Runtime.InteropServices;
using System.Windows.Forms;
using log4net;
using Newtonsoft.Json;
using Newtonsoft.Json.Linq;
using SparkClient.Model.Entity.ApiEntity;
using SparkClient.Model.Helper;
using System.Configuration;
using SparkClient.Model.Common;

namespace SparkClient.Model.Services
{
    public class AlgorithmServer
    {
        private static readonly ILog Logger = LogManager.GetLogger(typeof(AlgorithmServer));
        // 导入 C++ DLL 中的 DetectDiamond 函数
        [DllImport("diamond_cut_inspector.dll", CharSet = CharSet.Ansi, CallingConvention = CallingConvention.Cdecl)]
        private static extern IntPtr DetectDiamond(string jsonData);
        // 导入 C++ DLL 中的 FreeMemory 函数
        [DllImport("diamond_cut_inspector.dll", CharSet = CharSet.Ansi, CallingConvention = CallingConvention.Cdecl)]
        private static extern void FreeString(IntPtr ptr);
        
        // 添加公共方法
        public AlgorithmResultEntity CallParseJsonAndReturnActions(string shape, string shape_mode, string image_files)
        { 
            try
            {
                //半圆
                string circleSql = $"SELECT VALUE FROM CUTTER_CONFIG WHERE KEY = 'half_circle'";
                DataTable circleTable = DataBaseHelper.ExecuteQuery(circleSql);
                object halfCircleValue = circleTable.Rows[0][0];
                bool.TryParse(halfCircleValue.ToString(), out bool boolResult);
                bool half_circle = boolResult;
       
                //算法配置参数
                string sql = $"SELECT JSON FROM ALGORITHM_CONFIG ORDER BY JSON_ORDER ASC";
                DataTable table = DataBaseHelper.ExecuteQuery(sql);
                object lightLevelValue = table.Rows[0][0];
                string algo_config = lightLevelValue.ToString() ?? throw new InvalidOperationException();
                
                //图片根目录
                string? image_file_base_path = ConfigurationManager.AppSettings["ImageFileBasePath"];

                // 将所有变量拼接成一个 JSON 对象
                JObject jsonData = new JObject(
                    new JProperty("shape", shape),
                    new JProperty("shape_mode", shape_mode),
                    new JProperty("image_file_base_path", image_file_base_path),
                    new JProperty("image_files", JToken.Parse(image_files)),
                    new JProperty("half_circle", half_circle),
                    new JProperty("algo_config", JObject.Parse(algo_config))
                );
                
                string jsonDataString = jsonData.ToString();
                // 调用 C++ DLL 函数解析 JSON
                IntPtr resultPtr = DetectDiamond(jsonDataString);
                
                string resultJson = Marshal.PtrToStringAnsi(resultPtr);
                // 释放 DLL 分配的内存
                FreeString(resultPtr);

                // 检查返回的 JSON 字符串是否为空或无效
                if (string.IsNullOrEmpty(resultJson))
                {
                    // 返回一个默认的 AlgorithmResultEntity 对象表示解析失败
                    return new AlgorithmResultEntity
                    {
                        facets = new List<Facet>(),
                        measurements = new Measurements()
                    };
                } 

                // 反序列化 JSON 字符串为 AlgorithmResultEntity 对象
                var result = JsonConvert.DeserializeObject<AlgorithmResultEntity>(resultJson);

                // 检查反序列化结果是否为 null
                if (result == null)
                {
                    // 返回一个默认的 AlgorithmResultEntity 对象表示解析失败
                    return new AlgorithmResultEntity
                    {
                        facets = new List<Facet>(),
                        measurements = new Measurements()
                    };
                }

                // 算法调用失败时，保存图片到历史记录文件夹
                if (result.status == StatusCodes.AlgorithmFailed)
                { 
                    // 从配置文件中读取 imageHistoryPath
                    // 定义历史记录文件夹路径
                    string? imageHistoryPath = ConfigurationManager.AppSettings["ImageHistoryPath"];
                    // 生成时间戳，格式为 yyyyMMddHHmmss
                    string timestamp = DateTime.Now.ToString("yyyy-MM-dd-HH-mm-ss");
                    // 组合新的文件夹路径
                    string newFolderPath = Path.Combine(imageHistoryPath, $"image-{timestamp}");

                    // 检查 D 盘内存空间
                    DriveInfo dDrive = new DriveInfo("D");
                    long requiredSpace = 10L * 1024 * 1024 * 1024; // 10GB
                    if (dDrive.TotalFreeSpace < requiredSpace)
                    {
                        // 如果 D 盘空间不足 10GB，删除最早创建的10个文件夹
                        DirectoryInfo historyDir = new DirectoryInfo(imageHistoryPath);//获取历史记录文件夹的信息
                        DirectoryInfo[] subDirs = historyDir.GetDirectories();//获取历史记录文件夹中的所有子文件夹
                        if (subDirs.Length > 0)
                        {
                            // 按创建时间排序子文件夹
                            var orderedSubDirs = subDirs.OrderBy(d => d.CreationTime).ToList();
                            int foldersToDelete = Math.Min(10, orderedSubDirs.Count);

                            // 删除最早的 10 个文件夹，如果不够 10 个则全删掉
                            for (int i = 0; i < foldersToDelete; i++)
                            {
                                orderedSubDirs[i].Delete(true);
                            }
                        }
                    }

                    // 创建新文件夹
                    Directory.CreateDirectory(newFolderPath);

                    // 保存图片到新文件夹
                    try
                    {
                        // 解析 image_files JSON 数组
                        JArray imageFilesArray = JArray.Parse(image_files);
                        string[] imageFiles = imageFilesArray.Select(token => token.ToString()).ToArray();

                        string? imageBasePath = ConfigurationManager.AppSettings["ImageFileBasePath"];// 图片根目录

                        foreach (string imageFile in imageFiles)
                        {
                            // 获取文件名
                            string fileName = Path.GetFileName(imageFile);
                            // 构建完整的源文件路径
                            string sourcePath = Path.Combine(imageBasePath, imageFile);
                            // 组合目标路径
                            string destinationPath = Path.Combine(newFolderPath, fileName);
                            // 复制文件到目标路径，如果目标文件已存在则覆盖
                            File.Copy(sourcePath, destinationPath, true);
                        }
                    }
                    catch (JsonException ex)
                    {
                        // 记录日志或处理异常
                        Logger.Error($"Error parsing image_files JSON: {ex.Message}");
                        Logger.Error($"Stack Trace: {ex.StackTrace}");
                    }
                }
                

                return result;
            }
            catch (Exception ex)
            {
                // 记录日志或处理异常
                Logger.Error($"Error in CallParseJsonAndReturnActions: {ex.Message}");
                Logger.Error($"Stack Trace: {ex.StackTrace}");
                // 如果有 InnerException，打印出来
                if (ex.InnerException != null)
                {
                    Logger.Error($"Inner Exception: {ex.InnerException.Message}");
                    Logger.Error($"Inner Stack Trace: {ex.InnerException.StackTrace}");
                }
                // 返回一个默认的 AlgorithmResultEntity 对象表示解析失败
                return new AlgorithmResultEntity
                {
                    facets = new List<Facet>(),
                    measurements = new Measurements()
                };
            }
        }
    }

}