de.uni.freiburg.iig.telematik.jagal.visualization.flexible.AbstractGraphComponent Maven / Gradle / Ivy

Go to download
package de.uni.freiburg.iig.telematik.jagal.visualization.flexible;

import java.awt.BorderLayout;
import java.util.HashMap;
import java.util.Map;

import javax.swing.JComponent;
import javax.swing.JPanel;

import com.mxgraph.layout.hierarchical.mxHierarchicalLayout;
import com.mxgraph.model.mxCell;
import com.mxgraph.swing.mxGraphComponent;
import com.mxgraph.util.mxConstants;
import com.mxgraph.view.mxGraph;

import de.invation.code.toval.validate.Validate;
import de.uni.freiburg.iig.telematik.jagal.graph.Edge;
import de.uni.freiburg.iig.telematik.jagal.graph.Vertex;
import de.uni.freiburg.iig.telematik.jagal.graph.abstr.AbstractGraph;

public abstract class AbstractGraphComponent,V extends Vertex,E extends Edge,U> extends JPanel {
	
	private static final long serialVersionUID = -5919436015610195204L;
	
	protected static final String sourceNodeColor = "#f6f5b1";
	protected static final String drainNodeColor = "#f2ddf8";
	
	private JComponent graphPanel = null;

	protected mxGraph visualGraph = null;
	protected G graph = null;
	protected Map vertices = new HashMap(); 
	
	protected boolean highlightSources = true;
	protected boolean highlightDrains = false;

	public AbstractGraphComponent(G graph) throws Exception {
		Validate.notNull(graph);
		this.graph = graph;
	}
	
	public void initialize() throws Exception{
		setupVisualGraph(graph);
		setLayout(new BorderLayout(20, 0));
		add(getGraphPanel(), BorderLayout.CENTER);
		setPreferredSize(getGraphPanel().getPreferredSize());
	}

	protected void setupVisualGraph(G graph) throws Exception{
		Validate.notNull(graph);
		visualGraph = new mxGraph();
		Object parent = visualGraph.getDefaultParent();

		visualGraph.getModel().beginUpdate();
		try{
			for (V graphVertex : graph.getVertices()) {
				String vertexName = graphVertex.getName();
				Object vertex = visualGraph.insertVertex(parent, vertexName, vertexName, 0, 0, 40, 40, getShapeForVertex(graphVertex));
				vertices.put(vertexName, vertex);
			}
			for (String vertexName : graph.getVertexNames()) {
				for (E outgoingEdge : graph.getOutgoingEdgesFor(vertexName)) {
					String followerVertexName = outgoingEdge.getTarget().getName();
					mxCell insertedEdge = (mxCell) visualGraph.insertEdge(parent, vertexName + "-" + followerVertexName, getEdgeLabel(outgoingEdge), vertices.get(vertexName), vertices.get(followerVertexName));
					Object[] cells = new Object[1];
					cells[0] = insertedEdge;
					visualGraph.setCellStyles(mxConstants.STYLE_LABEL_BACKGROUNDCOLOR, "#ffffff", cells);
				}
			}
		}
		finally{
			visualGraph.getModel().endUpdate();
		}
		
		if (highlightSources) {
			for (V sourceVertex : graph.getSources()) {
				setNodeColor(sourceNodeColor, sourceVertex.getName());
			}
		}
		
		if (highlightDrains) {
			for (V drainVertex : graph.getDrains()) {
				setNodeColor(drainNodeColor, drainVertex.getName());
			}
		}
		
		mxHierarchicalLayout layout = new mxHierarchicalLayout(visualGraph);
		layout.execute(visualGraph.getDefaultParent());
	}
	
	protected String getShapeForVertex(V graphVertex) {
		return "shape=ellipse";
	}

	protected String getEdgeLabel(E edge){
		return "";
	}
	
	private JComponent getGraphPanel() {
		if(graphPanel == null){
			graphPanel = new mxGraphComponent(visualGraph);
		}
		return graphPanel;
	}
	
	public void setNodeColor(String color, String... nodeNames){
		Object[] cells = new Object[nodeNames.length];
		for(int i=0; i