| gvSIG desktop 1

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

import java.awt.BorderLayout;

import java.awt.GridLayout;

import java.awt.event.ActionEvent;

import java.awt.event.ActionListener;

import java.io.IOException;

import java.sql.Types;

import java.util.ArrayList;

import javax.swing.BorderFactory;

import javax.swing.DefaultComboBoxModel;

import javax.swing.JComponent;

import javax.swing.JPanel;

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

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.symbols.ISymbol;

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

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

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

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

import com.iver.cit.gvsig.fmap.layers.layerOperations.AlphanumericData;

import com.iver.cit.gvsig.fmap.rendering.GraduatedSymbolLegend;

import com.iver.cit.gvsig.fmap.rendering.ILegend;

import com.iver.cit.gvsig.fmap.rendering.QuantityByCategoryLegend;

import com.iver.cit.gvsig.fmap.rendering.VectorialIntervalLegend;

import com.iver.utiles.swing.JComboBox;

public class QuantityByCategory extends JPanel implements ILegendPanel, ActionListener {

        private PictureMarkerSymbol previewSymbol;

        private JPanel pnlNorth;

        private GridBagLayoutPanel pnlFields;

        private JPanel pnlColorAndSymbol;

        private JComboBox cmbColorField;

        private JComboBox cmbGraduatedSymbolField;

        private JButton btnColor;

        private JButton btnSymbol;

        private QuantityByCategoryLegend legend;

        private FLayer layer;

        public QuantityByCategory() {

                super();

                initialize();

        }

        private void initialize() {

                this.setLayout(new BorderLayout());

        this.setSize(490, 300);

        this.add(getPnlNorth(), BorderLayout.NORTH);

        }

        private JPanel getPnlNorth() {

                if (pnlNorth == null) {

                        pnlNorth = new JPanel(new GridLayout(1, 2));

                        pnlNorth.add(getPnlFields());

                        pnlNorth.add(getPnlColorAndSymbol());

                }

                return pnlNorth;

        }

        private JPanel getPnlColorAndSymbol() {

                if (pnlColorAndSymbol == null) {

                        pnlColorAndSymbol = new JPanel();

                        pnlColorAndSymbol.setBorder(

                                        BorderFactory.createTitledBorder(

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

                        pnlColorAndSymbol.add(getBtnColor());

                        pnlColorAndSymbol.add(getBtnSymbol());

                }

                return pnlColorAndSymbol;

        }

        private JButton getBtnSymbol() {

                if (btnSymbol == null) {

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

                        btnSymbol.addActionListener(this);

                }

                return btnSymbol;

        }

        private JButton getBtnColor() {

                if (btnColor == null) {

                        btnColor = new JButton(PluginServices.getText(this, "color_ramp"));

                        btnColor.addActionListener(this);

                }

                return btnColor;

        }

        private GridBagLayoutPanel getPnlFields() {

                if (pnlFields == null) {

                        pnlFields = new GridBagLayoutPanel();

                        pnlFields.setBorder(

                                        BorderFactory.createTitledBorder(

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

                        pnlFields.addComponent(

                                        PluginServices.getText(this, "color_field"), getCmbColorField());

                        pnlFields.addComponent(

                                        PluginServices.getText(this, "symbol_field") ,getCmbGraduatedField());

                }

                return pnlFields;

        }

        private JComboBox getCmbGraduatedField() {

                if (cmbGraduatedSymbolField == null) {

                        cmbGraduatedSymbolField = new JComboBox();

                        cmbGraduatedSymbolField.addActionListener(this);

                }

                return cmbGraduatedSymbolField;

        }

        private JComboBox getCmbColorField() {

                if (cmbColorField == null) {

                        cmbColorField = new JComboBox();

                        cmbColorField.addActionListener(this);

                }

                return cmbColorField;

        }

        public void setData(FLayer lyr, ILegend legend) {

                this.layer = lyr;

                fillFieldNames();

                if (legend instanceof QuantityByCategoryLegend) {

                        this.legend = (QuantityByCategoryLegend) legend;

                        getCmbColorField().setSelectedItem(

                                        this.legend.getColorRampLegend().getFieldNames());

                        getCmbGraduatedField().setSelectedItem(

                                        this.legend.getGraduatedSymbolLegend().getFieldNames());

                } else {

                        this.legend = new QuantityByCategoryLegend();

                        this.legend.setFieldNames(

                                        new String[] {

                                                        (String) getCmbColorField().getSelectedItem(),

                                                        (String) getCmbGraduatedField().getSelectedItem()

                                        });

                        try {

                                this.legend.setShapeType(((FLyrVect)lyr).getShapeType());

                        } catch (ReadDriverException e) {

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

                        }

                }

        }

        public ILegend getLegend() {

                return legend;

        }

        public String getDescription() {

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

        }

        public ISymbol getPreviewSymbol() {

                if (previewSymbol == null) {

                        try {

                                previewSymbol = new PictureMarkerSymbol(

                                                this.getClass().getClassLoader()

                                                                .getResource("images/QuantitiesByCategory.png"));

                        } catch (IOException e) {

                                return null;

                        }

                }

                return previewSymbol;

        }

        public Class getParentClass() {

                return MultipleAttributes.class;

        }

        public String getTitle() {

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

        }

        public JPanel getPanel() {

                return this;

        }

        public Class getLegendClass() {

                return QuantityByCategoryLegend.class;

        }

        public void actionPerformed(ActionEvent e) {

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

                if (c.equals(getBtnColor())) {

                        VectorialIntervalLegend colorRamp = legend.getColorRampLegend();

                        String fieldName = (String) getCmbColorField().getSelectedItem();

                        if (!fieldName.equals(colorRamp.getFieldNames())) {

                                // if classification field has changed, clear the legend

                                colorRamp.setFieldNames(new String[] {fieldName});

                                colorRamp.clear();

                        }

                        // create a new modal window to edit the color ramp legend

                        VectorialInterval legPanel = new VectorialInterval();

                        legPanel.setData(layer, legend.getColorRampLegend());

                        InnerWindow window = new InnerWindow(legPanel);

                        PluginServices.getMDIManager().addWindow(window);

                        legend.setColorRampLegend(legPanel.getLegend());

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

                        GraduatedSymbolLegend graduatedSymbol = legend.getGraduatedSymbolLegend();

                        String fieldName = (String) getCmbColorField().getSelectedItem();

                        if (!fieldName.equals(graduatedSymbol.getFieldNames())) {

                                // if classification field has changed, clear the legend

                                graduatedSymbol.setFieldNames(new String[] {fieldName});

                                graduatedSymbol.clear();

                        }

                        // create a new modal window to edit the graduated symbol legend

                        GraduatedSymbols legPanel = new GraduatedSymbols();

                        legPanel.setData(layer, legend.getGraduatedSymbolLegend());

                        InnerWindow window = new InnerWindow(legPanel);

                        PluginServices.getMDIManager().addWindow(window);

                        legend.setGraduateSymbolLegend(legPanel.getLegend());

                }

        }

        private class InnerWindow extends JPanel implements IWindow {

                private ActionListener okAction = new ActionListener() {

                        public void actionPerformed(ActionEvent e) {

                                legend = panel.getLegend();

                                PluginServices.getMDIManager().closeWindow(InnerWindow.this);

                        }

                }, cancelAction = new ActionListener() {

                        public void actionPerformed(ActionEvent e) {

                                PluginServices.getMDIManager().closeWindow(InnerWindow.this);

                        }

                };

                private ILegend legend;

                private ILegendPanel panel;

                private WindowInfo wi;

                public InnerWindow(ILegendPanel panel) {

                        this.panel = panel;

                        this.setLayout(new BorderLayout());

                        add((JComponent) panel, BorderLayout.NORTH);

                        add(new AcceptCancelPanel(okAction, cancelAction), BorderLayout.SOUTH);

                }

                public ILegend getLegend() {

                        return legend;

                }

                public WindowInfo getWindowInfo() {

                        if (wi == null) {

                                wi = new WindowInfo(WindowInfo.MODALDIALOG | WindowInfo.RESIZABLE);

                                JComponent c = (JComponent) panel;

                                wi.setWidth(c.getWidth());

                                wi.setHeight(c.getHeight());

                                wi.setTitle(panel.getDescription());

                        }

                        return wi;

                }

        }

    protected void fillFieldNames() {

        DataSource rs = null;

        ArrayList nomFields = null;

        try {

            rs = ((AlphanumericData) layer).getRecordset();

            rs.start();

            nomFields = new ArrayList();

            int type;

            for (int i = 0; i < rs.getFieldCount(); i++) {

                type = rs.getFieldType(i);

                if (type == Types.NULL) {

                    continue;

                }

                if ((type == Types.INTEGER) ||

                        (type == Types.DOUBLE) ||

                        (type == Types.FLOAT) ||

                        (type == Types.BIGINT)) {

                    nomFields.add(rs.getFieldName(i).trim());

                }

            }

            rs.stop();

        } catch (ReadDriverException e) {

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

        }

        DefaultComboBoxModel cM = new DefaultComboBoxModel(nomFields.toArray());

        cmbColorField.setModel(cM);

        cM = new DefaultComboBoxModel(nomFields.toArray());

        cmbGraduatedSymbolField.setModel(cM);

    }

    private boolean isNumericField(int fieldType) {

                switch (fieldType) {

                case Types.BIGINT:

                case Types.DECIMAL:

                case Types.DOUBLE:

                case Types.FLOAT:

                case Types.INTEGER:

                case Types.NUMERIC:

                case Types.REAL:

                case Types.SMALLINT:

                case Types.TINYINT:

                        return true;

                default:

                        return false;

                }

        }

        public boolean isSuitableFor(FLayer layer) {

                if (layer instanceof FLyrVect) {

                        try {

                                FLyrVect lyr = (FLyrVect) layer;

                                if (lyr.getShapeType() != FShape.POLYGON)

                                        return false;

                                SelectableDataSource sds;

                                sds = ((FLyrVect) layer).getRecordset();

                                String[] fNames = sds.getFieldNames();

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

                                        if (isNumericField(sds.getFieldType(i))) {

                                                return true;

                                        }

                                }

                        } catch (ReadDriverException e) {

                                return false;

                        }

                }

                return false;

        }

}