public static Expr ExprFromExprContext(LatteParser.ExprContext context)
{
if (context == null)
{
return(null);
}
return(context switch
{
LatteParser.EAddOpContext eAddOpContext => new AddOp(eAddOpContext),
LatteParser.EAndContext eAndContext => new And(eAndContext),
LatteParser.EFalseContext eFalseContext => new Bool(eFalseContext),
LatteParser.EFunCallContext eFunCallContext => new FunCall(eFunCallContext),
LatteParser.EIdContext eIdContext => new ID(eIdContext),
LatteParser.EIntContext eIntContext => new Int(eIntContext),
LatteParser.EMethodCallContext eMethodCallContext => new MethodCall(eMethodCallContext),
LatteParser.EMulOpContext eMulOpContext => new MulOp(eMulOpContext),
LatteParser.ENewObjectContext eNewObjectContext => new NewObject(eNewObjectContext),
LatteParser.ENullContext _ => new Null(),
LatteParser.ENullCastContext _ => new Null(),
LatteParser.EObjectFieldContext eObjectFieldContext => new ObjectField(eObjectFieldContext),
LatteParser.EOrContext eOrContext => new Or(eOrContext),
LatteParser.EParenContext eParenContext => ExprFromExprContext(eParenContext.expr()),
LatteParser.ERelOpContext eRelOpContext => new RelOp(eRelOpContext),
LatteParser.EStrContext eStrContext => new Str(eStrContext),
LatteParser.ETrueContext eTrueContext => new Bool(eTrueContext),
LatteParser.EUnOpContext eUnOpContext => new UnOp(eUnOpContext),
_ => throw new ArgumentOutOfRangeException(nameof(context))
});
public UnOp(LatteParser.EUnOpContext context)
{
Operator = context.unOp() switch
{
LatteParser.UnaryMinusContext _ => Unary.Minus,
LatteParser.UnaryNegContext _ => Unary.Neg
};
Expr = Utils.ExprFromExprContext(context.expr());
}
public override LatteParser.TypeContext VisitEUnOp(LatteParser.EUnOpContext context)
{
var expr = Visit(context.expr());
switch (context.unOp())
{
case LatteParser.UnaryMinusContext _:
if (!expr.Equals(new LatteParser.TIntContext()))
{
Utils.StateUtils.InterruptWithMessage(context.start.Line, ErrorMessages.UnaryMinusToNotInt);
}
return(new LatteParser.TIntContext());
case LatteParser.UnaryNegContext _:
if (!expr.Equals(new LatteParser.TBoolContext()))
{
Utils.StateUtils.InterruptWithMessage(context.start.Line, ErrorMessages.UnaryNegToNotBool);
}
return(new LatteParser.TBoolContext());
default:
throw new NotSupportedException();
}
}
