/*
* Copyright 2003-2007 the original author or authors.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.codehaus.groovy.ast.expr;
import org.codehaus.groovy.ast.ClassNode;
import org.codehaus.groovy.ast.GroovyCodeVisitor;
import org.codehaus.groovy.ast.Variable;
import org.codehaus.groovy.syntax.Token;
import org.codehaus.groovy.syntax.Types;
/**
* Represents two expressions and an operation
*
* @author <a href="mailto:james@coredevelopers.net">James Strachan</a>
* @version $Revision: 20590 $
*/
public class BinaryExpression extends Expression {
private Expression leftExpression;
private Expression rightExpression;
private final Token operation;
public BinaryExpression(Expression leftExpression,
Token operation,
Expression rightExpression) {
this.leftExpression = leftExpression;
this.operation = operation;
this.rightExpression = rightExpression;
}
public String toString() {
return super.toString() + "[" + leftExpression + operation + rightExpression + "]";
}
public void visit(GroovyCodeVisitor visitor) {
visitor.visitBinaryExpression(this);
}
public Expression transformExpression(ExpressionTransformer transformer) {
Expression ret = new BinaryExpression(transformer.transform(leftExpression), operation, transformer.transform(rightExpression));
ret.setSourcePosition(this);
return ret;
}
public Expression getLeftExpression() {
return leftExpression;
}
public void setLeftExpression(Expression leftExpression) {
this.leftExpression = leftExpression;
}
public void setRightExpression(Expression rightExpression) {
this.rightExpression = rightExpression;
}
public Token getOperation() {
return operation;
}
public Expression getRightExpression() {
return rightExpression;
}
public String getText() {
if (operation.getType() == Types.LEFT_SQUARE_BRACKET) {
return leftExpression.getText() + "[" + rightExpression.getText() + "]";
}
return "(" + leftExpression.getText() + " " + operation.getText() + " " + rightExpression.getText() + ")";
}
/**
* Creates an assignment expression in which the specified expression
* is written into the specified variable name.
*/
public static BinaryExpression newAssignmentExpression(Variable variable, Expression rhs) {
VariableExpression lhs = new VariableExpression(variable);
Token operator = Token.newPlaceholder(Types.ASSIGN);
return new BinaryExpression(lhs, operator, rhs);
}
/**
* Creates variable initialization expression in which the specified expression
* is written into the specified variable name.
*/
public static BinaryExpression newInitializationExpression(String variable, ClassNode type, Expression rhs) {
VariableExpression lhs = new VariableExpression(variable);
if (type != null) {
lhs.setType(type);
}
Token operator = Token.newPlaceholder(Types.ASSIGN);
return new BinaryExpression(lhs, operator, rhs);
}
}