edu.uci.ics.jung.visualization.layout.LayoutTransition Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of jung-visualization Show documentation

Core visualization support for the JUNG project

The newest version!

package edu.uci.ics.jung.visualization.layout;

import java.awt.geom.Point2D;

import edu.uci.ics.jung.algorithms.layout.Layout;
import edu.uci.ics.jung.algorithms.layout.StaticLayout;
import edu.uci.ics.jung.algorithms.layout.util.Relaxer;
import edu.uci.ics.jung.algorithms.layout.util.VisRunner;
import edu.uci.ics.jung.algorithms.util.IterativeContext;
import edu.uci.ics.jung.graph.Graph;
import edu.uci.ics.jung.visualization.VisualizationViewer;

public class LayoutTransition implements IterativeContext {
	
	protected Layout startLayout;
	protected Layout endLayout;
	protected Layout transitionLayout;
	protected boolean done = false;
	protected int count = 20;
	protected int counter = 0;
	protected VisualizationViewer vv;

	public LayoutTransition(VisualizationViewer vv, Layout startLayout, Layout endLayout) {
		this.vv = vv;
		this.startLayout = startLayout;
		this.endLayout = endLayout;
		if(endLayout instanceof IterativeContext) {
			Relaxer relaxer = new VisRunner((IterativeContext)endLayout);
			relaxer.prerelax();
		}
		this.transitionLayout =
			new StaticLayout(startLayout.getGraph(), startLayout);
		vv.setGraphLayout(transitionLayout);
	}

	public boolean done() {
		return done;
	}

	public void step() {
		Graph g = transitionLayout.getGraph();
		for(V v : g.getVertices()) {
			Point2D tp = transitionLayout.apply(v);
			Point2D fp = endLayout.apply(v);
			double dx = (fp.getX()-tp.getX())/(count-counter);
			double dy = (fp.getY()-tp.getY())/(count-counter);
			transitionLayout.setLocation(v, 
					new Point2D.Double(tp.getX()+dx,tp.getY()+dy));
		}
		counter++;
		if(counter >= count) {
			done = true;
			vv.setGraphLayout(endLayout);
		}
		vv.repaint();
	}
}