| gvSIG desktop 1

/*

 * Created on 01-jun-2004

 *

 */

/* gvSIG. Sistema de Informaci?n Geogr?fica de la Generalitat Valenciana

 *

 * Copyright (C) 2004 IVER T.I. and Generalitat Valenciana.

 *

 * 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., 59 Temple Place - Suite 330, Boston, MA  02111-1307,USA.

 *

 * For more information, contact:

 *

 *  Generalitat Valenciana

 *   Conselleria d'Infraestructures i Transport

 *   Av. Blasco Ib??ez, 50

 *   46010 VALENCIA

 *   SPAIN

 *

 *      +34 963862235

 *   gvsig@gva.es

 *      www.gvsig.gva.es

 *

 *    or

 *

 *   IVER T.I. S.A

 *   Salamanca 50

 *   46005 Valencia

 *   Spain

 *

 *   +34 963163400

 *   dac@iver.es

 */

package com.iver.cit.gvsig.project.documents.view.legend.gui;

import java.awt.BorderLayout;

import java.awt.Component;

import java.awt.Dimension;

import java.awt.FlowLayout;

import java.awt.Font;

import java.awt.GridLayout;

import java.awt.event.ActionEvent;

import java.awt.event.ActionListener;

import java.util.ArrayList;

import java.util.Hashtable;

import java.util.Iterator;

import javax.swing.BorderFactory;

import javax.swing.JCheckBox;

import javax.swing.JComponent;

import javax.swing.JLabel;

import javax.swing.JOptionPane;

import javax.swing.JPanel;

import javax.swing.JTextField;

import org.gvsig.gui.beans.AcceptCancelPanel;

import org.gvsig.gui.beans.swing.GridBagLayoutPanel;

import org.gvsig.gui.beans.swing.JButton;

import com.hardcode.gdbms.driver.exceptions.ReadDriverException;

import com.hardcode.gdbms.engine.data.driver.DriverException;

import com.iver.andami.PluginServices;

import com.iver.andami.messages.NotificationManager;

import com.iver.andami.ui.mdiManager.IWindow;

import com.iver.andami.ui.mdiManager.WindowInfo;

import com.iver.cit.gvsig.fmap.core.FShape;

import com.iver.cit.gvsig.fmap.core.ILabelable;

import com.iver.cit.gvsig.fmap.core.styles.IStyle;

import com.iver.cit.gvsig.fmap.core.symbols.ISymbol;

import com.iver.cit.gvsig.fmap.core.symbols.ITextSymbol;

import com.iver.cit.gvsig.fmap.layers.FLayer;

import com.iver.cit.gvsig.fmap.layers.FLyrVect;

import com.iver.cit.gvsig.fmap.rendering.styling.ILabelingMethod;

import com.iver.cit.gvsig.fmap.rendering.styling.ILabelingStrategy;

import com.iver.cit.gvsig.fmap.rendering.styling.IPlacementConstraints;

import com.iver.cit.gvsig.fmap.rendering.styling.IZoomConstraints;

import com.iver.cit.gvsig.fmap.rendering.styling.LabelClass;

import com.iver.cit.gvsig.fmap.rendering.styling.LabelingFactory;

import com.iver.cit.gvsig.gui.styling.LabelStyleSelector;

import com.iver.cit.gvsig.gui.styling.SelectorFilter;

import com.iver.cit.gvsig.gui.styling.SymbolPreviewer;

import com.iver.cit.gvsig.gui.styling.SymbolSelector;

import com.iver.cit.gvsig.gui.utils.FontChooser;

import com.iver.utiles.swing.JComboBox;

public class LabelingManager extends AbstractThemeManagerPage implements ActionListener {

        private static int newClassSuffix = 0;

        private static Hashtable methods = new Hashtable();

        private static final Font DEFAULT_FONT =

                  new Font("SansSerif", Font.PLAIN, 9);

        private JCheckBox chkApplyLabels;

        private JComboBox cmbMethod;

        private GridBagLayoutPanel classesPanel;

        private JComboBox cmbClasses;

        private JCheckBox chkLabel;

        private JButton btnRenameClass;

        private JButton btnSQLQuery;

        private JButton btnDelClass;

        private JButton btnAddClass;

        private JComboBox cmbExpressions;

        private JButton btnEditExpression;

        private SymbolPreviewer previewer;

        private JButton btnFont;

        protected Font labelFont;

        private JButton btnPlacement;

        private JButton btnLabelStyles;

        private JCheckBox chkTextOnly;

        private ILabelable layer;

        private JButton btnVisualization;

        private ArrayList expressions;

        private GridBagLayoutPanel content;

        private IPlacementConstraints placementConstraints;

        private IZoomConstraints zoomConstraints;

        private JButton btnSymbol;

        private class MethodItem {

                private String name;

                private ILabelingMethod method;

                private MethodItem(String name, ILabelingMethod method) {

                        this.name = name;

                        this.method = method;

                }

                public String toString() {

                        return name;

                }

        }

        private class ExprEditorPanel extends JPanel implements IWindow {

                private ActionListener action = new ActionListener() {

                        public void actionPerformed(ActionEvent e) {

                                if ("OK".equals(e.getActionCommand())) {

                                        if (validateExpr())

                                                PluginServices.getMDIManager().closeWindow(

                                                                ExprEditorPanel.this);

                                        else

                                                JOptionPane.showMessageDialog(ExprEditorPanel.this,

                                                                PluginServices.getText(this, "syntax_error"),

                                                                PluginServices.getText(this, "error"),

                                                                JOptionPane.ERROR_MESSAGE);

                                } else {

                                        getTxtExpression().setText(lastExtression);

                                        PluginServices.getMDIManager().closeWindow(

                                                        ExprEditorPanel.this);

                                }

                        }

                };

                private JTextField txtExpression;

                private String lastExtression;

                public ExprEditorPanel(String currentExpression) {

                        super();

                        initialize();

                        lastExtression = currentExpression;

                        getTxtExpression().setText(currentExpression);

                }

                private void initialize() {

                        GridBagLayoutPanel content = new GridBagLayoutPanel();

                        content.addComponent(new JLabel(" "+PluginServices.getText(this, "label_expression_help")));

                        content.addComponent(

                                        " "+PluginServices.getText(this, "expression")+":",

                                        getTxtExpression());

                        setLayout(new BorderLayout(5, 5));

                        add(content, BorderLayout.CENTER);

                        add(new AcceptCancelPanel(action, action), BorderLayout.SOUTH);

                        setSize(new Dimension(300, 80));

                }

                private JTextField getTxtExpression() {

                        if (txtExpression == null) {

                                txtExpression = new JTextField(25);

                        }

                        return txtExpression;

                }

                public String getExpression() {

                        return getTxtExpression().getText();

                }

                public WindowInfo getWindowInfo() {

                        WindowInfo viewInfo = new WindowInfo(WindowInfo.MODALDIALOG);

                        viewInfo.setWidth(getWidth());

                        viewInfo.setHeight(getHeight());

                        viewInfo.setTitle(PluginServices.getText(this,"label_expression_editor"));

                        return viewInfo;

                }

                private boolean validateExpr() {

                        // TODO : implement it

                        return true;

                }

        }

        public LabelingManager() {

                super();

                initialize();

        }

        private void initialize() {

                content = new GridBagLayoutPanel();

                content.addComponent(getChkApplyLabels());

                JPanel aux = new JPanel(new FlowLayout(FlowLayout.LEADING, 5,0));

                aux.add(new JLabel(PluginServices.getText(this, "method")+":"));

                aux.add(getCmbMethod());

                aux.setPreferredSize(new Dimension(550, 40));

                content.addComponent(aux);

                classesPanel = new GridBagLayoutPanel();

                aux = new JPanel(new FlowLayout(FlowLayout.LEADING, 5,0));

                aux.add(new JLabel(PluginServices.getText(this, "class")+":"));

                aux.add(getCmbClasses());

                aux.add(getChkLabelFeatures());

                aux.setPreferredSize(new Dimension(530, 40));

                classesPanel.addComponent(aux);

                aux = new JPanel(new FlowLayout(FlowLayout.LEADING, 5, 0));

                aux.add(getBtnAddClass());

                aux.add(getBtnDelClass());

                aux.add(getBtnRenameClass());

                aux.add(getBtnSQLQuery());

                aux.setPreferredSize(new Dimension(530, 30));

                classesPanel.addComponent(aux);

                classesPanel.setBorder(BorderFactory.createTitledBorder(null, PluginServices.getText(this, "classes")));

                content.addComponent(classesPanel);

                aux = new JPanel(new FlowLayout(FlowLayout.LEADING, 5, 0));

                aux.setBorder(BorderFactory.createTitledBorder(null, PluginServices.getText(this, "text_string")));

                aux.add(new JLabel(PluginServices.getText(this, "expression")+":"));

                aux.add(getCmbExpressions());

                aux.add(getBtnEditExpression());

                aux.setPreferredSize(new Dimension(540, 60));

                content.addComponent(aux);

                aux = new JPanel(new FlowLayout(FlowLayout.LEADING, 5, 0));

                aux.setBorder(BorderFactory.createTitledBorder(null, PluginServices.getText(this, "text_symbol")));

                aux.add(getSymbolPreviewer());

                JPanel aux2 = new JPanel();

                aux2.add(getBtnFont());

                aux2.add(getBtnSymbol());

                aux2.setLayout(new GridLayout(aux2.getComponentCount(), 1));

                aux.add(aux2);

                content.addComponent(aux);

                aux = new JPanel(new FlowLayout(FlowLayout.LEADING, 5, 0));

                aux.setBorder(BorderFactory.createTitledBorder(null, PluginServices.getText(this, "options")));

                aux.add(getBtnVisualization());

                aux.add(getBtnPlacement());

                aux.add(getBtnLabelStyles());

                aux.setPreferredSize(new Dimension(540, 60));

                content.addComponent(aux);

                add(content);

        }

        private Component getBtnSymbol() {

                if (btnSymbol == null) {

                        btnSymbol = new JButton(PluginServices.getText(this, "symbol"));

                        btnSymbol.addActionListener(this);

                }

                return btnSymbol;

        }

        private JCheckBox getChkApplyLabels() {

                if (chkApplyLabels == null) {

                        chkApplyLabels = new JCheckBox(PluginServices.getText(this, "enable_labelling"));

                        chkApplyLabels.addActionListener(this);

                }

                return chkApplyLabels;

        }

        public static void addLabelingMethod(String name, Class iLabelingMethodClass) {

                methods.put(name, iLabelingMethodClass);

        }

        private JButton getBtnVisualization() {

                if (btnVisualization == null) {

                        btnVisualization = new JButton(PluginServices.getText(this, "visualization")+"...");

                        btnVisualization.addActionListener(this);

                }

                return btnVisualization;

        }

        private JButton getBtnLabelStyles() {

                if (btnLabelStyles == null) {

                        btnLabelStyles = new JButton(PluginServices.getText(this, "label_styles")+"...");

                        btnLabelStyles.addActionListener(this);

                }

                return btnLabelStyles;

        }

        private JButton getBtnPlacement() {

                if (btnPlacement == null) {

                        btnPlacement = new JButton(PluginServices.getText(this, "placement")+"...");

                        btnPlacement.addActionListener(this);

                }

                return btnPlacement;

        }

        private JButton getBtnFont() {

                if (btnFont == null) {

                        btnFont = new JButton();

                        btnFont.setText(PluginServices.getText(this,"fuente")+"...");

                        btnFont.addActionListener(this);

                }

                return btnFont;

        }

        private SymbolPreviewer getSymbolPreviewer() {

                if (previewer == null) {

                        previewer = new SymbolPreviewer();

                        previewer.setPreferredSize(new Dimension(420, 65));

                }

                return previewer;

        }

        private JComboBox getCmbMethod() {

                if (cmbMethod == null) {

                        Iterator it = methods.keySet().iterator();

                        ArrayList aux = new ArrayList();

                        while (it.hasNext()) {

                                String name = (String) it.next();

                                Class methodClass = (Class) methods.get(name);

                            ILabelingMethod method;

                                try {

                                        method = (ILabelingMethod) methodClass.newInstance();

                                        MethodItem newItem = new MethodItem(name, method);

                                        aux.add(newItem);

                                } catch (InstantiationException e) {

                                        NotificationManager.addError("Trying to instantiate an interface" +

                                                        " or abstract class + "+methodClass.getName(), e);

                                } catch (IllegalAccessException e) {

                                        NotificationManager.addError("IllegalAccessException: does " +

                                                        methodClass.getName()        + " class have an anonymous" +

                                                                        " constructor?", e);

                                }

                        }

                        cmbMethod = new JComboBox(aux.toArray());

                        cmbMethod.setSize(new Dimension(300, 22));

                        cmbMethod.addActionListener(this);

                }

                return cmbMethod;

        }

        private JButton getBtnEditExpression() {

                if (btnEditExpression == null) {

                        btnEditExpression = new JButton(

                                        PluginServices.getText(this, "edit_expression")+"...");

                        btnEditExpression.addActionListener(this);

                }

                return btnEditExpression;

        }

        private JComboBox getCmbExpressions() {

                if (cmbExpressions == null) {

                        cmbExpressions = new JComboBox();

                        cmbExpressions.setPreferredSize(new Dimension(150, 20));

                        cmbExpressions.addActionListener(this);

                }

                return cmbExpressions;

        }

        private JButton getBtnSQLQuery() {

                if (btnSQLQuery == null) {

                        btnSQLQuery = new JButton(PluginServices.getText(this, "SQL_query"));

                        btnSQLQuery.addActionListener(this);

                }

                return btnSQLQuery;

        }

        private JButton getBtnRenameClass() {

                if (btnRenameClass == null) {

                        btnRenameClass = new JButton(PluginServices.getText(this, "remane_class"));

                        btnRenameClass.addActionListener(this);

                }

                return btnRenameClass;

        }

        private JButton getBtnDelClass() {

                if (btnDelClass == null) {

                        btnDelClass = new JButton(PluginServices.getText(this, "delete_class"));

                        btnDelClass.addActionListener(this);

                }

                return btnDelClass;

        }

        private JButton getBtnAddClass() {

                if (btnAddClass == null) {

                        btnAddClass = new JButton(PluginServices.getText(this, "add_class"));

                        btnAddClass.addActionListener(this);

                }

                return btnAddClass;

        }

        private JCheckBox getChkLabelFeatures() {

                if (chkLabel == null) {

                        chkLabel = new JCheckBox();

                        chkLabel.setText(PluginServices.getText(this, "label_features_in_this_class"));

                        chkLabel.addActionListener(this);

                }

                return chkLabel;

        }

        private JComboBox getCmbClasses() {

                if (cmbClasses == null) {

                        cmbClasses = new JComboBox();

                        cmbClasses.setPreferredSize(new Dimension(150, 20));

                        cmbClasses.addActionListener(this);

                }

                return cmbClasses;

        }

        private void setComponentEnabled(Component c, boolean b) {

                if (c instanceof JComponent) {

                        JComponent c1 = (JComponent) c;

                        for (int i = 0; i < c1.getComponentCount(); i++) {

                                setComponentEnabled(c1.getComponent(i), b);

                        }

                }

                c.setEnabled(b);

        }

        public void setModel(FLayer layer) throws IllegalArgumentException {

                if (layer instanceof ILabelable) {

                        this.layer = (ILabelable) layer;

                        // get the labeling strategy

                        ILabelingStrategy strategy = this.layer.getLabelingStrategy();

                        if (strategy == null) {

                                // set a new (simplest) one

                                try {

                                        strategy = LabelingFactory.createDefaultStrategy(layer);

                                } catch (ReadDriverException e) {

                                        strategy = null;

                                }

                        }

                        if (strategy != null)

                                setMethod(strategy.getLabelingMethod());

                        setComponentEnabled(this, true);

                        refreshControls();

                        actionPerformed(new ActionEvent(chkApplyLabels, 0, null));

                } else {

                        setComponentEnabled(this, false);

                }

        }

        private void setMethod(ILabelingMethod labelingMethod) {

                ArrayList aux = new ArrayList();

                for (int i = 0; i < getCmbMethod().getItemCount(); i++) {

                        aux.add(((MethodItem) getCmbMethod().getItemAt(i)).method);

                }

                int i = aux.indexOf(labelingMethod);

                if (i == -1)

                        i = 0; // index of the simplest method: EquallyLabeled (must be installed)

                getCmbMethod().setSelectedIndex(i);

        }

        private ILabelingMethod getMethod() {

                return ((MethodItem) getCmbMethod().getSelectedItem()).method;

        }

        private void refreshControls() {

                if (layer == null) return;

                // enables labeling

                getChkApplyLabels().setSelected(layer.isLabeled());

                // fires an event from the method's combobox

                actionPerformed(new ActionEvent(getCmbMethod(), 0, null));

                // expressions combobox

                refreshCmbExpressions();

        }

        private void refreshCmbExpressions() {

                ArrayList exp = getExpressions();

                getCmbExpressions().removeAllItems();

                for (int i = 0; i < exp.size(); i++)

                        getCmbExpressions().addItem(exp.get(i));

                LabelClass lc = (LabelClass) getCmbClasses().getSelectedItem();

                if (lc != null)

                        getCmbExpressions().setSelectedItem(lc.getLabelExpression());

        }

        private ArrayList getExpressions() {

                if (expressions == null) {

                        expressions = new ArrayList();

                        try {

                                String[] n =  ((FLyrVect) layer).getRecordset().getFieldNames();

                                for (int i = 0; i < n.length; i++) {

                                        expressions.add("[ "+n[i]+" ]");

                                }

                        } catch (Exception e) {

                                NotificationManager.addError(

                                                PluginServices.getText(

                                                                this, "could_not_retreive_layer_names"), e);

                        }

                }

                return expressions;

        }

        public void actionPerformed(ActionEvent e) {

                JComponent c = (JComponent)e.getSource();

                if (c.equals(chkApplyLabels)) {

                        boolean b = chkApplyLabels.isSelected();

                        // enables/disables all components

                        for (int i = 0; i < content.getComponentCount(); i++) {

                                Component c1 = content.getComponent(i);

                                if (!c1.equals(c))

                                        setComponentEnabled(c1, b);

                        }

                        if (b) actionPerformed(new ActionEvent(getCmbMethod(), 0, null));

                } else if (c.equals(btnAddClass)) {

                        LabelClass newClass = new LabelClass();

                        newClass.setName(PluginServices.getText(this, "labeling")+String.valueOf(++newClassSuffix));

                        getMethod().addLabelClass(newClass);

                        refreshCmbClasses();

                        getCmbClasses().setSelectedItem(newClass);

                } else if (c.equals(btnDelClass)) {

                        LabelClass clazz = getActiveClass();

                        getMethod().deleteLabelClass(clazz);

                        refreshCmbClasses();

                } else if (c.equals(btnEditExpression)) {

                        ExprEditorPanel ep = new ExprEditorPanel((String) getCmbExpressions().

                                        getSelectedItem());

                        PluginServices.getMDIManager().addWindow( ep );

                        getCmbExpressions().addItem(ep.getExpression());

                        getCmbExpressions().setSelectedItem(ep.getExpression());

                } else if (c.equals(btnFont)) {

                        Font newFont = FontChooser.showDialog(PluginServices.getText(this, "font"), labelFont);

                if (newFont != null) {

                    labelFont = newFont;

                }

                } else if (c.equals(btnLabelStyles)) {

                        LabelClass clazz = getActiveClass();

                        // here open symbol selector

                        LabelStyleSelector stySel = new LabelStyleSelector(

                                        clazz.getLabelSymbol(),

                                        FShape.TEXT );

                        PluginServices.getMDIManager().addWindow(stySel);

                        IStyle sty = (IStyle) stySel.getSelectedObject();

//                        if (sty != null)

//                                // gather the symbol and apply to the class

//                                clazz.setSymbol(sty);

                } else if (c.equals(btnPlacement)) {

                        try {

                                PlacementProperties pp = new PlacementProperties(

                                                (FLyrVect) layer,

                                                getPlacementConstraints());

                                PluginServices.getMDIManager().addWindow(pp);

                                placementConstraints = pp.getPlacementConstraints();

                        } catch (ClassCastException ccEx) {

                                NotificationManager.addError("Placement constraints not prepared for:"

                                                +layer.getClass().getName(),

                                                ccEx);

                        } catch (ReadDriverException dEx) {

                                NotificationManager.addWarning("Should be unreachable code", dEx);

                                NotificationManager.addError(PluginServices.getText(this, "usupported_layer_type"), dEx);

                        }

                } else if (c.equals(btnRenameClass)) {

                        LabelClass clazz = getActiveClass();

                        String newName = JOptionPane.showInputDialog(

                                        PluginServices.getText(this, "enter_new_name"));

                        if (newName != null)

                                getMethod().renameLabelClass(clazz, newName);

                        refreshCmbClasses();

                } else if (c.equals(btnSQLQuery)) {

                } else if (c.equals(btnVisualization)) {

                } else if (c.equals(chkLabel)) {

                        LabelClass lc = (LabelClass) getCmbClasses().getSelectedItem();

                        if (lc == null)

                                lc = getMethod().getDefaultLabelClass();

                        lc.setVisible(chkLabel.isSelected());

                } else if (c.equals(chkTextOnly)) {

                } else if (c.equals(cmbClasses)) {

                        // refresh expressions

                        LabelClass lc = (LabelClass) cmbClasses.getSelectedItem();

                        if (lc != null && !getExpressions().contains(lc.getLabelExpression())) {

                                getExpressions().add(0, lc.getLabelExpression());

                        }

                        refreshCmbExpressions();

                } else if (c.equals(cmbExpressions)) {

                        LabelClass lc = (LabelClass) getCmbClasses().getSelectedItem();

                        if (lc == null)

                                lc = getMethod().getDefaultLabelClass();

                        lc.setLabelExpression((String) cmbExpressions.getSelectedItem());

                } else if (c.equals(cmbMethod)) {

                        // disable components in class panel

                        // multiple class or not enables or disables the class panel

                        setComponentEnabled(classesPanel, getMethod().allowsMultipleClass());

                        refreshCmbClasses();

                } else if (c.equals(btnSymbol)) {

                        LabelClass clazz = getActiveClass();

                        // here open symbol selector

                        SymbolSelector symSel = new SymbolSelector(

                                        clazz.getLabelSymbol(),

                                        FShape.TEXT,

                                        new SelectorFilter() {

                                                public boolean accepts(Object obj) {

                                                        if (obj instanceof ISymbol) {

                                                                ISymbol sym = (ISymbol) obj;

                                                                return sym.getSymbolType() == FShape.TEXT;

                                                        }

                                                        return false;

                                                }

                                        });

                        PluginServices.getMDIManager().addWindow(symSel);

                        ISymbol sym = (ISymbol) symSel.getSelectedObject();

                        if (sym != null)

                                // gather the symbol and apply to the class

                                clazz.setSymbol((ITextSymbol) symSel.getSelectedObject());

                }

        }

        private void refreshCmbClasses() {

                // label classes

                getCmbClasses().removeAllItems();

                LabelClass[] lClasses = getMethod().getLabelClasses();

                for (int i = 0; i < lClasses.length; i++) {

                        getCmbClasses().addItem(lClasses[i]);

                }

                refreshCmbExpressions();

        }

        private LabelClass getActiveClass() {

                return (LabelClass) getCmbClasses().getSelectedItem();

        }

        public void acceptAction() {

        }

        public void cancelAction() {

        }

        public void applyAction() {

                if (layer != null) { // in other case the layer is not labelable

                        layer.setIsLabeled(getChkApplyLabels().isSelected());

                        try {

                                layer.setLabelingStrategy(

                                                LabelingFactory.

                                                createStrategy((FLayer) layer,

                                                                getMethod(),

                                                                getPlacementConstraints(),

                                                                getZoomConstraints()));

                        } catch (ReadDriverException e) {

                                // TODO Auto-generated catch block

                                NotificationManager.addError(PluginServices.getText(this, "XXXX"), e);

                        }

                }

        }

        private IZoomConstraints getZoomConstraints() {

                return zoomConstraints;

        }

        private IPlacementConstraints getPlacementConstraints() {

                return placementConstraints;

        }

        public String getName() {

                return PluginServices.getText(this,"Etiquetados");

        }

}