public Expression DecompileDynArrFunction(String name, bool secondArg = false, bool withoutMemOffs = false, bool withoutTrailingByte = false)
{
PopByte();
var arr = DecompileExpression();
if (arr == null)
return null;
if (!withoutMemOffs)
ReadInt16(); // MemSize
var args = new List<Expression>();
if (secondArg)
{
var prop = DecompileExpression();
if (prop == null)
return null;
args.Add(prop);
}
var value = DecompileExpression();
if (value == null)
return null;
args.Add(value);
if (!withoutTrailingByte)
PopByte(); // EndFuncParms
var builder = new CodeBuilderVisitor(); // what a wonderful hack, TODO.
arr.AcceptVisitor(builder);
StartPositions.Pop();
// TODO: ugly solution, should be reworked once dynarrays are in the AST.
return new FunctionCall(new SymbolReference(null, null, null, builder.GetCodeString() + "." + name), args, null, null);
}
public Expression DecompileContext(bool isClass = false)
{
PopByte();
var left = DecompileExpression();
if (left == null)
return null; // ERROR
ReadInt16(); // discard MemSize value. (size of expr-right in half-bytes)
ReadObject(); // discard RetValRef.
ReadByte(); // discard unknown byte.
isInClassContext = isClass;
var right = DecompileExpression();
if (right == null)
return null; // ERROR
isInClassContext = false;
if (isClass)
{
var builder = new CodeBuilderVisitor(); // what a wonderful hack, TODO.
left.AcceptVisitor(builder);
var str = builder.GetCodeString() + ".static";
left = new SymbolReference(null, null, null, str);
}
StartPositions.Pop();
return new CompositeSymbolRef(left, right, null, null);
}
public void BasicClassTest()
{
var source =
"class Test Deprecated Transient; \n" +
"var enum ETestnumeration {\n" +
" TEST_value1,\n" +
" TEST_value2,\n" +
" TEST_value3,\n" +
"} inlineNumeration, testnum2;\n" +
"var private deprecated int X; \n" +
"VAR INT Y, Z; \n" +
"var ETestnumeration testnum;\n" +
"struct transient testStruct\n" +
"{ var float a, b, c; };\n" +
"var private struct transient twoStruct extends testStruct\n" +
"{\n" +
" var etestnumeration num;\n" +
"} structA, structB;\n" +
"function float funcB( testStruct one, float two ) \n" +
"{\n" +
" local float c;" +
" one.a = 1.3 * c;" +
" while (true)" +
" {" +
" c = c - c - c;" +
" }" +
" if (false) {" +
" switch(one.a) {" +
" case one.a:" +
" c = one.a;" +
" break;" +
" case 1.2:" +
" case 1.3:" +
" c = c + 0.5;" +
" break;" +
" default:" +
" c = 6.6;" +
" }" +
" }" +
" return one.a + 0.33 * (0.66 + 0.1) * 1.5;\n" +
"}\n" +
"private simulated function float MyFunc( out testStruct one, coerce optional float two ) \n" +
"{\n" +
" return one.b + funcB(one, two);\n" +
"}\n" +
"auto state MyState\n" +
"{\n" +
"ignores MyFunc;\n" +
"function StateFunc()\n" +
"{\n" +
"}\n" +
"\n" +
"Begin:\n" +
" moredragons\n" +
"}\n" +
"\n" +
"final static operator(254) int >>>( coerce float left, coerce float right )\n" +
"{\n" +
" all the dragons\n" +
"}\n" +
"\n" +
"\n" +
"\n";
var parser = new ClassOutlineParser(new TokenStream<String>(new StringLexer(source)), log);
var symbols = new SymbolTable();
Class obj = new Class("Object", null, null, null, null, null, null, null, null, null, null);
obj.OuterClass = obj;
symbols.PushScope(obj.Name);
symbols.AddSymbol(obj.Name, obj);
VariableType integer = new VariableType("int", null, null);
symbols.AddSymbol(integer.Name, integer);
VariableType floatingpoint = new VariableType("float", null, null);
symbols.AddSymbol(floatingpoint.Name, floatingpoint);
InOpDeclaration plus_float = new InOpDeclaration("+", 20, false, null, floatingpoint, new FunctionParameter(floatingpoint, null, null, null, null),
new FunctionParameter(floatingpoint, null, null, null, null), null, null, null);
symbols.AddOperator(plus_float);
InOpDeclaration sub_float = new InOpDeclaration("-", 20, false, null, floatingpoint, new FunctionParameter(floatingpoint, null, null, null, null),
new FunctionParameter(floatingpoint, null, null, null, null), null, null, null);
symbols.AddOperator(sub_float);
InOpDeclaration mult_float = new InOpDeclaration("*", 16, false, null, floatingpoint, new FunctionParameter(floatingpoint, null, null, null, null),
new FunctionParameter(floatingpoint, null, null, null, null), null, null, null);
symbols.AddOperator(mult_float);
Class node = (Class)parser.ParseDocument();
var ClassValidator = new ClassValidationVisitor(log, symbols);
node.AcceptVisitor(ClassValidator);
symbols.GoDirectlyToStack(node.GetInheritanceString());
foreach (Function f in node.Functions)
{
symbols.PushScope(f.Name);
var p = new CodeBodyParser(new TokenStream<String>(new StringLexer(source)), f.Body, symbols, f, log);
var b = p.ParseBody();
symbols.PopScope();
}
var CodeBuilder = new CodeBuilderVisitor();
node.AcceptVisitor(CodeBuilder);
Console.Write(CodeBuilder.GetCodeString());
Assert.IsTrue(log.AllErrors.Count == 0);
return;
}
// TODO: guess for loop, probably requires a large restructure
public Statement DecompileForEach(bool isDynArray = false)
{
PopByte();
var scopeStatements = new List<Statement>();
var iteratorFunc = DecompileExpression();
if (iteratorFunc == null)
return null;
Expression dynArrVar = null;
Expression dynArrIndex = null;
bool unknByte = false;
if (isDynArray)
{
dynArrVar = DecompileExpression();
unknByte = Convert.ToBoolean(ReadByte());
dynArrIndex = DecompileExpression();
}
var scopeEnd = ReadUInt16(); // MemOff
ForEachScopes.Push(scopeEnd);
Scopes.Add(scopeStatements);
CurrentScope.Push(Scopes.Count - 1);
while (Position < Size)
{
if (CurrentIs(StandardByteCodes.IteratorNext) && PeekByte == (byte)StandardByteCodes.IteratorPop)
{
PopByte(); // IteratorNext
PopByte(); // IteratorPop
break;
}
var current = DecompileStatement();
if (current == null)
return null; // ERROR ?
scopeStatements.Add(current);
}
CurrentScope.Pop();
ForEachScopes.Pop();
if (isDynArray)
{
var builder = new CodeBuilderVisitor(); // what a wonderful hack, TODO.
iteratorFunc.AcceptVisitor(builder);
var arrayName = new SymbolReference(null, null, null, builder.GetCodeString());
var parameters = new List<Expression>() { dynArrVar, dynArrIndex };
iteratorFunc = new FunctionCall(arrayName, parameters, null, null);
}
var statement = new ForEachLoop(iteratorFunc, new CodeBody(scopeStatements, null, null), null, null);
StatementLocations.Add(StartPositions.Pop(), statement);
return statement;
}
private void DecompileDefaultParameterValues(List<Statement> statements)
{
OptionalParams = new Stack<FunctionParameter>();
var func = DataContainer as ME3Function;
if (func != null) // Gets all optional params for default value parsing
{
for (int n = 0; n < Parameters.Count; n++)
{
if (func.Parameters[n].PropertyFlags.HasFlag(PropertyFlags.OptionalParm))
OptionalParams.Push(Parameters[n]);
}
}
while (CurrentByte == (byte)StandardByteCodes.DefaultParmValue
|| CurrentByte == (byte)StandardByteCodes.Nothing)
{
StartPositions.Push((UInt16)Position);
var token = PopByte();
if (token == (byte)StandardByteCodes.DefaultParmValue) // default value assigned
{
ReadInt16(); //MemSize of value
var value = DecompileExpression();
PopByte(); // end of value
var builder = new CodeBuilderVisitor(); // what a wonderful hack, TODO.
value.AcceptVisitor(builder);
if (OptionalParams.Count != 0)
{
var parm = OptionalParams.Pop();
parm.Variables.First().Name += " = " + builder.GetCodeString();
StartPositions.Pop();
}
else
{ // TODO: weird, research how to deal with this
var comment = new SymbolReference(null, null, null, "// Orphaned Default Parm: " + builder.GetCodeString());
var statement = new ExpressionOnlyStatement(null, null, comment);
StatementLocations.Add(StartPositions.Pop(), statement);
statements.Add(statement);
}
}
}
}