private Translate(RuleNode ruleNode, ScopeGroup root, VarCollection varCollection, UserMethod[] userMethods)
{
Root = root;
VarCollection = varCollection;
UserMethods = userMethods;
Rule = new Rule(ruleNode.Name, ruleNode.Event, ruleNode.Team, ruleNode.Player);
IsGlobal = Rule.IsGlobal;
ContinueSkip = new ContinueSkip(IsGlobal, Actions, varCollection);
ParseConditions(ruleNode.Conditions);
ParseBlock(root.Child(), ruleNode.Block, false, null);
Rule.Actions = Actions.ToArray();
Rule.Conditions = Conditions.ToArray();
// Fufill remaining skips
foreach (var skip in ReturnSkips)
{
skip.ParameterValues = new IWorkshopTree[] { new V_Number(Actions.Count - ReturnSkips.IndexOf(skip)) }
}
;
ReturnSkips.Clear();
}
void ParseConditions(IExpressionNode[] expressions)
{
foreach (var expr in expressions)
{
Element parsedIf = ParseExpression(Root, expr);
// If the parsed if is a V_Compare, translate it to a condition.
// Makes "(value1 == value2) == true" to just "value1 == value2"
if (parsedIf is V_Compare)
{
Element left = (Element)parsedIf.ParameterValues[0];
if (!left.ElementData.IsValue)
{
throw SyntaxErrorException.InvalidMethodType(true, left.Name, ((Node)expr).Range);
}
Element right = (Element)parsedIf.ParameterValues[2];
if (!right.ElementData.IsValue)
{
throw SyntaxErrorException.InvalidMethodType(true, right.Name, ((Node)expr).Range);
}
Conditions.Add(
new Condition(
left,
(EnumMember)parsedIf.ParameterValues[1],
right
)
);
}
// If not, just do "parsedIf == true"
else
{
if (!parsedIf.ElementData.IsValue)
{
throw SyntaxErrorException.InvalidMethodType(true, parsedIf.Name, ((Node)expr).Range);
}
Conditions.Add(new Condition(
parsedIf, EnumData.GetEnumValue(Operators.Equal), new V_True()
));
}
}
}
Element ParseExpression(ScopeGroup scope, IExpressionNode expression)
{
switch (expression)
{
// Math and boolean operations.
case OperationNode operationNode:
{
Element left = ParseExpression(scope, operationNode.Left);
Element right = ParseExpression(scope, operationNode.Right);
/*
* if (Constants.BoolOperations.Contains(operationNode.Operation))
* {
* if (left.ElementData.ValueType != Elements.ValueType.Any && left.ElementData.ValueType != Elements.ValueType.Boolean)
* throw new SyntaxErrorException($"Expected boolean, got {left .ElementData.ValueType.ToString()} instead.", ((Node)operationNode.Left).Range);
*
* if (right.ElementData.ValueType != Elements.ValueType.Any && right.ElementData.ValueType != Elements.ValueType.Boolean)
* throw new SyntaxErrorException($"Expected boolean, got {right.ElementData.ValueType.ToString()} instead.", ((Node)operationNode.Right).Range);
* }
*/
switch (operationNode.Operation)
{
// Math: ^, *, %, /, +, -
case "^":
return(Element.Part <V_RaiseToPower>(left, right));
case "*":
return(Element.Part <V_Multiply>(left, right));
case "%":
return(Element.Part <V_Modulo>(left, right));
case "/":
return(Element.Part <V_Divide>(left, right));
case "+":
return(Element.Part <V_Add>(left, right));
case "-":
return(Element.Part <V_Subtract>(left, right));
// BoolCompare: &, |
case "&":
return(Element.Part <V_And>(left, right));
case "|":
return(Element.Part <V_Or>(left, right));
// Compare: <, <=, ==, >=, >, !=
case "<":
return(Element.Part <V_Compare>(left, EnumData.GetEnumValue(Operators.LessThan), right));
case "<=":
return(Element.Part <V_Compare>(left, EnumData.GetEnumValue(Operators.LessThanOrEqual), right));
case "==":
return(Element.Part <V_Compare>(left, EnumData.GetEnumValue(Operators.Equal), right));
case ">=":
return(Element.Part <V_Compare>(left, EnumData.GetEnumValue(Operators.GreaterThanOrEqual), right));
case ">":
return(Element.Part <V_Compare>(left, EnumData.GetEnumValue(Operators.GreaterThan), right));
case "!=":
return(Element.Part <V_Compare>(left, EnumData.GetEnumValue(Operators.NotEqual), right));
}
throw new Exception($"Operation {operationNode.Operation} not implemented.");
}
// Number
case NumberNode numberNode:
return(new V_Number(numberNode.Value));
// Bool
case BooleanNode boolNode:
if (boolNode.Value)
{
return(new V_True());
}
else
{
return(new V_False());
}
// Not operation
case NotNode notNode:
return(Element.Part <V_Not>(ParseExpression(scope, notNode.Value)));
// Strings
case StringNode stringNode:
Element[] stringFormat = new Element[stringNode.Format?.Length ?? 0];
for (int i = 0; i < stringFormat.Length; i++)
{
stringFormat[i] = ParseExpression(scope, stringNode.Format[i]);
}
return(V_String.ParseString(stringNode.Range, stringNode.Value, stringFormat));
// Null
case NullNode nullNode:
return(new V_Null());
// TODO check if groups need to be implemented here
// Methods
case MethodNode methodNode:
return(ParseMethod(scope, methodNode, true));
// Variable
case VariableNode variableNode:
return(scope.GetVar(variableNode.Name, variableNode.Range, Diagnostics)
.GetVariable(variableNode.Target != null ? ParseExpression(scope, variableNode.Target) : null));
// Get value in array
case ValueInArrayNode viaNode:
return(Element.Part <V_ValueInArray>(ParseExpression(scope, viaNode.Value), ParseExpression(scope, viaNode.Index)));
// Create array
case CreateArrayNode createArrayNode:
{
Element prev = null;
Element current = null;
for (int i = 0; i < createArrayNode.Values.Length; i++)
{
current = new V_Append()
{
ParameterValues = new IWorkshopTree[2]
};
if (prev != null)
{
current.ParameterValues[0] = prev;
}
else
{
current.ParameterValues[0] = new V_EmptyArray();
}
current.ParameterValues[1] = ParseExpression(scope, createArrayNode.Values[i]);
prev = current;
}
return(current ?? new V_EmptyArray());
}
case EnumNode enumNode:
return(EnumData.Special(enumNode.EnumMember)
?? throw SyntaxErrorException.EnumCantBeValue(enumNode.Type, enumNode.Range));
// Seperator
}
throw new Exception();
}
Element ParseMethod(ScopeGroup scope, MethodNode methodNode, bool needsToBeValue)
{
methodNode.RelatedScopeGroup = scope;
// Get the kind of method the method is (Method (Overwatch), Custom Method, or User Method.)
var methodType = GetMethodType(UserMethods, methodNode.Name);
// Throw exception if the method does not exist.
if (methodType == null)
{
throw SyntaxErrorException.NonexistentMethod(methodNode.Name, methodNode.Range);
}
Element method;
switch (methodType)
{
case MethodType.Method:
{
Type owMethod = Element.GetMethod(methodNode.Name);
method = (Element)Activator.CreateInstance(owMethod);
Parameter[] parameterData = owMethod.GetCustomAttributes <Parameter>().ToArray();
List <IWorkshopTree> parsedParameters = new List <IWorkshopTree>();
for (int i = 0; i < parameterData.Length; i++)
{
if (methodNode.Parameters.Length > i)
{
// Parse the parameter.
parsedParameters.Add(ParseParameter(scope, methodNode.Parameters[i], methodNode.Name, parameterData[i]));
}
else
{
if (parameterData[i].ParameterType == ParameterType.Value && parameterData[i].DefaultType == null)
{
// Throw exception if there is no default method to fallback on.
throw SyntaxErrorException.MissingParameter(parameterData[i].Name, methodNode.Name, methodNode.Range);
}
else
{
parsedParameters.Add(parameterData[i].GetDefault());
}
}
}
method.ParameterValues = parsedParameters.ToArray();
break;
}
case MethodType.CustomMethod:
{
MethodInfo customMethod = CustomMethods.GetCustomMethod(methodNode.Name);
Parameter[] parameterData = customMethod.GetCustomAttributes <Parameter>().ToArray();
object[] parsedParameters = new Element[parameterData.Length];
for (int i = 0; i < parameterData.Length; i++)
{
if (methodNode.Parameters.Length > i)
{
parsedParameters[i] = ParseParameter(scope, methodNode.Parameters[i], methodNode.Name, parameterData[i]);
}
else
{
// Throw exception if there is no default method to fallback on.
throw SyntaxErrorException.MissingParameter(parameterData[i].Name, methodNode.Name, methodNode.Range);
}
}
MethodResult result = (MethodResult)customMethod.Invoke(null, new object[] { IsGlobal, VarCollection, parsedParameters });
switch (result.MethodType)
{
case CustomMethodType.Action:
if (needsToBeValue)
{
throw SyntaxErrorException.InvalidMethodType(true, methodNode.Name, methodNode.Range);
}
break;
case CustomMethodType.MultiAction_Value:
case CustomMethodType.Value:
if (!needsToBeValue)
{
throw SyntaxErrorException.InvalidMethodType(false, methodNode.Name, methodNode.Range);
}
break;
}
// Some custom methods have extra actions.
if (result.Elements != null)
{
Actions.AddRange(result.Elements);
}
method = result.Result;
break;
}
case MethodType.UserMethod:
{
if (!AllowRecursion)
{
UserMethod userMethod = UserMethod.GetUserMethod(UserMethods, methodNode.Name);
if (MethodStackNoRecursive.Contains(userMethod))
{
throw SyntaxErrorException.RecursionNotAllowed(methodNode.Range);
}
var methodScope = Root.Child();
// Add the parameter variables to the scope.
DefinedVar[] parameterVars = new DefinedVar[userMethod.Parameters.Length];
for (int i = 0; i < parameterVars.Length; i++)
{
if (methodNode.Parameters.Length > i)
{
// Create a new variable using the parameter input.
parameterVars[i] = VarCollection.AssignDefinedVar(methodScope, IsGlobal, userMethod.Parameters[i].Name, methodNode.Range);
Actions.Add(parameterVars[i].SetVariable(ParseExpression(scope, methodNode.Parameters[i])));
}
else
{
throw SyntaxErrorException.MissingParameter(userMethod.Parameters[i].Name, methodNode.Name, methodNode.Range);
}
}
var returns = VarCollection.AssignVar($"{methodNode.Name}: return temp value", IsGlobal);
MethodStackNoRecursive.Add(userMethod);
var userMethodScope = methodScope.Child();
userMethod.Block.RelatedScopeGroup = userMethodScope;
ParseBlock(userMethodScope, userMethod.Block, true, returns);
MethodStackNoRecursive.Remove(userMethod);
// No return value if the method is being used as an action.
if (needsToBeValue)
{
method = returns.GetVariable();
}
else
{
method = null;
}
break;
}
else
{
UserMethod userMethod = UserMethod.GetUserMethod(UserMethods, methodNode.Name);
MethodStack lastMethod = MethodStack.FirstOrDefault(ms => ms.UserMethod == userMethod);
if (lastMethod != null)
{
ContinueSkip.Setup();
for (int i = 0; i < lastMethod.ParameterVars.Length; i++)
{
if (methodNode.Parameters.Length > i)
{
Actions.Add(lastMethod.ParameterVars[i].Push(ParseExpression(scope, methodNode.Parameters[i])));
}
}
// ?--- Multidimensional Array
Actions.Add(
Element.Part <A_SetGlobalVariable>(EnumData.GetEnumValue(Variable.B), lastMethod.ContinueSkipArray.GetVariable())
);
Actions.Add(
Element.Part <A_ModifyGlobalVariable>(EnumData.GetEnumValue(Variable.B), EnumData.GetEnumValue(Operation.AppendToArray), new V_Number(ContinueSkip.GetSkipCount() + 4))
);
Actions.Add(
lastMethod.ContinueSkipArray.SetVariable(Element.Part <V_GlobalVariable>(EnumData.GetEnumValue(Variable.B)))
);
// ?---
ContinueSkip.SetSkipCount(lastMethod.ActionIndex);
Actions.Add(Element.Part <A_Loop>());
if (needsToBeValue)
{
method = lastMethod.Return.GetVariable();
}
else
{
method = null;
}
}
else
{
var methodScope = Root.Child();
// Add the parameter variables to the scope.
ParameterVar[] parameterVars = new ParameterVar[userMethod.Parameters.Length];
for (int i = 0; i < parameterVars.Length; i++)
{
if (methodNode.Parameters.Length > i)
{
// Create a new variable using the parameter input.
parameterVars[i] = VarCollection.AssignParameterVar(Actions, methodScope, IsGlobal, userMethod.Parameters[i].Name, methodNode.Range);
Actions.Add(parameterVars[i].Push(ParseExpression(scope, methodNode.Parameters[i])));
}
else
{
throw SyntaxErrorException.MissingParameter(userMethod.Parameters[i].Name, methodNode.Name, methodNode.Range);
}
}
var returns = VarCollection.AssignVar($"{methodNode.Name}: return temp value", IsGlobal);
Var continueSkipArray = VarCollection.AssignVar($"{methodNode.Name}: continue skip temp value", IsGlobal);
var stack = new MethodStack(userMethod, parameterVars, ContinueSkip.GetSkipCount(), returns, continueSkipArray);
MethodStack.Add(stack);
var userMethodScope = methodScope.Child();
userMethod.Block.RelatedScopeGroup = userMethodScope;
ParseBlock(userMethodScope, userMethod.Block, true, returns);
// No return value if the method is being used as an action.
if (needsToBeValue)
{
method = returns.GetVariable();
}
else
{
method = null;
}
Actions.Add(Element.Part <A_Wait>(new V_Number(Constants.MINIMUM_WAIT)));
for (int i = 0; i < parameterVars.Length; i++)
{
parameterVars[i].Pop();
}
ContinueSkip.Setup();
ContinueSkip.SetSkipCount(Element.Part <V_LastOf>(continueSkipArray.GetVariable()));
// ?--- Multidimensional Array
Actions.Add(
Element.Part <A_SetGlobalVariable>(EnumData.GetEnumValue(Variable.B), continueSkipArray.GetVariable())
);
Actions.Add(
continueSkipArray.SetVariable(
Element.Part <V_ArraySlice>(
Element.Part <V_GlobalVariable>(EnumData.GetEnumValue(Variable.B)),
new V_Number(0),
Element.Part <V_Subtract>(
Element.Part <V_CountOf>(Element.Part <V_GlobalVariable>(EnumData.GetEnumValue(Variable.B))),
new V_Number(1)
)
)
)
);
// ?---
Actions.Add(
Element.Part <A_LoopIf>(
Element.Part <V_Compare>(
Element.Part <V_CountOf>(continueSkipArray.GetVariable()),
EnumData.GetEnumValue(Operators.NotEqual),
new V_Number(0)
)
)
);
ContinueSkip.ResetSkip();
MethodStack.Remove(stack);
}
break;
}
}
default: throw new NotImplementedException();
}
methodNode.RelatedElement = method;
return(method);
}
IWorkshopTree ParseParameter(ScopeGroup scope, IExpressionNode node, string methodName, Parameter parameterData)
{
IWorkshopTree value = null;
switch (node)
{
case EnumNode enumNode:
/*
* if (parameterData.ParameterType != ParameterType.Enum)
* throw SyntaxErrorException.ExpectedType(true, parameterData.ValueType.ToString(), methodName, parameterData.Name, enumNode.Range);
*
* if (enumNode.Type != parameterData.EnumType.Name)
* throw SyntaxErrorException.ExpectedType(false, parameterData.EnumType.ToString(), methodName, parameterData.Name, enumNode.Range);
*/
value = (IWorkshopTree)EnumData.Special(enumNode.EnumMember) ?? (IWorkshopTree)enumNode.EnumMember;
//if (value == null)
// throw SyntaxErrorException.InvalidEnumValue(enumNode.Type, enumNode.Value, enumNode.Range);
break;
default:
if (parameterData.ParameterType != ParameterType.Value)
{
throw SyntaxErrorException.ExpectedType(false, parameterData.EnumType.Name, methodName, parameterData.Name, ((Node)node).Range);
}
value = ParseExpression(scope, node);
Element element = value as Element;
ElementData elementData = element.GetType().GetCustomAttribute <ElementData>();
if (elementData.ValueType != Elements.ValueType.Any &&
!parameterData.ValueType.HasFlag(elementData.ValueType))
{
throw SyntaxErrorException.InvalidType(parameterData.ValueType, elementData.ValueType, ((Node)node).Range);
}
break;
}
if (value == null)
{
throw new Exception("Failed to parse parameter.");
}
return(value);
}
public bool WasCalled(IApplyBlock callable) => MethodStack.Any(ms => ms.Function == this);
override public Element Get(TranslateRule context, ScopeGroup scope, MethodNode methodNode, IWorkshopTree[] parameters)
{
Element result;
if (!IsRecursive)
{
// Check the method stack if this method was already called.
// Throw a syntax error if it was.
if (context.MethodStackNotRecursive.Contains(this))
{
throw SyntaxErrorException.RecursionNotAllowed(methodNode.Location);
}
var methodScope = scope.Root().Child();
// Add the parameter variables to the scope.
context.AssignParameterVariables(methodScope, Parameters, parameters, methodNode);
// The variable that stores the return value.
IndexedVar returns = null;
if (DoesReturn)
{
returns = IndexedVar.AssignVar(context.VarCollection, scope, $"{methodNode.Name} return", context.IsGlobal, null);
returns.Type = Type;
}
// Add the method to the method stack
context.MethodStackNotRecursive.Add(this);
Block.RelatedScopeGroup = methodScope;
// Parse the block of the method
context.ParseBlock(methodScope, methodScope, Block, true, returns);
// Take the method scope out of scope.
methodScope.Out(context);
// Remove the method from the stack.
context.MethodStackNotRecursive.Remove(this);
if (DoesReturn)
{
result = returns.GetVariable();
}
else
{
result = new V_Null();
}
}
else
{
// Check the method stack if this method was already called. It will be null if it wasn't called.
MethodStack lastMethod = context.MethodStackRecursive.FirstOrDefault(ms => ms.UserMethod == this);
if (lastMethod != null)
{
context.ContinueSkip.Setup();
// Re-push the paramaters.
for (int i = 0; i < lastMethod.ParameterVars.Length; i++)
{
if (lastMethod.ParameterVars[i] is RecursiveVar)
{
context.Actions.AddRange
(
((RecursiveVar)lastMethod.ParameterVars[i]).InScope((Element)parameters[i])
);
}
}
// Add to the continue skip array.
context.Actions.AddRange(
lastMethod.ContinueSkipArray.SetVariable(
Element.Part <V_Append>(lastMethod.ContinueSkipArray.GetVariable(), new V_Number(context.ContinueSkip.GetSkipCount() + 3))
)
);
// Loop back to the start of the method.
context.ContinueSkip.SetSkipCount(lastMethod.ActionIndex);
context.Actions.Add(Element.Part <A_Loop>());
result = lastMethod.Return.GetVariable();
}
else
{
var methodScope = scope.Root().Child(true);
// Add the parameter variables to the scope.
Var[] parameterVars = new Var[Parameters.Length];
for (int i = 0; i < parameterVars.Length; i++)
{
if (parameters[i] is Element)
{
// Create a new variable using the parameter input.
parameterVars[i] = (RecursiveVar)IndexedVar.AssignVar(context.VarCollection, methodScope, Parameters[i].Name, context.IsGlobal, methodNode);
((RecursiveVar)parameterVars[i]).Type = ((Element)parameters[i]).SupportedType?.Type;
context.Actions.AddRange
(
((RecursiveVar)parameterVars[i]).InScope((Element)parameters[i])
);
}
else if (parameters[i] is EnumMember)
{
parameterVars[i] = new ElementReferenceVar(Parameters[i].Name, methodScope, methodNode, parameters[i]);
}
else
{
throw new NotImplementedException();
}
}
var returns = IndexedVar.AssignInternalVarExt(context.VarCollection, null, $"{methodNode.Name}: return", context.IsGlobal);
returns.Type = Type;
// Setup the continue skip array.
IndexedVar continueSkipArray = IndexedVar.AssignInternalVar(context.VarCollection, null, $"{methodNode.Name} sca", context.IsGlobal);
var stack = new MethodStack(this, parameterVars, context.ContinueSkip.GetSkipCount(), returns, continueSkipArray);
// Add the method to the stack.
context.MethodStackRecursive.Add(stack);
Block.RelatedScopeGroup = methodScope;
// Parse the method block
context.ParseBlock(methodScope, methodScope, Block, true, returns);
// No return value if the method is being used as an action.
result = returns.GetVariable();
// Take the method out of scope.
//Actions.AddRange(methodScope.RecursiveMethodStackPop());
methodScope.Out(context);
// Setup the next continue skip.
context.ContinueSkip.Setup();
context.ContinueSkip.SetSkipCount(Element.Part <V_LastOf>(continueSkipArray.GetVariable()));
// Remove the last continue skip.
context.Actions.AddRange(
continueSkipArray.SetVariable(
Element.Part <V_ArraySlice>(
continueSkipArray.GetVariable(),
new V_Number(0),
Element.Part <V_CountOf>(continueSkipArray.GetVariable()) - 1
)
)
);
// Loop if the method goes any deeper by checking the length of the continue skip array.
context.Actions.Add(
Element.Part <A_LoopIf>(
Element.Part <V_Compare>(
Element.Part <V_CountOf>(continueSkipArray.GetVariable()),
EnumData.GetEnumValue(Operators.NotEqual),
new V_Number(0)
)
)
);
// Reset the continue skip.
context.ContinueSkip.ResetSkip();
context.Actions.AddRange(continueSkipArray.SetVariable(0));
// Remove the method from the stack.
context.MethodStackRecursive.Remove(stack);
}
}
return(result);
}
