private static List <ASTNode> ParseCodeBlock()
{
List <ASTNode> codeBlock = new List <ASTNode>();
if (currentToken < currentTokens.Count)
{
if (currentTokens[currentToken].Token == "{")
{
currentToken++;
while (currentToken < currentTokens.Count && currentTokens[currentToken].Token != "}")
{
if (currentTokens[currentToken] is ReservedWordToken && currentTokens[currentToken].Token == "if")
{
ASTIf ifNode = ParseIf();
codeBlock.Add(ifNode);
}
else if (currentTokens[currentToken] is ReservedWordToken && currentTokens[currentToken].Token == "for")
{
ASTFor forNode = ParseFor();
codeBlock.Add(forNode);
}
else if (currentTokens[currentToken] is ReservedWordToken && currentTokens[currentToken].Token == "while")
{
ASTWhile whileNode = ParseWhile();
codeBlock.Add(whileNode);
}
else
{
ASTStatement statement = ParseStatement();
if (statement != null)
{
codeBlock.Add(statement);
}
}
}
if (currentToken == currentTokens.Count)
{
throw new ParserException("SourceCode can't end in Code Block.", -1);
}
return(codeBlock);
}
// No '{' means just one line of code
else
{
ASTStatement statement = ParseStatement();
if (statement != null)
{
codeBlock.Add(statement);
}
return(codeBlock);
}
}
throw new ParserException("Expected Code Block found End of File.", -1);
}
public void Visit(ASTIf astIf, Function function)
{
// Compile condition
astIf.Condition.Accept(this, function);
// If condition evaluates to false, branch past the if part
function.Code.Write(OpCodeFactory.BranchIfFalse(0));
var index = function.Code.Count - 1;
// Compile if-part
astIf.IfPart.Accept(this, function);
// Set the offset after compiling the if-part
function.Code[index].As <BranchIfFalse>().Argument = (uint)function.Code.Count;
if (astIf.ElsePart != null)
{
astIf.ElsePart.Accept(this, function);
}
}
ASTStatement ParseStatement()
{
if (lexer.PeekNext() == TokenType.Identifier) // Function call
{
ASTFuncCall funcCall = ParseExpression() as ASTFuncCall;
return(new ASTInlineCall(funcCall));
}
Token first = lexer.NextToken();
switch (first.Type)
{
case TokenType.Set:
{
Token varNameToken = MatchNext(TokenType.Identifier);
List <ASTExpression> indexers = new List <ASTExpression>();
while (lexer.PeekNext() == TokenType.SquareBraceOpen)
{
MatchNext(TokenType.SquareBraceOpen);
indexers.Add(ParseExpression());
MatchNext(TokenType.SquareBraceClose);
}
MatchNext(TokenType.To);
ASTExpression expression = ParseExpression();
return(new ASTSet(varNameToken.Source, expression, indexers));
}
case TokenType.If:
{
ASTExpression check = ParseExpression();
ASTStatements statements = new ASTStatements();
ASTIf ifStmnt = new ASTIf(check, statements);
MatchNext(TokenType.Do);
while (lexer.PeekNext() != TokenType.End)
{
if (lexer.PeekNext() == TokenType.Else)
{
lexer.NextToken();
statements = new ASTStatements();
ifStmnt.SetElse(statements);
MatchNext(TokenType.Do);
}
else if (lexer.PeekNext() == TokenType.Elif)
{
lexer.NextToken();
statements = new ASTStatements();
ASTIf elseifStmnt = new ASTIf(ParseExpression(), statements);
ifStmnt.SetElseIf(elseifStmnt);
MatchNext(TokenType.Do);
}
statements.AddStatement(ParseStatement());
}
lexer.NextToken();
return(ifStmnt);
}
case TokenType.While:
{
ASTExpression check = ParseExpression();
ASTStatements statements = new ASTStatements();
ASTWhile whileStmnt = new ASTWhile(check, statements);
MatchNext(TokenType.Do);
while (lexer.PeekNext() != TokenType.End)
{
statements.AddStatement(ParseStatement());
}
lexer.NextToken();
return(whileStmnt);
}
case TokenType.Return:
{
return(new ASTReturn(ParseExpression()));
}
default:
throw new Exception("Unexpected token " + first);
}
return(null);
}
public bool Visit(ASTIf node)
{
throw new NotImplementedException();
}
public void SetElseIf(ASTIf ifStmnt)
{
elseifs.Add(ifStmnt);
}
private static void PrintNode(ASTNode node, int incident)
{
if (node == null)
{
return;
}
for (int i = 0; i < incident; i++)
{
sw.Write("\t");
}
if (node is ASTString)
{
sw.WriteLine("ASTString: " + ((ASTString)node).String.Replace("\n", "\\n"));
}
else if (node is ASTNumber)
{
sw.WriteLine("ASTNumber: " + ((ASTNumber)node).Number);
}
else if (node is ASTIdentifier)
{
sw.WriteLine("ASTIdentifier: " + ((ASTIdentifier)node).Identifier);
}
else if (node is ASTOperator)
{
ASTOperator op = (ASTOperator)node;
sw.WriteLine("ASTOperator: " + op.Operator);
PrintNode(op.Left, incident + 1);
PrintNode(op.Right, incident + 1);
}
else if (node is ASTStatement)
{
ASTStatement stm = (ASTStatement)node;
sw.WriteLine("ASTStatement: ");
PrintNode(stm.Statement, incident + 1);
}
else if (node is ASTFunctionCall)
{
ASTFunctionCall call = (ASTFunctionCall)node;
sw.WriteLine("ASTFunctionCall: " + call.Scope + "." + call.FunctionName);
foreach (ASTNode param in call.Parameter)
{
PrintNode(param, incident + 1);
}
}
else if (node is ASTClassParameter)
{
ASTClassParameter classP = (ASTClassParameter)node;
sw.WriteLine("ASTClassParameter: " + classP.Scope);
PrintNode(classP.Parameter, incident + 1);
}
else if (node is ASTIf)
{
ASTIf ifNode = (ASTIf)node;
sw.WriteLine("ASTIf:");
for (int i = 0; i < incident; i++)
{
sw.Write("\t");
}
sw.WriteLine("\tCondition:");
PrintNode(ifNode.Condition, incident + 2);
for (int i = 0; i < incident; i++)
{
sw.Write("\t");
}
sw.WriteLine("\tCodeBlock:");
foreach (ASTNode codeBlock in ifNode.CodeBlock)
{
PrintNode(codeBlock, incident + 2);
}
if (ifNode.ElseCodeBlock != null)
{
for (int i = 0; i < incident; i++)
{
sw.Write("\t");
}
sw.WriteLine("\tElse:");
foreach (ASTNode codeBlock in ifNode.ElseCodeBlock)
{
PrintNode(codeBlock, incident + 2);
}
}
}
else if (node is ASTFor)
{
ASTFor forNode = (ASTFor)node;
sw.WriteLine("ASTFor:");
for (int i = 0; i < incident; i++)
{
sw.Write("\t");
}
sw.WriteLine("\tInit:");
PrintNode(forNode.Initialise, incident + 2);
for (int i = 0; i < incident; i++)
{
sw.Write("\t");
}
sw.WriteLine("\tCondition:");
PrintNode(forNode.Condition, incident + 2);
for (int i = 0; i < incident; i++)
{
sw.Write("\t");
}
sw.WriteLine("\tCounter:");
PrintNode(forNode.Count, incident + 2);
for (int i = 0; i < incident; i++)
{
sw.Write("\t");
}
sw.WriteLine("\tCodeBlock:");
foreach (ASTNode codeBlock in forNode.CodeBlock)
{
PrintNode(codeBlock, incident + 2);
}
}
else if (node is ASTWhile)
{
ASTWhile wNode = (ASTWhile)node;
sw.WriteLine("ASTWhile:");
for (int i = 0; i < incident; i++)
{
sw.Write("\t");
}
sw.WriteLine("\tCondition:");
PrintNode(wNode.Condition, incident + 2);
for (int i = 0; i < incident; i++)
{
sw.Write("\t");
}
sw.WriteLine("\tCodeBlock:");
foreach (ASTNode codeBlock in wNode.CodeBlock)
{
PrintNode(codeBlock, incident + 2);
}
}
else if (node is ASTFunction)
{
ASTFunction func = (ASTFunction)node;
if (node is ASTConstructor)
{
sw.WriteLine("ASTConstructor:");
}
else
{
sw.WriteLine("ASTFunction: " + func.ReturnValue.ToString() + " " + func.Name);
}
for (int i = 0; i < incident; i++)
{
sw.Write("\t");
}
sw.WriteLine("\tParameter:");
foreach (ASTNode n in func.Parameter)
{
PrintNode(n, incident + 2);
}
for (int i = 0; i < incident; i++)
{
sw.Write("\t");
}
sw.WriteLine("\tCodeBlock:");
foreach (ASTNode codeBlock in func.CodeBlock)
{
PrintNode(codeBlock, incident + 2);
}
}
else if (node is ASTFunctionParameter)
{
ASTFunctionParameter param = (ASTFunctionParameter)node;
switch (param.Type)
{
case VMType.Object:
sw.WriteLine("Object: " + param.Name);
break;
case VMType.String:
sw.WriteLine("String: " + param.Name);
break;
case VMType.Number:
sw.WriteLine("Number: " + param.Name);
break;
}
}
else if (node is ASTLocalVariable)
{
ASTLocalVariable param = (ASTLocalVariable)node;
switch (param.Type)
{
case VMType.Object:
sw.WriteLine("var Object: " + param.Name);
break;
case VMType.String:
sw.WriteLine("var String: " + param.Name);
break;
case VMType.Number:
sw.WriteLine("var Number: " + param.Name);
break;
}
}
else if (node is ASTReturn)
{
sw.WriteLine("ASTReturn:");
PrintNode(((ASTReturn)node).Return, incident + 1);
}
else if (node is ASTClass)
{
ASTClass cNode = (ASTClass)node;
sw.WriteLine("ASTClass: " + cNode.Name);
sw.WriteLine("\tAttributes:");
foreach (ASTNode n in cNode.Attributes)
{
PrintNode(n, incident + 2);
}
sw.WriteLine("\tConstructors:");
foreach (ASTNode n in cNode.Constructors)
{
PrintNode(n, incident + 2);
}
sw.WriteLine("\tFunctions:");
foreach (ASTNode n in cNode.Functions)
{
PrintNode(n, incident + 2);
}
}
}
