IfStatement.java

package com.form.language.ast.statement;

import java.util.List;

import javax.swing.JPanel;

import com.form.language.ast.expression.Expression;
import com.form.language.ast.type.Type;
import com.form.language.gui.components.FormComponent;
import com.form.language.issue.Error;
import com.form.language.issue.QLToken;
import com.form.language.memory.Context;

public class IfStatement extends Statement {
	private Expression conditions;
	private StatementCollection thenStatements;

	public IfStatement(Expression conditions, List<Statement> thenStatements,
			QLToken tokenInfo) {
		super(tokenInfo);
		this.conditions = conditions;
		this.thenStatements = new StatementCollection(thenStatements);
	}

	@Override
	public boolean checkType(Context context) {
		thenStatements.checkTypes(context);
		return checkConditionType(context);
	}

	private boolean checkConditionType(Context context) {
		Type conditionType = conditions.getType(context);
		if (conditionType.isBoolType()) {
			return true;
		} else {
			context.addError(new Error(tokenInfo,
					"The conditions in an if statement should evaluate to a Boolean"));
			return false;
		}
	}

	@Override
	public void initMemory(Context mem) {
		/*
		 * Nothing to be initialized here
		 */
	}

	@Override
	public void createGUIComponent(FormComponent formComponent, JPanel panel,
			Context rm) {
		formComponent.setIfCondition(conditions);
		for (Statement s : thenStatements) {
			s.createGUIComponent(formComponent, panel, rm);
		}
	}

}