public override void EnterExpression(JavaParser.ExpressionContext context)
{
if (context.Parent is JavaParser.ForControlContext)
{
if (IsInCurrentContext(context))
{
_associatedTrees.Add(context);
}
}
}
public Ust VisitExpression(JavaParser.ExpressionContext context)
{
var textSpan = context.GetTextSpan();
var child0Terminal = context.GetChild(0) as ITerminalNode;
Expression target;
Expression result;
if (child0Terminal != null)
{
switch (child0Terminal.Symbol.Type)
{
case JavaParser.NEW:
result = (Expression)Visit(context.creator());
return(result);
case JavaParser.LPAREN: // '(' type ')' expression
var type = (TypeToken)Visit(context.typeType());
target = (Expression)Visit(context.expression(0));
result = new CastExpression(type, target, textSpan);
return(result);
default: // unary operator ('+', '-', '++', '--', '~', '!')
UnaryOperator op = UnaryOperatorLiteral.PrefixTextUnaryOperator[child0Terminal.GetText()];
var opLiteral = new UnaryOperatorLiteral(op, child0Terminal.GetTextSpan());
target = (Expression)Visit(context.expression(0));
result = new UnaryOperatorExpression(opLiteral, target, textSpan);
return(result);
}
}
ArgsUst args;
var child1Terminal = context.GetChild(1) as ITerminalNode;
if (child1Terminal != null)
{
switch (child1Terminal.Symbol.Type)
{
case JavaParser.DOT: // '.'
target = (Expression)Visit(context.expression(0));
if (context.methodCall() != null)
{
var invocation = (InvocationExpression)Visit(context.methodCall());
return(new InvocationExpression(
new MemberReferenceExpression(target, invocation.Target, target.TextSpan.Union(invocation.Target.TextSpan)),
invocation.Arguments,
textSpan));
}
// TODO: implement base processing
Expression rightPart = null;
if (context.IDENTIFIER() != null)
{
rightPart = (IdToken)Visit(context.IDENTIFIER());
}
else if (context.THIS() != null)
{
rightPart = new ThisReferenceToken(context.THIS().GetTextSpan());
}
if (rightPart != null)
{
return(new MemberReferenceExpression(target, rightPart, textSpan));
}
if (context.innerCreator() != null)
{
return(Visit(context.innerCreator()));
}
if (context.explicitGenericInvocation() != null)
{
return(VisitChildren(context.explicitGenericInvocation()).ToExpressionIfRequired());
}
break;
case JavaParser.LBRACK: // '['
target = (Expression)Visit(context.expression(0));
Expression expr = (Expression)Visit(context.expression(1));
args = new ArgsUst(new Expression[] { expr }, expr.TextSpan);
result = new IndexerExpression(target, args, textSpan);
return(result);
case JavaParser.INSTANCEOF: // x instanceof y -> (y)x != null
var expression = (Expression)Visit(context.expression(0));
var type = (TypeToken)Visit(context.typeType());
result = new BinaryOperatorExpression(context.GetTextSpan())
{
Left = new CastExpression(type, expression, context.GetTextSpan()),
Operator = new BinaryOperatorLiteral(BinaryOperator.NotEqual, default),
Right = new NullLiteral(default),
Root = root
};
return(result);
case JavaParser.QUESTION: // '?'
var condition = (Expression)Visit(context.expression(0));
var trueExpr = (Expression)Visit(context.expression(1));
var falseExpr = (Expression)Visit(context.expression(2));
result = new ConditionalExpression(condition, trueExpr, falseExpr, textSpan);
return(result);
case JavaParser.COLONCOLON:
return(VisitChildren(context));
default: // binary operator
var left = (Expression)Visit(context.expression(0));
JavaParser.ExpressionContext expr1 = context.expression(1);
if (expr1 != null)
{
string opText = string.Concat(context.children.Skip(1).Take(context.children.Count - 2));
var right = (Expression)Visit(expr1);
// TODO: fix via grammar refactoring (alternative labels).
if (opText == "=" ||
(opText.EndsWith("=") && BinaryOperatorLiteral.TextBinaryOperator.ContainsKey(opText.Remove(opText.Length - 1))))
{
result = UstUtils.CreateAssignExpr(left, right, context.GetTextSpan(),
opText, child1Terminal.GetTextSpan());
}
else
{
BinaryOperator op = BinaryOperatorLiteral.TextBinaryOperator[opText];
var opLiteral = new BinaryOperatorLiteral(op, child1Terminal.GetTextSpan());
result = new BinaryOperatorExpression(left, opLiteral, right, textSpan);
}
}
else
{
// post increment or decrement.
UnaryOperator op = UnaryOperatorLiteral.PostfixTextUnaryOperator[context.postfix.Text];
var opLiteral = new UnaryOperatorLiteral(op, child1Terminal.GetTextSpan());
result = new UnaryOperatorExpression(opLiteral, left, textSpan);
return(result);
}
return(result);
}
}
return(VisitChildren(context));
}
public UstNode VisitExpression(JavaParser.ExpressionContext context)
{
var textSpan = context.GetTextSpan();
var child0Terminal = context.GetChild(0) as ITerminalNode;
Expression target;
Expression result;
if (child0Terminal != null)
{
switch (child0Terminal.Symbol.Type)
{
case JavaParser.NEW:
result = (Expression)Visit(context.creator());
return(result);
case JavaParser.LPAREN: // '(' type ')' expression
var type = (TypeToken)Visit(context.typeType());
target = (Expression)Visit(context.expression(0));
result = new CastExpression(type, target, textSpan, FileNode);
return(result);
default: // unary operator ('+', '-', '++', '--', '~', '!')
UnaryOperator op = UnaryOperatorLiteral.PrefixTextUnaryOperator[child0Terminal.GetText()];
var opLiteral = new UnaryOperatorLiteral(op, child0Terminal.GetTextSpan(), FileNode);
target = (Expression)Visit(context.expression(0));
result = new UnaryOperatorExpression(opLiteral, target, textSpan, FileNode);
return(result);
}
}
ArgsNode args;
var child1Terminal = context.GetChild(1) as ITerminalNode;
if (child1Terminal != null)
{
switch (child1Terminal.Symbol.Type)
{
case JavaParser.DOT: // '.'
target = (Expression)Visit(context.expression(0));
var id = context.IDENTIFIER();
if (id != null)
{
result = new MemberReferenceExpression(target, (IdToken)Visit(id), textSpan, FileNode);
return(result);
}
var explicitGenericInvocation = context.explicitGenericInvocation();
if (explicitGenericInvocation != null)
{
return(VisitChildren(context).ToExpressionIfRequired());
}
var child2Terminal = context.GetChild <ITerminalNode>(1);
// TODO: implement
switch (child2Terminal.Symbol.Type)
{
case JavaParser.THIS:
break;
case JavaParser.NEW:
break;
case JavaParser.SUPER:
break;
}
break;
case JavaParser.LBRACK: // '['
target = (Expression)Visit(context.expression(0));
Expression expr = (Expression)Visit(context.expression(1));
args = new ArgsNode(new Expression[] { expr }, expr.TextSpan, FileNode);
result = new IndexerExpression(target, args, textSpan, FileNode);
return(result);
case JavaParser.LPAREN: // '('
target = (Expression)Visit(context.expression(0));
// TODO: fix with ArgsNode
JavaParser.ExpressionListContext expressionList = context.expressionList();
if (expressionList != null)
{
args = (ArgsNode)Visit(expressionList);
}
else
{
args = new ArgsNode();
}
result = new InvocationExpression(target, args, textSpan, FileNode);
return(result);
case JavaParser.INSTANCEOF: // x instanceof y -> (y)x != null
var expression = (Expression)Visit(context.expression(0));
var type = (TypeToken)Visit(context.typeType());
result = new BinaryOperatorExpression
{
Left = new CastExpression(type, expression, context.GetTextSpan(), FileNode),
Operator = new BinaryOperatorLiteral(BinaryOperator.NotEqual, default(TextSpan), FileNode),
Right = new NullLiteral(default(TextSpan), FileNode),
TextSpan = context.GetTextSpan(),
FileNode = FileNode
};
return(result);
case JavaParser.QUESTION: // '?'
var condition = (Expression)Visit(context.expression(0));
var trueExpr = (Expression)Visit(context.expression(1));
var falseExpr = (Expression)Visit(context.expression(2));
result = new ConditionalExpression(condition, trueExpr, falseExpr, textSpan, FileNode);
return(result);
default: // binary operator
string text = child1Terminal.GetText();
var left = (Expression)Visit(context.expression(0));
JavaParser.ExpressionContext expr1 = context.expression(1);
if (expr1 != null)
{
var right = (Expression)Visit(expr1);
if (text == "=")
{
result = new AssignmentExpression(left, right, textSpan, FileNode);
}
else if (BinaryOperatorLiteral.TextBinaryAssignmentOperator.Contains(text))
{
BinaryOperator op;
if (text == ">>>=")
{
op = BinaryOperator.ShiftRight; // TODO: fix shift operator.
}
else
{
op = BinaryOperatorLiteral.TextBinaryOperator[text.Remove(text.Length - 1)];
}
result = ConverterHelper.ConvertToAssignmentExpression(left, op, child1Terminal.GetTextSpan(), right,
context.GetTextSpan(), FileNode);
}
else
{
BinaryOperator op;
if (text == ">>>")
{
op = BinaryOperator.ShiftRight; // TODO: fix shift operator.
}
else
{
op = BinaryOperatorLiteral.TextBinaryOperator[text];
}
var opLiteral = new BinaryOperatorLiteral(op, child1Terminal.GetTextSpan(), FileNode);
result = new BinaryOperatorExpression(left, opLiteral, right, textSpan, FileNode);
}
}
else
{
// post increment or decrement.
UnaryOperator op = UnaryOperatorLiteral.PostfixTextUnaryOperator[text];
var opLiteral = new UnaryOperatorLiteral(op, child1Terminal.GetTextSpan(), FileNode);
result = new UnaryOperatorExpression(opLiteral, left, textSpan, FileNode);
return(result);
}
return(result);
}
}
return(Visit(context.GetChild(0)));
}
