net.sf.opendse.realtime.et.TimingGraphViewer Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of opendse-realtime Show documentation
The real-time module of OpenDSE
The newest version!
/*******************************************************************************
 * Copyright (c) 2015 OpenDSE
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in all
 * copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 *******************************************************************************/
package net.sf.opendse.realtime.et;

import java.awt.Dimension;
import java.awt.Paint;

import javax.swing.JFrame;

import net.sf.opendse.realtime.et.graph.TimingDependency;
import net.sf.opendse.realtime.et.graph.TimingDependencyTrigger;
import net.sf.opendse.realtime.et.graph.TimingElement;
import net.sf.opendse.realtime.et.graph.TimingGraph;
import net.sf.opendse.visualization.Graphics;
import net.sf.opendse.visualization.algorithm.DistanceFlowLayout;

import org.apache.commons.collections15.Transformer;

import edu.uci.ics.jung.algorithms.layout.Layout;
import edu.uci.ics.jung.visualization.DefaultVisualizationModel;
import edu.uci.ics.jung.visualization.GraphZoomScrollPane;
import edu.uci.ics.jung.visualization.RenderContext;
import edu.uci.ics.jung.visualization.VisualizationModel;
import edu.uci.ics.jung.visualization.VisualizationViewer;
import edu.uci.ics.jung.visualization.control.DefaultModalGraphMouse;

public class TimingGraphViewer {

	public static void view(TimingGraph tg) {
		
		/*TimingGraph tgFilter = FilterUtils.createInducedSubgraph(tg.getVertices(), tg);
		for(TimingDependency td: tg.getEdges()){
			if(td instanceof TimingDependencyPriority){
				tgFilter.removeEdge(td);
			}
		}*/
		
		Layout layout = new DistanceFlowLayout(tg);

		VisualizationModel vm = new DefaultVisualizationModel(layout,
				new Dimension(800, 600));
		VisualizationViewer vv = new VisualizationViewer(vm);

		DefaultModalGraphMouse graphMouse = new DefaultModalGraphMouse();
		vv.setGraphMouse(graphMouse);

		RenderContext ctx = vv.getRenderContext();

		ctx.setVertexLabelTransformer(new Transformer() {
			public String transform(TimingElement timingEntity) {
				return timingEntity.toString();
			}
		});

		ctx.setVertexFillPaintTransformer(new Transformer() {
			@Override
			public Paint transform(TimingElement entity) {
				//if (entity.isCommunication()) {
					return Graphics.ALICEBLUE;
				//} else {
				//	return Graphics.ROSYBROWN;
				//}
			}
		});
		
		ctx.setEdgeDrawPaintTransformer(new Transformer() {
			@Override
			public Paint transform(TimingDependency td) {
				if(td instanceof TimingDependencyTrigger){
					return Graphics.BLACK;
				} else {
					return Graphics.BLUE;
				}
			}
			
		});

		final GraphZoomScrollPane panel = new GraphZoomScrollPane(vv);
		JFrame frame = new JFrame("TimingEntity Graph");
		frame.add(panel);

		frame.pack();
		frame.setVisible(true);

	}

	protected static class VertexSizeTransformer implements Transformer {
		@Override
		public Integer transform(TimingElement vertex) {
			return 10;
		}
	}

	protected static class VertexRatioTransformer implements Transformer {
		@Override
		public Float transform(TimingElement vertex) {
			return 1.0f;
		}
	}

}