| gvSIG desktop 1

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

 *

 * Copyright (C) 2005 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 org.gvsig.symbology.gui.layerproperties;

import java.awt.BorderLayout;

import java.awt.Component;

import java.awt.Dimension;

import java.awt.FlowLayout;

import java.awt.event.ActionEvent;

import java.awt.event.ActionListener;

import java.awt.event.MouseAdapter;

import java.awt.event.MouseEvent;

import java.util.ArrayList;

import java.util.Hashtable;

import java.util.Iterator;

import java.util.TreeSet;

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.JScrollPane;

import org.gvsig.gui.beans.controls.dnd.JDnDList;

import org.gvsig.gui.beans.controls.dnd.JDnDListModel;

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

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

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

import org.gvsig.symbology.fmap.labeling.ExtendedLabelingFactory;

import org.gvsig.symbology.fmap.labeling.GeneralLabelingStrategy;

import org.gvsig.symbology.fmap.labeling.LabelClassComparatorByPriority;

import org.gvsig.symbology.gui.styling.LabelClassRenderingProperties;

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

import com.iver.andami.PluginServices;

import com.iver.andami.messages.NotificationManager;

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

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

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

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

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

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

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

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

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

import com.iver.cit.gvsig.project.documents.view.legend.gui.ILabelingStrategyPanel;

import com.iver.utiles.swing.JComboBox;

public class GeneralLabeling extends JPanel implements ILabelingStrategyPanel, ActionListener {

        private static final long serialVersionUID = 8864709758980903351L;

        private static Hashtable<String, Class<? extends ILabelingMethod>> methods 

                = new Hashtable<String, Class<? extends ILabelingMethod>>();

        private static int newClassSuffix = 0;

        private JButton btnVisualization;

        private JButton btnRenameClass;

        private JButton btnSQLQuery;

        private JButton btnDelClass;

        private JButton btnAddClass;

        private JButton btnPlacement;

        private JComboBoxLabelingMethod cmbMethod;

        private GridBagLayoutPanel classesPanel;

        private JComboBox cmbClasses;

        private IPlacementConstraints placementConstraints;

        private IZoomConstraints zoomConstraints;

        private long minScaleView = -1, maxScaleView = -1;

        private boolean noEvent;

        private JCheckBox chkLabel;

        private JCheckBox chkTextOnly;

        private FLyrVect layer;

        private String[] fieldNames;

        private JDnDList lstClassPriorities;

        private JCheckBox chkDefinePriorities;

        private JScrollPane scrlPan;

        private LabelClassRenderingProperties labelClassRenderingProperties;

        private JCheckBox chkAllowLabelOverlapping;

        public GeneralLabeling() {

                initialize();

        }

        private void initialize() {

                setLayout(new BorderLayout());

                GridBagLayoutPanel left = new GridBagLayoutPanel();

                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(605, 40));

                left.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(602, 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(602, 30));

                classesPanel.addComponent(aux);

                classesPanel.setBorder(BorderFactory.createTitledBorder(

                                null, PluginServices.getText(this, "classes")));

                left.addComponent(classesPanel);

                left.addComponent(labelClassRenderingProperties = new LabelClassRenderingProperties());

                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(getChkAllowLabelOverlapping());

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

                left.addComponent(aux);        

                add(left, BorderLayout.CENTER);

                JPanel right = new JPanel(new BorderLayout(15, 15));

                aux = new JPanel(new BorderLayout(15, 15));

                aux.add(getChkDefinePriorities(), BorderLayout.NORTH);

                aux.add(getScrlClassPriorities(), BorderLayout.CENTER);

                right.add(new JBlank(10, 10), BorderLayout.NORTH);

                right.add(aux, BorderLayout.CENTER);

                right.add(new JBlank(10, 10), BorderLayout.SOUTH);

                right.add(new JBlank(10, 10), BorderLayout.EAST);

                add(right, BorderLayout.EAST);

        }

        private JCheckBox getChkAllowLabelOverlapping() {

                if (chkAllowLabelOverlapping == null) {

                        chkAllowLabelOverlapping = new JCheckBox(PluginServices.getText(this, "allow_label_overlapping"));

                }

                return chkAllowLabelOverlapping;

        }

        private Component getScrlClassPriorities() {

                if (scrlPan == null) {

                        scrlPan = new JScrollPane();

                        scrlPan.setViewportView(getLstClassPriorities());

                        scrlPan.setPreferredSize(new Dimension(180, 300));

                }

                return scrlPan;

        }

        private JCheckBox getChkDefinePriorities() {

                if (chkDefinePriorities == null) {

                        chkDefinePriorities = new JCheckBox(PluginServices.getText(this, "label_priority"));

                        chkDefinePriorities.addActionListener(this);

                        chkDefinePriorities.setName("CHK_DEFINE_PRIORITIES");

                }

                return chkDefinePriorities;

        }

        private JDnDList getLstClassPriorities() {

                if (lstClassPriorities == null) {

                        lstClassPriorities = new JDnDList();

                        lstClassPriorities.setName("CLASS_PRIORITY_LIST");

                        lstClassPriorities.addMouseListener(new MouseAdapter() {

                                @Override

                                public void mouseReleased(MouseEvent e) {

                                        JDnDListModel m = (JDnDListModel) lstClassPriorities.getModel();

                                        for (int i = 0; i < m.getSize(); i++) {

                                                ((LabelClass) m.getElementAt(i)).setPriority(i);

                                        }

                                }

                        });

                }

                return lstClassPriorities;

        }

        private void refreshControls() {

                // classes combo box

                refreshClasses();

                // fires an event from the methods combo box

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

        }

        private JButton getBtnVisualization() {

                if (btnVisualization == null) {

                        btnVisualization = new JButton(

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

                        btnVisualization.setName("BTNVISUALIZATION");

                        btnVisualization.addActionListener(this);

                }

                return btnVisualization;

        }

        private JButton getBtnPlacement() {

                if (btnPlacement == null) {

                        btnPlacement = new JButton(

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

                        btnPlacement.setName("BTNPLACEMENT");

                        btnPlacement.addActionListener(this);

                }

                return btnPlacement;

        }

        private JComboBox getCmbMethod() {

                if (cmbMethod == null) {

                        Iterator<String> it = methods.keySet().iterator();

                        ArrayList<MethodItem> aux = new ArrayList<MethodItem>();

                        while (it.hasNext()) {

                                String name = it.next();

                                Class<? extends ILabelingMethod> methodClass = (Class<? extends ILabelingMethod>) 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 JComboBoxLabelingMethod(aux.toArray(new MethodItem[0]));

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

                        cmbMethod.setName("CMBMETHOD");

                        cmbMethod.addActionListener(this);

                }

                return cmbMethod;

        }

        private JButton getBtnSQLQuery() {

                if (btnSQLQuery == null) {

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

                        btnSQLQuery.setName("BTNSQLQUERY");

                        btnSQLQuery.addActionListener(this);

                }

                return btnSQLQuery;

        }

        private JButton getBtnRenameClass() {

                if (btnRenameClass == null) {

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

                        btnRenameClass.setName("BTNRENAMECLASS");

                        btnRenameClass.addActionListener(this);

                }

                return btnRenameClass;

        }

        private JButton getBtnDelClass() {

                if (btnDelClass == null) {

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

                        btnDelClass.setName("BTNDELCLASS");

                        btnDelClass.addActionListener(this);

                }

                return btnDelClass;

        }

        private JButton getBtnAddClass() {

                if (btnAddClass == null) {

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

                        btnAddClass.setName("BTNADDCLASS");

                        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.setName("CHKLABEL");

                        chkLabel.addActionListener(this);

                }

                return chkLabel;

        }

        private JComboBox getCmbClasses() {

                if (cmbClasses == null) {

                        cmbClasses = new JComboBox();

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

                        cmbClasses.setName("CMBCLASSES");

                        cmbClasses.addActionListener(this);

                }

                return cmbClasses;

        }

        public ILabelingStrategy getLabelingStrategy() {

                ILabelingStrategy st = ExtendedLabelingFactory.

                                                createStrategy((FLayer) layer,

                                                                getMethod(),

                                                                getPlacementConstraints(),

                                                                getZoomConstraints());

                if (st instanceof GeneralLabelingStrategy) {

                        GeneralLabelingStrategy gStr = (GeneralLabelingStrategy) st;

                        gStr.setAllowOverlapping(getChkAllowLabelOverlapping().isSelected());

                        gStr.setMinScaleView(minScaleView);

                        gStr.setMaxScaleView(maxScaleView);

                }

                return st;

        }

        public void setModel(FLayer layer, ILabelingStrategy str) {

                if (layer instanceof FLyrVect) {

                        this.layer = (FLyrVect) layer;

                        FLyrVect lv = (FLyrVect) layer;

                        try {

                                fieldNames = lv.getRecordset().getFieldNames();

                                labelClassRenderingProperties.setFieldNames(fieldNames);

                                if (str instanceof GeneralLabelingStrategy) {

                                        try {

                                                GeneralLabelingStrategy gStr = (GeneralLabelingStrategy) LabelingFactory.createStrategyFromXML(str.getXMLEntity(), layer.cloneLayer());

                                                setMethod(str.getLabelingMethod());

                                                placementConstraints = str.getPlacementConstraints();

                                                getChkAllowLabelOverlapping().setSelected(gStr.isAllowingOverlap());

                                                minScaleView = gStr.getMinScaleView();

                                                maxScaleView = gStr.getMaxScaleView();

                                        } catch (ReadDriverException e) {

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

                                        } catch (Exception e) {

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

                                        }

                                }

                                refreshControls();

                                labelClassRenderingProperties.setModel(getActiveClass());

                        } catch (ReadDriverException e) {

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

                        }

                }

        }

        public static void addLabelingMethod(String name, Class<? extends ILabelingMethod> iLabelingMethodClass) {

                methods.put(name, iLabelingMethodClass);

        }

        private void setMethod(ILabelingMethod labelingMethod) {

                getCmbMethod().setSelectedItem(new MethodItem(null, labelingMethod));

        }

        private ILabelingMethod getMethod() {

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

        }

        private void refreshClasses() {

                // label classes

                getCmbClasses().removeAllItems();

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

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

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

                }

                labelClassRenderingProperties.setModel(

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

                // expressions combo box

//                refreshCmbExpressions();

                // panel priorities

                refreshPnlPriorities();

        }

        private void refreshPnlPriorities() {

                TreeSet<LabelClass> ts = new TreeSet<LabelClass>(new LabelClassComparatorByPriority());

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

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

                        ts.add(lClasses[i]);

                }

                // refresh label priority panel

                getChkDefinePriorities().setSelected(getMethod().definesPriorities());

                JDnDListModel m = new JDnDListModel();

                for (LabelClass labelClass : ts) {

                        m.addElement(labelClass);

                }

                getLstClassPriorities().setModel(m);

        }

        private LabelClass getActiveClass() {

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

        }

        private IZoomConstraints getZoomConstraints() {

                return zoomConstraints;

        }

        private IPlacementConstraints getPlacementConstraints() {

                return placementConstraints;

        }

        public void actionPerformed(ActionEvent e) {

                if (noEvent) return;

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

                if (c.equals(btnAddClass)) {

                        LabelClass newClass = new LabelClass();

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

                        getMethod().addLabelClass(newClass);

                        refreshClasses();

                        getCmbClasses().setSelectedItem(newClass);

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

                        LabelClass clazz = getActiveClass();

                        getMethod().deleteLabelClass(clazz);

                        refreshClasses();

                } 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);

                        refreshClasses();

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

                        LabelClass clazz = getActiveClass();

                        String query = clazz.getSQLQuery();

                        query = JOptionPane.showInputDialog(null,

                                          "select from "+layer.getName()+" where ",

                                          "SQL query",

                                          JOptionPane.QUESTION_MESSAGE);

                        if (!query.equals("")) {

                                clazz.setSQLQuery(query);

                        }

                } 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 && lc.getLabelExpression() != null && !getExpressions().contains(lc.getLabelExpression())) {

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

//                        }

//                        refreshCmbExpressions();

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

                        try {

                                IPlacementProperties pp = PlacementProperties.createPlacementProperties(

                                                getPlacementConstraints(),

                                                ((FLyrVect) layer).getShapeType());

                                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(btnVisualization)) {

                        LabelScaleRange lsr = new LabelScaleRange(minScaleView, maxScaleView);

                        PluginServices.getMDIManager().addWindow(lsr);

                        minScaleView = lsr.getMinScale();

                        maxScaleView = lsr.getMaxScale();

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

                        // disable components in class panel

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

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

                        refreshClasses();

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

                        getMethod().setDefinesPriorities(chkDefinePriorities.isSelected());

                        refreshPnlPriorities();

                } 

        }

        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 String getLabelingStrategyName() {

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

        }

        public Class getLabelingStrategyClass() {

                return GeneralLabelingStrategy.class;

        }

        private class JComboBoxLabelingMethod extends JComboBox {

                private static final long serialVersionUID = 5935267402200698145L;

                public JComboBoxLabelingMethod(MethodItem[] items) {

                        super(items);

                }

                @Override

                public void setSelectedItem(Object anObject) {

                        if (anObject instanceof MethodItem) {

                                MethodItem methodItem = (MethodItem) anObject;

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

                                        MethodItem aux = (MethodItem) getItemAt(i);

                                        if (aux.equals(methodItem)) {

                                                aux.method=methodItem.method;

                                        }

                                }

                                super.setSelectedItem(methodItem);

//                                ((MethodItem) super.getSelectedItem()).method = methodItem.method;

                        }

                }

        }

        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;

                }

                @Override

                public boolean equals(Object obj) {

                        if (obj != null && obj instanceof MethodItem) {

                                MethodItem methodItem = (MethodItem) obj;

                                return methodItem.method.getClass().equals(this.method.getClass());

                        }

                        return false;

                }

        }

}