DefaultHashMapGraph.java

package kr.ac.snu.selab.soot.graph;

import java.util.Collection;
import java.util.HashMap;
import java.util.HashSet;

public class DefaultHashMapGraph<N extends Node> implements Graph<N> {
	protected HashMap<String, HashSet<N>> sourceMap;
	protected HashMap<String, HashSet<N>> targetMap;

	public DefaultHashMapGraph() {
		sourceMap = new HashMap<String, HashSet<N>>();
		targetMap = new HashMap<String, HashSet<N>>();
	}

	public HashMap<String, HashSet<N>> getSourceMap() {
		return sourceMap;
	}

	public HashMap<String, HashSet<N>> getTargetMap() {
		return targetMap;
	}

	@Override
	public Collection<N> sourceNodes(N aNode) {
		HashSet<N> result = new HashSet<N>();
		String key = aNode.key();
		if (sourceMap.containsKey(key)) {
			result = sourceMap.get(key);
		}
		return result;
	}

	@Override
	public Collection<N> targetNodes(N aNode) {
		HashSet<N> result = new HashSet<N>();
		String key = aNode.key();
		if (targetMap.containsKey(key)) {
			result = targetMap.get(key);
		}
		return result;
	}
}