Application: gvSIG desktop » gvSIG tools: Vector editing

import java.util.ArrayList;

import java.util.Iterator;

import java.util.List;

import org.gvsig.andami.PluginsLocator;

import org.gvsig.andami.actioninfo.ActionInfoManager;

import org.gvsig.andami.plugins.IExtension;

import org.gvsig.andami.plugins.status.IExtensionStatus;

import org.gvsig.andami.plugins.status.IUnsavedData;

import org.gvsig.andami.plugins.status.UnsavedData;

import org.gvsig.andami.ui.mdiManager.IWindow;

import org.gvsig.andami.ui.mdiManager.SingletonWindow;

import org.gvsig.app.project.DocumentsContainer;

import org.gvsig.app.project.documents.Document;

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

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

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

import org.gvsig.fmap.mapcontext.MapContext;

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

import org.gvsig.utils.swing.threads.IMonitorableTask;

                logger.info("Changed selected tool to '{}'", name);

    private List<FLyrVect> getEditedLayers(){

        List<FLyrVect> editedLayers = new ArrayList<FLyrVect>();

        ApplicationManager application = ApplicationLocator.getManager();

        IWindow[] windows = application.getUIManager().getOrderedWindows();

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

            IWindow window = windows[i];

            if (window instanceof SingletonWindow && window instanceof IDocumentWindow) {

                // Recorremos las ventanas.

                // Cogemos el model de aquellas que sea de tipo Document.

                // Si el model es de tipo View,

                // comprobamos que adem?s sean de tipo View.

                // Le pedimos al mapContext de la View las layers

                // y a?adimos a editedLayers aquellas layers vectoriales que est?n en edici?n.

                Document document;

                try {

                    document = (Document) ((SingletonWindow) window).getWindowModel();

                } catch (ClassCastException e) {

                    // Do nothing, skip this window

                    continue;

                }

                if (ViewDocument.class.isAssignableFrom(document.getClass())) {

                    ViewDocument view = (ViewDocument) document;

                    MapContext mapContext = view.getMapContext();

                    if (mapContext != null) {

                        FLayers layers = mapContext.getLayers();

                        for (int j = 0; j < layers.getLayersCount(); j++) {

                            FLayer lyr = layers.getLayer(j);

                            if (lyr instanceof FLyrVect) {

                                FLyrVect layerVect = (FLyrVect) lyr;

                                if (layerVect.isEditing()) {

                                    editedLayers.add(layerVect);

                                }

                            }

                        }

                    }

                }

            }

        }

        return editedLayers;

    }

    private List<IUnsavedData> getUnsavedData(List<FLyrVect> editedLayers){

        List<IUnsavedData> unsavedData = new ArrayList<IUnsavedData>();

        for (Iterator iterator = editedLayers.iterator(); iterator.hasNext();) {

            FLyrVect editedLayer = (FLyrVect) iterator.next();

            IUnsavedData unsavedDataLayer = new UnsavedLayer(this, editedLayer);

            unsavedData.add(unsavedDataLayer);

        }

        return unsavedData;

    }

    private static class UnsavedLayer extends UnsavedData {

        private FLyrVect layer;

        public UnsavedLayer(IExtension extension, FLyrVect layer) {

            super(extension);

            this.layer = layer;

        }

        @Override

        public String getDescription() {

            return layer.getName();

        }

        @Override

        public String getResourceName() {

            return layer.getFeatureStore().getFullName();

        }

        @Override

        public boolean saveData() {

            FeatureStore featureStore = layer.getFeatureStore();

            if(featureStore.isEditing()){

                try {

                    featureStore.finishEditing();

                    return true;

                } catch (DataException e) {

                    throw new RuntimeException(e);

                }

            }

            return true;

        }

    }

    @Override

    public IExtensionStatus getStatus() {

        List<FLyrVect> editedLayers = getEditedLayers();

        final List<IUnsavedData> unsavedData = getUnsavedData(editedLayers);

        return new IExtensionStatus() {

            public boolean hasUnsavedData() {

                if(unsavedData == null){

                    return false;

                }

                return !unsavedData.isEmpty();

            }

            public boolean hasRunningProcesses() {

                return false;

            }

            public IUnsavedData[] getUnsavedData() {

                if(unsavedData == null){

                    return null;

                }

                return unsavedData.toArray(new IUnsavedData[0]);

            }

            public IMonitorableTask[] getRunningProcesses() {

                return null;

            }

        };

    }