public override void EnterRet(LatteParser.RetContext context)
{
var func = _environment.CurrentFunction;
var exprType = new ExpressionTypeVisitor().Visit(context.expr());
if (!func.Type.Equals(exprType))
{
_errorState.AddErrorMessage(new ErrorMessage(
context.start.Line,
context.start.Column,
ErrorMessages.WrongReturn(
exprType.ToString(),
func.Type.ToString(),
func.Id)));
}
}
public static Stmt StmtFromStmtContext(LatteParser.StmtContext context)
{
return(context switch
{
LatteParser.AssContext assContext => new Ass(assContext),
LatteParser.BlockStmtContext blockStmtContext => new Block(blockStmtContext),
LatteParser.CondContext condContext => new Cond(condContext),
LatteParser.CondElseContext condElseContext => new CondElse(condElseContext),
LatteParser.DeclContext declContext => new Decl(declContext),
LatteParser.DecrContext decrContext => new Decr(decrContext),
LatteParser.EmptyContext _ => new Empty(),
LatteParser.IncrContext incrContext => new Incr(incrContext),
LatteParser.RetContext retContext => new Ret(retContext),
LatteParser.SExpContext sExpContext => new ExpStmt(sExpContext),
LatteParser.StructAssContext structAssContext => new StructAss(structAssContext),
LatteParser.StructDecrContext structDecrContext => new StructDecr(structDecrContext),
LatteParser.StructIncrContext structIncrContext => new StructIncr(structIncrContext),
LatteParser.VRetContext vRetContext => new Ret(vRetContext),
LatteParser.WhileContext whileContext => new While(whileContext),
_ => throw new ArgumentOutOfRangeException(nameof(context))
});
public Ret(LatteParser.RetContext context)
{
Expr = Exprs.Utils.ExprFromExprContext(context.expr());
}