EscudeTools/EscudeLSF/Program.cs
2024-10-10 18:22:22 +08:00

278 lines
10 KiB
C#
Raw Permalink Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

using ImageMagick;
using System.Runtime.InteropServices;
namespace EscudeLSF
{
[StructLayout(LayoutKind.Sequential, Pack = 1)]
public struct LSFHDR
{
[MarshalAs(UnmanagedType.ByValArray, SizeConst = 4)]
public byte[] signature; // 4 bytes
public uint unknown1; // 4 bytes
public uint unknown2; // 4 bytes
public uint width; // 4 bytes
public uint height; // 4 bytes
public uint unknown_width; // 4 bytes
public uint unknown_height; // 4 bytes
}
[StructLayout(LayoutKind.Sequential, Pack = 1)]
public struct LSFENTRY
{
[MarshalAs(UnmanagedType.ByValArray, SizeConst = 128)]
public char[] name; // 128 bytes
public uint x; // 4 bytes
public uint y; // 4 bytes
public uint baseWidth; // 4 bytes
public uint baseHeight; // 4 bytes
public uint unknown3; // 4 bytes
public uint unknown4; // 4 bytes
public byte type; // 1 byte
public byte id; // 1 byte
public byte unknown5; // 1 byte
public byte unknown6; // 1 byte
public uint unknown7; // 4 bytes
public uint unknown8; // 4 bytes
}
internal class Program
{
private static readonly byte[] Signature = [0x4C, 0x53, 0x46, 0x00];
static void Main(string[] args)
{
if (args.Length == 0 || args.Length > 2)
{
Console.WriteLine("Invalid arguments. Use -h for help.");
return;
}
string? workDirectory;
string? outputDirectory;
switch (args[0])
{
case "-h":
DisplayHelp();
return;
case "-r":
case "-d":
case "-s":
workDirectory = GetWorkDirectory(args);
outputDirectory = GetOutputDirectory(workDirectory);
if (args[0] == "-d")
{
ProcessBatch(args[1], workDirectory, outputDirectory);
}
else
{
CombineIMG(ReadLSF(args[1]), workDirectory, outputDirectory);
}
break;
default:
if (!IsValidFileArgument(args))
{
Console.WriteLine("Invalid arguments. Use -h for help.");
return;
}
workDirectory = Path.GetDirectoryName(args[0]);
outputDirectory = GetOutputDirectory(workDirectory);
CombineIMG(ReadLSF(args[0]), workDirectory, outputDirectory);
break;
}
}
static void DisplayHelp()
{
Console.WriteLine("Usage: EscudeLSF.exe [-r <filepath>] [-d <directory>] [-s <filepath>] [-h]");
Console.WriteLine("Options:");
Console.WriteLine(" <filepath> Single lsf process");
Console.WriteLine(" -r <filepath> Read single lsf file");
Console.WriteLine(" -d <directory> Process all lsf files in directory");
Console.WriteLine(" -s <filepath> Same as <filepath>");
Console.WriteLine(" -h Display help info");
}
static string GetOutputDirectory(string workDirectory)
{
string outputDirectory = Path.Combine(Path.GetDirectoryName(workDirectory), "output");
if (!Directory.Exists(outputDirectory))
{
Directory.CreateDirectory(outputDirectory);
}
return outputDirectory;
}
static void ProcessBatch(string directory, string workDirectory, string outputDirectory)
{
foreach (string file in Directory.GetFiles(directory, "*.lsf"))
{
CombineIMG(ReadLSF(file), workDirectory, outputDirectory);
}
}
static string GetWorkDirectory(string[] args)
{
return args[0] == "-d" ? args[1] : Path.GetDirectoryName(args[1]);
}
static bool IsValidFileArgument(string[] args)
{
return args.Length == 2 || (args.Length == 1 && !Path.Exists(args[0]));
}
static List<LSFENTRY> ReadLSF(string filePath)
{
LSFHDR header;
List<LSFENTRY> entries = [];
using (FileStream fs = new(filePath, FileMode.Open, FileAccess.Read))
using (BinaryReader reader = new(fs))
{
header = ReadLSFHDR(reader);
if (!header.signature.SequenceEqual(Signature))
{
Console.WriteLine("Invalid LSF file: Signature does not match.");
return [];
}
while (fs.Position < fs.Length)
{
LSFENTRY entry = ReadLSFENTRY(reader);
// 过滤某些类型(主要是表情,还有审查
// 没找到合适的处理方法有主意的老哥欢迎Pull Request
if (entry.type == 0x0B || entry.type == 0x15 || entry.type == 0xFF || (entry.type == 0x00 && entry.unknown5 == 0x03))
{
continue;
}
// 过滤重复项(没看出重复的意义,也许是前面过滤掉的某个类型需要
if (entries.Any(e => e.name.SequenceEqual(entry.name)))
{
continue;
}
entries.Add(entry);
}
}
return entries;
}
static LSFHDR ReadLSFHDR(BinaryReader reader)
{
byte[] headerData = reader.ReadBytes(Marshal.SizeOf(typeof(LSFHDR)));
GCHandle handle = GCHandle.Alloc(headerData, GCHandleType.Pinned);
try
{
return (LSFHDR)Marshal.PtrToStructure(handle.AddrOfPinnedObject(), typeof(LSFHDR));
}
finally
{
handle.Free();
}
}
static LSFENTRY ReadLSFENTRY(BinaryReader reader)
{
byte[] entryData = reader.ReadBytes(Marshal.SizeOf(typeof(LSFENTRY)));
GCHandle handle = GCHandle.Alloc(entryData, GCHandleType.Pinned);
try
{
return (LSFENTRY)Marshal.PtrToStructure(handle.AddrOfPinnedObject(), typeof(LSFENTRY));
}
finally
{
handle.Free();
}
}
static void CombineIMG(List<LSFENTRY> entrys, string workDirectory, string storeDir)
{
// 检查是否有多个角色
bool isMultiCharacter = entrys.Any(e => e.type == 0x14);
foreach (var entry in entrys.Where(e => e.type == 0x00))
{
string baseName = new string(entry.name).Trim('\0');
string targetPath = Path.Combine(storeDir, baseName);
string baseImagePath = Path.Combine(workDirectory, baseName + ".png");
if (!File.Exists(baseImagePath))
{
Console.WriteLine($"Error, File not found: {baseName}");
continue;
}
foreach (var eyeEntry in entrys.Where(e => e.type == 0x0A))
{
using var baseImage = new MagickImage(baseImagePath);
ProcessEyeImage(entrys, workDirectory, baseImage, eyeEntry, targetPath, isMultiCharacter);
}
}
}
private static void ProcessEyeImage(List<LSFENTRY> entrys, string workDirectory, MagickImage baseImage, LSFENTRY eyeEntry, string targetPath, bool isMultiCharacter)
{
string eyeName = new string(eyeEntry.name).Trim('\0');
string eyeImagePath = Path.Combine(workDirectory, eyeName + ".png");
if (!File.Exists(eyeImagePath))
{
Console.WriteLine($"Error, File not found: {eyeName}");
return;
}
using var eyeImage = new MagickImage(eyeImagePath);
var tmpImage = RealProcessIMG(baseImage, eyeImage, (int)eyeEntry.x, (int)eyeEntry.y);
string target2Path = targetPath + "_" + eyeName;
if (isMultiCharacter)
{
foreach (var eye2Entry in entrys.Where(e => e.type == 0x14))
{
using var tmpImageClone = (MagickImage)tmpImage.Clone();
ProcessOverlayImage(eye2Entry, target2Path, workDirectory, tmpImageClone);
}
}
else
{
tmpImage.Write(target2Path + ".png");
}
}
private static void ProcessOverlayImage(LSFENTRY eye2Entry, string target2Path, string workDirectory, MagickImage tmpImage)
{
string eye2Name = new string(eye2Entry.name).Trim('\0');
string eye2ImagePath = Path.Combine(workDirectory, eye2Name + ".png");
if (!File.Exists(eye2ImagePath))
{
Console.WriteLine($"Error, File not found: {eye2Name}");
return;
}
using var overlayImage = new MagickImage(eye2ImagePath);
RealProcessIMG(tmpImage, overlayImage, (int)eye2Entry.x, (int)eye2Entry.y).Write($"{target2Path}_{eye2Name}.png");
}
static MagickImage RealProcessIMG(MagickImage b, MagickImage o, int x, int y)
{
o.Alpha(AlphaOption.Set);
// Not working
// 原先是设计给处理那些不透明的图片,但好像有问题
//if (args[4] == "y")
//{
// //overlayImage.ColorFuzz = new Percentage(6); // For Face
// //overlayImage.Transparent(MagickColors.White); // For Face
// //overlayImage.Blur(0, 4); // For Face
//}
b.Composite(o, x, y, CompositeOperator.Over);
return b;
}
}
}