public static void main(String[] argv) {
// -- 1. load the data ------------------------------------------------
// load the socialnet.xml file. it is assumed that the file can be
// found at the root of the java classpath
Graph graph = null;
try {
graph = new GraphMLReader().readGraph(System.in);
} catch (DataIOException e) {
e.printStackTrace();
System.err.println("Error loading graph. Exiting...");
System.exit(1);
}
// -- 2. the visualization --------------------------------------------
// add the graph to the visualization as the data group "graph"
// nodes and edges are accessible as "graph.nodes" and "graph.edges"
Visualization vis = new Visualization();
vis.add("graph", graph);
vis.setInteractive("graph.edges", null, false);
// -- 3. the renderers and renderer factory ---------------------------
// draw the "name" label for NodeItems
LabelRenderer r = new LabelRenderer("label");
r.setRoundedCorner(8, 8); // round the corners
EdgeRenderer eRender =
new EdgeRenderer(Constants.EDGE_TYPE_CURVE, Constants.EDGE_ARROW_FORWARD);
// create a new default renderer factory
// return our name label renderer as the default for all non-EdgeItems
// includes straight line edges for EdgeItems by default
vis.setRendererFactory(new DefaultRendererFactory(r, eRender));
// -- 4. the processing actions ---------------------------------------
// create our nominal color palette
// pink for females, baby blue for males
int[] palette =
new int[] {
ColorLib.rgb(190, 190, 255), ColorLib.rgb(255, 180, 180), ColorLib.rgb(128, 128, 128),
};
// map nominal data values to colors using our provided palette
DataColorAction fill =
new DataColorAction(
"graph.nodes", "type", Constants.NOMINAL, VisualItem.FILLCOLOR, palette);
// use black for node text
ColorAction text = new ColorAction("graph.nodes", VisualItem.TEXTCOLOR, ColorLib.gray(0));
// use light grey for edges
ColorAction edges = new ColorAction("graph.edges", VisualItem.STROKECOLOR, ColorLib.gray(200));
ColorAction arrows = new ColorAction("graph.edges", VisualItem.FILLCOLOR, ColorLib.gray(200));
// create an action list containing all color assignments
ActionList color = new ActionList();
color.add(fill);
color.add(text);
color.add(edges);
color.add(arrows);
// Add a size action
DataSizeAction size = new DataSizeAction("graph.nodes", "type", 2, Constants.LINEAR_SCALE);
vis.putAction("size", size);
// create an action list with an animated layout
ActionList layout = new ActionList(Activity.INFINITY);
layout.add(new ForceDirectedLayout("graph"));
layout.add(new RepaintAction());
// add the actions to the visualization
vis.putAction("color", color);
vis.putAction("layout", layout);
// -- 5. the display and interactive controls -------------------------
Display d = new Display(vis);
d.setSize(720, 500); // set display size
// drag individual items around
d.addControlListener(new DragControl());
// pan with left-click drag on background
d.addControlListener(new PanControl());
// zoom with right-click drag
d.addControlListener(new ZoomControl());
// -- 6. launch the visualization -------------------------------------
// create a new window to hold the visualization
JFrame frame = new JFrame("IFDS Visualization");
// ensure application exits when window is closed
frame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
frame.add(d);
frame.pack(); // layout components in window
frame.setVisible(true); // show the window
// assign the colors
vis.run("color");
// start up the animated layout
vis.run("layout");
}
/* (non-Javadoc)
* @see org.lukep.javavis.visualisation.views.AbstractVisualisationView#visit(org.lukep.javavis.visualisation.visualisers.PrefuseVisualiser, org.lukep.javavis.ui.swing.WorkspaceContext, prefuse.Display)
*/
@Override
public void visit(PrefuseVisualiser visualiser, WorkspaceContext wspContext, Display display) {
display.reset();
// -- 1. load the data ------------------------------------------------
Graph graph = new Graph();
graph.addColumn("type", String.class);
graph.addColumn("name", String.class);
graph.addColumn("model", IGenericModelNode.class);
graph.addColumn("metricMeasurement", double.class);
// create package vertices
ProjectModel modelStore = wspContext.getModelStore();
HashMap<IGenericModelNode, Node> parentNodeMap =
new HashMap<IGenericModelNode, Node>(modelStore.getPackageMap().size() + 10);
Node curNode;
for (PackageModel pkg : modelStore.getPackageMap().values()) {
// add the package node
curNode = graph.addNode();
curNode.setString("type", pkg.getModelTypeName());
curNode.setString("name", pkg.getSimpleName());
curNode.set("model", pkg);
// link to the parent with an edge
IGenericModelNode parentPackage = pkg.getParent();
if (parentPackage != null
&& parentPackage instanceof PackageModel
&& parentNodeMap.containsKey(parentPackage)) {
// link to the parent package with an edge
graph.addEdge(parentNodeMap.get(parentPackage), curNode);
}
parentNodeMap.put(pkg, curNode);
}
// create class and method nodes
Queue<IGenericModelNode> modelQueue =
new LinkedList<IGenericModelNode>(modelStore.getClassMap().values());
IGenericModelNode model;
while (modelQueue.peek() != null) {
model = modelQueue.remove();
curNode = graph.addNode();
curNode.setString("type", model.getModelTypeName());
curNode.setString("name", model.getSimpleName());
curNode.set("model", model);
// add children to iterModels
if (model.getChildCount() > 0) {
for (Relationship r : model.getChildren())
if (r.getRelationshipType() == RelationshipType.ENCLOSED_IN)
modelQueue.add(r.getTarget());
parentNodeMap.put(model, curNode);
}
// link to the parent package
if (parentNodeMap.containsKey(model.getParent())) {
graph.addEdge(parentNodeMap.get(model.getParent()), curNode);
}
// grab metric measurement if applicable
if (model instanceof IMeasurableNode
&& wspContext.getMetric().testAppliesTo(model.getModelTypeName())) {
curNode.setDouble(
"metricMeasurement",
((IMeasurableNode) (model)).getMetricMeasurement(wspContext.getMetric()).getResult());
}
}
// -- 2. the visualisation --------------------------------------------
// TODO implement a Visualization cache
Visualization vis = new Visualization();
vis.add("graph", graph);
vis.setInteractive("graph.edges", null, false);
// -- 3. the renderers and renderer factory ---------------------------
LabelRenderer r = new LabelRenderer("name");
r.setRoundedCorner(8, 8);
vis.setRendererFactory(new DefaultRendererFactory(r));
// -- 4. the processing actions ---------------------------------------
// create our nominal color palette
// pink for classes, baby blue for packages
int[] palette = ColorLib.getCoolPalette();
// create a metric palette
int[] mPalette = ColorLib.getInterpolatedPalette(0xFF00AA6D, 0xFFFFAA6D);
// TODO replace
int[] mPalette2 = new int[mPalette.length + 1];
mPalette2[0] = Color.lightGray.getRGB();
for (int i = 1; i < mPalette.length; i++) mPalette2[i] = mPalette[i];
// map nominal data values to colors using our provided palette
DataColorAction fill =
new DataColorAction(
"graph.nodes", "type", Constants.NOMINAL, VisualItem.FILLCOLOR, palette);
DataColorAction fillMetrics =
new DataColorAction(
"graph.nodes",
"metricMeasurement",
Constants.NUMERICAL,
VisualItem.FILLCOLOR,
mPalette2);
// use black for node text
ColorAction text = new ColorAction("graph.nodes", VisualItem.TEXTCOLOR, ColorLib.gray(0));
// use light grey for edges
ColorAction edges = new ColorAction("graph.edges", VisualItem.STROKECOLOR, ColorLib.gray(200));
// create an action list containing all color assignments
ActionList colour = new ActionList();
colour.add(fillMetrics);
colour.add(text);
colour.add(edges);
// create an action list with an animated layout
ActionList layout = new ActionList(Activity.INFINITY);
// layout.add(new ForceDirectedLayout("graph"));
layout.add(new NodeLinkTreeLayout("graph"));
layout.add(new RepaintAction());
// add the actions to the visualisation
vis.putAction("colour", colour);
vis.putAction("layout", layout);
// -- 5. the display and interactive controls -------------------------
display.setVisualization(vis);
// -- 6. launch the visualization -------------------------------------
vis.run("colour");
vis.run("layout");
}