Application: gvSIG desktop

package com.iver.ai2.gvsig3d.legend;

import java.awt.BorderLayout;

import java.awt.Component;

import java.awt.FlowLayout;

import java.awt.event.ActionEvent;

import java.awt.event.ActionListener;

import java.awt.event.ItemEvent;

import java.awt.event.ItemListener;

import java.awt.event.KeyEvent;

import java.awt.event.KeyListener;

import java.sql.Types;

import java.util.ArrayList;

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 javax.swing.JTextField;

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

import org.gvsig.raster.datastruct.ColorItem;

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

import com.hardcode.gdbms.engine.data.DataSource;

import com.hardcode.gdbms.engine.instruction.IncompatibleTypesException;

import com.hardcode.gdbms.engine.values.DoubleValue;

import com.hardcode.gdbms.engine.values.NullValue;

import com.hardcode.gdbms.engine.values.Value;

import com.hardcode.gdbms.engine.values.ValueFactory;

import com.iver.ai2.gvsig3d.legend.symbols.BaseExtrusionSymbol;

import com.iver.ai2.gvsig3d.map3d.layers.Layer3DProps;

import com.iver.andami.PluginServices;

import com.iver.andami.messages.NotificationManager;

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

/**

 * 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 ExtrusionLegendPanel extends JPanel implements ILegendPanel,ActionListener,KeyListener,ItemListener {

    /**

         * 

         */

        private static final long serialVersionUID = 6470946030927461473L;

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

    private VectorialUniqueValueLegend theLegend;

    private ClassifiableVectorial layer;

    private SymbolTableExtrusion symbolTable;

    private 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 JTextField jtvariable;

        private JComboBoxColorScheme cmbColorScheme;

        private double extrusionValue = 1;

        private JTextField jtfija;

        private JCheckBox fixedOption;

        private double extrusionFixedValue = 1;

        private Value aux; 

    /**

     *

     */

    public ExtrusionLegendPanel() {

        super();

        initialize();

    }

    /**

     * DOCUMENT ME!

     */

    protected void initialize() {

        /* 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, "Nivel_de_extrusion")+":"));

        jtvariable = new JTextField(PluginServices.getText(this, "1"),3);

        jtvariable.setColumns(3);

        jtvariable.setHorizontalAlignment(JTextField.RIGHT); 

        jtvariable.setToolTipText(PluginServices.getText(this, "Valor_inicial") + " * " + PluginServices.getText(this, "Nivel_de_extrusion"));

        jtvariable.addKeyListener(this);

        pAux1.add(jtvariable);

        pnlNorth.add(pAux1);

        pAux1 = new JPanel();

        fixedOption = new JCheckBox(PluginServices.getText(this, "Altura_fija")+":",false);

        fixedOption.setHorizontalAlignment(JTextField.RIGHT); 

        fixedOption.setEnabled(true);

        fixedOption.setSelected(false);

        pAux1.add(fixedOption);

        jtfija = new JTextField(PluginServices.getText(this, "1"),3);

        jtfija.setColumns(10);

        jtfija.setHorizontalAlignment(JTextField.RIGHT); 

        jtfija.addKeyListener(this);

        jtfija.setToolTipText(PluginServices.getText(this, "Valor_inicial")+ " = "+ PluginServices.getText(this, "Altura_fija"));

        jtfija.setEnabled(false);

        pAux1.add(jtfija);

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

        fixedOption.addItemListener(this);

        this.setLayout(new BorderLayout());

        this.add(pnlNorth, BorderLayout.NORTH);

        this.add(pnlCenter, BorderLayout.CENTER);

        this.add(pnlButtons, BorderLayout.SOUTH);

    }

    /**

     * DOCUMENT ME!

     */

    private void fillTableValues() {

        DataSource elRs;

        try {

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

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

            auxLegend = new ExtrusionLegend(layer.getShapeType());

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

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

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

                        aux = clave.producto(ValueFactory.createValue(extrusionValue));

                    theSymbol.setDescription(aux.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

            Object[] extrudedValue = auxLegend.getValues();

            double defaultValue = 1;

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

                    if(jtvariable.isEnabled()){

                            try {

                                         extrudedValue[i] = ((Value)auxLegend.getValues()[i]).producto(ValueFactory.createValue(extrusionValue));

                                         theSymbol.setDescription(extrudedValue[i].toString());

                                 }catch (IncompatibleTypesException e) {

                                         extrudedValue[i] = ValueFactory.createValue(defaultValue);

                                         }

                    }else

                                         extrudedValue[i] = ValueFactory.createValue(extrusionFixedValue);

            }

//            symbolTable.fillTableFromSymbolList(auxLegend.getSymbols(),

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

            symbolTable.fillTableFromSymbolList(auxLegend.getSymbols(),

                                        auxLegend.getValues(), extrudedValue);

            elRs.stop();

        } catch (ReadDriverException e) {

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

        } catch (IncompatibleTypesException e) {

                        // TODO Auto-generated catch block

                        e.printStackTrace();

                }

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

        Value clave;

        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 {

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

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

        } catch (ReadDriverException 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 = (Value) symbolTable.getFieldValue(row, 0);

            DoubleValue extrusion = null;

            try{

                    extrusion = (DoubleValue) symbolTable.getFieldValue(row, 1);

                    theSymbol = new BaseExtrusionSymbol(extrusion.getValue());

                theSymbol.setDescription(extrusion.toString());

            }

            catch (Exception e) {

                     theSymbol = new BaseExtrusionSymbol(1);

                 theSymbol.setDescription("1");

                        }

            theLegend.addSymbol(clave,theSymbol);

        }

        if (bRestoValores) {

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

            theLegend.setDefaultSymbol(theSymbol);

        }

    }

    /**

     * DOCUMENT ME!

     */

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

                cmbFields.setModel(cM);

                symbolTable.removeAllItems();

        }

    public void setData(FLayer layer, ILegend legend) {

            this.layer = (ClassifiableVectorial) layer;

              int shapeType = 0;

              try {

                      shapeType = this.layer.getShapeType();

              } catch (ReadDriverException e) {

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

                }

              if (symbolTable != null)

                      pnlCenter.remove(symbolTable);

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

              pnlCenter.add(symbolTable, BorderLayout.CENTER);

        fillFieldNames();

        symbolTable.removeAllItems();

        if (legend instanceof VectorialUniqueValueLegend) {

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

                auxLegend.setClassifyingFieldTypes(theLegend.getClassifyingFieldTypes());

                LegendContentsChangedListener[] l  = theLegend.getListeners();

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

                        auxLegend.addLegendListener(l[i]);

                        };

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

            }

            // 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()),

                                            ValueFactory.createValue(0.0), "0 - 0");

                    } catch (ReadDriverException 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,PluginServices.getText(this,"info_extrusion"));

        }

        public ImageIcon getIcon() {

                return null;

        }

        public Class getParentClass() {

                return null;

        }

        public String getTitle() {

                return ("<html><b>"+ PluginServices.getText(this,"Extrusion")+ "</b></html>");

        }

        public JPanel getPanel() {

                return this;

        }

        public Class getLegendClass() {

                return ExtrusionLegend.class;

        }

        public boolean isSuitableFor(FLayer layer) {

                Layer3DProps props3D = Layer3DProps.getLayer3DProps(layer);

                return ((props3D != null) && (props3D.getType() == Layer3DProps.layer3DVector));

        }

        public void keyPressed(KeyEvent e) {

                // TODO Auto-generated method stub

        }

        public void keyReleased(KeyEvent e) {

                try {

                        extrusionValue = new Double(this.jtvariable.getText()).doubleValue();

                        extrusionFixedValue = new Double(this.jtfija.getText()).doubleValue();

                }

                catch (Exception e2) {

//                        String s = " ";

//                        if(!this.jtvariable.getText().equals(s)) {

//                                JOptionPane.showMessageDialog(null, PluginServices.getText(this, "Introduce un dato numerico"),

//                                                PluginServices.getText(this, "Dato incorrecto"), JOptionPane.WARNING_MESSAGE);

//        

//                                jtvariable.setText(Double.toString(extrusionValue));

//                                jtfija.setText(Double.toString(extrusionFixedValue));

//                        }

                }

        }

        public void keyTyped(KeyEvent e) {

                // TODO Auto-generated method stub

        }

        public void itemStateChanged(ItemEvent e) {

                        jtfija.setEnabled(fixedOption.isSelected());

                        jtvariable.setEnabled(!fixedOption.isSelected());

        }

}