package de.ugoe.cs.eventbench.commands; import java.awt.Dimension; import java.awt.Rectangle; import java.awt.Shape; import java.security.InvalidParameterException; import java.util.List; import javax.swing.JFrame; import org.apache.commons.collections15.Transformer; import de.ugoe.cs.eventbench.data.GlobalDataContainer; import de.ugoe.cs.eventbench.models.Trie; import de.ugoe.cs.eventbench.models.Trie.Edge; import de.ugoe.cs.eventbench.models.Trie.TrieVertex; import de.ugoe.cs.eventbench.models.TrieBasedModel; import de.ugoe.cs.util.console.Command; import de.ugoe.cs.util.console.Console; import edu.uci.ics.jung.algorithms.layout.Layout; import edu.uci.ics.jung.algorithms.layout.TreeLayout; import edu.uci.ics.jung.graph.Tree; import edu.uci.ics.jung.visualization.BasicVisualizationServer; import edu.uci.ics.jung.visualization.decorators.ToStringLabeller; import edu.uci.ics.jung.visualization.renderers.Renderer.VertexLabel.Position; /** *

* Command that visualizes the {@link Trie} of a {@link TrieBasedModel}. *

* * @author Steffen Herbold * @version 1.0 */ public class CMDshowTrie implements Command { /* * (non-Javadoc) * * @see de.ugoe.cs.util.console.Command#help() */ @Override public void help() { Console.println("Usage: showTrie "); } /* * (non-Javadoc) * * @see de.ugoe.cs.util.console.Command#run(java.util.List) */ @Override public void run(List parameters) { String modelname; try { modelname = (String) parameters.get(0); } catch (Exception e) { throw new InvalidParameterException(); } Object dataObject = GlobalDataContainer.getInstance() .getData(modelname); if (dataObject == null) { Console.printerrln("Object " + modelname + " not found in storage."); return; } if (!(dataObject instanceof TrieBasedModel)) { Console.printerr("Object " + modelname + " not of type TrieBasedModel."); } TrieBasedModel model = (TrieBasedModel) dataObject; Tree graph = model.getTrieGraph(); Layout layout = new TreeLayout( graph, 60); // The BasicVisualizationServer is parameterized by the edge // types BasicVisualizationServer vv = new BasicVisualizationServer( layout); vv.setPreferredSize(new Dimension(1100, 850)); // Sets the viewing // area size final Rectangle rect = new Rectangle(40, 20); Transformer vertexShapeTransformer = new Transformer() { public Shape transform(TrieVertex s) { return rect; } }; vv.getRenderer().getVertexLabelRenderer().setPosition(Position.CNTR); vv.getRenderContext().setVertexShapeTransformer(vertexShapeTransformer); vv.getRenderContext().setVertexLabelTransformer( new ToStringLabeller()); JFrame frame = new JFrame("Trie"); frame.setDefaultCloseOperation(JFrame.DISPOSE_ON_CLOSE); frame.getContentPane().add(vv); frame.pack(); frame.setVisible(true); } }