source: src/main/java/weka/clusterers/forMetisMQI/MQI.java @ 9

package weka.clusterers.forMetisMQI;

import java.awt.Dimension;
import java.util.Iterator;

import javax.swing.JFrame;

import edu.uci.ics.jung.algorithms.layout.KKLayout;
import edu.uci.ics.jung.algorithms.layout.Layout;
import edu.uci.ics.jung.graph.DirectedSparseGraph;
import edu.uci.ics.jung.graph.Graph;
import edu.uci.ics.jung.visualization.BasicVisualizationServer;
import edu.uci.ics.jung.visualization.decorators.ToStringLabeller;

public class MQI {
        
        public static void viewGraph(Graph<Node, Edge> g){
                Layout<Node, Edge> layout = new KKLayout<Node, Edge>(g);
                layout.setSize(new Dimension(300,300)); // sets the initial size of the space
                // The BasicVisualizationServer<V,E> is parameterized by the edge types
                BasicVisualizationServer<Node,Edge> vv =
                new BasicVisualizationServer<Node,Edge>(layout);
                vv.setPreferredSize(new Dimension(350,350)); //Sets the viewing area size
                vv.getRenderContext().setVertexLabelTransformer(new ToStringLabeller<Node>());
                JFrame frame = new JFrame("Simple Graph View");
                frame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
                frame.getContentPane().add(vv);
                frame.pack();
                frame.setVisible(true);
        }
        
        static public void start(KLPartition partition) {
                Subgraph A = null;
                Subgraph B = null;
                if(partition.getSubgraph().getVertexCount() > partition.getComplement().getVertexCount()) {
                        A = partition.getSubgraph();
                        B = partition.getComplement();
                }
                else {
                        A = partition.getComplement();
                        B = partition.getSubgraph();
                }
                int a = A.getVertexCount();
                int c = partition.edgeCut();
                
                
                Graph<Node,Edge> g = new DirectedSparseGraph<Node, Edge>();
                Iterator<Node> nodes =  A.iterator();
                while(nodes.hasNext()) {
                        Node u = nodes.next();
                        g.addVertex(u);
                }
                
                nodes =  A.iterator();
                int id = 0;
                while(nodes.hasNext()) {
                        Node u = nodes.next();
                        Iterator<Node> neighbors = A.getNeighbors(u).iterator();
                        while(neighbors.hasNext()) {
                                Node v = neighbors.next();
                                g.addEdge(new Edge(Integer.toString(id),A.getWeight(u, v),a),u,v);
                                id++;
                        }
                }
                
                Node source = new Node("S");
                Node sink = new Node("T");
                g.addVertex(source);
                g.addVertex(sink);
                
                
                nodes =  B.iterator();
                while(nodes.hasNext()) {
                        Node u = nodes.next();
                        Iterator<Node> neighbors = B.getGraph().getNeighbors(u).iterator();
                        while(neighbors.hasNext()) {
                                Node v = neighbors.next();
                                if(A.contains(v)) {
                                        g.addEdge(new Edge(Integer.toString(id),1,a),source,v);
                                        id++;
                                }
                        }
                }
                
                nodes =  A.iterator();
                while(nodes.hasNext()) {
                        Node u = nodes.next();
                        g.addEdge(new Edge(Integer.toString(id),1,c),u,sink);
                        id++;
                }
                
                viewGraph(g);
                System.out.println(g.toString());
                
        }
        
}