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.gui.project;

import java.util.BitSet;

import javax.swing.table.AbstractTableModel;

import com.hardcode.gdbms.engine.data.driver.DriverException;

import com.iver.andami.PluginServices;

import com.iver.andami.messages.NotificationManager;

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

import com.iver.cit.gvsig.project.ProjectTable;

/**

 * Modelo para la JTable de MapProperties

 *

 * @author Fernando Gonz?lez Cort?s

 */

public class TableModel extends AbstractTableModel {

    private BitSet visibles = new BitSet();

    private String[] alias;

    private ProjectTable table;

    /**

     * Creates a new MyTableModel object.

     *

     * @param t Tabla del proyecto que se quiere mostrar

     * @throws DriverException

     */

    public TableModel(ProjectTable t) throws DriverException {

        SelectableDataSource ds = t.getOriginal();

        if (ds == null) {

            ds = t.getModelo();

        }

        alias = t.getAliases();

        int[] mapping = t.getMapping();

        visibles = new BitSet();

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

            visibles.set(mapping[i]);

        }

        table = t;

    }

    /**

     * @see javax.swing.table.TableModel#getColumnCount()

     */

    public int getColumnCount() {

        return 3;

    }

    /* (non-Javadoc)

     * @see javax.swing.table.TableModel#getRowCount()

     */

    public int getRowCount() {

       try {

                return table.getModelo().getFieldCount();

            } catch (DriverException e) {

                NotificationManager.addError("No se pudo leer la informaci?n de la tabla", e);

                return 0;

            }

    }

    /* (non-Javadoc)

     * @see javax.swing.table.TableModel#getValueAt(int, int)

     */

    public Object getValueAt(int rowIndex, int columnIndex) {

        if (columnIndex == 0) {

            return new Boolean(visibles.get(rowIndex));

        } else if (columnIndex == 1) {

                try {

                return table.getModelo().getFieldName(rowIndex);

            } catch (DriverException e) {

                NotificationManager.addError("Error accediendo al nombre del campo", e);

                return "error!";

            }

        } else {

                return alias[rowIndex];

        }

    }

    /* (non-Javadoc)

     * @see javax.swing.table.TableModel#getColumnClass(int)

     */

    public Class getColumnClass(int columnIndex) {

        if (columnIndex == 0) {

            return Boolean.class;

        }

        return String.class;

    }

    /* (non-Javadoc)

     * @see javax.swing.table.TableModel#isCellEditable(int, int)

     */

    public boolean isCellEditable(int rowIndex, int columnIndex) {

        return (columnIndex != 1);

    }

    /* (non-Javadoc)

     * @see javax.swing.table.TableModel#setValueAt(java.lang.Object, int, int)

     */

    public void setValueAt(Object aValue, int rowIndex, int columnIndex) {

        if (columnIndex == 0) {

            visibles.set(rowIndex, ((Boolean) aValue).booleanValue());

        } else {

            alias[rowIndex] = aValue.toString();

        }

    }

    /* (non-Javadoc)

     * @see javax.swing.table.TableModel#getColumnName(int)

     */

    public String getColumnName(int column) {

        String ret = null;

        switch (column) {

        case 0:

            ret = PluginServices.getText(this, "visible");

            break;

        case 1:

            ret = PluginServices.getText(this, "campo");

            break;

        case 2:

            ret = PluginServices.getText(this, "alias");

            break;

        }

        return ret;

    }

    public String[] getAliases() {

        return alias;

    }

    public int[] getFieldMapping() {

        int[] mapping = new int[visibles.cardinality()];

        int index = 0;

        for(int i=visibles.nextSetBit(0); i>=0; i=visibles.nextSetBit(i+1)) {

            mapping[index] = i;

            index++;

        }

        return mapping;

    }

}