graph.java

来自「Java Op Processor java vhdl processor」· Java 代码 · 共 63 行

package com.jopdesign.util;

import java.util.*;

/**
 * Utility class for manipulation of graphs
 * TODO: find a library that does exactly this and
 * is not too big.
 * @author martin, rasmus
 *
 */
public class Graph {

	HashMap vmap = new HashMap();
	
	public Vertex addVertex(Vertex v) {
		vmap.put(v, v);
		return v;
	}
	
	public Vertex findVertex(Object data) {
		return (Vertex) vmap.get(data);
	}
	
	public void addEdge(Vertex from, Vertex to) {
		
		from.succ.add(to);
		to.pred.add(from);
		System.out.println(from.succ);
	}
	
	/**
	 * Generate dot graph
	 * 
	 * @return
	 */
	public String printGraph() {
		
		StringBuffer sb = new StringBuffer();
	    sb.append("digraph G {\n");
	    sb.append("size = \"10,7.5\"\n");
	    
	    System.out.println("******************");
	    
	    Iterator iter = vmap.values().iterator();
	    while (iter.hasNext()) {
	    	Vertex v = (Vertex) iter.next();
	    	System.out.println(v);
	    	System.out.println("\t"+v.succ);
	    	for (Iterator eit = v.getSucc().iterator(); eit.hasNext(); ) {
	    		Vertex suc = (Vertex) eit.next();
	    		sb.append("\t\""+v.toDotString()+"\" -> \""+suc.toDotString()+"\"\n");
	    	}
	    }

	    
		sb.append("}\n");
		return sb.toString();
	}
}