MainWindow.java

import java.awt.Dimension;
import java.awt.EventQueue;

import javax.swing.JFrame;
import javax.swing.JOptionPane;

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.File;
import java.io.FileReader;
import java.io.FileWriter;

import net.miginfocom.swing.MigLayout;

public class MainWindow {

    private CanvasPanel canvasPanel;
    private DrawPanel drawPanel;
    private JFrame frame;
    private MouseMode mouseMode;

    public static void main(String[] args) {
        EventQueue.invokeLater(new Runnable() {
            public void run() {
                try {
                    MainWindow window = new MainWindow();
                    window.frame.setVisible(true);
                } catch (Exception e) {
                    e.printStackTrace();
                }
            }
        });
    }

    public MainWindow() {
        initialize();
    }
    
    public JFrame getFrame(){
        return frame;
    }
    
    public MouseMode getMouseMode(){
        return mouseMode;
    }
    
    public void setMouseMode(MouseMode mouseMode){
        this.mouseMode = mouseMode;
    }

    public void bundleEdges(int stiffness, boolean animate){
        canvasPanel.bundleEdges(stiffness, animate);
    }
    
    public void clearCanvas(){
        canvasPanel.clearAll();
    }
    
    public void connectAll(){
        canvasPanel.connectAll();
    }

    public void loadGraph(File file){
        try {
            BufferedReader br = new BufferedReader(new FileReader(file));
            String line = br.readLine();
            canvasPanel.clearAll();
            while (line != null) {
                String[] parts = line.split(";");
                if(parts[0].equals("node")){
                    canvasPanel.addNode(Integer.parseInt(parts[1]), Integer.parseInt(parts[2]));
                }
                else if(parts[0].equals("edge")){
                    canvasPanel.addEdge(Integer.parseInt(parts[1]), Integer.parseInt(parts[2]));
                }
                line = br.readLine();
            }
            br.close();
            canvasPanel.paint(canvasPanel.getGraphics());
        } catch (Exception e) {
            JOptionPane.showMessageDialog(frame, "Could not load file.");
        }
    }
    
    public void saveGraph(File file){
        try {
            if (!file.getName().endsWith(".gdf"))
                file = new File(file.getAbsolutePath() + ".gdf");
            BufferedWriter bw = new BufferedWriter(new FileWriter(file));
            for(Node node : canvasPanel.getNodes())
                bw.write("node;" + node.getXInt() + ";" + node.getYInt() + "\n");
            for(Edge edge : canvasPanel.getEdges())
                bw.write("edge;" + canvasPanel.getNodes().indexOf(edge.getStartNode()) + ";" 
                            + canvasPanel.getNodes().indexOf(edge.getEndNode()) + "\n");
            bw.flush();
            bw.close();
        } catch (Exception e) {
            JOptionPane.showMessageDialog(frame, "Could not save file.");
        }
    }
    
    public void unbundleEdges(){
        canvasPanel.unbundleEdges();
    }
    
    public void updateProgress(int progress){
        drawPanel.updateProgress(progress);
    }

    private void initialize() {
        mouseMode = MouseMode.CURSOR;
        
        frame = new JFrame("Force-Directed Edge Bundling");
        frame.setBounds(100, 100, 700, 400);
        frame.setMinimumSize(new Dimension(700,400));
        frame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
        frame.getContentPane().setLayout(new MigLayout("", "[grow]", "[grow]"));
        
        drawPanel = new DrawPanel(this);
        frame.getContentPane().add(drawPanel, "cell 0 0, aligny top");
        canvasPanel = new CanvasPanel(this);
        canvasPanel.setMinimumSize(new Dimension(700-drawPanel.getWidth(), 400));
        frame.getContentPane().add(canvasPanel, "cell 1 0,alignx left,grow");
    }

}