BrowseArchetypesDialog.java

Go to the documentation of this file.

/*
 * Gridarta MMORPG map editor for Crossfire, Daimonin and similar games.
 * Copyright (C) 2000-2015 The Gridarta Developers.
 *
 * 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 net.sf.gridarta.gui.dialog.browsearchetypes;

import java.awt.BorderLayout;
import java.awt.Component;
import java.awt.Dimension;
import java.awt.event.ItemEvent;
import java.awt.event.ItemListener;
import java.awt.event.MouseEvent;
import java.awt.event.MouseListener;
import java.io.File;
import java.io.IOException;
import java.util.Collection;
import java.util.TreeSet;
import javax.swing.BoxLayout;
import javax.swing.DefaultComboBoxModel;
import javax.swing.ImageIcon;
import javax.swing.JButton;
import javax.swing.JComboBox;
import javax.swing.JDialog;
import javax.swing.JFileChooser;
import javax.swing.JOptionPane;
import javax.swing.JPanel;
import javax.swing.JScrollPane;
import javax.swing.JTable;
import javax.swing.ListSelectionModel;
import javax.swing.ScrollPaneConstants;
import javax.swing.SwingUtilities;
import javax.swing.WindowConstants;
import javax.swing.event.ListSelectionEvent;
import javax.swing.event.ListSelectionListener;
import javax.swing.table.JTableHeader;
import javax.swing.table.TableColumnModel;
import net.sf.gridarta.gui.panel.archetypechooser.ArchetypeChooserControl;
import net.sf.gridarta.gui.panel.objectchooser.ObjectChooser;
import net.sf.gridarta.gui.utils.GUIConstants;
import net.sf.gridarta.model.archetype.Archetype;
import net.sf.gridarta.model.archetypeset.ArchetypeSet;
import net.sf.gridarta.model.gameobject.GameObject;
import net.sf.gridarta.model.maparchobject.MapArchObject;
import net.sf.gridarta.utils.ActionBuilderUtils;
import net.sf.gridarta.utils.FileChooserUtils;
import net.sf.gridarta.utils.FileFilters;
import net.sf.japi.swing.action.ActionBuilder;
import net.sf.japi.swing.action.ActionBuilderFactory;
import net.sf.japi.swing.action.ActionMethod;
import org.jetbrains.annotations.NotNull;

public class BrowseArchetypesDialog<G extends GameObject<G, A, R>, A extends MapArchObject<A>, R extends Archetype<G, A, R>> extends JOptionPane {

    private static final long serialVersionUID = 1L;

    @NotNull
    private static final ActionBuilder ACTION_BUILDER = ActionBuilderFactory.getInstance().getActionBuilder("net.sf.gridarta");

    @NotNull
    private final ArchetypeChooserControl<G, A, R> archetypeChooserControl;

    @NotNull
    private final ObjectChooser<G, A, R> objectChooser;

    @NotNull
    private final Collection<String> availableAttributes = new TreeSet<>();

    @NotNull
    private final BrowseArchetypesTableModel<G, A, R> archetypesTableModel;

    @NotNull
    private final DefaultComboBoxModel<String> addAttributeComboBoxModel = new DefaultComboBoxModel<>();

    @NotNull
    private final JTable archetypesTable;

    @NotNull
    private final JDialog dialog;

    @NotNull
    private final JButton addAttributeButton;

    @NotNull
    private final JComboBox<String> addAttributeComboBox;

    @NotNull
    private final JFileChooser chooser = new JFileChooser();

    public BrowseArchetypesDialog(@NotNull final Component parentComponent, @NotNull final ArchetypeChooserControl<G, A, R> archetypeChooserControl, @NotNull final ObjectChooser<G, A, R> objectChooser, @NotNull final ArchetypeSet<G, A, R> archetypeSet, @NotNull final ImageIcon closingIcon) {
        this.archetypeChooserControl = archetypeChooserControl;
        this.objectChooser = objectChooser;
        archetypesTableModel = new BrowseArchetypesTableModel<>(archetypeSet);
        archetypesTable = new JTable(archetypesTableModel);

        for (final Archetype<?, ?, ?> archetype : archetypeSet.getArchetypes()) {
            availableAttributes.addAll(archetype.getAttributeKeys());
        }
        if (availableAttributes.remove("value")) {
            archetypesTableModel.addAttribute("value");
        }

        archetypesTable.setFillsViewportHeight(true);
        archetypesTable.setAutoCreateRowSorter(true);
        archetypesTable.setSelectionMode(ListSelectionModel.SINGLE_SELECTION);
        archetypesTable.setRowSelectionAllowed(true);
        archetypesTable.setColumnSelectionAllowed(false);
        final ListSelectionListener listSelectionListener = new ListSelectionListener() {

            @Override
            public void valueChanged(@NotNull final ListSelectionEvent e) {
                highlightSelectedEntry();
            }

        };
        archetypesTable.getSelectionModel().addListSelectionListener(listSelectionListener);
        final JTableHeader tableHeader = archetypesTable.getTableHeader();
        tableHeader.setReorderingAllowed(true);
        final TableHeaderCellRenderer renderer = new TableHeaderCellRenderer(closingIcon, tableHeader.getDefaultRenderer());
        tableHeader.setDefaultRenderer(renderer);
        final MouseListener mouseListener = new MouseListener() {

            private boolean pressed;

            @Override
            public void mouseClicked(@NotNull final MouseEvent e) {
                if (pressed) {
                    pressed = false;
                    if (SwingUtilities.isLeftMouseButton(e)) {
                        final TableColumnModel columnModel = archetypesTable.getColumnModel();
                        final int viewColumn = columnModel.getColumnIndexAtX(e.getX());
                        final int columnIndex = archetypesTable.convertColumnIndexToModel(viewColumn);
                        final int offsetX = renderer.getX(columnIndex);
                        int cx = 0;
                        for (int i = 0; i < columnIndex; i++) {
                            cx += columnModel.getColumn(i).getWidth();
                        }
                        final int ex = e.getX();
                        if (offsetX >= 0 && ex - cx >= offsetX) {
                            final String attribute = archetypesTableModel.removeAttribute(columnIndex);
                            if (attribute != null) {
                                availableAttributes.add(attribute);
                                updateComboBox();
                                addAttributeComboBoxModel.setSelectedItem(attribute);
                            }
                        }
                    }
                }
            }

            @Override
            public void mousePressed(@NotNull final MouseEvent e) {
                pressed = true;
            }

            @Override
            public void mouseReleased(@NotNull final MouseEvent e) {
            }

            @Override
            public void mouseEntered(@NotNull final MouseEvent e) {
            }

            @Override
            public void mouseExited(@NotNull final MouseEvent e) {
            }

        };
        tableHeader.addMouseListener(mouseListener);
        addAttributeButton = new JButton(ACTION_BUILDER.createAction(false, "browseArchetypesAddAttribute", this));
        addAttributeComboBox = new JComboBox<>(addAttributeComboBoxModel);
        final ItemListener itemListener = new ItemListener() {

            @Override
            public void itemStateChanged(@NotNull final ItemEvent e) {
                updateActions();
            }

        };
        addAttributeComboBox.addItemListener(itemListener);
        setMessage(createPanel());

        final JPanel addAttributePanel = new JPanel();
        addAttributePanel.setLayout(new BoxLayout(addAttributePanel, BoxLayout.LINE_AXIS));
        addAttributePanel.add(addAttributeButton);
        addAttributePanel.add(addAttributeComboBox);
        addAttributePanel.add(new JButton(ACTION_BUILDER.createAction(false, "browseArchetypesSave", this)));
        setOptions(new Object[] { addAttributePanel });

        updateComboBox();
        addAttributeComboBox.setSelectedIndex(Math.min(0, addAttributeComboBoxModel.getSize() - 1));

        dialog = createDialog(parentComponent, ActionBuilderUtils.getString(ACTION_BUILDER, "browseArchetypes.title"));
        dialog.setDefaultCloseOperation(WindowConstants.DISPOSE_ON_CLOSE);
        dialog.setModal(false);
        dialog.setResizable(true);
        dialog.pack();
        dialog.setMinimumSize(new Dimension(200, 200));
        dialog.setSize(new Dimension(800, 600));
        dialog.setLocationRelativeTo(parentComponent);
        setInitialValue(archetypesTable);
        updateActions();

        chooser.setDialogTitle(ActionBuilderUtils.getString(ACTION_BUILDER, "browseArchetypesSave.title"));
        chooser.setFileFilter(FileFilters.CSV_FILE_FILTER);
    }

    public void showDialog() {
        dialog.setVisible(true);
        dialog.toFront();
    }

    @NotNull
    private JPanel createPanel() {
        final JPanel mainPanel = new JPanel(new BorderLayout());

        mainPanel.setBorder(GUIConstants.DIALOG_BORDER);

        final JScrollPane scrollPane = new JScrollPane(archetypesTable);
        scrollPane.setHorizontalScrollBarPolicy(ScrollPaneConstants.HORIZONTAL_SCROLLBAR_AS_NEEDED);
        scrollPane.setVerticalScrollBarPolicy(ScrollPaneConstants.VERTICAL_SCROLLBAR_AS_NEEDED);

        mainPanel.add(scrollPane, BorderLayout.CENTER);
        return mainPanel;
    }

    @ActionMethod
    public void browseArchetypesAddAttribute() {
        doAddAttribute(true);
    }

    @ActionMethod
    public void browseArchetypesSave() {
        chooser.setFileSelectionMode(JFileChooser.FILES_ONLY);
        chooser.setMultiSelectionEnabled(false);
        FileChooserUtils.sanitizeCurrentDirectory(chooser);
        final int returnVal = chooser.showSaveDialog(this);
        if (returnVal == JFileChooser.APPROVE_OPTION) {
            File file = chooser.getSelectedFile();
            try {
                if (chooser.getFileFilter() == FileFilters.CSV_FILE_FILTER && !file.getName().endsWith(".csv")) {
                    file = new File(file.getParentFile(), file.getName() + ".csv");
                }
                archetypesTableModel.saveAsCsv(file);
            } catch (final IOException ex) {
                ACTION_BUILDER.showMessageDialog(this, "browseArchetypesSaveIOException", file, ex.getMessage());
            }
        }
    }

    private boolean doAddAttribute(final boolean performAction) {
        final String attributeName = (String) addAttributeComboBox.getSelectedItem();
        if (attributeName == null) {
            return false;
        }

        if (performAction) {
            if (availableAttributes.remove(attributeName)) {
                archetypesTableModel.addAttribute(attributeName);
                final int selectedIndex = Math.max(addAttributeComboBox.getSelectedIndex(), 0);
                updateComboBox();
                addAttributeComboBox.setSelectedIndex(Math.min(selectedIndex, addAttributeComboBoxModel.getSize() - 1));
            }
            updateActions();
        } else if (!availableAttributes.contains(attributeName)) {
            return false;
        }

        return true;
    }

    private void updateActions() {
        addAttributeButton.setEnabled(doAddAttribute(false));
        addAttributeComboBox.setEnabled(!availableAttributes.isEmpty());
    }

    private void updateComboBox() {
        addAttributeComboBoxModel.removeAllElements();
        for (final String attribute : availableAttributes) {
            addAttributeComboBoxModel.addElement(attribute);
        }
    }

    private void highlightSelectedEntry() {
        final int index = archetypesTable.getSelectedRow();
        if (index == -1) {
            return;
        }

        final R archetype = archetypesTableModel.get(archetypesTable.getRowSorter().convertRowIndexToModel(index));
        objectChooser.moveArchetypeChooserToFront();
        archetypeChooserControl.selectArchetype(archetype);
    }

}