protected void initSimulator(ForceSimulator fsim) {
// make sure we have force items to work with
TupleSet t = (TupleSet) m_vis.getGroup(m_group);
t.addColumns(ANCHORITEM_SCHEMA);
t.addColumns(FORCEITEM_SCHEMA);
Iterator iter = m_vis.visibleItems(m_nodeGroup);
while (iter.hasNext()) {
VisualItem item = (VisualItem) iter.next();
// get force item
ForceItem fitem = (ForceItem) item.get(FORCEITEM);
if (fitem == null) {
fitem = new ForceItem();
item.set(FORCEITEM, fitem);
}
fitem.location[0] = (float) item.getEndX();
fitem.location[1] = (float) item.getEndY();
fitem.mass = getMassValue(item);
// get spring anchor
ForceItem aitem = (ForceItem) item.get(ANCHORITEM);
if (aitem == null) {
aitem = new ForceItem();
item.set(ANCHORITEM, aitem);
aitem.location[0] = fitem.location[0];
aitem.location[1] = fitem.location[1];
}
fsim.addItem(fitem);
fsim.addSpring(fitem, aitem, 0);
}
}
public void clear() {
// System.err.println("run clear");
ForceSimulator fsim = getForceSimulator();
fsim.clear();
// make sure we have force items to work with
TupleSet t = (TupleSet) m_vis.getGroup(m_nodeGroup);
Iterator iter = m_vis.visibleItems(m_nodeGroup);
while (iter.hasNext()) {
VisualItem item = (VisualItem) iter.next();
if (!item.canGet(ANCHORITEM, ForceItem.class)) break;
// get force item
ForceItem fitem = (ForceItem) item.get(FORCEITEM);
if (fitem != null) {
fitem.location[0] = (float) item.getEndX();
fitem.location[1] = (float) item.getEndY();
fitem.mass = getMassValue(item);
// get spring anchor
ForceItem aitem = (ForceItem) item.get(ANCHORITEM);
// only reset if an anchor exists
if (aitem != null) {
aitem.location[0] = fitem.location[0];
aitem.location[1] = fitem.location[1];
fsim.addItem(fitem);
fsim.addSpring(fitem, aitem, 0);
}
}
}
}
public AnchoredForceDirectedLayout(String nodes, boolean enforceBounds, float springForce) {
super(nodes, enforceBounds, false);
ForceSimulator fsim = new ForceSimulator();
fsim.addForce(new SpringForce(springForce, 0f));
fsim.addForce(new DragForce());
setForceSimulator(fsim);
m_nodeGroup = nodes;
m_edgeGroup = null;
}
public DataMountainForceLayout(boolean enforceBounds) {
super("data", enforceBounds, false);
ForceSimulator fsim = new ForceSimulator();
fsim.addForce(new NBodyForce(-0.4f, 25f, NBodyForce.DEFAULT_THETA));
fsim.addForce(new SpringForce(1e-5f, 0f));
fsim.addForce(new DragForce());
setForceSimulator(fsim);
m_nodeGroup = "data";
m_edgeGroup = null;
}
public static JComponent demo(Graph g, String label) {
// create a new, empty visualization for our data
final Visualization vis = new Visualization();
VisualGraph vg = vis.addGraph(graph, g);
vis.setValue(edges, null, VisualItem.INTERACTIVE, Boolean.FALSE);
int[] palette =
new int[] {
ColorLib.rgb(77, 175, 74),
ColorLib.rgb(55, 126, 184),
ColorLib.rgb(228, 26, 28),
ColorLib.rgb(152, 78, 163),
ColorLib.rgb(255, 127, 0)
};
int[] shape_palette =
new int[] {Constants.SHAPE_ELLIPSE, Constants.SHAPE_RECTANGLE, Constants.SHAPE_TRIANGLE_UP};
TupleSet focusGroup = vis.getGroup(Visualization.FOCUS_ITEMS);
focusGroup.addTupleSetListener(
new TupleSetListener() {
public void tupleSetChanged(TupleSet ts, Tuple[] add, Tuple[] rem) {
for (int i = 0; i < rem.length; ++i) ((VisualItem) rem[i]).setFixed(false);
for (int i = 0; i < add.length; ++i) {
((VisualItem) add[i]).setFixed(false);
((VisualItem) add[i]).setFixed(true);
}
vis.run("draw");
}
});
// set up the renderers
/* ShapeRenderer s = new ShapeRenderer();
s.rectangle(0, 0, 30, 30);
vis.setRendererFactory(new DefaultRendererFactory(s));*/
MyRenderer tr = new MyRenderer(label);
// MyRenderer tr = new MyRenderer(label);
DefaultRendererFactory d = new DefaultRendererFactory(tr);
EdgeRenderer MyEdgeRenderer = new EdgeRenderer();
d.add(new InGroupPredicate(edges), MyEdgeRenderer);
vis.setRendererFactory(d);
// -- set up the actions ----------------------------------------------
int maxhops = 1024, hops = 1024;
final GraphDistanceFilter filter = new GraphDistanceFilter(graph, hops);
ActionList draw = new ActionList();
draw.add(filter);
// draw.add(new ColorAction(nodes, VisualItem.FILLCOLOR, ColorLib.rgb(180,180,255)));
draw.add(new ColorAction(nodes, VisualItem.STROKECOLOR, 0));
draw.add(new ColorAction(nodes, VisualItem.TEXTCOLOR, ColorLib.rgb(0, 0, 0)));
draw.add(new ColorAction(edges, VisualItem.FILLCOLOR, ColorLib.gray(100)));
draw.add(new ColorAction(edges, VisualItem.STROKECOLOR, ColorLib.gray(170)));
draw.add(new DataShapeAction(nodes, "type", shape_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));
// create an action list containing all color assignments
ActionList color = new ActionList();
color.add(fill);
color.add(text);
color.add(edges);
// ColorAction fill = new ColorAction(nodes,
// VisualItem.FILLCOLOR, ColorLib.rgb(200,200,255));
fill.add("_fixed", ColorLib.rgb(255, 100, 100));
fill.add("_highlight", ColorLib.rgb(255, 200, 125));
// Instead have to use GridLayout here - how do i create
ForceDirectedLayout fdl = new ForceDirectedLayout(graph);
// GridLayout fdl = new GridLayout(graph);
ForceSimulator fsim = fdl.getForceSimulator();
fsim.getForces()[0].setParameter(0, -5f);
ActionList animate = new ActionList(Activity.INFINITY);
// ActionList animate = new ActionList(vis);
animate.add(fdl);
animate.add(fill);
animate.add(new RepaintAction());
// finally, we register our ActionList with the Visualization.
// we can later execute our Actions by invoking a method on our
// Visualization, using the name we've chosen below.
vis.putAction("draw", draw);
vis.putAction("layout", animate);
vis.runAfter("draw", "layout");
vis.putAction("color", color);
// vis.putAction("layout", layout);
// --------------------------------------------------------------------
// STEP 4: set up a display to show the visualization
Display display = new Display(vis);
display.setSize(2000, 2000);
display.setForeground(Color.GRAY);
display.setBackground(Color.WHITE);
// main display controls
display.addControlListener(new FocusControl(1));
display.addControlListener(new DragControl());
display.addControlListener(new PanControl());
display.addControlListener(new ZoomControl());
display.addControlListener(new WheelZoomControl());
display.addControlListener(new ZoomToFitControl());
display.addControlListener(new NeighborHighlightControl());
display.setForeground(Color.GRAY);
display.setBackground(Color.WHITE);
// --------------------------------------------------------------------
// STEP 5: launching the visualization
// create a panel for editing force values
final JForcePanel fpanel = new JForcePanel(fsim);
final JValueSlider slider = new JValueSlider("Distance", 0, maxhops, hops);
slider.addChangeListener(
new ChangeListener() {
public void stateChanged(ChangeEvent e) {
filter.setDistance(slider.getValue().intValue());
vis.run("draw");
}
});
slider.setBackground(Color.WHITE);
slider.setPreferredSize(new Dimension(300, 30));
slider.setMaximumSize(new Dimension(300, 30));
Box cf = new Box(BoxLayout.Y_AXIS);
cf.add(slider);
cf.setBorder(BorderFactory.createTitledBorder("Connectivity Filter"));
fpanel.add(cf);
fpanel.add(Box.createVerticalGlue());
// create a new JSplitPane to present the interface
JSplitPane split = new JSplitPane();
split.setLeftComponent(display);
split.setRightComponent(fpanel);
split.setOneTouchExpandable(true);
split.setContinuousLayout(false);
split.setDividerLocation(530);
split.setDividerLocation(800);
// position and fix the default focus node
NodeItem focus = (NodeItem) vg.getNode(0);
PrefuseLib.setX(focus, null, 400);
PrefuseLib.setY(focus, null, 250);
focusGroup.setTuple(focus);
// now we run our action list and return
return split;
}
