| 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 com.iver.cit.gvsig.project.documents.view.legend.gui;

import java.awt.Dimension;

import java.awt.FlowLayout;

import java.awt.Font;

import java.awt.event.ActionEvent;

import java.awt.event.ActionListener;

import java.util.ArrayList;

import java.util.Iterator;

import javax.swing.BorderFactory;

import javax.swing.ButtonGroup;

import javax.swing.JButton;

import javax.swing.JPanel;

import javax.swing.JRadioButton;

import javax.swing.JTextField;

import org.gvsig.fmap.dal.DataTypes;

import org.gvsig.fmap.dal.exception.ReadException;

import org.gvsig.fmap.dal.feature.FeatureAttributeDescriptor;

import org.gvsig.fmap.dal.feature.FeatureType;

import org.gvsig.fmap.mapcontext.layers.FLayer;

import org.gvsig.fmap.mapcontext.layers.vectorial.FLyrVect;

import org.gvsig.fmap.mapcontext.rendering.legend.styling.AttrInTableLabelingStrategy;

import org.gvsig.fmap.mapcontext.rendering.legend.styling.ILabelingStrategy;

import org.gvsig.fmap.mapcontext.rendering.symbols.SymbologyFactory;

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

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

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

import com.iver.andami.PluginServices;

import com.iver.andami.messages.NotificationManager;

import com.iver.cit.gvsig.gui.JComboBoxUnits;

import com.iver.cit.gvsig.gui.panels.ColorChooserPanel;

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

import com.iver.cit.gvsig.project.Project;

import com.iver.utiles.swing.JComboBox;

public class AttrInTableLabeling extends JPanel implements  ILabelingStrategyPanel{

        private static final long serialVersionUID = 8229927418031917075L;

        private static final String NO_FIELD_ITEM = "-- " +

                        PluginServices.getText(LabelingManager.class, "none") + " --";

        private String[] fieldNames;

        private String[] numericFieldNames;

        private String[] integerFieldNames;

        private JRadioButton rdBtnFixedHeight;

        private JRadioButton rdBtnHeightField;

        private JRadioButton rdBtnFixedColor;

        private JRadioButton rdBtnColorField;

        private JComboBox cmbTextField;

        private JComboBox cmbHeightField;

        private JComboBox cmbRotationField;

        private JComboBoxUnits cmbUnits;

        private JTextField txtHeightField;

        private FLyrVect layer;

        private JComboBoxFonts cmbFonts;

        private ColorChooserPanel colorChooser;

        private JComboBox cmbColorField;

        private JButton chooseFontBut;

        private Font labelFont = SymbologyFactory.DefaultTextFont;

        public AttrInTableLabeling() {

                initialize();

        }

        private void initialize() {

                setLayout(new FlowLayout(FlowLayout.LEFT, 10, 10));

                GridBagLayoutPanel panel = new GridBagLayoutPanel();

                GridBagLayoutPanel aux = new GridBagLayoutPanel();

                aux.addComponent(PluginServices.getText(this, "field_to_be_labeled") + ":", getCmbTextField());

                aux.addComponent(getRdBtnHeightField(), getCmbHeightField());

                aux.addComponent(getRdBtnFixedHeight(), getTxtHeightField());

                aux.addComponent(PluginServices.getText(this, "rotation_height") + ":", getCmbRotationField());

                aux.addComponent(PluginServices.getText(this, "units") + ":", getCmbUnits());

                panel.add(aux);

                aux = new GridBagLayoutPanel();

                aux.addComponent(getChooseFontBut(),new JBlank(20,20));

                GridBagLayoutPanel aux2 = new GridBagLayoutPanel();

                aux2.setBorder(BorderFactory.createTitledBorder(null,PluginServices.getText(this,"color")));

                aux2.addComponent(getRdBtnFixedColor(),colorChooser = new ColorChooserPanel(true));

                aux2.addComponent(getRdBtnColorField(),getCmbColorField());

                aux.addComponent(aux2);

                panel.add(new JBlank(20,20));

                panel.add(aux);

                add(panel);

                ButtonGroup group = new ButtonGroup();

                group.add(getRdBtnFixedHeight());

                group.add(getRdBtnHeightField());

                ButtonGroup colorGroup = new ButtonGroup();

                colorGroup.add(getRdBtnFixedColor());

                colorGroup.add(getRdBtnColorField());

                getRdBtnHeightField().setEnabled(true);

        }

        private JButton getChooseFontBut() {

                if(chooseFontBut == null){

                        chooseFontBut = new JButton(PluginServices.getText(this,"font"));

                        chooseFontBut.setPreferredSize(new Dimension(80,10));

                        chooseFontBut.addActionListener(new ActionListener(){

                                public void actionPerformed(ActionEvent e) {

                                        Font newFont;

                                        newFont = FontChooser.showDialog("Choose Font", labelFont);

                                        if (newFont == null) {

                                                return;

                                        }

                                        labelFont = newFont;

                                }

                        });

                }

                return chooseFontBut;

        }

        private JRadioButton getRdBtnFixedHeight() {

                if (rdBtnFixedHeight == null) {

                        rdBtnFixedHeight = new JRadioButton(PluginServices.getText(this, "fixed_height") + ":");

                        rdBtnFixedHeight.setSelected(false);

                        rdBtnFixedHeight.setName("RDFIXEDHEIGHT");

                }

                return rdBtnFixedHeight;

        }

        private JRadioButton getRdBtnHeightField() {

                if (rdBtnHeightField == null) {

                        rdBtnHeightField = new JRadioButton(PluginServices.getText(this, "text_height_field") + ":");

                        rdBtnHeightField.setSelected(true);

                        rdBtnHeightField.setName("RDHEIGHTFIELD");

                }

                return rdBtnHeightField;

        }

        private JRadioButton getRdBtnFixedColor() {

                if (rdBtnFixedColor == null) {

                        rdBtnFixedColor = new JRadioButton(PluginServices.getText(this, "fixed_color") + ":");

                        rdBtnFixedColor.setSelected(false);

                        rdBtnFixedColor.setName("RDFIXEDCOLOR");

                }

                return rdBtnFixedColor;

        }

        private JRadioButton getRdBtnColorField() {

                if (rdBtnColorField == null) {

                        rdBtnColorField = new JRadioButton(PluginServices.getText(this, "color_field") + ":");

                        rdBtnColorField.setSelected(true);

                        rdBtnColorField.setName("RDCOLORFIELD");

                }

                return rdBtnColorField;

        }

        private JComboBoxUnits getCmbUnits() {

                if (cmbUnits == null) {

                        cmbUnits = new JComboBoxUnits();

                        cmbUnits.setSelectedIndex(Project.getDefaultDistanceUnits());

                        cmbUnits.setName("CMBUNITS");

                }

                return cmbUnits;

        }

        private JComboBox getCmbColorField() {

                if (cmbColorField == null) {

                        cmbColorField = new JComboBox();

                        cmbColorField.setName("CMBCOLOR");

                }

                return cmbColorField;

        }

        private void refreshControls() {

                // When the attributes are in the table -----

                //      field with the text

                refreshCmbTextField();

                //      field with the rotation

                refreshCmbRotationField();

                //      field with the text height or the text size

                refreshTextHeight();

                //                the text size unit name

                refreshCmbUnits();

                //                the font for the text

                refreshFont();

//

                //                the color for the font

                refreshColorFont();

        }

        private JComboBox getCmbRotationField() {

                if (cmbRotationField == null) {

                        cmbRotationField = new JComboBox();

                        cmbRotationField.setPreferredSize(new Dimension(200, 20));

                        cmbRotationField.setName("CMBROTATIONFIELD");

                }

                return cmbRotationField;

        }

        private JComboBox getCmbHeightField() {

                if (cmbHeightField == null) {

                        cmbHeightField = new JComboBox();

                        cmbHeightField.setPreferredSize(new Dimension(200, 20));

                        cmbHeightField.setName("CMBHEIGHTFIELD");

                }

                return cmbHeightField;

        }

        private JComboBox getCmbTextField() {

                if (cmbTextField == null) {

                        cmbTextField = new JComboBox();

                        cmbTextField.setPreferredSize(new Dimension(200, 20));

                        cmbTextField.setName("CMBTEXTFIELD");

                }

                return cmbTextField;

        }

        private JTextField getTxtHeightField() {

                if (txtHeightField == null) {

                        txtHeightField = new JTextField(10);

                        txtHeightField.setText("10");

                        txtHeightField.setName("TXTHEIGHTFIELD");

                }

                return txtHeightField;

        }

        public void actionPerformed(ActionEvent e) {

                // TODO Auto-generated method stub

                throw new Error("Not yet implemented!");

        }

        public ILabelingStrategy getLabelingStrategy() {

                // user selected to define each label attributes from values

                // contained in the table for each feature row.

                double fixedSize;

                try {

                        fixedSize = Double.parseDouble(getTxtHeightField().getText());

                } catch (Exception e) {

                        fixedSize = 10;

                }

                AttrInTableLabelingStrategy strategy = new AttrInTableLabelingStrategy();

                strategy.setLayer(layer);

                if(getCmbHeightField().getItemCount() > 0) {

                        strategy.setHeightField(

                                (String) getCmbHeightField().getSelectedItem());

                }

                if(getCmbRotationField().getItemCount() > 0) {

                        if(!getCmbRotationField().getSelectedItem().equals(NO_FIELD_ITEM)) {

                                strategy.setRotationField(

                                                (String) getCmbRotationField().getSelectedItem());

                        } else {

                                strategy.setRotationField(null);

                        }

                }

                if(getCmbTextField().getItemCount() > 0) {

                        strategy.setTextField(

                                (String) getCmbTextField().getSelectedItem());

                }

                strategy.setUsesFixedSize(getRdBtnFixedHeight().isSelected());

                strategy.setFixedSize(fixedSize);

                if(getCmbUnits().getItemCount() > 0) {

                        strategy.setUnit(getCmbUnits().getSelectedUnitIndex());

                }

                strategy.setUsesFixedColor(getRdBtnFixedColor().isSelected());

                strategy.setFixedColor(colorChooser.getColor());

                if(getCmbColorField().getItemCount() > 0) {

                        strategy.setColorField((String) getCmbColorField().getSelectedItem());

                }

                strategy.setFont(labelFont);

                return strategy;

        }

        public void setModel(FLayer layer, ILabelingStrategy str) {

                this.layer = (FLyrVect) layer;

                // to allow the labeling of non-FLyrVect layers

                if (layer instanceof FLyrVect) {

                        FLyrVect lv = (FLyrVect) layer;

                        try {

                                FeatureType featureType=(FeatureType)lv.getFeatureStore().getFeatureTypes().get(0);

                                fieldNames = featureType.getFields();

                                Iterator iterator=featureType.iterator();

                                ArrayList<String> l = new ArrayList<String>();

                                ArrayList<String> lColors = new ArrayList<String>();

                                while (iterator.hasNext()) {

                                        FeatureAttributeDescriptor descriptor = (FeatureAttributeDescriptor) iterator.next();

                                        if (descriptor.getDataType() == DataTypes.DOUBLE ||

                                                        descriptor.getDataType() == DataTypes.FLOAT ||

                                                        descriptor.getDataType() == DataTypes.LONG)/*||

                                                        descriptor.getDataType().equals(FeatureAttributeDescriptor.TYPE_INT)*/){

                                                l.add(descriptor.getName());

                                        }

                                        if (descriptor.getDataType() == DataTypes.INT) {

                                                l.add(descriptor.getName());

                                                lColors.add(descriptor.getName());

                                        }

                                // detect the numeric fields

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

//                                        switch (lv.getRecordset().getFieldType(i)) {

//                                        case Types.DECIMAL:

//                                        case Types.NUMERIC:

//                                        case Types.FLOAT:

//                                        case Types.DOUBLE:

//                                        case Types.INTEGER:

//                                        case Types.REAL:

//                                        case Types.SMALLINT:

//                                        case Types.TINYINT:

//                                        case Types.BIGINT:

//                                                l.add(fieldNames[i]);

//                                                break;

//                                        }

                                }

                                numericFieldNames = l.toArray(new String[l.size()]);

                                integerFieldNames = lColors.toArray(new String[lColors.size()]);

                        } catch (ReadException e) {

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

                        }

                        refreshControls();

                }

        }

        private void refreshCmbFont(){

                if (layer.getLabelingStrategy() instanceof AttrInTableLabelingStrategy) {

                        AttrInTableLabelingStrategy aux = (AttrInTableLabelingStrategy) layer.getLabelingStrategy();

                        cmbFonts.setFont(aux.getFont());

                }

        }

        private void refreshColorFont(){

                getCmbColorField().removeAllItems();

                boolean enabled = integerFieldNames.length>0;

                getCmbColorField().setEnabled(enabled);

                getRdBtnColorField().setEnabled(enabled);

                if (!enabled) {

                        getRdBtnFixedColor().setSelected(true);

                }

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

                        getCmbColorField().addItem(integerFieldNames[i]);

                }

                if (layer.getLabelingStrategy() instanceof AttrInTableLabelingStrategy) {

                        AttrInTableLabelingStrategy aux = (AttrInTableLabelingStrategy) layer.getLabelingStrategy();

                        try {

                                getRdBtnFixedColor().setSelected(aux.usesFixedColor());

                                getRdBtnColorField().setSelected(!aux.usesFixedColor());

                                String item = aux.getColorField();

                                getCmbColorField().setSelectedItem(item);

                                colorChooser.setColor(aux.getFixedColor());

                        } catch (ReadException e) {

                                // should never happen

                                NotificationManager.addWarning(PluginServices.getText(this, "could_not_restore_color_field"), e);

                        }

                }

        }

        private void refreshFont(){

                if (layer.getLabelingStrategy() instanceof AttrInTableLabelingStrategy) {

                        AttrInTableLabelingStrategy aux = (AttrInTableLabelingStrategy) layer.getLabelingStrategy();

                        labelFont = aux.getFont();

                }

        }

        private void refreshCmbUnits() {

                getCmbUnits().setSelectedUnitIndex(Project.getDefaultDistanceUnits());

                if (layer.getLabelingStrategy() instanceof AttrInTableLabelingStrategy) {

                        AttrInTableLabelingStrategy aux = (AttrInTableLabelingStrategy) layer.getLabelingStrategy();

                        getCmbUnits().setSelectedUnitIndex(aux.getUnit());

                }

        }

        private void refreshTextHeight() {

                getCmbHeightField().removeAllItems();

                boolean enabled = numericFieldNames.length>0;

                getCmbHeightField().setEnabled(enabled);

                getRdBtnHeightField().setEnabled(enabled);

                if (!enabled) {

                        getRdBtnFixedHeight().setSelected(true);

                }

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

                        getCmbHeightField().addItem(numericFieldNames[i]);

                }

                if (layer.getLabelingStrategy() instanceof AttrInTableLabelingStrategy) {

                        AttrInTableLabelingStrategy aux = (AttrInTableLabelingStrategy) layer.getLabelingStrategy();

                        try {

                                getTxtHeightField().setText(String.valueOf(aux.getFixedSize()));

                                getRdBtnFixedHeight().setSelected(aux.usesFixedSize());

                                getRdBtnHeightField().setSelected(!aux.usesFixedSize());

                                String item = aux.getHeightField();

                                getCmbHeightField().setSelectedItem(item);

                        } catch (ReadException e) {

                                // should never happen

                                NotificationManager.addWarning(PluginServices.getText(this, "could_not_restore_text_height_field"), e);

                        }

                }

        }

        private void refreshCmbRotationField() {

                getCmbRotationField().removeAllItems();

                getCmbRotationField().addItem(NO_FIELD_ITEM);

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

                        getCmbRotationField().addItem(numericFieldNames[i]);

                }

                if (layer.getLabelingStrategy() instanceof AttrInTableLabelingStrategy) {

                        AttrInTableLabelingStrategy aux = (AttrInTableLabelingStrategy) layer.getLabelingStrategy();

                        try {

                                String item = aux.getRotationField();

                                getCmbRotationField().setSelectedItem(item != null? item : NO_FIELD_ITEM);

                        } catch (ReadException e) {

                                // should never happen

                                NotificationManager.addWarning(PluginServices.getText(this, "could_not_restore_rotation_field"), e);

                        }

                }

        }

        private void refreshCmbTextField() {

                getCmbTextField().removeAllItems();

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

                        getCmbTextField().addItem(fieldNames[i]);

                }

                if (layer.getLabelingStrategy() instanceof AttrInTableLabelingStrategy) {

                        AttrInTableLabelingStrategy aux = (AttrInTableLabelingStrategy) layer.getLabelingStrategy();

                        try {

                                String item = aux.getTextField();

                                getCmbTextField().setSelectedItem(item != null? item : NO_FIELD_ITEM);

                        } catch (ReadException e) {

                                // should never happen

                                NotificationManager.addWarning(PluginServices.getText(this, "could_not_restore_text_field"), e);

                        }

                }

        }

        public String getLabelingStrategyName() {

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

        }

        public Class getLabelingStrategyClass() {

                return AttrInTableLabeling.class;

        }

}