public void CenterGraph(int borderSpacing, boolean _repaint) {
List<Functionality.Node> ln = myGraph.getNodes();
double xsum = 0;
double ysum = 0;
for (int i = 0; i < ln.size(); i++) {
xsum += layout.getX(ln.get(i));
ysum += layout.getY(ln.get(i));
}
double xmean = xsum / ln.size();
double ymean = ysum / ln.size();
double xcoef = (layout_width - borderSpacing * 2) / (double) layout_width;
double ycoef = (layout_height - borderSpacing * 2) / (double) layout_height;
for (int i = 0; i < ln.size(); i++) {
double xnew = layout.getX(ln.get(i)) + (layout_width / 2 - xmean);
double ynew = layout.getY(ln.get(i)) + (layout_height / 2 - ymean);
layout.setLocation(ln.get(i), xnew, ynew);
}
for (int i = 0; i < ln.size(); i++) {
double xnew = borderSpacing + layout.getX(ln.get(i)) * xcoef;
double ynew = borderSpacing + layout.getY(ln.get(i)) * ycoef;
layout.setLocation(ln.get(i), xnew, ynew);
}
if (_repaint == true) vv.repaint();
}
private void InitializeClusterGraphLayout() {
int numClusters = myGraph.GetNumberOfClusters();
double radius = Math.min(layout_width, layout_height) / 3.5;
double angleShift = (2 * Math.PI) / numClusters;
List<Functionality.Node> ln = myGraph.getNodes();
Node n;
int clusterID;
for (int i = 0; i < ln.size(); i++) {
n = ln.get(i);
clusterID = n.GetCluster();
if (clusterID == -1) continue;
double xnew =
(layout_width / 2)
+ Math.cos(angleShift * clusterID) * radius
+ (50 - Math.random() * 100);
double ynew =
(layout_height / 2)
+ Math.sin(angleShift * clusterID) * radius
+ (50 - Math.random() * 100);
layout.setLocation(n, xnew, ynew);
}
}
public JComponent actualizeGraph(boolean lockLayout) {
// myGraph = g;
System.err.println("Location: x = " + vv.getLocation().x + ", y = " + vv.getLocation().y);
MultiLayerTransformer mlt = vv.getRenderContext().getMultiLayerTransformer();
Map<Functionality.Node, Point2d> locSet = new HashMap<Functionality.Node, Point2d>();
PickedState<Node> psn = vv.getPickedVertexState();
List<Functionality.Node> ln = DataModule.displayedGraph.getNodes();
for (int i = 0; i < ln.size(); i++) {
locSet.put(ln.get(i), new Point2d(layout.getX(ln.get(i)), layout.getY(ln.get(i))));
}
init(DataModule.displayedGraph, layout_width, layout_height);
drawGraph();
for (int i = 0; i < ln.size(); i++) {
layout.setLocation(ln.get(i), locSet.get(ln.get(i)).x, locSet.get(ln.get(i)).y);
}
PickedState<Node> newPsn = vv.getPickedVertexState();
for (int i = 0; i < ln.size(); i++) {
newPsn.pick(ln.get(i), psn.isPicked(ln.get(i)));
}
if (vdp instanceof VertexDisplayPredicateDistance) {
((VertexDisplayPredicateDistance) vdp).Actualise();
}
vv.getRenderContext().setVertexIncludePredicate(vdp);
vv.getRenderContext().setEdgeIncludePredicate(eip);
vv.setPickedVertexState(newPsn);
vv.getRenderContext().setMultiLayerTransformer(mlt);
vv.getRenderContext().setVertexFillPaintTransformer(VertexColorTransformer.vctInstance);
layout.lock(lockLayout);
MyGraphZoomScrollPane graphPane = new MyGraphZoomScrollPane(vv);
return graphPane;
}
public JComponent drawGraph() {
// System.out.println("The graph qt"+qt.toString()); // DEBUG
// ISOMLayout
layout = new SpringLayoutWeighted(this.qt);
layout.setSize(new Dimension(layout_width, layout_height));
// layout.setSize(new Dimension(layout_width, layout_height));
vv = new MyVisualizationViewer<Functionality.Node, Functionality.Edge>(layout);
vv.setBackground(Color.WHITE);
vv.setPreferredSize(new Dimension(layout_width, layout_height));
vv.setIgnoreRepaint(true);
// Show vertex and edge labels
vv.getRenderContext().setVertexLabelTransformer(new DefaultToStringLabeller());
// BasicVertexLabelRenderer<Functionality.Node, Functionality.Edge> bvlr = new
// (BasicVertexLabelRenderer<Functionality.Node,
// Functionality.Edge>)(vv.getRenderContext().getVertexLabelRenderer());
BasicRenderer<Functionality.Node, Functionality.Edge> br = new BasicRenderer<Node, Edge>();
// br.setVertexLabelRenderer(new BasicVertexLabelRenderer<Functionality.Node,
// Functionality.Edge>(Position.CNTR));
vv.setRenderer(br);
// MyVertexLabelRenderer dvlr = new MyVertexLabelRenderer(Color.red);
// vv.getRenderContext().setVertexLabelRenderer(dvlr);
// bvlr.setPosition(Position.CNTR);
// vv.getRenderContext().setEdgeLabelTransformer(new ToStringLabeller());
vv.setDoubleBuffered(true);
// layout = new SpringLayout2<Functionality.Node, Functionality.Edge>(this.qt);
layout.initialize();
if (myGraph.GetNumberOfClusters() > 1) {
InitializeClusterGraphLayout();
}
layout.step();
layout.step();
layout.step();
layout.step();
layout.step();
layout.step();
layout.step();
layout.step();
layout.step();
layout.step();
layout.step();
layout.step();
layout.lock(true);
CenterGraph(10, false);
// Create a graph mouse and add it to the visualization component
// DefaultModalGraphMouse gm = new DefaultModalGraphMouse();
DefaultModalGraphMouse gm = new DefaultModalGraphMouse();
gm.setMode(ModalGraphMouse.Mode.PICKING);
vv.addMouseListener(this);
VertexMouseMenu vMouseMenu = new VertexMouseMenu();
JPopupMenu vertexMenu = new MyVertexMenu();
vMouseMenu.setVertexPopup(vertexMenu);
gm.add(vMouseMenu);
vv.setGraphMouse(gm);
setEdgeWeightStrokeFunction();
vv.getRenderContext()
.setVertexShapeTransformer(
new VertexShapeSizeAspect<Functionality.Node, Functionality.Edge>(myGraph));
vv.getRenderContext().setVertexFillPaintTransformer(VertexColorTransformer.vctInstance);
MyGraphZoomScrollPane graphPane = new MyGraphZoomScrollPane(vv);
return graphPane;
}
public void LayoutStop() {
layout.lock(true);
vv.setRedrawing(false);
}
