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

import java.awt.Dimension;

import java.awt.GridLayout;

import java.util.HashMap;

import java.util.Map;

import javax.swing.JPanel;

import javax.swing.JScrollPane;

import javax.swing.JTable;

import javax.swing.table.DefaultTableModel;

import javax.swing.table.TableCellEditor;

import javax.swing.table.TableColumn;

import org.gvsig.andami.PluginServices;

import org.gvsig.app.project.documents.gui.SymbolCellEditor;

import org.gvsig.app.project.documents.gui.TableSymbolCellRenderer;

import org.gvsig.app.project.documents.view.legend.edition.gui.IntervalCellEditor;

import org.gvsig.app.project.documents.view.legend.edition.gui.ValueCellEditor;

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

import org.gvsig.utils.swing.jtable.TextFieldCellEditor;

/**

 * JPanel que contiene la tabla con los s?mbolos intervalos o valores y

 * etiquetado de estos valores.

 *

 * @author Vicente Caballero Navarro

 */

public class SymbolTable extends JPanel {

        private static final long serialVersionUID = -8694846716328735113L;

        public static final String VALUES_TYPE = "values";

        public static final String INTERVALS_TYPE = "intervals";

  private static final Map<String,TableCellEditor> CELL_EDITORS = new HashMap<>();

  private final JTable table;

        private final String type;

        private final int shapeType;

        public SymbolTable(Component ownerComponent, String type, int shapeType) {

    this(new org.gvsig.utils.swing.jtable.JTable(), ownerComponent, type, shapeType);

  }

        /**

         * Crea un nuevo FSymbolTable.

         *

   * @param theTable

   * @param ownerComponent

         * @param type, tipo de valor si es intervalo: "intervals" y si es por valores: "values".

   * @param shapeType

         */

        public SymbolTable(JTable theTable, Component ownerComponent, String type, int shapeType) {

                super(new GridLayout(1, 0));

                this.type = type;

                this.shapeType = shapeType;

                table = theTable;

                table.setModel(new MyTableModel());

                table.setPreferredScrollableViewportSize(new Dimension(480, 110));

                initializeCellEditors();

                // Create the scroll pane and add the table to it.

                JScrollPane scrollPane = new JScrollPane(table);

                // Set up column sizes.

                // initColumnSizes(table);

                setUpSymbolColumn(table, table.getColumnModel().getColumn(0));

                if(CELL_EDITORS.get(type) == null)

                        throw new Error("Symbol table type not set!");

                setUpValueColumn(table, table.getColumnModel().getColumn(1),CELL_EDITORS.get(this.type));

                setUpLabelColumn(table, table.getColumnModel().getColumn(2));

                // Add the scroll pane to this panel.

                add(scrollPane);

                table.setRowSelectionAllowed(true);

                // default is 16

                table.setRowHeight(24);

        }

  public void setCellEditor(TableCellEditor cellEditor) {

    TableColumn column = table.getColumnModel().getColumn(1);

    column.setCellEditor(cellEditor);

  }

        /**

         * Inicializa los valores de los CellEditors que la SymbolTable poseer? por defecto

         */

        private void initializeCellEditors() {

                this.CELL_EDITORS.put(this.INTERVALS_TYPE,new IntervalCellEditor());

                this.CELL_EDITORS.put(this.VALUES_TYPE, new ValueCellEditor());

        }

        /**

         * A?ade un nuevo CellEditor a la lista de disponibles

         *

         * @param key String con el nombre identificativo del CellEditor

         * @param cellEditor CellEditor que va a ser a?adido

         */

        public static void addNewCellEditor(String key,TableCellEditor cellEditor ) {

                CELL_EDITORS.put(key, cellEditor);

        }

        /**

         * Obtiene el valor de los elementos de una fila seleccionada

         *

         * @return Object[] Array con los objetos de cada una de las columnas de la fila seleccionada

         */

        public Object[] getSelectedRowElements() {

                Object[] values = new Object[3];

                MyTableModel m = (MyTableModel) table.getModel();

                int[] selectedRows = table.getSelectedRows();

                if(selectedRows.length != 1)

                        return null;

                for (int i = 0; i < 3; i++) {

                        values[i] = m.getValueAt(selectedRows[0], i);

                }

                return values;

        }

        /**

         * A?ade una fila al modelo.

         *

         * @param vector

         *            Fila en forma de vector de Object para a?adir al modelo.

         */

        public void addRow(Object[] vector) {

                MyTableModel m = (MyTableModel) table.getModel();

                m.addRow(vector);

        }

        /**

         * Elimina la fila que tiene como clave el objeto que se pasa como

         * par?metro.

         *

         * @param obj

         *            clave del objeto a eliminar.

         */

        public void removeRow(Object obj) {

                MyTableModel m = (MyTableModel) table.getModel();

                for (int i = 0; i < m.getRowCount(); i++) {

//                        if (m.getValueAt(i, 1) instanceof NullUniqueValue

//                                        || m.getValueAt(i, 1) instanceof NullIntervalValue) {

//                                m.removeRow(i);

//                        }

                        if (m.getValueAt(i, 1) == null) {

                                m.removeRow(i);

                        }

                }

        }

        /**

         * Elimina las filas que est?n seleccionadas.

         */

        public void removeSelectedRows() {

                if (table.getCellEditor() != null) {

                        table.getCellEditor().cancelCellEditing();

                }

                MyTableModel m = (MyTableModel) table.getModel();

                int[] selectedRows = table.getSelectedRows();

                for (int i = selectedRows.length - 1; i >= 0; i--) {

                        m.removeRow(selectedRows[i]);

                }

        }

        /**

         * Rellena la tabla con los s?mbolos valores y descripciones que se pasan

         * como par?metro.

         *

         * @param symbols

         *            Array de s?mbolos

         * @param values

         *            Array de valores.

         * @param descriptions

         *            Array de descripciones.

         */

        public void fillTableFromSymbolList(ISymbol[] symbols, Object[] values,

                        String[] descriptions) {

                ISymbol theSymbol;

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

                        theSymbol = symbols[i];

//                        if(!(values[i] instanceof NullIntervalValue) && !(values[i] instanceof NullUniqueValue))

//                                addTableRecord(theSymbol, values[i], descriptions[i]);

                        if(!(values[i] == null))

                                addTableRecord(theSymbol, values[i], descriptions[i]);

                }

        }

        /**

         * A?ade una fila con los objetos que se pasan como par?metros.

         *

         * @param symbol

         *            s?mbolo de la fila.

         * @param value

         *            Valor de la fila.

         * @param description

         *            Descripci?n.

         */

        public void addTableRecord(ISymbol symbol, Object value, String description) {

                Object[] theRow = new Object[3];

                theRow[0] = symbol;

                theRow[1] = value;

                theRow[2] = description;

                addRow(theRow);

        }

        /**

         * Devuelve el valor a partie del n?mero de fila y columna.

         *

         * @param row

         *            n?mero de fila.

         * @param col

         *            n?mero de columna.

         *

         * @return Objeto.

         */

        public Object getFieldValue(int row, int col) {

                MyTableModel m = (MyTableModel) table.getModel();

                return m.getValueAt(row, col);

        }

        /**

         * Devuelve el n?mero total de filas que contiene el modelo.

         *

         * @return N?mero de filas.

         */

        public int getRowCount() {

                MyTableModel m = (MyTableModel) table.getModel();

                return m.getRowCount();

        }

        /**

         * Elimina todas las filas del modelo.

         */

        public void removeAllItems() {

                table.setModel(new MyTableModel());

                setUpSymbolColumn(table, table.getColumnModel().getColumn(0));

                setUpValueColumn(table, table.getColumnModel().getColumn(1),CELL_EDITORS.get(this.type));

                setUpLabelColumn(table, table.getColumnModel().getColumn(2));

        }

        /**

         * Inicializa el cell editor de tipo descripci?n de la columna que se pasa

         * como par?metro.

         *

         * @param table2

         *            Tabla.

         * @param column

         *            Columna.

         */

        public void setUpLabelColumn(JTable table2, TableColumn column) {

                TextFieldCellEditor labeleditor = new TextFieldCellEditor();

                column.setCellEditor(labeleditor);

        }

        /**

         * Inicializa el cell editor de tipo valor de la columna que se pasa como

         * par?metro.

         *

         * @param table2

         *            Tabla.

         * @param column

         *            Columna.

         * @param tableCellEditor

         */

        public void setUpValueColumn(JTable table2,TableColumn column, TableCellEditor tableCellEditor) {

                column.setCellEditor(tableCellEditor);

        }

        /**

         * Inicializa el cell editor de tipo s?mbolo de la columna que se pasa como

         * par?metro.

         *

         * @param table2

         *            Tabla.

         * @param column

         *            Columna.

         */

        public void setUpSymbolColumn(JTable table2, TableColumn column) {

                // Set up the editor

                column.setMaxWidth(100);

                column.setWidth(60);

                column.setPreferredWidth(60);

                column.setMinWidth(50);

                // FSymbolCellEditor symboleditor = new FSymbolCellEditor();

                SymbolCellEditor symboleditor = new SymbolCellEditor(shapeType);

                column.setCellEditor(symboleditor);

                TableSymbolCellRenderer renderer = new TableSymbolCellRenderer(true);

                column.setCellRenderer(renderer);

        }

        class MyTableModel extends DefaultTableModel {

                private final String[] columnNames = {

                                PluginServices.getText(this, "Simbolo"),

                                PluginServices.getText(this, "Valor"),

                                PluginServices.getText(this, "Etiqueta") };

    @Override

                public int getColumnCount() {

                        return columnNames.length;

                }

    @Override

                public String getColumnName(int col) {

                        return columnNames[col];

                }

    @Override

                public Class getColumnClass(int c) {

            switch (c) {

            case 0:

                return ISymbol.class;

            case 1:

                // TODO: take the value from the related FeatureType attribute

                Object value = getValueAt(0, c);

                return value == null ? Object.class : value.getClass();

            case 2:

                return String.class;

            default:

                return Object.class;

            }

                }

    @Override

                public boolean isCellEditable(int row, int col) {

                        return true;

                }

                @Override

                public Object getValueAt(int row, int column) {

                        if(column == 2)

                                return ((ISymbol)getValueAt(row,0)).getDescription();

                        return super.getValueAt(row, column);

                }

                @Override

                public void setValueAt(Object aValue, int row, int column) {

                        if(column == 2){

                                ISymbol symbol = (ISymbol) getValueAt(row,0);

                                symbol.setDescription((String) aValue);

                                setValueAt(symbol,row,0);

                        }

                        super.setValueAt(aValue, row, column);

                }

        }

    public void moveDownRows(int startPos, int endPos, int numOfElements) {

        if(startPos > endPos)

            return;

        if(startPos >= getRowCount()-1 )

            return;

        if(startPos == getRowCount()-1)

            return;

        Object[][] values = new Object[getRowCount()][3];

        for (int i = 0; i < getRowCount(); i++) {

            values[i][0] = table.getModel().getValueAt(i,0);

            values[i][1] = table.getModel().getValueAt(i,1);

            values[i][2] = table.getModel().getValueAt(i,2);

        }

        Object[][]aux = new Object[numOfElements][3];

        for (int i = 0; i < numOfElements; i++) {

            aux[numOfElements - i - 1][0] = values[startPos - i][0];

            aux[numOfElements - i - 1][1] = values[startPos - i][1];

            aux[numOfElements - i - 1][2] = values[startPos - i][2];

        }

        Object [][] targetVal = {{values[endPos][0],values[endPos][1],values[endPos][2]}};

        values[startPos - numOfElements + 1][0] = targetVal[0][0];

        values[startPos - numOfElements + 1][1] = targetVal[0][1];

        values[startPos - numOfElements + 1][2] = targetVal[0][2];

        for (int i = 0; i < numOfElements; i++) {

            values[endPos - i][0] = aux[numOfElements - i - 1][0];

            values[endPos - i][1] = aux[numOfElements - i - 1][1];

            values[endPos - i][2] = aux[numOfElements - i - 1][2];

        }

        ISymbol[] symbols = new ISymbol[getRowCount()];

        Object[] objects = new Object[getRowCount()];

        String[] cads = new String[getRowCount()];

        for (int i = 0; i < getRowCount(); i++) {

            symbols[i] = (ISymbol) values[i][0];

            objects[i] = values[i][1];

            cads[i] = (String) values[i][2];

        }

        removeAllItems();

        fillTableFromSymbolList(symbols,objects,cads);

        table.addRowSelectionInterval(endPos-numOfElements+1,endPos);

    }

    public void moveUpRows(int startPos, int endPos, int numOfElements) {

        if(startPos == 0)

            return;

        if(endPos > startPos)

            return;

        Object[][] values = new Object[getRowCount()][3];

        for (int i = 0; i < getRowCount(); i++) {

            values[i][0] = table.getModel().getValueAt(i,0);

            values[i][1] = table.getModel().getValueAt(i,1);

            values[i][2] = table.getModel().getValueAt(i,2);

        }

        Object[][]aux = new Object[numOfElements][3];

        for (int i = 0; i < numOfElements; i++) {

            aux[i][0] = values[startPos + i][0];

            aux[i][1] = values[startPos + i][1];

            aux[i][2] = values[startPos + i][2];

        }

        Object [][] targetVal = {{values[endPos][0],values[endPos][1],values[endPos][2]}};

        values[startPos + numOfElements - 1][0] = targetVal[0][0];

        values[startPos + numOfElements - 1][1] = targetVal[0][1];

        values[startPos + numOfElements - 1][2] = targetVal[0][2];

        for (int i = 0; i < numOfElements; i++) {

            values[endPos + i][0] = aux[i][0];

            values[endPos + i][1] = aux[i][1];

            values[endPos + i][2] = aux[i][2];

        }

        ISymbol[] symbols = new ISymbol[getRowCount()];

        Object[] objects = new Object[getRowCount()];

        String[] cads = new String[getRowCount()];

        for (int i = 0; i < getRowCount(); i++) {

            symbols[i] = (ISymbol) values[i][0];

            objects[i] = values[i][1];

            cads[i] = (String) values[i][2];

        }

        removeAllItems();

        fillTableFromSymbolList(symbols,objects,cads);

        table.addRowSelectionInterval(endPos,endPos+numOfElements-1);

    }

    public int[] getSelectedRows(){

        return table.getSelectedRows();

    }

}