mirror of
https://github.com/ncblakely/GiantsTools
synced 2024-11-16 19:55:36 +01:00
182 lines
6.2 KiB
C#
182 lines
6.2 KiB
C#
namespace Giants.EffectCompiler
|
|
{
|
|
using System;
|
|
using System.CodeDom.Compiler;
|
|
using System.Collections.Generic;
|
|
using System.IO;
|
|
using System.Text;
|
|
using Giants.BinTools;
|
|
using Giants.BinTools.Macro;
|
|
using NLog;
|
|
|
|
/// <summary>
|
|
/// Effect binary decompiler.
|
|
/// </summary>
|
|
public class FxDecompiler
|
|
{
|
|
private static readonly Logger logger = LogManager.GetLogger(nameof(FxDecompiler));
|
|
|
|
private readonly Dictionary<string, int> tableOfContents = new Dictionary<string, int>();
|
|
|
|
private FxBinaryData binaryData;
|
|
private FxMacroDefinitionTable macroDefinitionTable;
|
|
private int doneOpcode;
|
|
private int pushObjOpcode;
|
|
private int popObjOpcode;
|
|
|
|
/// <summary>
|
|
/// Initializes a new instance of the <see cref="FxDecompiler"/> class.
|
|
/// </summary>
|
|
/// <param name="macroDefinitionTable">The table of macro and symbol definitions.</param>
|
|
public FxDecompiler(FxMacroDefinitionTable macroDefinitionTable)
|
|
{
|
|
this.macroDefinitionTable = macroDefinitionTable;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Decompiles an effect binary file to the specified output path.
|
|
/// </summary>
|
|
/// <param name="inputPath">The path to the effect binary file.</param>
|
|
/// <param name="outputPath">The path to write to.</param>
|
|
public void Decompile(
|
|
string inputPath,
|
|
string outputPath)
|
|
|
|
{
|
|
if (!File.Exists(inputPath))
|
|
{
|
|
throw new InvalidOperationException($"The input file {inputPath} does not exist.");
|
|
}
|
|
|
|
this.ReadBinaryData(inputPath);
|
|
|
|
using var memoryStream = new MemoryStream(this.binaryData.Data);
|
|
using var binaryReader = new BinaryReader(memoryStream);
|
|
using var streamWriter = new StreamWriter(outputPath, false);
|
|
using var textWriter = new IndentedTextWriter(streamWriter);
|
|
|
|
this.BuildTableOfContents(binaryReader);
|
|
|
|
// Get constants for known symbols
|
|
this.doneOpcode = Utilities.GetFxSymbolValue(this.macroDefinitionTable.SymbolTable, "FxDone");
|
|
this.pushObjOpcode = Utilities.GetFxSymbolValue(this.macroDefinitionTable.SymbolTable, "FxPushObj");
|
|
this.popObjOpcode = Utilities.GetFxSymbolValue(this.macroDefinitionTable.SymbolTable, "FxPopObj");
|
|
|
|
this.ProcessEffects(binaryReader, textWriter);
|
|
}
|
|
|
|
private void ReadBinaryData(string path)
|
|
{
|
|
using var fileStream = File.OpenRead(path);
|
|
this.binaryData = new FxBinaryData(fileStream);
|
|
}
|
|
|
|
private void BuildTableOfContents(BinaryReader binaryReader)
|
|
{
|
|
for (int i = 0; i < this.binaryData.EffectCount; i++)
|
|
{
|
|
string fxName = binaryReader.ReadCString();
|
|
int offset = binaryReader.ReadInt32();
|
|
|
|
if (this.tableOfContents.ContainsKey(fxName))
|
|
{
|
|
logger.Warn($"TOC already contains fx '{fxName}'; this may be a bug");
|
|
continue;
|
|
}
|
|
|
|
this.tableOfContents.Add(fxName, offset);
|
|
}
|
|
}
|
|
|
|
private void ProcessEffects(BinaryReader binaryReader, IndentedTextWriter fileStream)
|
|
{
|
|
try
|
|
{
|
|
foreach (var kvp in this.tableOfContents)
|
|
{
|
|
this.WriteHeader(fileStream);
|
|
this.ProcessEffect(binaryReader, fileStream, kvp.Key, kvp.Value);
|
|
}
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
logger.Error(e.Message);
|
|
return;
|
|
}
|
|
}
|
|
|
|
private void WriteHeader(IndentedTextWriter fileStream)
|
|
{
|
|
fileStream.WriteLine("; Generated by FX Compiler. Comments will be lost.");
|
|
}
|
|
|
|
private void ProcessEffect(
|
|
BinaryReader reader,
|
|
IndentedTextWriter fileStream,
|
|
string name,
|
|
int offset)
|
|
{
|
|
logger.Info($"Processing '{name}'");
|
|
|
|
reader.BaseStream.Position = offset;
|
|
|
|
fileStream.WriteLine($"fxdef {name}");
|
|
fileStream.Indent = 1;
|
|
|
|
int lastOpcode = 0;
|
|
try
|
|
{
|
|
while (true)
|
|
{
|
|
int opcode = reader.ReadByte();
|
|
if (!this.macroDefinitionTable.MacroDefinitions.TryGetValue(opcode, out FxMacroDefinition macroDefinition))
|
|
{
|
|
throw new InvalidOperationException("Unable to map opcode to macro definition");
|
|
}
|
|
|
|
if (opcode == this.doneOpcode)
|
|
{
|
|
fileStream.Indent--;
|
|
break;
|
|
}
|
|
|
|
if (lastOpcode == this.pushObjOpcode)
|
|
{
|
|
fileStream.Indent++;
|
|
}
|
|
|
|
var lineBuilder = new StringBuilder();
|
|
lineBuilder.Append(macroDefinition.Name);
|
|
lineBuilder.Append(" ");
|
|
foreach (var line in macroDefinition.FxDefGroup)
|
|
{
|
|
string stringValue = line switch
|
|
{
|
|
DataDefinitionMacroLine primitiveMacroLine => primitiveMacroLine.Deserialize(reader, this.macroDefinitionTable.SymbolTable),
|
|
_ => throw new NotSupportedException()
|
|
};
|
|
|
|
lineBuilder.Append(stringValue);
|
|
lineBuilder.Append(" ");
|
|
}
|
|
|
|
if (opcode == this.popObjOpcode)
|
|
{
|
|
fileStream.Indent--;
|
|
}
|
|
|
|
fileStream.WriteLine(lineBuilder);
|
|
fileStream.Flush();
|
|
|
|
lastOpcode = opcode;
|
|
}
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
logger.Error($"Exception processing {name}, last opcode was {lastOpcode}. Trace: {e}");
|
|
}
|
|
|
|
fileStream.WriteLine("fxdone" + Environment.NewLine);
|
|
}
|
|
}
|
|
} |