添加 Test2.txt

Test2.txt

@@ -0,0 +1,366 @@
+using System;
+using System.Collections.Generic;
+using System.IO;
+using System.Text;
+using System.Text.RegularExpressions;
+using NPOI.SS.UserModel;
+using NPOI.XSSF.UserModel;
+using UnityEngine;
+using UnityEditor;
+using UnityEditor.Compilation;
+
+public class ExcelToEntityGenerator : EditorWindow
+{
+    private static string ExcelFolder = "Assets/Data/Excels";
+    private static string ScriptsFolder = "Assets/Scripts/Data";
+    private static string AssetFolder = "Assets/Data/Containers";
+
+    public static void LoadAll(string excelFolder, string scriptsFolder, string assetFolder)
+    {
+        ExcelFolder = excelFolder;
+        ScriptsFolder = scriptsFolder;
+        AssetFolder = assetFolder;
+        // 确保输出目录存在
+        if (!Directory.Exists(ScriptsFolder)) Directory.CreateDirectory(ScriptsFolder);
+        if (!Directory.Exists(AssetFolder)) Directory.CreateDirectory(AssetFolder);
+
+        // 加载所有 Excel 文件
+        var allData = ExcelBatchLoader.LoadAllXlsxFiles(ExcelFolder);
+        if (allData.Count == 0)
+        {
+            Debug.LogWarning("未找到任何 .xlsx 文件");
+            return;
+        }
+
+        // 为每个文件生成数据行类和容器类
+        foreach (var kv in allData)
+        {
+            string fileName = kv.Key; // 不含扩展名
+            var sheet = kv.Value[0];   // 只处理第一个工作表
+
+            // 解析字段信息
+            IRow nameRow = sheet.GetRow(0);
+            IRow typeRow = sheet.GetRow(1);
+            List<FieldInfo> fields = ParseFieldInfo(nameRow, typeRow);
+
+            // 生成数据行类
+            string rowClassPath = $"{ScriptsFolder}/{fileName}Container.cs";
+            var rowClassContent = GenerateRowClass(fields, fileName, rowClassPath);
+
+            // 生成容器类
+            string containerClassPath = $"{ScriptsFolder}/{fileName}Container.cs";
+            GenerateContainerClass(fileName, containerClassPath, rowClassContent);
+        }
+        AssetDatabase.SaveAssets();
+        AssetDatabase.Refresh();
+    }
+
+    public static void CreateAllContainers()
+    {
+        // 重新加载 Excel 文件（此时数据行类已编译）
+        var allData = ExcelBatchLoader.LoadAllXlsxFiles(ExcelFolder);
+        foreach (var kv in allData)
+        {
+            string fileName = kv.Key;
+            var sheet = kv.Value[0];
+            CreateContainerFromSheet(sheet, fileName);
+        }
+        AssetDatabase.SaveAssets();
+        AssetDatabase.Refresh();
+        Debug.Log("所有 ScriptableObject 容器已生成完毕");
+    }
+
+    private static List<FieldInfo> ParseFieldInfo(IRow nameRow, IRow typeRow)
+    {
+        List<FieldInfo> fields = new List<FieldInfo>();
+        for (int i = 0; i < nameRow.LastCellNum; i++)
+        {
+            string rawName = nameRow.GetCell(i)?.ToString()?.Trim();
+            string rawType = typeRow.GetCell(i)?.ToString()?.Trim();
+            if (string.IsNullOrEmpty(rawName) || string.IsNullOrEmpty(rawType))
+                continue;
+
+            string fieldName = SanitizeName(rawName);
+            string fieldType = MapType(rawType); // 映射为C#类型字符串
+            fields.Add(new FieldInfo { Name = fieldName, RawType = rawType, Type = fieldType });
+        }
+        return fields;
+    }
+
+    private static string SanitizeName(string name)
+    {
+        // 移除非法字符，并确保首字母大写
+        string sanitized = Regex.Replace(name, @"[^a-zA-Z0-9_]", "");
+        if (string.IsNullOrEmpty(sanitized)) return "Field";
+        return char.ToUpper(sanitized[0]) + sanitized.Substring(1);
+    }
+
+    private static string MapType(string typeStr)
+    {
+        switch (typeStr.ToLower())
+        {
+            case "int": return "int";
+            case "float": return "float";
+            case "bool": return "bool";
+            case "string": return "string";
+            case "intlist": return "List<int>";
+            case "floatlist": return "List<float>";
+            case "boollist": return "List<bool>";
+            case "stringlist": return "List<string>";
+            default: return "string";
+        }
+    }
+
+    private static string GenerateRowClass(List<FieldInfo> fields, string className, string outputPath)
+    {
+        StringBuilder sb = new StringBuilder();
+        sb.AppendLine("//由程序自动生成");
+        sb.AppendLine("using System;");
+        sb.AppendLine("using System.Collections.Generic;");
+        sb.AppendLine("using UnityEngine;");
+        sb.AppendLine();
+        sb.AppendLine("[System.Serializable]");
+        sb.AppendLine($"public class {className}");
+        sb.AppendLine("{");
+
+        foreach (var field in fields)
+        {
+            sb.AppendLine($"    public {field.Type} {field.Name};");
+        }
+
+        sb.AppendLine("}");
+
+        // WriteFile(outputPath, sb.ToString());
+        return sb.ToString();
+    }
+
+    private static void GenerateContainerClass(string rowClassName, string outputPath, string rowClassContent)
+    {
+        
+        string containerName = rowClassName + "Container";
+        StringBuilder sb = new StringBuilder();
+        // sb.AppendLine("using System.Collections.Generic;");
+        // sb.AppendLine("using UnityEngine;");
+        sb.AppendLine(rowClassContent);
+        sb.AppendLine();
+        sb.AppendLine("[CreateAssetMenu(fileName = \"" + containerName + "\", menuName = \"Data/" + containerName + "\")]");
+        sb.AppendLine($"public class {containerName} : ScriptableObject");
+        sb.AppendLine("{");
+        sb.AppendLine($"    public List<{rowClassName}> items;");
+        sb.AppendLine("}");
+
+        WriteFile(outputPath, sb.ToString());
+    }
+
+    private static void WriteFile(string path, string content)
+    {
+        // 如果文件已存在，删除后重新创建（避免残留）
+        if (File.Exists(path))
+        {
+            File.Delete(path);
+        }
+        File.WriteAllText(path, content, Encoding.UTF8);
+    }
+
+    private static void CreateContainerFromSheet(ISheet sheet, string fileName)
+    {
+        try
+        {
+            // 获取已编译的类型
+            string rowClassName = fileName;
+            string containerClassName = fileName + "Container";
+            System.Reflection.Assembly assembly = System.Reflection.Assembly.Load("Assembly-CSharp");
+            Type rowType = assembly.GetType(rowClassName);
+            Type containerType = assembly.GetType(containerClassName);
+            if (rowType == null || containerType == null)
+            {
+                Debug.LogError($"无法找到类型: {rowClassName} 或 {containerClassName}，请检查编译是否成功");
+                return;
+            }
+
+            // 解析字段信息（与生成类时一致）
+            IRow nameRow = sheet.GetRow(0);
+            IRow typeRow = sheet.GetRow(1);
+            List<FieldInfo> fields = ParseFieldInfo(nameRow, typeRow);
+
+            // 创建容器实例
+            ScriptableObject container = ScriptableObject.CreateInstance(containerType);
+            if (container == null)
+            {
+                Debug.LogError($"无法创建容器实例: {containerClassName}");
+                return;
+            }
+
+            // 获取 items 字段并创建 List<T>
+            System.Reflection.FieldInfo itemsField = containerType.GetField("items");
+            Type listType = typeof(List<>).MakeGenericType(rowType);
+            object listInstance = Activator.CreateInstance(listType);
+
+            // 遍历数据行（从第3行开始，索引2）
+            for (int rowIdx = 2; rowIdx <= sheet.LastRowNum; rowIdx++)
+            {
+                IRow dataRow = sheet.GetRow(rowIdx);
+                if (dataRow == null) continue;
+
+                // 创建数据行对象
+                object rowObj = Activator.CreateInstance(rowType);
+
+                bool hasData = false; // 标记该行是否有有效数据
+                for (int i = 0; i < fields.Count; i++)
+                {
+                    var field = fields[i];
+                    var cell = dataRow.GetCell(i);
+                    string cellValue = cell?.ToString()?.Trim();
+                    if (string.IsNullOrEmpty(cellValue)) continue;
+
+                    hasData = true;
+                    // 转换并赋值
+                    object convertedValue = ConvertValue(cellValue, field.RawType);
+                    System.Reflection.FieldInfo rowField = rowType.GetField(field.Name);
+                    if (rowField != null)
+                    {
+                        rowField.SetValue(rowObj, convertedValue);
+                    }
+                }
+
+                if (hasData)
+                {
+                    // 添加到列表
+                    listType.GetMethod("Add").Invoke(listInstance, new[] { rowObj });
+                }
+            }
+
+            // 将列表赋值给容器
+            itemsField.SetValue(container, listInstance);
+
+            // 保存资产
+            string assetPath = $"{AssetFolder}/{fileName}Container.asset";
+            // 如果已存在，先删除
+            if (File.Exists(assetPath))
+            {
+                AssetDatabase.DeleteAsset(assetPath);
+            }
+            AssetDatabase.CreateAsset(container, assetPath);
+            Debug.Log($"生成容器资产: {assetPath}");
+        }
+        catch (Exception e)
+        {
+            Debug.LogError($"处理文件 {fileName} 时出错: {e.Message}\n{e.StackTrace}");
+        }
+    }
+
+    private static object ConvertValue(string value, string rawType)
+    {
+        string lowerType = rawType.ToLower();
+        try
+        {
+            switch (lowerType)
+            {
+                case "int":
+                    return int.Parse(value);
+                case "float":
+                    return float.Parse(value);
+                case "bool":
+                    return bool.Parse(value);
+                case "string":
+                    return value;
+                case "intlist":
+                    return ParseList<int>(value, int.Parse);
+                case "floatlist":
+                    return ParseList<float>(value, float.Parse);
+                case "boollist":
+                    return ParseList<bool>(value, bool.Parse);
+                case "stringlist":
+                    return ParseList<string>(value, s => s);
+                default:
+                    return value;
+            }
+        }
+        catch (Exception e)
+        {
+            Debug.LogWarning($"转换值 '{value}' 到类型 {rawType} 失败: {e.Message}，使用默认值。");
+            return GetDefaultValue(rawType);
+        }
+    }
+
+    private static List<T> ParseList<T>(string value, Func<string, T> parser)
+    {
+        List<T> list = new List<T>();
+        if (string.IsNullOrEmpty(value)) return list;
+        string[] parts = value.Split(new[] { '#' }, StringSplitOptions.RemoveEmptyEntries);
+        foreach (string part in parts)
+        {
+            try
+            {
+                list.Add(parser(part.Trim()));
+            }
+            catch
+            {
+                // 忽略解析失败的项
+            }
+        }
+        return list;
+    }
+
+    private static object GetDefaultValue(string rawType)
+    {
+        switch (rawType.ToLower())
+        {
+            case "int": return 0;
+            case "float": return 0f;
+            case "bool": return false;
+            case "string": return "";
+            case "intlist": return new List<int>();
+            case "floatlist": return new List<float>();
+            case "boollist": return new List<bool>();
+            case "stringlist": return new List<string>();
+            default: return null;
+        }
+    }
+
+    private class FieldInfo
+    {
+        public string Name;
+        public string RawType;   // Excel中的原始类型标识
+        public string Type;       // 映射后的C#类型字符串
+    }
+}
+
+public static class ExcelBatchLoader
+{
+    public static Dictionary<string, List<ISheet>> LoadAllXlsxFiles(string directoryPath, SearchOption searchOption = SearchOption.TopDirectoryOnly)
+    {
+        var result = new Dictionary<string, List<ISheet>>();
+
+        if (!Directory.Exists(directoryPath))
+        {
+            Debug.LogError($"目录不存在: {directoryPath}");
+            return result;
+        }
+
+        string[] files = Directory.GetFiles(directoryPath, "*.xlsx", searchOption);
+        foreach (string filePath in files)
+        {
+            try
+            {
+                using (FileStream fs = new FileStream(filePath, FileMode.Open, FileAccess.Read))
+                {
+                    IWorkbook workbook = new XSSFWorkbook(fs);
+                    List<ISheet> sheets = new List<ISheet>();
+                    for (int i = 0; i < workbook.NumberOfSheets; i++)
+                    {
+                        sheets.Add(workbook.GetSheetAt(i));
+                    }
+                    var fileName = Path.GetFileNameWithoutExtension(filePath);
+                    result[fileName] = sheets;
+                }
+            }
+            catch (Exception e)
+            {
+                Debug.LogError($"加载文件失败: {filePath}\n{e.Message}");
+            }
+        }
+
+        return result;
+    }
+}