private Node addNode(String label, int size, int index) {
Node node = graph.addNode();
node.setString(LABEL, label);
node.setInt(Index, index);
node.setInt(SIZE, size);
return node;
}
private Node addNode(Class type) {
if (type != null && !nodes.containsKey(type)) {
Node node = graph.addNode();
node.set("type", type);
nodes.put(type, node);
}
return (Node) nodes.get(type);
}
private void initDataGroups() {
// create sample graph
// 9 nodes broken up into 3 interconnected cliques
Graph g = new Graph();
for (int i = 0; i < 3; ++i) {
Node n1 = g.addNode();
Node n2 = g.addNode();
Node n3 = g.addNode();
g.addEdge(n1, n2);
g.addEdge(n1, n3);
g.addEdge(n2, n3);
}
g.addEdge(0, 3);
g.addEdge(3, 6);
g.addEdge(6, 0);
// add visual data groups
VisualGraph vg = m_vis.addGraph(GRAPH, g);
m_vis.setInteractive(EDGES, null, false);
m_vis.setValue(NODES, null, VisualItem.SHAPE, new Integer(Constants.SHAPE_ELLIPSE));
at = m_vis.addAggregates(AGGR);
at.addColumn(VisualItem.POLYGON, float[].class);
at.addColumn("id", int.class);
// add nodes to aggregates
// create an aggregate for each 3-clique of nodes
Iterator nodes = vg.nodes();
for (int i = 0; i < 3; ++i) {
AggregateItem aitem = (AggregateItem) at.addItem();
aitem.setInt("id", i);
for (int j = 0; j < 3; ++j) {
aitem.addItem((VisualItem) nodes.next());
}
}
}
private void makeGraph(Graph displayed, rndalg.Graph graphToShow, Vector<Vertex> cut) {
HashMap<rndalg.Graph.Vertex, Node> nodeMap = new HashMap<rndalg.Graph.Vertex, Node>();
for (rndalg.Graph.Vertex v : graphToShow.vertices) {
Node n = displayed.addNode();
n.set("id", v.id + "");
nodeMap.put(v, n);
}
for (rndalg.Graph.Edge e : graphToShow.getEdges()) {
Edge edge = displayed.addEdge(nodeMap.get(e.a), nodeMap.get(e.b));
edge.set("isCut", isFromCut(e, cut));
}
}
/* (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");
}
