Application: gvSIG desktop

/**

 * gvSIG. Desktop Geographic Information System.

 *

 * Copyright (C) 2007-2013 gvSIG Association.

 *

 * 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 3

 * 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., 51 Franklin Street, Fifth Floor, Boston,

 * MA  02110-1301, USA.

 *

 * For any additional information, do not hesitate to contact us

 * at info AT gvsig.com, or visit our website www.gvsig.com.

 */

package org.gvsig.app.project.documents.view.legend.gui;

import java.awt.BorderLayout;

import java.awt.Color;

import java.awt.Component;

import java.awt.Dimension;

import java.awt.GridLayout;

import java.awt.event.ActionEvent;

import java.awt.event.ActionListener;

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 org.gvsig.andami.IconThemeHelper;

import org.gvsig.andami.PluginServices;

import org.gvsig.app.gui.styling.JComboBoxColorScheme;

import org.gvsig.app.gui.styling.SymbolLevelsWindow;

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

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

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

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

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

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

import org.gvsig.fmap.mapcontext.MapContextLocator;

import org.gvsig.fmap.mapcontext.MapContextManager;

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

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

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

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

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

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

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

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.mapcontext.rendering.legend.impl.AbstractClassifiedVectorLegend;

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

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.dispose.DisposableIterator;

import org.gvsig.tools.dispose.DisposeUtils;

import org.gvsig.tools.i18n.I18nManager;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

/**

 * 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 final Logger LOGGER = LoggerFactory

          .getLogger(VectorialUniqueValue.class);

  protected VectorialUniqueValueLegend theLegend;

  private ClassifiableVectorial layer;

  private SymbolTable symbolTable;

  protected JComboBox cmbFields;

  private JButton btnRemoveAll;

  private JButton btnRemove;

  private JCheckBox chbUseDefault = null;

  private JSymbolPreviewButton defaultSymbolPrev;

  private VectorialUniqueValueLegend auxLegend;

  private JPanel pnlCenter;

  private JButton btnOpenSymbolLevelsEditor;

  private JComboBoxColorScheme cmbColorScheme;

  private final GridBagLayoutPanel defaultSymbolPanel = new GridBagLayoutPanel();

  private final MapContextManager mapContextManager = MapContextLocator.getMapContextManager();

  /**

   *

   */

  public VectorialUniqueValue() {

    super();

    initComponents();

  }

  protected void initComponents() {

    I18nManager i18n = ToolsLocator.getI18nManager();

    JPanel pnlButtons = new JPanel();

    JButton btnAddAll = new JButton(i18n.getTranslation("Anadir_todos"));

    btnAddAll.setActionCommand("ADD_ALL_VALUES");

    btnAddAll.addActionListener(this);

    pnlButtons.add(btnAddAll);

    JButton btnAdd = new JButton(i18n.getTranslation("Anadir"));

    btnAdd.setActionCommand("ADD_VALUE");

    btnAdd.addActionListener(this);

    pnlButtons.add(btnAdd);

    btnRemoveAll = new JButton(i18n.getTranslation("Quitar_todos"));

    btnRemoveAll.setActionCommand("REMOVE_ALL");

    btnRemoveAll.addActionListener(this);

    pnlButtons.add(btnRemoveAll);

    btnRemove = new JButton(i18n.getTranslation("Quitar"));

    btnRemove.setActionCommand("REMOVE");

    btnRemove.addActionListener(this);

    pnlButtons.add(btnRemove);

    btnOpenSymbolLevelsEditor = new JButton(i18n.getTranslation("symbol_levels"));

    btnOpenSymbolLevelsEditor.addActionListener(this);

    btnOpenSymbolLevelsEditor.setActionCommand("OPEN_SYMBOL_LEVEL_EDITOR");

    pnlButtons.add(btnOpenSymbolLevelsEditor);

    btnOpenSymbolLevelsEditor.setEnabled(symbolTable != null && symbolTable.getRowCount() > 0);

    pnlCenter = new JPanel();

    pnlCenter.setLayout(new BorderLayout());

    cmbFields = new JComboBox();

    cmbFields.setActionCommand("FIELD_SELECTED");

    cmbFields.addActionListener(this);

    cmbFields.setVisible(true);

    JPanel pnlNorth = new JPanel();

    pnlNorth.setLayout(new GridLayout(0, 2));

    GridBagLayoutPanel auxPanel = new GridBagLayoutPanel();

    JLabel lblFieldClassification = new JLabel(i18n.getTranslation("Campo_de_clasificacion") + ": ");

    auxPanel.add(lblFieldClassification);

    auxPanel.add(cmbFields);

    pnlNorth.add(auxPanel);

    auxPanel = new GridBagLayoutPanel();

    auxPanel.add(new JLabel(i18n.getTranslation("color_scheme") + ": "));

    cmbColorScheme = new JComboBoxColorScheme(false);

    cmbColorScheme.addActionListener(this);

    auxPanel.add(cmbColorScheme);

    pnlNorth.add(auxPanel);

    defaultSymbolPanel.add(getChbUseDefault(), null);

    pnlNorth.add(defaultSymbolPanel);

    pnlNorth.add(new JBlank(0, 30));

    this.setLayout(new BorderLayout());

    this.add(pnlNorth, BorderLayout.NORTH);

    this.add(pnlCenter, BorderLayout.CENTER);

    this.add(pnlButtons, BorderLayout.SOUTH);

  }

  private void fillTableValues() {

    FeatureStore elRs;

    try {

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

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

      if (fieldName == null) {

        I18nManager i18n = ToolsLocator.getI18nManager();

        JOptionPane.showMessageDialog(

                (Component) PluginServices.getMainFrame(),

                i18n.getTranslation("no_hay_campo_seleccionado")

        );

        return;

      }

      symbolTable.removeAllItems();

      int numSymbols = 0;

      ISymbol theSymbol;

      auxLegend = (VectorialUniqueValueLegend) MapContextLocator

              .getMapContextManager().createLegend(

                      IVectorialUniqueValueLegend.LEGEND_NAME);

      auxLegend.setShapeType(layer.getShapeType());

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

      FeatureAttributeDescriptor attr = elRs.getDefaultFeatureTypeQuietly().getAttributeDescriptor(fieldName);

      int fieldType = attr.getDataType().getType();

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

      Object clave;

      //Object resul;

      if (chbUseDefault.isSelected()) {

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

        auxLegend.addSymbol(null, auxLegend.getDefaultSymbol());

      }

      Color[] colorScheme = cmbColorScheme.getSelectedColors();

      Color[] colors = new Color[colorScheme.length];

      System.arraycopy(colorScheme, 0, colors, 0, colorScheme.length);

      auxLegend.setColorScheme(colors);

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

      FeatureSet set = null;

      DisposableIterator iterator = null;

      try {

        set = elRs.getFeatureSet();

        int count = 0;

        iterator = set.fastIterator();

        while (iterator.hasNext()) {

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

          clave = feature.get(fieldName);

          if (clave == null) {

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

            if (count == 100) {

              I18nManager i18n = ToolsLocator.getI18nManager();

              int resp = JOptionPane.showConfirmDialog(this,

                      i18n.getTranslation("mas_de_100_simbolos"),

                      i18n.getTranslation("quiere_continuar"),

                      JOptionPane.YES_NO_OPTION,

                      JOptionPane.WARNING_MESSAGE);

              if ((resp == JOptionPane.NO_OPTION)

                      || (resp == JOptionPane.DEFAULT_OPTION)) {

                break;

              }

            }

            // si no esta creado el simbolo se crea

            // jaume (moved to ISymbol); theSymbol = new

            // FSymbol(layer.getShapeType());

            theSymbol = mapContextManager.getSymbolManager().createSymbol(

                    layer.getShapeType(),

                    colorScheme[rand.nextInt(colorScheme.length)]

            );

            theSymbol.setDescription(clave.toString());

            auxLegend.addSymbol(clave, theSymbol);

            count++;

          }

        } // for

      } finally {

        DisposeUtils.disposeQuietly(iterator);

        DisposeUtils.disposeQuietly(set);

      }

      Object[] values = auxLegend.getValues();

      String[] descriptions = new String[values.length];

      ISymbol[] symbols = new ISymbol[values.length];

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

        Object value = values[i];

        symbols[i] = auxLegend.getSymbolByValue(value);

        descriptions[i] = symbols[i].getDescription();

      }

      symbolTable.fillTableFromSymbolList(symbols,

              values, descriptions);

    } catch (DataException e) {

      LOGGER.warn("recovering_recordset", e);

    }

    btnRemoveAll.setEnabled(true);

    btnRemove.setEnabled(true);

  }

  private boolean compareClassifyingFieldNames(String[] a, String[] b) {

    if (a == b) {

      return true;

    }

    if (a == null || b == null) {

      return false;

    }

    if (a.length != b.length) {

      return false;

    }

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

      if (!a[i].equals(b[i])) {

        return false;

      }

    }

    return true;

  }

  private boolean compareClassifyingFieldTypes(int[] a, int[] b) {

    if (a == b) {

      return true;

    }

    if (a == null || b == null) {

      return false;

    }

    if (a.length != b.length) {

      return false;

    }

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

      if (a[i] != b[i]) {

        return false;

      }

    }

    return true;

  }

  /**

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

    ISymbol theSymbol;

    ArrayList visitedKeys = new ArrayList();

    boolean changedLegend = false;

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

    String[] classifyingFieldNames = new String[]{fieldName};

    if (auxLegend != null) {

      if (!compareClassifyingFieldNames(classifyingFieldNames, auxLegend.getClassifyingFieldNames())) {

        auxLegend.setClassifyingFieldNames(classifyingFieldNames);

        changedLegend = true;

      }

    } else {

      auxLegend.setClassifyingFieldNames(classifyingFieldNames);

      changedLegend = true;

    }

    FLyrVect m = (FLyrVect) layer;

    try {

      int fieldType = m.getFeatureStore().getDefaultFeatureType()

              .getAttributeDescriptor(cmbFields.getSelectedIndex())

              .getType();

      int[] classifyingFieldTypes = new int[]{fieldType};

      if (auxLegend != null) {

        if (!compareClassifyingFieldTypes(classifyingFieldTypes, auxLegend.getClassifyingFieldTypes())) {

          auxLegend.setClassifyingFieldTypes(classifyingFieldTypes);

          changedLegend = true;

        }

      } else {

        auxLegend.setClassifyingFieldTypes(classifyingFieldTypes);

        changedLegend = true;

      }

    } catch (DataException e) {

      LOGGER.warn("could_not_setup_legend", e);

    } catch (Exception e) {

      LOGGER.warn("could_not_setup_legend", e);

    }

    if (changedLegend) {

      auxLegend.clear();

    }

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

      clave = symbolTable.getFieldValue(row, 1);

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

      String description = (String) symbolTable.getFieldValue(row, 2);

      theSymbol.setDescription(description);

      ISymbol legendSymbol = null;

      if (auxLegend != null) {

        legendSymbol = auxLegend.getSymbolByValue(clave);

      }

      if (legendSymbol == null || (auxLegend.isUseDefaultSymbol() && legendSymbol == auxLegend.getDefaultSymbol())) {

        if (auxLegend != null) {

          auxLegend.addSymbol(clave, theSymbol);

        }

      } else {

        // avoid same instance

        if (legendSymbol != theSymbol) {

          /*

                                 * Perhaps use this "if" when implementations of ISymbol

                                 * properly implement the equals method. 

           */

          // if (!legendSymbol.equals(theSymbol)) {

          auxLegend.replace(legendSymbol, theSymbol);

          // }

        }

      }

      visitedKeys.add(clave);

    }

    if (auxLegend != null) {

      Object[] keys = auxLegend.getValues();

      for (Object key : keys) {

        if (!visitedKeys.contains(key)) {

          auxLegend.delSymbol(key);

        }

      }

    }

    clave = null;

    if (chbUseDefault.isSelected()) {

      theSymbol = defaultSymbolPrev.getSymbol();

      if (theSymbol != null) {

        I18nManager i18n = ToolsLocator.getI18nManager();

        String description = i18n.getTranslation("default");

        theSymbol.setDescription(description);

        ISymbol legendSymbol = null;

        if (auxLegend != null) {

          legendSymbol = auxLegend.getSymbolByValue(clave);

        }

        if (legendSymbol == null) {

          auxLegend.addSymbol(clave, theSymbol);

        } else {

          if (legendSymbol != theSymbol) {

            auxLegend.replace(legendSymbol, theSymbol);

          }

        }

      }

    } else {

      if (auxLegend != null) {

        ISymbol legendSymbol = auxLegend.getSymbolByValue(clave);

        if (legendSymbol != null) {

          auxLegend.replace(legendSymbol, null);

        }

      }

    }

  }

  private void fillFieldNames() {

    FeatureStore rs;

    try {

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

      ArrayList names = new ArrayList();

      for (FeatureAttributeDescriptor descriptor : rs.getDefaultFeatureType()) {

        Class fieldClass = descriptor.getDataType().getDefaultClass();

        if (Comparable.class.isAssignableFrom(fieldClass)) {

          names.add(descriptor.getName());

        }

      }

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

      cmbFields.setModel(cM);

    } catch (DataException e) {

      LOGGER.warn("recovering_recordset", e);

    }

  }

  @Override

  public void setData(FLayer layer, ILegend legend) {

    this.layer = (ClassifiableVectorial) layer;

    int shapeType = 0;

    try {

      shapeType = this.layer.getShapeType();

    } catch (ReadException e) {

      LOGGER.warn("generating_intervals", e);

    }

    getDefaultSymbolPrev(shapeType);

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

      auxLegend = (VectorialUniqueValueLegend) legend.cloneLegend();

      getChbUseDefault().setSelected(auxLegend.isUseDefaultSymbol());

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

      setColorScheme();

      Object[] values = auxLegend.getValues();

      String[] descriptions = new String[values.length];

      ISymbol[] symbols = new ISymbol[values.length];

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

        Object value = values[i];

        symbols[i] = auxLegend.getSymbolByValue(value);

        descriptions[i] = symbols[i].getDescription();

      }

      symbolTable.fillTableFromSymbolList(symbols, values, descriptions);

      chbUseDefault.setSelected(auxLegend.isUseDefaultSymbol());

    } else {

      auxLegend = new VectorialUniqueValueLegend(shapeType);

    }

    defaultSymbolPrev.setSymbol(auxLegend.getDefaultSymbol());

    btnOpenSymbolLevelsEditor.setEnabled(symbolTable != null && symbolTable.getRowCount() > 0);

  }

  private void setColorScheme() {

    if (auxLegend.getColorScheme() != null) {

      Color[] colorScheme = auxLegend.getColorScheme();

      cmbColorScheme.setSelectedColors(colorScheme);

    }

  }

  private void getDefaultSymbolPrev(int shapeType) {

    if (defaultSymbolPrev == null) {

      defaultSymbolPrev = new JSymbolPreviewButton(shapeType);

      defaultSymbolPrev.setPreferredSize(new Dimension(110, 20));

      defaultSymbolPrev.addActionListener(this);

      defaultSymbolPanel.add(defaultSymbolPrev, null);

    }

  }

  @Override

  public ILegend getLegend() {

    fillSymbolListFromTable();

    if (auxLegend != null) {

      ISymbol defaultSymbolLegend = auxLegend.getDefaultSymbol();

      ISymbol symbol = defaultSymbolPrev.getSymbol();

      if (symbol != null) {

        if (symbol != defaultSymbolLegend) {

          auxLegend.setDefaultSymbol(symbol);

        }

      }

      auxLegend.useDefaultSymbol(chbUseDefault.isSelected());

      theLegend = (VectorialUniqueValueLegend) auxLegend.cloneLegend();

      theLegend.setZSort(auxLegend.getZSort());

    }

    return theLegend;

  }

  private JCheckBox getChbUseDefault() {

    if (chbUseDefault == null) {

      I18nManager i18n = ToolsLocator.getI18nManager();

      chbUseDefault = new JCheckBox();

      chbUseDefault.setSelected(false);

      chbUseDefault.addActionListener((java.awt.event.ActionEvent e) -> {

        if (chbUseDefault.isSelected()) {

          auxLegend.useDefaultSymbol(true);

        } else {

          auxLegend.useDefaultSymbol(false);

        }

      });

      chbUseDefault.setText(i18n.getTranslation("resto_valores") + ": ");

    }

    return chbUseDefault;

  }

  private void addDefault() {

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

    auxLegend.addSymbol(null, auxLegend.getDefaultSymbol());

    symbolTable.addTableRecord(

            auxLegend.getDefaultSymbol(),

            null,

            auxLegend.getDefaultSymbol().getDescription()

    );

    symbolTable.repaint();

  }

  private void delDefault() {

    auxLegend.delSymbol(null);

    symbolTable.removeRow(null);

    symbolTable.repaint();

  }

  @Override

  public void actionPerformed(ActionEvent e) {

    //modificar el combobox de valor

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

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

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

      symbolTable.removeAllItems();

      btnOpenSymbolLevelsEditor.setEnabled(false);

    }

    // add all elements by value

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

      fillTableValues();

      btnOpenSymbolLevelsEditor.setEnabled(symbolTable != null && symbolTable.getRowCount() > 0);

    }

    // add only one value

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

      try {

        ISymbol symbol = mapContextManager.getSymbolManager().createSymbol(

                layer.getShapeType()

        );

        Double clave = 0.0;

        symbolTable.addTableRecord(

                symbol, clave, "0 - 0"

        );

        btnOpenSymbolLevelsEditor.setEnabled(true);

      } catch (ReadException ex) {

        LOGGER.warn("getting_shape_type", ex);

      }

    }

    //Vacia la tabla

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

      symbolTable.removeAllItems();

      auxLegend.setZSort(null);

      btnOpenSymbolLevelsEditor.setEnabled(false);

    }

    //Quitar solo el elemento seleccionado

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

      symbolTable.removeSelectedRows();

      btnOpenSymbolLevelsEditor.setEnabled(symbolTable.getRowCount() > 0);

    }

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

      ZSort myZSort = null;

      if (auxLegend != null) {

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

        if (myZSort == null) {

          myZSort = new ZSort(auxLegend);

        }

      }

      if (myZSort == null && theLegend != null) {

        myZSort = new ZSort(theLegend);

      }

      SymbolLevelsWindow sl = new SymbolLevelsWindow(myZSort);

      PluginServices.getMDIManager().addWindow(sl);

      auxLegend.setZSort(sl.getZSort());

    }

  }

  @Override

  public String getDescription() {

    I18nManager i18n = ToolsLocator.getI18nManager();

    return i18n.getTranslation("Dado_un_campo_de_atributos") + ","

            + i18n.getTranslation("muestra_los_elementos_de_la_capa_usando_un_simbolo_por_cada_valor_unico") + ".";

  }

  @Override

  public ImageIcon getIcon() {

    return IconThemeHelper.getImageIcon(

            "legend-overview-vectorial-unique-value");

  }

  @Override

  public Class getParentClass() {

    return Categories.class;

  }

  @Override

  public String getTitle() {

    I18nManager i18n = ToolsLocator.getI18nManager();

    return i18n.getTranslation("Valores_unicos");

  }

  @Override

  public JPanel getPanel() {

    return this;

  }

  @Override

  public Class getLegendClass() {

    return VectorialUniqueValueLegend.class;

  }

  @Override

  public boolean isSuitableFor(FLayer layer) {

    return (layer instanceof FLyrVect);

  }

}