This commit is contained in:
Chenx221 2024-11-01 16:57:49 +08:00
parent 119ecb9cf9
commit bbe55d4c56
Signed by: chenx221
GPG Key ID: D7A9EC07024C3021
3 changed files with 195 additions and 101 deletions

View File

@ -1,4 +1,7 @@
namespace ArtemisFgTools using System.Reflection;
using System.Xml.Linq;
namespace ArtemisFgTools
{ {
public class FgHelper public class FgHelper
{ {
@ -10,10 +13,35 @@
public List<List<string>> Pose { get; set; } = pose; public List<List<string>> Pose { get; set; } = pose;
public Dictionary<string, List<string>> Face { get; set; } = face; 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 static List<FgObject> FetchFgObjectsFromScript(string path) 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)
{ {
//检查path是否存在然后foreach获得文件夹下所有.ast文件循环
if (!Directory.Exists(path)) if (!Directory.Exists(path))
throw new DirectoryNotFoundException("The path does not exist."); throw new DirectoryNotFoundException("The path does not exist.");
else else
@ -33,45 +61,64 @@
throw new Exception("No valid fg script line found."); throw new Exception("No valid fg script line found.");
else else
{ {
List<FgObject> fgObjects = []; HashSet<FgRecord> fgRecords = [];
foreach (var line in fgScriptLine) foreach (var line in fgScriptLine)
{ {
FgObject fgObject = ParseScriptFGLine(line); FgRecord? fgRecord = ParseScriptFGLine(line);
// TODO:如果fgObject未被添加到fgObjects中添加 if (fgRecord != null)
fgObjects.Add(fgObject); fgRecords.Add(fgRecord);
} }
if (fgObjects.Count == 0) if (fgRecords.Count == 0)
throw new Exception("No valid fg object found."); throw new Exception("No valid fg object found.");
else else
return fgObjects; return fgRecords;
} }
} }
} }
public static FgObject ParseScriptFGLine(string input) public static FgRecord? ParseScriptFGLine(string input)
{ {
FgRecord fgRecord = new();
input = input.Trim('{', '}'); input = input.Trim('{', '}');
var pairs = input.Split(','); var pairs = input.Split(',');
var result = new Dictionary<string, string>();
foreach (var pair in pairs) foreach (var pair in pairs)
{ {
var keyValue = pair.Split(['=', '='], 2); var keyValue = pair.Split(['=', '='], 2);
if (keyValue.Length == 2) if (keyValue.Length == 2)
{ {
string key = keyValue[0].Trim(); string key = keyValue[0].Trim();
string value = keyValue[1].Trim().Trim('"'); // 去掉引号 string value = keyValue[1].Trim().Trim('"').Trim('}');
result[key] = value; PropertyInfo? property = typeof(FgRecord).GetProperty(char.ToUpper(key[0]) + key[1..]);
} if (property != null)
}
foreach (var kv in result)
{ {
Console.WriteLine($"{kv.Key}: {kv.Value}"); 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;
} }
throw new NotImplementedException(); public static string GetCharacterEngName(string input)
{
int underscoreIndex = input.IndexOf('_');
if (underscoreIndex > 0)
return input[..underscoreIndex];
throw new Exception("Not supported character name format.");
} }
} }
} }

View File

@ -6,66 +6,65 @@ namespace ArtemisFgTools
{ {
internal class Program internal class Program
{ {
static void Main(string[] args)
static void Main()
{ {
Console.WriteLine("请输入立绘fg文件夹的所在路径无需\"\""); if (args.Length == 1)
string? fgImagePath = Console.ReadLine();
Console.WriteLine("有找到exlist吗(y/n) ");
string spModeStr = Console.ReadLine() ?? throw new Exception("Invalid input");
bool spMode = (spModeStr == "n") || (spModeStr == "y" ? false : throw new Exception("Invalid input"));
Console.WriteLine("请输入保存位置:");
string? savePath = Console.ReadLine();
if (!Directory.Exists(savePath))
Directory.CreateDirectory(savePath);
if (spMode)
{ {
Console.WriteLine("请输入script文件夹的所在路径"); if (args[0] == "-h")
string? scriptPath = Console.ReadLine(); Console.WriteLine("Usage: tools.exe -c <fgPath> (-s <scriptPath> | -t <luaTablePath>) -o <outputPath>");
if (string.IsNullOrEmpty(fgImagePath) || string.IsNullOrEmpty(savePath) || string.IsNullOrEmpty(scriptPath)) else
{ Console.WriteLine("Invalid arguments, Please check the usage via -h");
Console.WriteLine("路径不能为空");
return;
} }
if (!Directory.Exists(fgImagePath) || !Directory.Exists(scriptPath)) 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("路径不存在"); if (!Directory.Exists(args[5]))
return; Directory.CreateDirectory(args[5]);
if (!Directory.Exists(args[1]))
Console.WriteLine("Invalid fg path");
else if (args[2] == "-s")
{
if (!Directory.Exists(args[3]))
Console.WriteLine("Invalid script path");
else
PreProcess2(args[1], args[5], args[3]);
}
else
{
if (!File.Exists(args[3]))
Console.WriteLine("Invalid lua table path");
else
PreProcess(args[1], args[5], args[3]);
} }
List<FgObject> fgObjects = FetchFgObjectsFromScript(scriptPath);
Process(fgImagePath, savePath, size, fgObjects);
//我忘了size是干啥的了...
return;
} }
Console.WriteLine("请输入exlist的文件路径"); Console.WriteLine("Press any key to exit...");
string? luaFilePath = Console.ReadLine(); Console.ReadKey();
if (string.IsNullOrEmpty(fgImagePath) || string.IsNullOrEmpty(savePath) || string.IsNullOrEmpty(luaFilePath))
{
Console.WriteLine("路径不能为空");
return;
}
if (!Directory.Exists(fgImagePath) || !File.Exists(luaFilePath))
{
Console.WriteLine("路径不存在");
return;
} }
Dictionary<object, object>? dictionary = ParseLuaTable(luaFilePath); private static void PreProcess2(string fgImagePath, string savePath, string scriptPath)
if (dictionary != null)
{ {
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 (dictionary["fg"] is Dictionary<object, object> fgDictionary)
{ {
if (fgDictionary["size"] is not List<object> size || size.Count == 0) if (fgDictionary["size"] is not List<object> size || size.Count == 0)
throw new Exception("size not found or empty"); throw new Exception("size not found or empty");
fgDictionary.Remove("size"); fgDictionary.Remove("size");
//convert to FgObject
List<FgObject> fgObjects = []; List<FgObject> fgObjects = [];
foreach (var fg in fgDictionary) foreach (var fg in fgDictionary)
{ {
@ -76,23 +75,52 @@ namespace ArtemisFgTools
var face = ConvertToStringDictionary(fgValue["face"] as Dictionary<object, object>); var face = ConvertToStringDictionary(fgValue["face"] as Dictionary<object, object>);
//check null //check null
if (fgValue["path"] is not string path || fgValue["head"] is not string head || fuku == null || pose == null || face == 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/../ path = path[4..]; // remove :fg/../
fgObjects.Add(new FgObject(path, head, fuku, pose, face)); fgObjects.Add(new FgObject(path, head, fuku, pose, face));
} }
} }
//jmp
Process(fgImagePath, savePath, size, fgObjects); Process(fgImagePath, savePath, size, fgObjects);
} }
else else
Console.WriteLine("fg not found"); 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) private static void Process(string fgImagePath, string savePath, List<object> size, List<FgObject> fgObjects)
{ {
foreach (var fgObject in fgObjects) foreach (var fgObject in fgObjects)
{ {
@ -174,6 +202,19 @@ namespace ArtemisFgTools
} }
} }
private static void ProcessAndSaveLite(string baseImg, string faceImg, string target)
{
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) private static void ProcessAndSave(string baseImg, string layerImg, string layer2Img, string target, bool special)
{ {
if (File.Exists(target)) if (File.Exists(target))
@ -223,9 +264,7 @@ namespace ArtemisFgTools
LuaTable luaTable = lua.GetTable("exfgtable"); LuaTable luaTable = lua.GetTable("exfgtable");
if (luaTable != null) if (luaTable != null)
{
return (Dictionary<object, object>?)LuaTableToSs(luaTable); return (Dictionary<object, object>?)LuaTableToSs(luaTable);
}
else else
{ {
Console.WriteLine("Lua table not found"); Console.WriteLine("Lua table not found");

View File

@ -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"
}
}
}