JavaTXCompilerInJavaTX/test/unify/FiniteClosureBuilder.java

package unify;

import java.util.HashSet;
import java.util.Set;

import de.dhbwstuttgart.typeinference.unify.interfaces.IFiniteClosure;
import de.dhbwstuttgart.typeinference.unify.model.FiniteClosure;
import de.dhbwstuttgart.typeinference.unify.model.MPair;
import de.dhbwstuttgart.typeinference.unify.model.UnifyType;
import de.dhbwstuttgart.typeinference.unify.model.MPair.PairOperator;

public class FiniteClosureBuilder {
	
	private Set<MPair> pairs = new HashSet<>();
	
	public void add(UnifyType sub, UnifyType sup) {
		pairs.add(new MPair(sub, sup, PairOperator.SMALLER));
	}
	
	public IFiniteClosure getFiniteClosure() {
		return new FiniteClosure(pairs);
	}
	
	public void clear() {
		pairs = new HashSet<>();
	}
	
	public IFiniteClosure getCollectionExample() {
		TypeFactory tf = new TypeFactory();
		
		UnifyType collection = tf.getSimpleType("Collection");
		UnifyType set = tf.getSimpleType("Set", "T");
		UnifyType sortedSet = tf.getSimpleType("Set", "T");
		UnifyType TreeSet = tf.getSimpleType("TreeSet", "T");
		UnifyType hashSet = tf.getSimpleType("HashSet", "T");
		UnifyType linkedHashSet = tf.getSimpleType("LinkedHashSet", "T");
		UnifyType queue = tf.getSimpleType("Queue", "T");
		UnifyType deque = tf.getSimpleType("Deque", "T");
		UnifyType linkedList = tf.getSimpleType("LinkedList", "T");
		UnifyType list = tf.getSimpleType("List", "T");
		UnifyType vector = tf.getSimpleType("Vector", "T");
		UnifyType stack = tf.getSimpleType("Stack", "T");
		UnifyType arrayList = tf.getSimpleType("ArrayList", "T");
		
		add(set, collection);
		add(sortedSet, set);
		add(TreeSet, sortedSet);
		add(hashSet, set);
		add(linkedHashSet, set);
		
		add(queue, collection);
		add(deque, queue);
		add(linkedList, deque);
		add(list, collection);
		add(linkedList, list);
		add(vector, list);
		add(arrayList, list);
		add(stack, vector);
		
		IFiniteClosure fc = getFiniteClosure();
		clear();
		return fc;
	}
}