edu.uci.ics.jung.samples.MinimumSpanningTreeDemo Maven / Gradle / Ivy
Show all versions of jung-samples Show documentation
/*
* Copyright (c) 2003, The JUNG Authors
* All rights reserved.
*
* This software is open-source under the BSD license; see either "license.txt"
* or https://github.com/jrtom/jung/blob/master/LICENSE for a description.
*
*/
package edu.uci.ics.jung.samples;
import java.awt.BorderLayout;
import java.awt.Color;
import java.awt.Container;
import java.awt.Dimension;
import java.awt.Font;
import java.awt.GridLayout;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import javax.swing.BorderFactory;
import javax.swing.JApplet;
import javax.swing.JButton;
import javax.swing.JFrame;
import javax.swing.JLabel;
import javax.swing.JPanel;
import com.google.common.base.Functions;
import edu.uci.ics.jung.algorithms.layout.KKLayout;
import edu.uci.ics.jung.algorithms.layout.Layout;
import edu.uci.ics.jung.algorithms.layout.StaticLayout;
import edu.uci.ics.jung.algorithms.layout.TreeLayout;
import edu.uci.ics.jung.algorithms.shortestpath.MinimumSpanningForest2;
import edu.uci.ics.jung.graph.DelegateForest;
import edu.uci.ics.jung.graph.DelegateTree;
import edu.uci.ics.jung.graph.Forest;
import edu.uci.ics.jung.graph.Graph;
import edu.uci.ics.jung.graph.util.TestGraphs;
import edu.uci.ics.jung.visualization.DefaultVisualizationModel;
import edu.uci.ics.jung.visualization.GraphZoomScrollPane;
import edu.uci.ics.jung.visualization.VisualizationModel;
import edu.uci.ics.jung.visualization.VisualizationViewer;
import edu.uci.ics.jung.visualization.control.CrossoverScalingControl;
import edu.uci.ics.jung.visualization.control.DefaultModalGraphMouse;
import edu.uci.ics.jung.visualization.control.ScalingControl;
import edu.uci.ics.jung.visualization.decorators.EdgeShape;
import edu.uci.ics.jung.visualization.decorators.PickableEdgePaintTransformer;
import edu.uci.ics.jung.visualization.decorators.PickableVertexPaintTransformer;
import edu.uci.ics.jung.visualization.decorators.ToStringLabeller;
import edu.uci.ics.jung.visualization.picking.MultiPickedState;
import edu.uci.ics.jung.visualization.picking.PickedState;
import edu.uci.ics.jung.visualization.renderers.Renderer;
import edu.uci.ics.jung.visualization.transform.MutableTransformer;
/**
* Demonstrates a single graph with 3 layouts in 3 views.
* The first view is an undirected graph using KKLayout
* The second view show a TreeLayout view of a MinimumSpanningTree
* of the first graph. The third view shows the complete graph
* of the first view, using the layout positions of the
* MinimumSpanningTree tree view.
*
* @author Tom Nelson
*
*/
@SuppressWarnings("serial")
public class MinimumSpanningTreeDemo extends JApplet {
/**
* the graph
*/
Graph graph;
Forest tree;
/**
* the visual components and renderers for the graph
*/
VisualizationViewer vv0;
VisualizationViewer vv1;
VisualizationViewer vv2;
/**
* the normal Function
*/
MutableTransformer layoutTransformer;
Dimension preferredSize = new Dimension(300,300);
Dimension preferredLayoutSize = new Dimension(400,400);
Dimension preferredSizeRect = new Dimension(500,250);
/**
* create an instance of a simple graph in two views with controls to
* demo the zoom features.
*
*/
public MinimumSpanningTreeDemo() {
// create a simple graph for the demo
// both models will share one graph
graph =
TestGraphs.getDemoGraph();
MinimumSpanningForest2 prim =
new MinimumSpanningForest2(graph,
new DelegateForest(), DelegateTree.getFactory(),
Functions.constant(1.0));
tree = prim.getForest();
// create two layouts for the one graph, one layout for each model
Layout layout0 = new KKLayout(graph);
layout0.setSize(preferredLayoutSize);
Layout layout1 = new TreeLayout(tree);
Layout layout2 = new StaticLayout(graph, layout1);
// create the two models, each with a different layout
VisualizationModel vm0 =
new DefaultVisualizationModel(layout0, preferredSize);
VisualizationModel vm1 =
new DefaultVisualizationModel(layout1, preferredSizeRect);
VisualizationModel vm2 =
new DefaultVisualizationModel(layout2, preferredSizeRect);
// create the two views, one for each model
// they share the same renderer
vv0 = new VisualizationViewer(vm0, preferredSize);
vv1 = new VisualizationViewer(vm1, preferredSizeRect);
vv2 = new VisualizationViewer(vm2, preferredSizeRect);
vv1.getRenderContext().setMultiLayerTransformer(vv0.getRenderContext().getMultiLayerTransformer());
vv2.getRenderContext().setMultiLayerTransformer(vv0.getRenderContext().getMultiLayerTransformer());
vv1.getRenderContext().setEdgeShapeTransformer(EdgeShape.line(graph));
vv0.addChangeListener(vv1);
vv1.addChangeListener(vv2);
vv0.getRenderContext().setVertexLabelTransformer(new ToStringLabeller());
vv2.getRenderContext().setVertexLabelTransformer(new ToStringLabeller());
Color back = Color.decode("0xffffbb");
vv0.setBackground(back);
vv1.setBackground(back);
vv2.setBackground(back);
vv0.getRenderer().getVertexLabelRenderer().setPosition(Renderer.VertexLabel.Position.CNTR);
vv0.setForeground(Color.darkGray);
vv1.getRenderer().getVertexLabelRenderer().setPosition(Renderer.VertexLabel.Position.CNTR);
vv1.setForeground(Color.darkGray);
vv2.getRenderer().getVertexLabelRenderer().setPosition(Renderer.VertexLabel.Position.CNTR);
vv2.setForeground(Color.darkGray);
// share one PickedState between the two views
PickedState ps = new MultiPickedState();
vv0.setPickedVertexState(ps);
vv1.setPickedVertexState(ps);
vv2.setPickedVertexState(ps);
PickedState pes = new MultiPickedState();
vv0.setPickedEdgeState(pes);
vv1.setPickedEdgeState(pes);
vv2.setPickedEdgeState(pes);
// set an edge paint function that will show picking for edges
vv0.getRenderContext().setEdgeDrawPaintTransformer(new PickableEdgePaintTransformer(vv0.getPickedEdgeState(), Color.black, Color.red));
vv0.getRenderContext().setVertexFillPaintTransformer(new PickableVertexPaintTransformer(vv0.getPickedVertexState(),
Color.red, Color.yellow));
vv1.getRenderContext().setEdgeDrawPaintTransformer(new PickableEdgePaintTransformer(vv1.getPickedEdgeState(), Color.black, Color.red));
vv1.getRenderContext().setVertexFillPaintTransformer(new PickableVertexPaintTransformer(vv1.getPickedVertexState(),
Color.red, Color.yellow));
// add default listeners for ToolTips
vv0.setVertexToolTipTransformer(new ToStringLabeller());
vv1.setVertexToolTipTransformer(new ToStringLabeller());
vv2.setVertexToolTipTransformer(new ToStringLabeller());
vv0.setLayout(new BorderLayout());
vv1.setLayout(new BorderLayout());
vv2.setLayout(new BorderLayout());
Font font = vv0.getFont().deriveFont(Font.BOLD, 16);
JLabel vv0Label = new JLabel("Original Graphusing KKLayout");
vv0Label.setFont(font);
JLabel vv1Label = new JLabel("Minimum Spanning Trees");
vv1Label.setFont(font);
JLabel vv2Label = new JLabel("Original Graph using TreeLayout");
vv2Label.setFont(font);
JPanel flow0 = new JPanel();
flow0.setOpaque(false);
JPanel flow1 = new JPanel();
flow1.setOpaque(false);
JPanel flow2 = new JPanel();
flow2.setOpaque(false);
flow0.add(vv0Label);
flow1.add(vv1Label);
flow2.add(vv2Label);
vv0.add(flow0, BorderLayout.NORTH);
vv1.add(flow1, BorderLayout.NORTH);
vv2.add(flow2, BorderLayout.NORTH);
Container content = getContentPane();
JPanel grid = new JPanel(new GridLayout(0,1));
JPanel panel = new JPanel(new BorderLayout());
panel.add(new GraphZoomScrollPane(vv0), BorderLayout.WEST);
grid.add(new GraphZoomScrollPane(vv1));
grid.add(new GraphZoomScrollPane(vv2));
panel.add(grid);
content.add(panel);
// create a GraphMouse for each view
DefaultModalGraphMouse gm0 = new DefaultModalGraphMouse();
DefaultModalGraphMouse gm1 = new DefaultModalGraphMouse();
DefaultModalGraphMouse gm2 = new DefaultModalGraphMouse();
vv0.setGraphMouse(gm0);
vv1.setGraphMouse(gm1);
vv2.setGraphMouse(gm2);
// create zoom buttons for scaling the Function that is
// shared between the two models.
final ScalingControl scaler = new CrossoverScalingControl();
vv0.scaleToLayout(scaler);
vv1.scaleToLayout(scaler);
vv2.scaleToLayout(scaler);
JButton plus = new JButton("+");
plus.addActionListener(new ActionListener() {
public void actionPerformed(ActionEvent e) {
scaler.scale(vv1, 1.1f, vv1.getCenter());
}
});
JButton minus = new JButton("-");
minus.addActionListener(new ActionListener() {
public void actionPerformed(ActionEvent e) {
scaler.scale(vv1, 1/1.1f, vv1.getCenter());
}
});
JPanel zoomPanel = new JPanel(new GridLayout(1,2));
zoomPanel.setBorder(BorderFactory.createTitledBorder("Zoom"));
JPanel modePanel = new JPanel();
modePanel.setBorder(BorderFactory.createTitledBorder("Mouse Mode"));
gm1.getModeComboBox().addItemListener(gm2.getModeListener());
gm1.getModeComboBox().addItemListener(gm0.getModeListener());
modePanel.add(gm1.getModeComboBox());
JPanel controls = new JPanel();
zoomPanel.add(plus);
zoomPanel.add(minus);
controls.add(zoomPanel);
controls.add(modePanel);
content.add(controls, BorderLayout.SOUTH);
}
public static void main(String[] args) {
JFrame f = new JFrame();
f.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
f.getContentPane().add(new MinimumSpanningTreeDemo());
f.pack();
f.setVisible(true);
}
}