ExportTreeDialog.java

/*
 * ExportTreeDialog.java
 *
 * Copyright (C) 2006-2014 Andrew Rambaut
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
 */

package figtree.application;

import jam.panels.OptionsPanel;

import javax.swing.*;
import javax.swing.border.EmptyBorder;
import java.awt.event.ItemListener;
import java.awt.event.ItemEvent;

/**
 * Dialog presenting options for exporting one or more trees from FigTree
 *
 * @author Andrew Rambaut
 * @version $Id$
 *
 * $HeadURL$
 *
 * $LastChangedBy$
 * $LastChangedDate$
 * $LastChangedRevision$
 */
public class ExportTreeDialog {

    public enum Format {
        NEXUS("NEXUS"),
        NEWICK("Newick"),
        JSON("JSON");

        Format(String name) {
            this.name = name;
        }

        public String toString() {
            return name;
        }

        private String name;
    }

    private final JDialog dialog;
    private final JOptionPane optionPane;

    private JComboBox formatCombo = null;
    private JCheckBox asDisplayedCheck = new JCheckBox("Save as currently displayed");
    private JCheckBox allTreesCheck = new JCheckBox("Save all trees");
    private JCheckBox includeFigTreeCheck = new JCheckBox("Include FigTree block (NEXUS only)");
    private JCheckBox includeAnnotationsCheck = new JCheckBox("Include Annotations (NEXUS & JSON only)");

    public ExportTreeDialog(JFrame frame) {

        OptionsPanel options = new OptionsPanel(12, 12);

        formatCombo = new JComboBox(Format.values());
        options.addComponentWithLabel("Tree file format: ", formatCombo);

        options.addComponent(asDisplayedCheck);
        options.addComponent(allTreesCheck);
        options.addComponent(includeFigTreeCheck);
        options.addComponent(includeAnnotationsCheck);

        optionPane = new JOptionPane(options,
                JOptionPane.QUESTION_MESSAGE,
                JOptionPane.OK_CANCEL_OPTION,
                null,
                null,
                null);
        optionPane.setBorder(new EmptyBorder(12, 12, 12, 12));

        dialog = optionPane.createDialog(frame, "Export Trees");
        dialog.pack();

        formatCombo.addItemListener(new ItemListener() {
            public void itemStateChanged(ItemEvent e) {
                Object item = formatCombo.getSelectedItem();

                includeFigTreeCheck.setEnabled(item.equals(Format.NEXUS));
                includeAnnotationsCheck.setEnabled(item.equals(Format.NEXUS) || item.equals(Format.JSON));
            }
        });
    }

    public int showDialog() {


        dialog.setVisible(true);

        int result = JOptionPane.CANCEL_OPTION;
        Integer value = (Integer)optionPane.getValue();
        if (value != null && value.intValue() != -1) {
            result = value.intValue();
        }

        if (result == JOptionPane.OK_OPTION) {

        }

        return result;
    }

    public Format getFormat() {
        return (Format)formatCombo.getSelectedItem();
    }

    public boolean asDisplayed() {
        return asDisplayedCheck.isSelected();
    }

    public boolean allTrees() {
        return allTreesCheck.isSelected();
    }

    public boolean includeFigTreeBlock() {
        return includeFigTreeCheck.isSelected();
    }

    public boolean includeAnnotations() {
        return includeAnnotationsCheck.isSelected();
    }


}