DrawPanel.java

import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.io.File;

import javax.swing.JFileChooser;
import javax.swing.JPanel;
import javax.swing.JToggleButton;
import javax.swing.JSlider;
import javax.swing.JButton;
import net.miginfocom.swing.MigLayout;
import javax.swing.ButtonGroup;
import javax.swing.filechooser.FileNameExtensionFilter;
import javax.swing.JProgressBar;
import javax.swing.BoxLayout;
import java.awt.Component;
import javax.swing.JLabel;
import javax.swing.JCheckBox;

public class DrawPanel extends JPanel {
    
    private boolean bundlingActive;
    private final ButtonGroup buttonGroup = new ButtonGroup();
    private MainWindow window;
    private JFileChooser fc;
    private JToggleButton tglbtnNode;
    private JToggleButton tglbtnEdge;
    private JToggleButton tglbtnDelete;
    private JButton btnLoad;
    private JButton btnSave;
    private JButton btnClearAll;
    private JButton btnConnectAll;
    private JToggleButton btnBundle;
    private JToggleButton btnUnbundle;
    private JSlider slider;
    private final ButtonGroup buttonGroup_1 = new ButtonGroup();
    private JProgressBar progressBar;
    private JLabel lblStiffness;
    private JCheckBox chckbxAnimation;

    public DrawPanel(MainWindow window) {
        this.window = window;
        fc = new JFileChooser();
        fc.setFileFilter(new FileNameExtensionFilter("GraphDraw File", "gdf"));
        fc.setCurrentDirectory(new File("./"));
        
        setLayout(new MigLayout("", "[90.00][]", "[][][][][][][][]"));
        
        tglbtnNode = new JToggleButton("Node");
        buttonGroup.add(tglbtnNode);
        add(tglbtnNode, "cell 0 0,growx");
        
        btnConnectAll = new JButton("Connect All");
        add(btnConnectAll, "cell 1 0,growx");
        btnConnectAll.setAlignmentX(Component.CENTER_ALIGNMENT);
        
        tglbtnEdge = new JToggleButton("Edge");
        buttonGroup.add(tglbtnEdge);
        add(tglbtnEdge, "cell 0 1,growx");
        
        btnClearAll = new JButton("Clear All");
        add(btnClearAll, "cell 1 1,growx");
        btnClearAll.setAlignmentX(Component.CENTER_ALIGNMENT);
        
        tglbtnDelete = new JToggleButton("Delete");
        buttonGroup.add(tglbtnDelete);
        add(tglbtnDelete, "flowy,cell 0 2,growx");
        bundlingActive = false;
        
        chckbxAnimation = new JCheckBox("Animation");
        add(chckbxAnimation, "cell 1 2,alignx center");
        
        lblStiffness = new JLabel("Stiffness:");
        add(lblStiffness, "cell 0 3");
        
        slider = new JSlider();
        slider.setPaintLabels(true);
        slider.setPaintTicks(true);
        slider.setMaximum(10);
        slider.setMinimum(1);
        slider.setMajorTickSpacing(1);
        slider.setSnapToTicks(true);
        slider.setValue(1);
        add(slider, "cell 0 4 2 1,growx");
        
        btnLoad = new JButton("Load");
        add(btnLoad, "cell 0 5,growx");
        
        btnSave = new JButton("Save");
        add(btnSave, "cell 1 5,growx");
        
        btnBundle = new JToggleButton("Bundle");
        buttonGroup_1.add(btnBundle);
        add(btnBundle, "cell 0 6,growx");
        
        btnUnbundle = new JToggleButton("Unbundle");
        btnUnbundle.setSelected(true);
        buttonGroup_1.add(btnUnbundle);
        add(btnUnbundle, "cell 1 6,growx");
        
        progressBar = new JProgressBar();
        add(progressBar, "cell 0 7 2 1,growx");

        addListeners();
    }
    
    public void updateProgress(int progress){
        progressBar.setValue(progress);
        progressBar.update(progressBar.getGraphics());
    }
    
    private void addListeners(){
        tglbtnNode.addActionListener(new ActionListener() {
            
            @Override
            public void actionPerformed(ActionEvent e) {
                window.setMouseMode(MouseMode.NODE);
            }
        });
        
        tglbtnEdge.addActionListener(new ActionListener() {
            
            @Override
            public void actionPerformed(ActionEvent e) {
                window.setMouseMode(MouseMode.EDGE);
            }
        });
        
        tglbtnDelete.addActionListener(new ActionListener() {
            
            @Override
            public void actionPerformed(ActionEvent e) {
                window.setMouseMode(MouseMode.DELETE);
            }
        });
        
        btnLoad.addActionListener(new ActionListener() {

            @Override
            public void actionPerformed(ActionEvent e) {
                int returnVal = fc.showOpenDialog(window.getFrame());
                if (returnVal == JFileChooser.APPROVE_OPTION){
                    btnUnbundle.setSelected(true);
                    tglbtnNode.setEnabled(true);
                    tglbtnEdge.setEnabled(true);
                    tglbtnDelete.setEnabled(true);
                    tglbtnNode.setSelected(false);
                    tglbtnEdge.setSelected(false);
                    tglbtnDelete.setSelected(false);
                    btnConnectAll.setEnabled(true);
                    btnClearAll.setEnabled(true);
                    chckbxAnimation.setEnabled(true);
                    window.unbundleEdges();
                    bundlingActive = false;
                    window.loadGraph(fc.getSelectedFile());
                }
            }
        });

        btnSave.addActionListener(new ActionListener() {

            @Override
            public void actionPerformed(ActionEvent e) {
                int returnVal = fc.showSaveDialog(window.getFrame());
                if (returnVal == JFileChooser.APPROVE_OPTION)
                    window.saveGraph(fc.getSelectedFile());
            }
        });
        
        btnConnectAll.addActionListener(new ActionListener() {

            @Override
            public void actionPerformed(ActionEvent e) {
                window.connectAll();
            }
        });

        btnClearAll.addActionListener(new ActionListener() {

            @Override
            public void actionPerformed(ActionEvent e) {
                window.clearCanvas();
            }
        });
        
        btnBundle.addActionListener(new ActionListener() {
            
            @Override
            public void actionPerformed(ActionEvent e) {
                if(!bundlingActive){
                    window.setMouseMode(MouseMode.CURSOR);
                    tglbtnNode.setEnabled(false);
                    tglbtnEdge.setEnabled(false);
                    tglbtnDelete.setEnabled(false);
                    btnConnectAll.setEnabled(false);
                    btnClearAll.setEnabled(false);
                    chckbxAnimation.setEnabled(false);
                    btnLoad.setEnabled(false);
                    btnSave.setEnabled(false);
                    btnBundle.setEnabled(false);
                    btnUnbundle.setEnabled(false);
                    slider.setEnabled(false);
                    update(getGraphics());
                    
                    window.bundleEdges(slider.getValue(), chckbxAnimation.isSelected());
                    bundlingActive = true;
                    
                    btnLoad.setEnabled(true);
                    btnSave.setEnabled(true);
                    btnBundle.setEnabled(true);
                    btnUnbundle.setEnabled(true);
                    slider.setEnabled(true);
                }
            }
        });
        
        btnUnbundle.addActionListener(new ActionListener() {
            
            @Override
            public void actionPerformed(ActionEvent e) {
                if(bundlingActive){
                    window.unbundleEdges();
                    bundlingActive = false;
                    tglbtnNode.setEnabled(true);
                    tglbtnEdge.setEnabled(true);
                    tglbtnDelete.setEnabled(true);
                    btnConnectAll.setEnabled(true);
                    btnClearAll.setEnabled(true);
                    chckbxAnimation.setEnabled(true);
                }
            }
        });
    }
}