test

C3032

2025-12-20 38c333f8091e5062be0bbdd81960315ebce69008

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using System.IO.Compression;
 
namespace SmartScanner
{
    public static class LBProjService
    {
        private const string ConfigFileName = "config.json";
        private const string ModelNameFileName = "model_name.txt";
        private const string ModelDir = "model";  // 固定模型目录
 
        /// <summary>
        /// 创建临时目录
        /// </summary>
        private static string CreateTempDirectory()
        {
            string tempDir = Path.Combine(Path.GetTempPath(), Path.GetRandomFileName());
            Directory.CreateDirectory(tempDir);
            return tempDir;
        }
 
        /// <summary>
        /// 获取所有可用模型
        /// </summary>
        public static string[] GetAvailableModels()
        {
            string modelPath = Path.Combine(AppDomain.CurrentDomain.BaseDirectory, ModelDir);
 
            if (!Directory.Exists(modelPath))
            {
                Directory.CreateDirectory(modelPath);
                return Array.Empty<string>();
            }
 
            return Directory.GetFiles(modelPath, "*.lb")
                           .Select(Path.GetFileName)
                           .ToArray();
        }
 
        /// <summary>
        /// 创建工程文件
        /// </summary>
        public static (bool Success, string Message) CreateLBProj(
            string paramFilePath,
            string modelFileName,
            string outputPath)
        {
            try
            {
                // 验证模型是否存在
                string fullModelPath = Path.Combine(ModelDir, modelFileName);
                if (!File.Exists(fullModelPath))
                {
                    return (false, $"模型文件 {modelFileName} 不存在于 {ModelDir} 目录");
                }
 
                // 创建临时目录并打包
                string tempDir = CreateTempDirectory();
                try
                {
                    // 1. 保存参数文件
                    File.Copy(paramFilePath, Path.Combine(tempDir, ConfigFileName), true);
 
                    // 2. 保存模型文件名（非路径）
                    File.WriteAllText(Path.Combine(tempDir, ModelNameFileName), modelFileName);
 
                    // 3. 创建ZIP包
                    ZipFile.CreateFromDirectory(tempDir, outputPath);
                    return (true, "工程文件创建成功");
                }
                finally
                {
                    Directory.Delete(tempDir, true);
                }
            }
            catch (Exception ex)
            {
                return (false, $"创建失败: {ex.Message}");
            }
        }
 
        /// <summary>
        /// 加载工程文件
        /// </summary>
        public static (string ModelName, string configPath, string ConfigContent, string Message) LoadLBProj(string projectFilePath)
        {
            try
            {
                string tempDir = CreateTempDirectory();
                try
                {
                    ZipFile.ExtractToDirectory(projectFilePath, tempDir);
 
                    // 1. 读取模型文件名
                    string modelNameFile = Path.Combine(tempDir, ModelNameFileName);
                    if (!File.Exists(modelNameFile))
                    {
                        return (null, null, null, "工程文件中缺少模型名称记录");
                    }
                    string modelName = File.ReadAllText(modelNameFile);
 
                    // 2. 验证模型是否存在
                    string fullModelPath = Path.Combine(ModelDir, modelName);
                    if (!File.Exists(fullModelPath))
                    {
                        return (null, null, null, $"模型 {modelName} 不存在于 {ModelDir} 目录");
                    }
 
                    // 3. 读取配置
                    string configPath = Path.Combine(tempDir, ConfigFileName);
                    string configContent = File.ReadAllText(configPath);
 
                    return (modelName, configPath, configContent, "加载成功");
                }
                finally
                {
                    Directory.Delete(tempDir, true);
                }
            }
            catch (Exception ex)
            {
                return (null, null, null, $"加载失败: 请输入正确的工程文件路径: {ex.Message}");
            }
        }
    }
}