BinaryExpression.java

package nl.uva.sc.encoders.ql.ast.expression;

import java.util.Set;

import nl.uva.sc.encoders.ql.ast.TextLocation;
import nl.uva.sc.encoders.ql.ast.operator.BinaryOperator;
import nl.uva.sc.encoders.ql.ast.type.DataType;
import nl.uva.sc.encoders.ql.ast.type.TypeMap;
import nl.uva.sc.encoders.ql.visitor.ExpressionVisitor;

public class BinaryExpression extends Expression {

	private BinaryOperator operator;

	private Expression leftHand;

	private Expression rightHand;

	public BinaryExpression(TextLocation textLocation, Expression leftHand, Expression rightHand, BinaryOperator operator) {
		super(textLocation);
		this.leftHand = leftHand;
		this.rightHand = rightHand;
		this.operator = operator;
	}

	public Expression getLeftHand() {
		return leftHand;
	}

	public Expression getRightHand() {
		return rightHand;
	}

	public BinaryOperator getOperator() {
		return operator;
	}

	@Override
	public void collectQuestionNames(Set<String> relatedQuestionNames) {
		leftHand.collectQuestionNames(relatedQuestionNames);
		rightHand.collectQuestionNames(relatedQuestionNames);
	}

	@Override
	public String toString() {
		StringBuilder builder = new StringBuilder();
		builder.append(leftHand);
		builder.append(" ");
		builder.append(operator);
		builder.append(" ");
		builder.append(rightHand);
		return builder.toString();
	}

	@Override
	public <T> T accept(ExpressionVisitor<T> visitor) {
		return visitor.visit(this);
	}

	@Override
	public DataType getType(TypeMap typeMap) {
		DataType leftHandType = leftHand.getType(typeMap);
		DataType rightHandType = rightHand.getType(typeMap);
		return operator.getType(leftHandType, rightHandType);
	}
}