| gvSIG desktop 1

/* gvSIG. Geographic Information System of the Valencian Government

 *

 * Copyright (C) 2007-2008 Infrastructures and Transports Department

 * of the Valencian Gobernment (CIT)

 * 

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

 * MA  02110-1301, USA.

 * 

 */

/*

 * AUTHORS (In addition to CIT):

 * 2008 {DiSiD Technologies}  {Create a JTable TableModel for a FeatureCollection}

 */

package org.gvsig.fmap.data.feature.swing.table;

import javax.swing.table.AbstractTableModel;

import org.gvsig.fmap.data.DataException;

import org.gvsig.fmap.data.ReadException;

import org.gvsig.fmap.data.feature.*;

import org.gvsig.fmap.data.feature.paging.FeaturePagingHelper;

import org.gvsig.fmap.data.feature.paging.FeaturePagingHelperImpl;

/**

 * TableModel to access data of a FeatureCollection.

 * 

 * @author <a href="mailto:cordin@disid.com">C?sar Ordi?ana</a>

 */

public class FeatureTableModel extends AbstractTableModel {

    private static final long serialVersionUID = -2488157521902851301L;

    private FeaturePagingHelper helper;

    /**

     * Constructs a TableModel from a FeatureCollection, with the default page

     * size.

     * 

     * @param featureCollection

     *            to extract data from

     * @throws ReadException

     *             if there is an error reading data from the FeatureStore

     */

    public FeatureTableModel(FeatureStore featureStore,

            FeatureQuery featureQuery) throws ReadException {

        this(featureStore, featureQuery, FeaturePagingHelper.DEFAULT_PAGE_SIZE);

    }

    /**

     * Constructs a TableModel from a FeatureCollection, and a given page size.

     * 

     * @param featureCollection

     *            to extract data from

     * @param pageSize

     *            the number of elements per page data

     * @throws ReadException

     *             if there is an error reading data from the FeatureStore

     */

    public FeatureTableModel(FeatureStore featureStore,

            FeatureQuery featureQuery, int pageSize) throws ReadException {

        this(new FeaturePagingHelperImpl(featureStore, featureQuery, pageSize));

    }

    /**

     * Constructs a TableModel from a FeatureCollection and a Paging helper.

     * 

     * @param featureCollection

     *            to extract data from

     * @param helper

     *            the paging helper

     * @throws ReadException

     *             if there is an error reading data from the FeatureStore

     */

    public FeatureTableModel(FeaturePagingHelper helper) {

        this.helper = helper;

        initialize();

    }

    public int getColumnCount() {

        // Return the number of fields of the Features

        FeatureType featureType = getFeatureType();

        return featureType.getFields().length;

    }

    public int getRowCount() {

        // Return the total size of the collection

        return getHelper().getTotalSize();

    }

    public Object getValueAt(int rowIndex, int columnIndex) {

        // Get the Feature at row "rowIndex", and return the value of the

        // attribute at "columnIndex"

        Feature feature = getFeatureAt(rowIndex);

        return getFeatureValue(feature, columnIndex);

    }

    @SuppressWarnings("unchecked")

    public Class getColumnClass(int columnIndex) {

        // Return the class of the FeatureAttributeDescriptor for the value

        FeatureAttributeDescriptor attributeDesc = getAttributeDescriptor(columnIndex);

        Class clazz = attributeDesc.getObjectClass();

        return (clazz == null ? super.getColumnClass(columnIndex) : clazz);

    }

    public String getColumnName(int column) {

        // Return the Feature attribute name

        FeatureAttributeDescriptor attributeDesc = getAttributeDescriptor(column);

        return attributeDesc.getName();

    }

    @Override

    public boolean isCellEditable(int rowIndex, int columnIndex) {

        return getFeatureStore().isEditing();

    }

    /**

     * TODO: the current implementation does not work, update to the last API

     * changes for Feature edition.

     */

    @Override

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

        // Get the feature at rowIndex

        Feature feature = getFeatureAt(rowIndex);

        // Only set the value if the feature exists

        if (feature != null) {

            // We only need to update if the value to set is not equal to the

            // current value

            Object currentValue = getFeatureValue(feature, columnIndex);

            if (value != currentValue

                    || (value != null && !value.equals(currentValue))) {

                try {

                    // getFeatureStore().startEditing();

                    //                    

                    // EditableFeature editable = feature.getEditableFeature();

                    // editable.set(columnIndex, value);

                    //                    

                    // getFeatureStore().update(editable);

                    // getFeatureStore().finishEditing();

                    getFeatureStore().startEditing();

                    feature.editing();

                    setFeatureValue(feature, columnIndex, value);

                    getFeatureStore().update(feature);

                    // getHelper().reload();

                    // TODO: REVISAR

                    // getHelper().setFeatureCollection(

                    // (FeatureCollection) getFeatureStore()

                    // .getDataCollection(getFeatureQuery()));

                    fireTableCellUpdated(rowIndex, columnIndex);

                } catch (DataException ex) {

                    throw new SetFeatureValueException(rowIndex, columnIndex,

                            value, ex);

                }

            }

        }

    }

    /**

     * Returns a reference to the Paging Helper used to load the data from the

     * DataStore.

     * 

     * @return the paging helper

     */

    public FeaturePagingHelper getHelper() {

        return helper;

    }

    /**

     * Initialize the TableModel

     */

    protected void initialize() {

        // Nothing to do

    }

    /**

     * Returns the value for a row position.

     * 

     * @param rowIndex

     *            the row position

     * @return the Feature

     */

    protected Feature getFeatureAt(int rowIndex) {

        return getHelper().getFeatureAt(rowIndex);

    }

    /**

     * Returns the value of a Feature attribute, at the given position.

     * 

     * @param feature

     *            the feature to get the value from

     * @param columnIndex

     *            the Feature attribute position

     * @return the value

     */

    protected Object getFeatureValue(Feature feature, int columnIndex) {

        return feature.get(columnIndex);

    }

    /**

     * Sets the value of an Feature attribute at the given position.

     * 

     * @param feature

     *            the feature to update

     * @param columnIndex

     *            the attribute position

     * @param value

     *            the value to set

     * @throws IsNotFeatureSettingException

     *             if there is an error setting the value

     */

    protected void setFeatureValue(Feature feature, int columnIndex,

            Object value) throws IsNotFeatureSettingException {

        feature.set(columnIndex, value);

    }

    /**

     * Returns the FeatureCollection used to get the data.

     * 

     * @return the FeatureCollection

     */

    protected FeatureCollection getFeatureCollection() {

        return getHelper().getFeatureCollection();

    }

    /**

     * Returns the FeatureStore of the Collection.

     * 

     * @return the FeatureStore

     */

    protected FeatureStore getFeatureStore() {

        return getHelper().getFeatureStore();

    }

    /**

     * Returns the FeatureQuery used to get the Features.

     * 

     * @return the FeatureQuery

     */

    protected FeatureQuery getFeatureQuery() {

        return getHelper().getFeatureQuery();

    }

    /**

     * Returns the type of the features.

     */

    private FeatureType getFeatureType() {

        return getFeatureCollection().getFeatureType();

    }

    /**

     * Returns the descriptor of a Feature attribute.

     */

    private FeatureAttributeDescriptor getAttributeDescriptor(int columnIndex) {

        return getFeatureType().getByOrder(columnIndex);

    }

}