public override AstNode VisitFunction(ALangParser.FunctionContext context)
{
var functionNode = new FunctionNode();
functionNode.Identifier = context.ID().GetText();
if (context.@params() != null)
{
functionNode.Params = context.@params().Accept(this);
}
functionNode.Type = Type.GetType(context);
functionNode.AdoptChildren(context.stmts().Accept(this));
return(functionNode);
}
public static LanguageType GetType(ALangParser.FunctionContext context)
{
var type = context.TYPE().GetText();
switch (type)
{
case "void":
return(LanguageType.Void);
case "int":
return(LanguageType.Int);
case "pin":
return(LanguageType.Pin);
case "time":
return(LanguageType.Time);
default:
return(LanguageType.Null);
}
}
/// <summary>
/// Enter a parse tree produced by <see cref="ALangParser.function"/>.
/// <para>The default implementation does nothing.</para>
/// </summary>
/// <param name="context">The parse tree.</param>
public virtual void EnterFunction([NotNull] ALangParser.FunctionContext context)
{
}
/// <summary>
/// Visit a parse tree produced by <see cref="ALangParser.function"/>.
/// <para>
/// The default implementation returns the result of calling <see cref="AbstractParseTreeVisitor{Result}.VisitChildren(IRuleNode)"/>
/// on <paramref name="context"/>.
/// </para>
/// </summary>
/// <param name="context">The parse tree.</param>
/// <return>The visitor result.</return>
public virtual Result VisitFunction([NotNull] ALangParser.FunctionContext context)
{
return(VisitChildren(context));
}
