Application: gvSIG desktop

/* gvSIG. Geographic Information System of the Valencian Government

 *

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

 * of the Valencian Government (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.

 * 

 */

package org.gvsig.app.project.documents.table;

import java.awt.Component;

import java.text.ParseException;

import java.util.ArrayList;

import java.util.Iterator;

import java.util.List;

import javax.swing.JOptionPane;

import javax.swing.event.TableModelListener;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

import org.gvsig.andami.PluginServices;

import org.gvsig.andami.messages.NotificationManager;

import org.gvsig.app.ApplicationLocator;

import org.gvsig.app.project.documents.table.gui.CreateNewAttributePanel;

import org.gvsig.app.project.documents.table.gui.FeatureTableDocumentPanel;

import org.gvsig.fmap.dal.DataTypes;

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

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

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

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

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

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

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

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

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

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

import org.gvsig.fmap.mapcontrol.dal.feature.swing.FeatureTable;

import org.gvsig.fmap.mapcontrol.dal.feature.swing.table.FeatureTableModel;

import org.gvsig.i18n.Messages;

import org.gvsig.tools.dispose.DisposableIterator;

/**

 * Feature Table Operations.

 * 

 * @author Vicente Caballero Navarro

 * 

 */

public class TableOperations {

    private static Logger logger = LoggerFactory.getLogger(TableOperations.class);

    public static final int MAX_FIELD_LENGTH = 254;

    private static TableOperations fto = null;

    private ArrayList<Feature> selectedFeatures = new ArrayList<Feature>();

    private boolean cutting = false;

    private FeatureTableDocumentPanel tablePanel = null;

    private FeatureStore featureStore;

    public static TableOperations getInstance() {

        if (fto == null) {

            fto = new TableOperations();

        }

        return fto;

    }

    public void setTablePanel(FeatureTableDocumentPanel tp) {

        tablePanel = tp;

        featureStore = tp.getModel().getStore();

    }

    public void copyFeatures() throws DataException {

        cutting = false;

        copy();

    }

    public boolean hasSelection() {

        return !selectedFeatures.isEmpty();

    }

    public void pasteFeatures() throws DataException {

        if (cutting) {

            delete();

            cutting = false;

        }

        Iterator<Feature> features = selectedFeatures.iterator();

        while (features.hasNext()) {

            Feature feature = features.next();

            featureStore.insert(feature.getEditable());

        }

    }

    public void cutFeatures() throws DataException {

        cutting = true;

        copy();

    }

    private void copy() throws DataException {

        DisposableIterator features = null;

        try {

            features =

                ((FeatureSelection) featureStore.getSelection()).fastIterator();

            selectedFeatures.clear();

            while (features.hasNext()) {

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

                selectedFeatures.add(feature);

            }

        } finally {

            if (features != null) {

                features.dispose();

            }

        }

    }

    private void delete() throws DataException {

        Iterator<Feature> features = selectedFeatures.iterator();

        while (features.hasNext()) {

            Feature feature = features.next();

            featureStore.delete(feature);

        }

    }

    public void deleteFeatures() throws DataException {

        FeatureTableModel _ftm = this.tablePanel.getTablePanel().getTableModel();

        List<TableModelListener> tmll = removeTableModelListeners(_ftm);

        DisposableIterator features = null;

        try {

            /*

            features =

                ((FeatureSelection) featureStore.getSelection()).fastIterator();

                */

            FeatureSet all_fset = featureStore.getFeatureSet();

            FeatureSelection sele = (FeatureSelection) featureStore.getSelection();

            features = all_fset.fastIterator();

            Feature item = null;

            while (features.hasNext()) {

                item = (Feature) features.next();

                if (sele.isSelected(item)) {

                    all_fset.delete(item);

                }

            }

            /*

            while (features.hasNext()) {

                features.remove();

            }

            */

        } finally {

            if (features != null) {

                features.dispose();

            }

            addTableModelListeners(_ftm, tmll);

        }

    }

    /**

     * @param _ftm

     * @param tmll

     */

    private void addTableModelListeners(

        FeatureTableModel _model,

        List<TableModelListener> _list) {

        Iterator<TableModelListener> iter = _list.iterator();

        while (iter.hasNext()) {

            _model.addTableModelListener(iter.next());

        }

        _model.fireTableDataChanged();

    }

    /**

     * @param ftm

     * @param class1

     * @return

     */

    private List<TableModelListener> removeTableModelListeners(FeatureTableModel ftm) {

        TableModelListener[] ll = ftm.getListeners(TableModelListener.class);

        List<TableModelListener> resp = new ArrayList<TableModelListener>();

        int n = ll.length;

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

            resp.add(ll[i]);

            ftm.removeTableModelListener(ll[i]);

        }

        return resp;

    }

    public void insertNewFeature() throws DataException {

        // if (getModel().getAssociatedTable()!=null){

        // JOptionPane.showMessageDialog((Component)PluginServices.getMainFrame(),"No se puede a?adir una fila a una tabla asociada a una capa.");

        // return;

        // }

        EditableFeature feature = featureStore.createNewFeature();

        featureStore.insert(feature);

    }

    public void deleteAttributes(FeatureTable table) throws DataException {

        EditableFeatureType eft =

            featureStore.getDefaultFeatureType().getEditable();

        FeatureAttributeDescriptor[] selecteds =

            table.getSelectedColumnsAttributeDescriptor();

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

            eft.remove(selecteds[i].getName());

        }

        featureStore.update(eft);

    }

    public void insertAttributes(FeatureTable table) throws DataException {

        EditableFeatureType eft =

            featureStore.getDefaultFeatureType().getEditable();

        try {

            CreateNewAttributePanel panelNewField =

                new CreateNewAttributePanel();

            EditableFeatureAttributeDescriptor ead =

                panelNewField.loadFieldDescription(eft);

            if (ead == null) {

                return;

            }

            if (ead.getType() == DataTypes.STRING

                && ead.getSize() > MAX_FIELD_LENGTH) {

                NotificationManager.showMessageInfo(

                    PluginServices.getText(this, "max_length_is") + ":"

                        + MAX_FIELD_LENGTH, null);

                ead.setSize(MAX_FIELD_LENGTH);

            }

            PluginServices.getMDIManager().closeWindow(panelNewField);

        } catch (ParseException e2) {

            NotificationManager.addError(e2);

        }

        featureStore.update(eft);

    }

    public void renameAttributes(FeatureTable table) throws DataException {

        FeatureType _ft = featureStore.getDefaultFeatureType();

        FeatureAttributeDescriptor[] selecteds =

            table.getSelectedColumnsAttributeDescriptor();

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

            String newName =

                JOptionPane.showInputDialog((Component) PluginServices

                    .getMDIManager().getActiveWindow(),

                    PluginServices.getText(

                    this, "_Please_insert_new_field_name_Cannot_be_undone"),

                    selecteds[i]

                    .getName());

            if (newName == null) {

                return;

            }

            if (_ft.getIndex(newName) != -1) {

                NotificationManager.showMessageInfo(

                    PluginServices.getText(this, "field_already_exists"), null);

                return;

            }

            renameAttribute(featureStore, selecteds[i].getName(), newName);

        }

        featureStore.finishEditing();

        // featureStore.edit(FeatureStore.MODE_FULLEDIT);

    }

    /**

     * This method renames a field in three steps:

     * 

     * (1) add new field using type and size of old field.

     * (2) copy value from old field to new field.

     * (3) remove old field.

     * 

     * @param fs

     * @param name

     * @param newName

     */

    private void renameAttribute(FeatureStore fs, String name, String newName) {

        try {

            // ========== add new field

            EditableFeatureType eft = fs.getDefaultFeatureType().getEditable();

            FeatureAttributeDescriptor fad = eft.getAttributeDescriptor(name);

            eft.add(newName, fad.getType(), fad.getSize());

            fs.update(eft);

            // ========== copy value old field -> new field

            FeatureSet fset = fs.getFeatureSet();

            DisposableIterator diter = fset.fastIterator();

            Feature feat = null;

            Object val = null;

            EditableFeature efeat = null;

            while (diter.hasNext()) {

                feat = (Feature) diter.next();

                val = feat.get(name);

                efeat = feat.getEditable();

                efeat.set(newName, val);

                fset.update(efeat);

            }

            diter.dispose();

            // ========== delete old field

            eft = fs.getDefaultFeatureType().getEditable();

            eft.remove(name);

            fs.update(eft);

        } catch (Exception ex) {

            logger.info("Unable to rename attribute (" + name + " --> " + newName + ")", ex);

            ApplicationLocator.getManager().message(

                Messages.getText("_Unable_to_rename_attribute"),

                JOptionPane.ERROR_MESSAGE);

        }

    }

}