public CompileResult Verify(AST ast)
{
VarScope vScope = new VarScope();
foreach (ASTNode node in ast.Root)
{
}
return(new CompileResult(true));
}
public CompileResult Vars(AST ast)
{
VarScope vScope = new VarScope();
foreach (ASTNode node in ast.Root)
{
// check node type etc... for declarations and other scopes...
CompileResult result = VarsNode(node, vScope);
if (!result)
{
return(result);
}
}
return(new CompileResult(true));
}
private CompileResult VarsNode(ASTNode node, VarScope scope)
{
switch (node)
{
case FuncDeclNode funcDeclNode:
VarScope funcScope = new VarScope();
foreach (ParamsNode.ParameterNode param in funcDeclNode.Params.Parameters)
{
param.Identifier.Index = funcScope.Enter(param.Identifier.Content);
}
if (funcDeclNode.Body is not null)
{
this.VarsNode(funcDeclNode.Body, funcScope);
funcDeclNode.Body.VarIndices = funcDeclNode.Body.VarIndices.Union(this.Exit(new VarScope(), funcScope)).ToArray();
}
scope.Enter(funcDeclNode.Name);
break;
case ClassDeclNode classDeclNode:
VarScope classScope = new VarScope();
foreach (FuncDeclNode methodDecl in classDeclNode.Methods)
{
var varsfunc = this.VarsNode(methodDecl, classScope);
if (!varsfunc)
{
return(varsfunc);
}
}
break;
case ScopeNode scopeNode:
VarScope subScope = new VarScope(scope);
foreach (ASTNode subNode in scopeNode.Nodes)
{
this.VarsNode(subNode, subScope);
}
scopeNode.VarIndices = this.Exit(scope, subScope);
scope.top = subScope.top;
break;
case IGroupedASTNode groupExprNode:
foreach (ASTNode subNode in groupExprNode.Nodes)
{
this.VarsNode(subNode, scope);
}
break;
case VarDeclNode vDeclNode:
this.VarsNode(vDeclNode.AssignToExpr, scope);
vDeclNode.EnterIndex = scope.Enter(vDeclNode.VarName);
break;
case BinOpNode binopNode:
this.VarsNode(binopNode.Left, scope);
this.VarsNode(binopNode.Right, scope);
break;
case UnaryOpNode unaryOpNode:
this.VarsNode(unaryOpNode.Expr, scope);
break;
case IdentifierNode iDNode:
if (scope.Lookup(iDNode.Content, out ushort idIndex))
{
iDNode.Index = idIndex;
}
else
{
return(new CompileResult(false, $"Variable '{iDNode.Content}' does not exist").SetOrigin(iDNode.Pos));
}
break;
case CallNode callNode:
this.VarsNode(callNode.IdentifierNode, scope);
foreach (ASTNode arg in callNode.Arguments.Arguments)
{
this.VarsNode(arg, scope);
}
if (callNode.IdentifierNode is IdentifierNode callIdNode)
{
callIdNode.IsFuncIdentifier = true;
} // else ...
break;
case MemberAccessNode memberAccessNode:
this.VarsNode(memberAccessNode.Left as ASTNode, scope);
break;
case NewObjectNode newObjNode:
foreach (ASTNode arg in newObjNode.CtorArguments.Arguments)
{
this.VarsNode(arg, scope);
}
break;
case NewArrayNode newArrNode:
foreach (ASTNode arg in newArrNode.Indexer.Nodes)
{
this.VarsNode(arg, scope);
}
break;
case ReturnStatement returnStatementNode:
this.VarsNode(returnStatementNode.Expression as ASTNode, scope);
break;
case ValueListInitializerNode valListInitNode:
foreach (IExpr valNode in valListInitNode)
{
this.VarsNode(valNode as ASTNode, scope);
}
break;
case LookupNode lookupNode:
this.VarsNode(lookupNode.Index, scope);
this.VarsNode(lookupNode.Left as ASTNode, scope);
break;
case IfStatement ifStatement:
this.VarsNode(ifStatement.Condition as ASTNode, scope);
this.VarsNode(ifStatement.Body as ASTNode, scope);
if (ifStatement.HasTrailingBranch)
{
this.VarsNode(ifStatement.Trail as ASTNode, scope);
}
break;
case ElseStatement elseStatement:
this.VarsNode(elseStatement.Body as ASTNode, scope);
break;
case ForStatement forStatement:
VarScope forScope = new VarScope(scope);
this.VarsNode(forStatement.Init, forScope);
this.VarsNode(forStatement.Condition as ASTNode, forScope);
this.VarsNode(forStatement.After as ASTNode, forScope);
this.VarsNode(forStatement.Body as ASTNode, forScope);
forStatement.VarIndices = this.Exit(scope, forScope);
scope.top = forScope.top;
break;
case LoopNode loopStatement:
this.VarsNode(loopStatement.Condition as ASTNode, scope);
this.VarsNode(loopStatement.Body as ASTNode, scope); // If Scope node it's handled properly
break;
case NamespaceDirectiveNode namespaceDirective:
this.VarsNode(namespaceDirective.Body, scope);
break;
case ThisNode:
case BaseNode:
case IntLitNode:
case BoolLitNode:
case IDirective: // Ignore directives
break;
default:
Log.WriteLine($"Vars not implemented for type: {node.GetType().Name}");
break;
}
return(new CompileResult(true));
}
private ushort[] Exit(VarScope current, VarScope sub)
{
(string key, PrimitiveStack <ushort> stack)[] subAsArray = sub.stack.ToArray();
public VarScope(VarScope copy)
{
this.stack = new LookupStack <string, ushort>(copy.stack);
this.top = copy.top;
}