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

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

import de.uni.freiburg.iig.telematik.jagal.ts.abstr.AbstractState;
import de.uni.freiburg.iig.telematik.jagal.ts.abstr.AbstractTransitionRelation;
import de.uni.freiburg.iig.telematik.jagal.ts.abstr.AbstractTransitionSystem;

public abstract class AbstractTransitionSystemComponent,S extends AbstractState, T extends AbstractTransitionRelation, O extends Object> extends AbstractGraphComponent {

	private static final long serialVersionUID = -6292113630339758126L;
	
	protected static final String endStateColor = "#f2ddf8";
	protected boolean highlightEndStates = true;

	public AbstractTransitionSystemComponent(G ts) throws Exception {
		super(ts);
	}

	@Override
	protected void setupVisualGraph(G ts) throws Exception {
		super.setupVisualGraph(ts);
		if(highlightEndStates){
			for(S endState: ts.getEndStates()){
				setNodeColor(endStateColor, endState.getName());
			}
		}

		for (S sourceVertex : graph.getStartStates()) {
			setNodeColor(sourceNodeColor, sourceVertex.getName());
		}
		for (S drainVertex : graph.getEndStates()) {
			setNodeColor(drainNodeColor, drainVertex.getName());
		}
//		mxHierarchicalLayout layout = new mxHierarchicalLayout(visualGraph);
//		layout.execute(visualGraph.getDefaultParent());
	}


	@Override
	protected String getShapeForVertex(S graphVertex) {
		if(graph.isEndState(graphVertex.getName()))
			return "shape=doubleEllipse";
		return "shape=ellipse";
	}

}