Application: gvSIG desktop

/* 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.FlowLayout;

import java.awt.event.ActionEvent;

import java.awt.event.ActionListener;

import java.util.ArrayList;

import java.util.Iterator;

import java.util.Random;

import javax.swing.DefaultComboBoxModel;

import javax.swing.ImageIcon;

import javax.swing.JCheckBox;

import javax.swing.JComboBox;

import javax.swing.JLabel;

import javax.swing.JOptionPane;

import javax.swing.JPanel;

import org.apache.log4j.Logger;

import org.gvsig.data.ReadException;

import org.gvsig.data.vectorial.Feature;

import org.gvsig.data.vectorial.FeatureAttributeDescriptor;

import org.gvsig.data.vectorial.FeatureStore;

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

import org.gvsig.fmap.mapcontext.layers.operations.AlphanumericData;

import org.gvsig.fmap.mapcontext.layers.operations.ClassifiableVectorial;

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

import org.gvsig.fmap.mapcontext.rendering.legend.AbstractClassifiedVectorLegend;

import org.gvsig.fmap.mapcontext.rendering.legend.ILegend;

import org.gvsig.fmap.mapcontext.rendering.legend.LegendFactory;

import org.gvsig.fmap.mapcontext.rendering.legend.NullUniqueValue;

import org.gvsig.fmap.mapcontext.rendering.legend.NullValue;

import org.gvsig.fmap.mapcontext.rendering.legend.VectorialUniqueValueLegend;

import org.gvsig.fmap.mapcontext.rendering.legend.ZSort;

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

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

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

import org.gvsig.raster.datastruct.ColorItem;

import com.iver.andami.PluginServices;

import com.iver.andami.messages.NotificationManager;

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

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

/**

 * DOCUMENT ME!

 *

 * @author fjp To change the template for this generated type comment go to

 *         Window>Preferences>Java>Code Generation>Code and

 *         Comments

 */

public class VectorialUniqueValue extends JPanel implements ILegendPanel, ActionListener{

    private static Logger logger = Logger.getLogger(VectorialUniqueValue.class.getName());

    protected VectorialUniqueValueLegend theLegend;

    private ClassifiableVectorial layer;

    private SymbolTable symbolTable;

    protected JComboBox cmbFields;

    private JButton btnRemoveAll;

    private JButton btnRemove;

    private JCheckBox chbUseDefault = null;

    private VectorialUniqueValueLegend auxLegend;

        private JPanel pnlCenter;

        private ZSort zSort;

        private JButton btnOpenSymbolLevelsEditor;

        private JComboBoxColorScheme cmbColorScheme;

    /**

     *

     */

    public VectorialUniqueValue() {

        super();

        initComponents();

    }

    /**

     * DOCUMENT ME!

     */

    protected void initComponents() {

        /* JLabel label = new JLabel();

           label.setIcon(new javax.swing.ImageIcon(Abrir.class.getClassLoader()

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

           limagen[1] = new JLabel();

           limagen[1] = label; */

        JPanel pnlButtons = new JPanel();

        JButton btnAddAll = new JButton(PluginServices.getText(this,

                    "Anadir_todos"));

        btnAddAll.setActionCommand("ADD_ALL_VALUES");

        btnAddAll.addActionListener(this);

        pnlButtons.add(btnAddAll);

        JButton btnAdd = new JButton(PluginServices.getText(this, "Anadir"));

        btnAdd.setActionCommand("ADD_VALUE");

        btnAdd.addActionListener(this);

        pnlButtons.add(btnAdd);

        btnRemoveAll = new JButton(PluginServices.getText(this, "Quitar_todos"));

        btnRemoveAll.setActionCommand("REMOVE_ALL");

        btnRemoveAll.addActionListener(this);

        pnlButtons.add(btnRemoveAll);

        btnRemove = new JButton(PluginServices.getText(this, "Quitar"));

        btnRemove.setActionCommand("REMOVE");

        btnRemove.addActionListener(this);

        pnlButtons.add(btnRemove);

        btnOpenSymbolLevelsEditor = new JButton(PluginServices.getText(this, "symbol_levels"));

        btnOpenSymbolLevelsEditor.addActionListener(this);

        btnOpenSymbolLevelsEditor.setActionCommand("OPEN_SYMBOL_LEVEL_EDITOR");

        pnlButtons.add(btnOpenSymbolLevelsEditor);

                pnlCenter = new JPanel();

        pnlCenter.setLayout(new BorderLayout());

        cmbFields = new JComboBox();

        cmbFields.setActionCommand("FIELD_SELECTED");

        cmbFields.addActionListener(this);

        cmbFields.setVisible(true);

        JPanel pnlNorth = new JPanel(new FlowLayout(FlowLayout.LEFT, 20, 0));

        JPanel pAux1 = new JPanel();

        JLabel lblFieldClassification = new JLabel(PluginServices.getText(

                    this, "Campo_de_clasificacion"));

        pAux1.add(lblFieldClassification);

        pAux1.add(cmbFields);

        pAux1.add(getChbUseDefault(), null);

        pnlNorth.add(pAux1);

        pAux1 = new JPanel();

        pAux1.add(new JLabel(PluginServices.getText(this, "color_scheme")+":"));

        cmbColorScheme = new JComboBoxColorScheme(false);

        pAux1.add(cmbColorScheme);

        pnlNorth.add(pAux1);

        this.setLayout(new BorderLayout());

        this.add(pnlNorth, BorderLayout.NORTH);

        this.add(pnlCenter, BorderLayout.CENTER);

        this.add(pnlButtons, BorderLayout.SOUTH);

    }

    /**

     * DOCUMENT ME!

     */

    private void fillTableValues() {

        FeatureStore elRs;

        try {

            elRs = ((FLyrVect) layer).getFeatureStore();

//            logger.debug("elRs.start()");

//            elRs.start();

            int idField = -1;

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

            if (fieldName==null) {

                    JOptionPane.showMessageDialog((Component)PluginServices.getMainFrame(),PluginServices.getText(this,"no_hay_campo_seleccionado"));

                    return;

            }

//            idField = elRs.getFieldIndexByName(fieldName);

            theLegend.setClassifyingFieldNames(new String[] {fieldName});

            //long numReg = elRs.getRowCount();

//            if (idField == -1) {

//                NotificationManager.addWarning(

//                                PluginServices.getText(this, "unrecognized_field_name")+" " + fieldName, null);

//

//                return;

//            }

            symbolTable.removeAllItems();

            int numSymbols = 0;

            ISymbol theSymbol = null;

            //auxLegend=(VectorialUniqueValueLegend)m_lyr.getLegend();

            auxLegend = LegendFactory.createVectorialUniqueValueLegend(layer.getShapeType());

            Object clave=null;

            //Object resul;

            if (chbUseDefault.isSelected()) {

                auxLegend.getDefaultSymbol().setDescription("Default");

                auxLegend.addSymbol(new NullUniqueValue(),

                    auxLegend.getDefaultSymbol());

            }

            ColorItem[] colorScheme = cmbColorScheme.getSelectedColors();

            Random rand = new Random(System.currentTimeMillis());

            Iterator iterator=elRs.getDataCollection().iterator();

            while (iterator.hasNext()) {

                                Feature feature = (Feature) iterator.next();

                                feature.get(fieldName);

//                        }

//            for (int j = 0; j < elRs.getRowCount(); j++) {

//                clave = elRs.getFieldValue(j, idField);

                if (clave instanceof NullValue) {

                    continue;

                }

                ////Comprobar que no esta repetido y no hace falta introducir en el hashtable el campo junto con el simbolo.

                if (auxLegend.getSymbolByValue(clave) == null) {

                    //si no esta creado el simbolo se crea

                    // jaume (moved to ISymbol); theSymbol = new FSymbol(layer.getShapeType());

                        theSymbol = SymbologyFactory.

                                createDefaultSymbolByShapeType(layer.getShapeType(),

                                                colorScheme[rand.nextInt(colorScheme.length)].getColor());

                    theSymbol.setDescription(clave.toString());

                    auxLegend.addSymbol(clave, theSymbol);

                    numSymbols++;

                    if (numSymbols == 100) {

                        int resp = JOptionPane.showConfirmDialog(this,

                                PluginServices.getText(this,

                                    "mas_de_100_simbolos"),

                                PluginServices.getText(this, "quiere_continuar"),

                                JOptionPane.YES_NO_OPTION,

                                JOptionPane.WARNING_MESSAGE);

                        if ((resp == JOptionPane.NO_OPTION) ||

                                (resp == JOptionPane.DEFAULT_OPTION)) {

                            return;

                        }

                    }

                }

            } // for

            symbolTable.fillTableFromSymbolList(auxLegend.getSymbols(),

                auxLegend.getValues(), auxLegend.getDescriptions());

//            elRs.stop();

        } catch (ReadException e) {

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

        }

        btnRemoveAll.setEnabled(true);

        btnRemove.setEnabled(true);

        //m_bCacheDirty = false;

    }

    /**

     * A partir de los registros de la tabla, regenera el FRenderer. (No solo

     * el symbolList, si no tambi�n el arrayKeys y el defaultRenderer

     */

    private void fillSymbolListFromTable() {

        Object clave=null;

        ISymbol theSymbol;

        // Borramos las anteriores listas:

        boolean bRestoValores = false; // PONERLO EN UN CHECKBOX

        int hasta;

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

        theLegend.setClassifyingFieldNames(new String[] {fieldName} );

        /////////////////////////////////////////PEPE

        FLyrVect m = (FLyrVect) layer;

        try {

                String fieldType=m.getFeatureStore().getDefaultFeatureType().getByOrder((int)cmbFields.getSelectedIndex()).getName();

//                int fieldType = m.getSource().getRecordset().getFieldType((int)cmbFields.getSelectedIndex());

                theLegend.setClassifyingFieldTypes(new String[] {fieldType});

        } catch (ReadException e) {

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

        }

        /////////////////////////////////////////PEPE

        if (bRestoValores) {

            hasta = symbolTable.getRowCount() - 1;

        } else {

            hasta = symbolTable.getRowCount();

        }

        for (int row = 0; row < symbolTable.getRowCount(); row++) {

            clave = (Object) symbolTable.getFieldValue(row, 1);

            theSymbol = (ISymbol) symbolTable.getFieldValue(row, 0);

            theSymbol.setDescription((String) symbolTable.getFieldValue(row, 2));

            theLegend.addSymbol(clave, theSymbol);

        }

        if (bRestoValores) {

            theSymbol = (ISymbol) symbolTable.getFieldValue(hasta, 0);

            theLegend.setDefaultSymbol(theSymbol);

        }

    }

    /**

     * DOCUMENT ME!

     */

    private void fillFieldNames() {

        FeatureStore rs;

        try {

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

//            logger.debug("rs.start()");

//            rs.start();

            ArrayList names=new ArrayList();

            Iterator iterator=rs.getDefaultFeatureType().iterator();

            while (iterator.hasNext()) {

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

                                names.add(descriptor.getDataType());

                        }

//            String[] nomFields = new String[rs.getFieldCount()];

//

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

//                nomFields[i] = rs.getFieldName(i).trim();

//            }

//

//            rs.stop();

            DefaultComboBoxModel cM = new DefaultComboBoxModel((String[])names.toArray(new String[0]));

            cmbFields.setModel(cM);

            // fieldsListValor.setSelectedIndex(0);

        } catch (ReadException e) {

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

        }

    }

    public void setData(FLayer layer, ILegend legend) {

            this.layer = (ClassifiableVectorial) layer;

              int shapeType = 0;

              try {

                      shapeType = this.layer.getShapeType();

              } catch (ReadException e) {

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

                }

              if (symbolTable != null)

                      pnlCenter.remove(symbolTable);

              symbolTable = new SymbolTable(this, SymbolTable.VALUES_TYPE, shapeType);

              pnlCenter.add(symbolTable, BorderLayout.CENTER);

        fillFieldNames();

        symbolTable.removeAllItems();

        if (VectorialUniqueValueLegend.class.equals(legend.getClass())) {

            theLegend = (VectorialUniqueValueLegend) legend;

            getChbUseDefault().setSelected(theLegend.isUseDefaultSymbol());

            cmbFields.getModel().setSelectedItem(theLegend.getClassifyingFieldNames()[0]);

            symbolTable.fillTableFromSymbolList(theLegend.getSymbols(),

                                                                                    theLegend.getValues(),

                                                                                    theLegend.getDescriptions());

            zSort = theLegend.getZSort();

        } else {

            theLegend = new VectorialUniqueValueLegend(shapeType);

        }

    }

    /* (non-Javadoc)

     * @see com.iver.cit.gvsig.gui.legendmanager.panels.ILegendPanel#getLegend()

     */

    public ILegend getLegend() {

        fillSymbolListFromTable();

        if (auxLegend != null) {

                // your settings that are not the set of symbols must be located here

                auxLegend.setClassifyingFieldNames(

                                new String[] {(String) cmbFields.getSelectedItem()});

            theLegend = auxLegend;

        }

        theLegend.setZSort(zSort);

        return theLegend;

    }

    private JCheckBox getChbUseDefault() {

        if (chbUseDefault == null) {

            chbUseDefault = new JCheckBox();

            chbUseDefault.setSelected(true);

            chbUseDefault.addActionListener(new java.awt.event.ActionListener() {

                    public void actionPerformed(java.awt.event.ActionEvent e) {

                            if (auxLegend==null)auxLegend=theLegend;

                            if (chbUseDefault.isSelected()) {

                                    addDefault();

                            } else {

                                    delDefault();

                            }

                    }

            });

            chbUseDefault.setText(PluginServices.getText(this, "resto_valores"));

        }

        return chbUseDefault;

    }

    /**

     * A�ade el resto de valores.

     */

    private void addDefault() {

        auxLegend.getDefaultSymbol().setDescription("Default");

        auxLegend.addSymbol(new NullUniqueValue(), auxLegend.getDefaultSymbol());

        symbolTable.addTableRecord(auxLegend.getDefaultSymbol(),

            new NullUniqueValue(), auxLegend.getDefaultSymbol().getDescription());

        symbolTable.repaint();

    }

    /**

     * Elimina el resto de valores que no estan representados por ning�n otro s�mbolo..

     */

    private void delDefault() {

        auxLegend.delSymbol(new NullUniqueValue());

        symbolTable.removeRow(new NullUniqueValue());

        symbolTable.repaint();

    }

    public void actionPerformed(ActionEvent e) {

            //modificar el combobox de valor

            if (e.getActionCommand() == "FIELD_SELECTED") {

                    JComboBox cb = (JComboBox) e.getSource();

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

                    symbolTable.removeAllItems();

//                    if (theLegend.getClassifyingFieldNames()!=null && fieldName != theLegend.getClassifyingFieldNames()[0]) {

//                            //m_bCacheDirty = true;

//                            theLegend.setClassifyingFieldNames(new String[] {fieldName});

//                    }

//

//                    //////////////////////////////////////////PEPE

//                    FLyrVect m = (FLyrVect) layer;

//                    try {

//                            int fieldType = m.getSource().getRecordset().getFieldType((int)cb.getSelectedIndex());

//                            if (theLegend.getClassifyingFieldTypes()!=null && fieldType != theLegend.getClassifyingFieldTypes()[0]) {

//                                    //m_bCacheDirty = true;

//                                    theLegend.setClassifyingFieldTypes(new int[] {fieldType});

//                            }

//                    } catch (ReadDriverException e1) {

//                            NotificationManager.addError(PluginServices.getText(this, "could_not_setup_legend"), e1);

//                    }

//                    /////////////////////////////////////////PEPE

            }

            // add all elements by value

            if (e.getActionCommand() == "ADD_ALL_VALUES") {

                    fillTableValues();

            }

            // add only one value

            if (e.getActionCommand() == "ADD_VALUE") {

                    try {

                            symbolTable.addTableRecord(SymbologyFactory.

                                            createDefaultSymbolByShapeType(layer.getShapeType()),

                                            new Double(0.0), "0 - 0");

                    } catch (ReadException ex) {

                            NotificationManager.addError(PluginServices.getText(this, "getting_shape_type"), ex);

                    }

            }

            //Vacia la tabla

            if (e.getActionCommand() == "REMOVE_ALL") {

                    symbolTable.removeAllItems();

                    theLegend.setZSort(null);

                    zSort = null;

            }

            //Quitar solo el elemento seleccionado

            if (e.getActionCommand() == "REMOVE") {

                    symbolTable.removeSelectedRows();

            }

            if (e.getActionCommand() == "OPEN_SYMBOL_LEVEL_EDITOR") {

                        if (theLegend != null) {

                                ZSort myZSort = ((AbstractClassifiedVectorLegend) getLegend()).getZSort();

                                if (myZSort == null) {

                                                myZSort = new ZSort(theLegend);

                                }

                                SymbolLevelsWindow sl = new SymbolLevelsWindow(myZSort);

                                PluginServices.getMDIManager().addWindow(sl);

                                zSort = sl.getZSort();

                                if (auxLegend != null)

                                        auxLegend.setZSort(zSort);

                        }

            }

    }

        public String getDescription() {

                return PluginServices.getText(this,"Dado_un_campo_de_atributos") + "," + PluginServices.getText(this,"muestra_los_elementos_de_la_capa_usando_un_simbolo_por_cada_valor_unico") + ".";

        }

        public ImageIcon getIcon() {

                return new ImageIcon(this.getClass().getClassLoader().

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

        }

        public Class getParentClass() {

                return Categories.class;

        }

        public String getTitle() {

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

        }

        public JPanel getPanel() {

                return this;

        }

        public Class getLegendClass() {

                return VectorialUniqueValueLegend.class;

        }

        public boolean isSuitableFor(FLayer layer) {

                return (layer instanceof FLyrVect);

        }

}