Merge branch 'master' of https://git.chenx221.cyou/chenx221/ArtemisFgTools

这下还有人敢说我漏东西了嘛

ArtemisFgTools/ArtemisFgTools.csproj

@@ -8,8 +8,8 @@
   </PropertyGroup>
 
   <ItemGroup>
-    <PackageReference Include="Magick.NET-Q16-AnyCPU" Version="13.10.0" />
+    <PackageReference Include="Magick.NET-Q16-AnyCPU" Version="14.0.0" />
-    <PackageReference Include="Magick.NET.Core" Version="13.10.0" />
+    <PackageReference Include="Magick.NET.Core" Version="14.0.0" />
     <PackageReference Include="NLua" Version="1.7.3" />
     <PackageReference Include="System.Threading.Tasks" Version="4.3.0" />
   </ItemGroup>

ArtemisFgTools/FgHelper.cs

@@ -0,0 +1,124 @@
+using System.Reflection;
+using System.Xml.Linq;
+
+namespace ArtemisFgTools
+{
+    public class FgHelper
+    {
+        public class FgObject(string path, string head, List<string> fuku, List<List<string>> pose, Dictionary<string, List<string>> face)
+        {
+            public string Path { get; set; } = path;
+            public string Head { get; set; } = head;
+            public List<string> Fuku { get; set; } = fuku;
+            public List<List<string>> Pose { get; set; } = pose;
+            public Dictionary<string, List<string>> Face { get; set; } = face;
+        }
+        //{"fg",ch="零",size="z1",mx=40,mode=3,resize=1,path=":fg/rei/z1/",file="rei_z1a0200",face="a0001",head="rei_z1a",lv=4,id=11},
+        public class FgRecord : IEquatable<FgRecord>
+        {
+            public string ChName { get; set; }
+            public string Size { get; set; }
+            public string File { get; set; }
+            public string Face { get; set; }
+
+            public override bool Equals(object? obj)
+            {
+                return Equals(obj as FgRecord);
+            }
+
+            public bool Equals(FgRecord? other)
+            {
+                return other != null &&
+                       ChName == other.ChName &&
+                       Size == other.Size &&
+                       File == other.File &&
+                       Face == other.Face;
+            }
+
+            public override int GetHashCode()
+            {
+                return HashCode.Combine(ChName, Size, File, Face);
+            }
+        }
+        public static HashSet<FgRecord> FetchFgObjectsFromScript(string path)
+        {
+            if (!Directory.Exists(path))
+                throw new DirectoryNotFoundException("The path does not exist.");
+            else
+            {
+                List<string> fgScriptLine = [];
+                foreach (string f in Directory.GetFiles(path, "*.ast"))
+                {
+                    using StreamReader sr = new(f);
+                    string? line;
+                    while ((line = sr.ReadLine()) != null)
+                    {
+                        if (line.Contains("{\"fg\","))
+                            fgScriptLine.Add(line);
+                    }
+                }
+                if (fgScriptLine.Count == 0)
+                    throw new Exception("No valid fg script line found.");
+                else
+                {
+                    HashSet<FgRecord> fgRecords = [];
+                    foreach (var line in fgScriptLine)
+                    {
+                        FgRecord? fgRecord = ParseScriptFGLine(line);
+                        if (fgRecord != null)
+                            fgRecords.Add(fgRecord);
+
+                    }
+                    if (fgRecords.Count == 0)
+                        throw new Exception("No valid fg object found.");
+                    else
+                        return fgRecords;
+                }
+            }
+        }
+
+        public static FgRecord? ParseScriptFGLine(string input)
+        {
+            FgRecord fgRecord = new();
+            input = input.Trim('{', '}');
+            var pairs = input.Split(',');
+            foreach (var pair in pairs)
+            {
+                var keyValue = pair.Split(['=', '='], 2);
+                if (keyValue.Length == 2)
+                {
+                    string key = keyValue[0].Trim();
+                    string value = keyValue[1].Trim().Trim('"').Trim('}');
+                    PropertyInfo? property = typeof(FgRecord).GetProperty(char.ToUpper(key[0]) + key[1..]);
+                    if (property != null)
+                    {
+                        if (property.PropertyType == typeof(int))
+                        {
+                            if (int.TryParse(value, out int intValue))
+                                property.SetValue(fgRecord, intValue);
+                            else
+                                throw new Exception($"Invalid integer value '{value}' for property '{key}'.");
+                        }
+                        else if (property.PropertyType == typeof(string))
+                        {
+                            property.SetValue(fgRecord, value);
+                        }
+                        // 其他类型的处理可以在这里添加
+                    }
+                }
+            }
+            //补个chname
+            if (fgRecord.File != null)
+                fgRecord.ChName = GetCharacterEngName(fgRecord.File);
+            return fgRecord.Size == null ? null : fgRecord;
+        }
+
+        public static string GetCharacterEngName(string input)
+        {
+            int underscoreIndex = input.IndexOf('_');
+            if (underscoreIndex > 0)
+                return input[..underscoreIndex];
+            throw new Exception("Not supported character name format.");
+        }
+    }
+}

ArtemisFgTools/Program.cs

@@ -1,58 +1,70 @@
 using ImageMagick;
 using NLua;
-using System.Reflection.Metadata.Ecma335;
 using System.Text.RegularExpressions;
+using static ArtemisFgTools.FgHelper;
 namespace ArtemisFgTools
 {
     internal class Program
     {
-        public class FgObject(string path, string head, List<string> fuku, List<List<string>> pose, Dictionary<string, List<string>> face)
+        static void Main(string[] args)
         {
-            public string Path { get; set; } = path;
+            if (args.Length == 1)
-            public string Head { get; set; } = head;
+            {
-            public List<string> Fuku { get; set; } = fuku;
+                if (args[0] == "-h")
-            public List<List<string>> Pose { get; set; } = pose;
+                    Console.WriteLine("Usage: tools.exe -c <fgPath> (-s <scriptPath> | -t <luaTablePath>) -o <outputPath>");
-            public Dictionary<string, List<string>> Face { get; set; } = face;
+                else
+                    Console.WriteLine("Invalid arguments, Please check the usage via -h");
             }
-        static void Main()
+            else if (args.Length != 6)
+                Console.WriteLine("Invalid arguments, Please check the usage via -h");
+            else if (args[0] != "-c" || !(args[2] == "-s" || args[2] == "-t") || args[4] != "-o")
+                Console.WriteLine("Invalid arguments, Please check the usage via -h");
+            else
             {
-            Console.WriteLine("请输入立绘fg文件夹的所在路径（无需\"\"）：");
+                if (!Directory.Exists(args[5]))
-            string? fgImagePath = Console.ReadLine();
+                    Directory.CreateDirectory(args[5]);
-
+                if (!Directory.Exists(args[1]))
-            Console.WriteLine("请输入exlist的文件路径：");
+                    Console.WriteLine("Invalid fg path");
-            string? luaFilePath = Console.ReadLine();
+                else if (args[2] == "-s")
-
-            Console.WriteLine("请输入保存位置：");
-            string? savePath = Console.ReadLine();
-
-            if (string.IsNullOrEmpty(fgImagePath) || string.IsNullOrEmpty(luaFilePath) || string.IsNullOrEmpty(savePath))
                 {
-                Console.WriteLine("路径不能为空");
+                    if (!Directory.Exists(args[3]))
-                return;
+                        Console.WriteLine("Invalid script path");
+                    else
+                        PreProcess2(args[1], args[5], args[3]);
                 }
-            if (!Directory.Exists(fgImagePath) || !File.Exists(luaFilePath))
+                else
                 {
-                Console.WriteLine("路径不存在");
+                    if (!File.Exists(args[3]))
-                return;
+                        Console.WriteLine("Invalid lua table path");
+                    else
+                        PreProcess(args[1], args[5], args[3]);
                 }
-            if (!Directory.Exists(savePath))
-            {
-                Directory.CreateDirectory(savePath);
             }
 
-            Dictionary<object, object>? dictionary = ParseLuaTable(luaFilePath);
+            Console.WriteLine("Press any key to exit...");
+            Console.ReadKey();
+        }
 
-            if (dictionary != null)
+        private static void PreProcess2(string fgImagePath, string savePath, string scriptPath)
         {
+            HashSet<FgRecord> fgRecords = FetchFgObjectsFromScript(scriptPath);
+            if (fgRecords.Count == 0)
+                throw new Exception("No valid fg object found.");
+            //重新写个，我也懒得将FgRecord转FgObject了
+            //Tips:如果有单独饰品素材，可能前面的解析会有遗漏 //反正遥かなるニライカナイ里没有戴眼镜的角色 (笑
+            Process2(fgImagePath, savePath, fgRecords);
+        }
+
+        private static void PreProcess(string fgImagePath, string savePath, string luaFilePath)
+        {
+            Dictionary<object, object> dictionary = ParseLuaTable(luaFilePath) ?? throw new Exception("Lua table parsing failed");
+
             if (dictionary["fg"] is Dictionary<object, object> fgDictionary)
             {
                 if (fgDictionary["size"] is not List<object> size || size.Count == 0)
-                    {
                     throw new Exception("size not found or empty");
-                    }
                 fgDictionary.Remove("size");
 
-                    //convert to FgObject
                 List<FgObject> fgObjects = [];
                 foreach (var fg in fgDictionary)
                 {
@@ -63,14 +75,53 @@ namespace ArtemisFgTools
                         var face = ConvertToStringDictionary(fgValue["face"] as Dictionary<object, object>);
                         //check null
                         if (fgValue["path"] is not string path || fgValue["head"] is not string head || fuku == null || pose == null || face == null)
-                            {
+                            throw new Exception("fg object has null value");
-                                Console.WriteLine("fg object has null value");
-                                continue;
-                            }
                         path = path[4..]; // remove :fg/../
                         fgObjects.Add(new FgObject(path, head, fuku, pose, face));
                     }
                 }
+                Process(fgImagePath, savePath, size, fgObjects);
+            }
+            else
+                Console.WriteLine("fg not found");
+        }
+
+        private static void Process2(string fgImagePath, string savePath, HashSet<FgRecord> fgRecords)
+        {
+            var parallelOptions = new ParallelOptions
+            {
+                MaxDegreeOfParallelism = 6 // 设置最大并行度
+            };
+
+            Parallel.ForEach(fgRecords, parallelOptions, fgRecord =>
+            {
+                string originImageBase = Path.Combine(fgImagePath, fgRecord.ChName, fgRecord.Size, fgRecord.File + ".png");
+                string originImageFace = Path.Combine(fgImagePath, fgRecord.ChName, fgRecord.Size, fgRecord.Face + ".png");
+                string targetFilename = $"{fgRecord.File}_{fgRecord.Face}.png";
+                string targetPath = Path.Combine(savePath, fgRecord.Size);
+
+                if (!File.Exists(originImageBase) || !File.Exists(originImageFace))
+                {
+                    Console.WriteLine("ERROR, Image not found. Details:");
+                    Console.WriteLine($"Base: {originImageBase}");
+                    Console.WriteLine($"Face: {originImageFace}");
+                    return;
+                }
+                if (!Directory.Exists(targetPath))
+                    Directory.CreateDirectory(targetPath);
+                targetPath = Path.Combine(targetPath, targetFilename);
+                if (File.Exists(targetPath))
+                {
+                    Console.WriteLine("File already exists, skipping...");
+                    return;
+                }
+                ProcessAndSaveLite(originImageBase, originImageFace, targetPath);
+            });
+
+        }
+
+        private static void Process(string fgImagePath, string savePath, List<object> size, List<FgObject> fgObjects)
+        {
             foreach (var fgObject in fgObjects)
             {
                 foreach (var siz in size)
@@ -89,7 +140,7 @@ namespace ArtemisFgTools
                                     return;
                                 }
                                 bool special = false;
-                                        string special_text="";
+                                string special_text = "";
                                 string fuku_current = fuku;
                                 int index = fuku_current.IndexOf('|');
                                 if (index != -1)
@@ -150,11 +201,18 @@ namespace ArtemisFgTools
                 }
             }
         }
-                else
+
+        private static void ProcessAndSaveLite(string baseImg, string faceImg, string target)
         {
-                    Console.WriteLine("fg not found");
+            using MagickImage firstImage = new(baseImg);
-                }
+            List<int> comment1 = ReadPngComment(firstImage.Comment); //base
-            }
+            using MagickImage secondImage = new(faceImg);
+            List<int> comment2 = ReadPngComment(secondImage.Comment); //face
+            int x = comment2[0] - comment1[0]; // face x - base x
+            int y = comment2[1] - comment1[1]; // face y - base y
+            firstImage.Composite(secondImage, x, y, CompositeOperator.Over);
+            firstImage.Write(target);
+            Console.WriteLine($"Image {Path.GetFileName(target)} processing completed.");
         }
 
         private static void ProcessAndSave(string baseImg, string layerImg, string layer2Img, string target, bool special)
@@ -206,9 +264,7 @@ namespace ArtemisFgTools
 
             LuaTable luaTable = lua.GetTable("exfgtable");
             if (luaTable != null)
-            {
                 return (Dictionary<object, object>?)LuaTableToSs(luaTable);
-            }
             else
             {
                 Console.WriteLine("Lua table not found");

ArtemisFgTools/Properties/launchSettings.json

@@ -0,0 +1,8 @@
+{
+  "profiles": {
+    "ArtemisFgTools": {
+      "commandName": "Project",
+      "commandLineArgs": "-c\r\nG:\\x221.local\\1\\lab\\fg\r\n-s\r\nG:\\x221.local\\1\\lab\\script\r\n-o\r\nG:\\x221.local\\1\\lab\\output"
+    }
+  }
+}