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 override LatteParser.TypeContext VisitERelOp(LatteParser.ERelOpContext context)
{
var lhs = Visit(context.expr()[0]);
var rhs = Visit(context.expr()[1]);
switch (context.relOp())
{
case LatteParser.LessThanContext _:
case LatteParser.LessEqualsContext _:
case LatteParser.GreaterThanContext _:
case LatteParser.GreaterEqualsContext _:
if (!lhs.Equals(rhs) || !lhs.Equals(new LatteParser.TIntContext()))
{
Utils.StateUtils.InterruptWithMessage(context.start.Line, ErrorMessages.CompOpToNotInt);
}
break;
case LatteParser.EqualsContext _:
case LatteParser.NotEqualsContext _:
if (!lhs.Equals(rhs))
{
Utils.StateUtils.InterruptWithMessage(context.start.Line, ErrorMessages.EqOpWrongTypes);
}
break;
}
return(new LatteParser.TBoolContext());
}
public RelOp(LatteParser.ERelOpContext context) :
base(context.expr()[0], context.expr()[1])
{
Rel = context.relOp() switch
{
LatteParser.LessThanContext _ => Rel.LessThan,
LatteParser.LessEqualsContext _ => Rel.LessEquals,
LatteParser.GreaterThanContext _ => Rel.GreaterThan,
LatteParser.GreaterEqualsContext _ => Rel.GreaterEquals,
LatteParser.EqualsContext _ => Rel.Equals,
LatteParser.NotEqualsContext _ => Rel.NotEquals
};
}
