Graph Script Code

package ls.graph.script;

import java.util.HashMap;
import java.util.Map;

import edu.uci.ics.jung.graph.ArchetypeVertex;
import edu.uci.ics.jung.graph.decorators.VertexStringer;

public class ScriptableVertexStringer implements VertexStringer
{

	private Map<ArchetypeVertex,String> vertexLabelMap = null;

	
	public ScriptableVertexStringer()
	{
		super();
		this.vertexLabelMap = new HashMap<ArchetypeVertex,String>();
	}
	
	public String getLabel(ArchetypeVertex v)
	{
		return v == null ? 
				"NULL" : 
				(this.vertexLabelMap.containsKey(v) ? this.vertexLabelMap.get(v) : "");
	}

	void setVertexLabel(ArchetypeVertex v,String label)
	{
		if (v != null)
			this.vertexLabelMap.put(v, label);
	}
}