FindDialog.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.find;

import java.awt.Component;
import java.awt.Container;
import java.awt.FlowLayout;
import java.awt.Point;
import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import javax.swing.AbstractButton;
import javax.swing.BorderFactory;
import javax.swing.Box;
import javax.swing.BoxLayout;
import javax.swing.JButton;
import javax.swing.JCheckBox;
import javax.swing.JDialog;
import javax.swing.JOptionPane;
import javax.swing.JPanel;
import javax.swing.JTextField;
import javax.swing.WindowConstants;
import javax.swing.text.JTextComponent;
import net.sf.gridarta.gui.map.mapview.MapView;
import net.sf.gridarta.gui.utils.TextComponentUtils;
import net.sf.gridarta.model.archetype.Archetype;
import net.sf.gridarta.model.gameobject.GameObject;
import net.sf.gridarta.model.maparchobject.MapArchObject;
import net.sf.gridarta.model.mapcontrol.MapControl;
import net.sf.gridarta.model.mapcursor.MapCursor;
import net.sf.gridarta.model.mapgrid.MapGrid;
import net.sf.gridarta.model.mapgrid.SelectionMode;
import net.sf.gridarta.model.mapmodel.MapModel;
import net.sf.gridarta.model.mapmodel.MapSquare;
import net.sf.gridarta.model.select.ArchetypeNameMatchCriteria;
import net.sf.gridarta.model.select.AttributeOtherValueMatchCriteria;
import net.sf.gridarta.model.select.AttributeValueMatchCriteria;
import net.sf.gridarta.model.select.MatchCriteria;
import net.sf.gridarta.model.select.ObjectNameMatchCriteria;
import net.sf.gridarta.utils.ActionBuilderUtils;
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 FindDialog<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 JDialog dialog;

    @NotNull
    private final Component parent;

    private boolean isBuilt;

    @NotNull
    private MapView<G, A, R> mapView;

    @NotNull
    private final JTextComponent findInput = new JTextField(20);

    @NotNull
    private final AbstractButton findNameCheckbox = new JCheckBox(ActionBuilderUtils.getString(ACTION_BUILDER, "findWhereName"));

    @NotNull
    private final AbstractButton findArchCheckbox = new JCheckBox(ActionBuilderUtils.getString(ACTION_BUILDER, "findWhereArch"));

    @NotNull
    private final AbstractButton findMsgCheckbox = new JCheckBox(ActionBuilderUtils.getString(ACTION_BUILDER, "findWhereMsg"));

    @NotNull
    private final AbstractButton findFaceCheckbox = new JCheckBox(ActionBuilderUtils.getString(ACTION_BUILDER, "findWhereFace"));

    @NotNull
    private final AbstractButton findSlayingCheckbox = new JCheckBox(ActionBuilderUtils.getString(ACTION_BUILDER, "findWhereSlaying"));

    @NotNull
    private final AbstractButton findOtherCheckbox = new JCheckBox(ActionBuilderUtils.getString(ACTION_BUILDER, "findWhereOther"));

    public FindDialog(@NotNull final Component parent) {
        dialog = createDialog(parent, "");
        dialog.setModal(false);
        dialog.setDefaultCloseOperation(WindowConstants.HIDE_ON_CLOSE);
        this.parent = parent;

        TextComponentUtils.setAutoSelectOnFocus(findInput);
    }

    public void display(@NotNull final MapView<G, A, R> mapView) {
        if (isBuilt) {
            this.mapView = mapView;

            dialog.pack();
            dialog.toFront();
        } else {
            this.mapView = mapView;
            final JPanel mainPanel = new JPanel();
            mainPanel.setLayout(new BoxLayout(mainPanel, BoxLayout.Y_AXIS));
            mainPanel.setBorder(BorderFactory.createEmptyBorder(5, 5, 2, 5));

            final Container lineFind = new JPanel(new FlowLayout(FlowLayout.LEFT));
            lineFind.add(ActionBuilderUtils.newLabel(ACTION_BUILDER, "findFind"));
            lineFind.add(Box.createVerticalStrut(3));
            lineFind.add(findInput);
            lineFind.add(Box.createVerticalStrut(3));
            mainPanel.add(lineFind);

            final Container lineWhere = new JPanel(new FlowLayout(FlowLayout.LEFT));
            lineWhere.add(ActionBuilderUtils.newLabel(ACTION_BUILDER, "findWhere"));
            lineWhere.add(Box.createVerticalStrut(5));

            final JPanel panelWhere = new JPanel();
            panelWhere.setLayout(new BoxLayout(panelWhere, BoxLayout.Y_AXIS));
            panelWhere.add(findNameCheckbox);
            panelWhere.add(findArchCheckbox);
            panelWhere.add(findMsgCheckbox);
            panelWhere.add(findFaceCheckbox);
            panelWhere.add(findSlayingCheckbox);
            panelWhere.add(findOtherCheckbox);
            lineWhere.add(panelWhere);
            mainPanel.add(lineWhere);

            final JButton okButton = new JButton(ACTION_BUILDER.createAction(false, "findOk", this));
            final JButton cancelButton = new JButton(ACTION_BUILDER.createAction(false, "findCancel", this));

            findNameCheckbox.setSelected(true);
            findMsgCheckbox.setSelected(true);
            findSlayingCheckbox.setSelected(true);

            setMessage(mainPanel);
            setOptions(new Object[] { okButton, cancelButton });
            dialog.getRootPane().setDefaultButton(okButton);
            dialog.pack();
            dialog.setLocationRelativeTo(parent);

            isBuilt = true;
        }
        dialog.setTitle(ActionBuilderUtils.format(ACTION_BUILDER, "findTitle", mapView.getMapControl().getMapModel().getMapArchObject().getMapName()));
        dialog.setVisible(true);
        findInput.requestFocusInWindow();
    }

    public boolean findAgain(@NotNull final MapView<G, A, R> mapView, final boolean forward, final boolean performAction) {
        if (this.mapView != mapView) {
            return false;
        }
        if (performAction) {
            doFind(forward);
        }
        return true;
    }

    @ActionMethod
    public void findOk() {
        if (doFind(true)) {
            dialog.setVisible(false);
        }
    }

    @ActionMethod
    public void findCancel() {
        dialog.setVisible(false);
    }

    private boolean doFind(final boolean forward) {
        final String findString = findInput.getText().trim();
        final Collection<MatchCriteria<G, A, R>> matchCriterias = new ArrayList<>();
        if (findNameCheckbox.isSelected()) {
            matchCriterias.add(new ObjectNameMatchCriteria<>(findString));
        }
        if (findArchCheckbox.isSelected()) {
            matchCriterias.add(new ArchetypeNameMatchCriteria<>(findString));
        }
        if (findMsgCheckbox.isSelected()) {
            matchCriterias.add(new AttributeValueMatchCriteria<>("msg", findString));
        }
        if (findFaceCheckbox.isSelected()) {
            matchCriterias.add(new AttributeValueMatchCriteria<>("face", findString));
            matchCriterias.add(new AttributeValueMatchCriteria<>("animation", findString));
        }
        if (findSlayingCheckbox.isSelected()) {
            matchCriterias.add(new AttributeValueMatchCriteria<>("slaying", findString));
        }
        if (findOtherCheckbox.isSelected()) {
            matchCriterias.add(new AttributeOtherValueMatchCriteria<>(findString, "msg", "face", "animation", "slaying"));
        }
        return !matchCriterias.isEmpty() && doFind(matchCriterias, forward) > 0;
    }

    private int doFind(@NotNull final Iterable<MatchCriteria<G, A, R>> matchCriterias, final boolean forward) {
        final List<G> matchingGameObjects = new ArrayList<>();
        final Collection<MapSquare<G, A, R>> matchingMapSquares = new ArrayList<>();
        final MapControl<G, A, R> mapControl = mapView.getMapControl();
        final MapModel<G, A, R> mapModel = mapControl.getMapModel();
        for (final MapSquare<G, A, R> mapSquare : mapModel) {
            boolean matchesMapSquare = false;
            for (final G gameObject : mapSquare.recursive()) {
                for (final MatchCriteria<G, A, R> matchCriteria : matchCriterias) {
                    if (matchCriteria.matches(gameObject)) {
                        matchingGameObjects.add(gameObject);
                        matchesMapSquare = true;
                    }
                }
            }
            if (matchesMapSquare) {
                matchingMapSquares.add(mapSquare);
            }
        }
        selectMapSquares(matchingMapSquares);
        setMapCursor(forward, matchingGameObjects);
        return matchingGameObjects.size();
    }

    private void selectMapSquares(@NotNull final Iterable<MapSquare<G, A, R>> mapSquares) {
        final MapGrid mapGrid = mapView.getMapGrid();
        mapGrid.unSelect();
        final Point p = new Point();
        for (final MapSquare<G, A, R> mapSquare : mapSquares) {
            p.x = mapSquare.getMapX();
            p.y = mapSquare.getMapY();
            mapGrid.select(p, SelectionMode.ADD);
        }
    }

    private void setMapCursor(final boolean forward, @NotNull final List<G> gameObjects) {
        if (gameObjects.isEmpty()) {
            return;
        }

        final MapCursor<G, A, R> mapCursor = mapView.getMapCursor();
        final int index;
        final G selectedGameObject = mapCursor.getGameObject();
        if (selectedGameObject == null) {
            index = 0;
        } else {
            final int selectedIndex = gameObjects.indexOf(selectedGameObject);
            if (selectedIndex == -1) {
                index = 0;
            } else if (forward) {
                index = selectedIndex + 1 < gameObjects.size() ? selectedIndex + 1 : 0;
            } else {
                index = selectedIndex > 0 ? selectedIndex - 1 : gameObjects.size() - 1;
            }
        }
        mapCursor.setGameObject(gameObjects.get(index));
    }

    public void dispose(@NotNull final MapView<G, A, R> mapView) {
        if (mapView == this.mapView) {
            dialog.setVisible(false);
        }
    }

}