Application: gvSIG desktop » gvSIG tools: Vector editing

/**

 * gvSIG. Desktop Geographic Information System.

 *

 * Copyright ? 2007-2014 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 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.

 *

 * 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.vectorediting.app.mainplugin;

import java.io.File;

import java.util.ArrayList;

import java.util.Iterator;

import java.util.List;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

import org.gvsig.andami.IconThemeHelper;

import org.gvsig.andami.PluginServices;

import org.gvsig.andami.PluginsLocator;

import org.gvsig.andami.actioninfo.ActionInfoManager;

import org.gvsig.andami.plugins.Extension;

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

import org.gvsig.app.ApplicationManager;

import org.gvsig.app.project.DocumentsContainer;

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

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

import org.gvsig.app.project.documents.view.ViewDocument;

import org.gvsig.app.project.documents.view.gui.DefaultViewPanel;

import org.gvsig.app.project.documents.view.gui.IView;

import org.gvsig.app.project.documents.view.toc.actions.EndEditingTocMenuEntry;

import org.gvsig.app.project.documents.view.toc.actions.StartEditingTocMenuEntry;

import org.gvsig.app.project.documents.view.toolListeners.StatusBarListener;

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

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

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

import org.gvsig.fmap.mapcontext.layers.vectorial.FLyrVect;

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

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

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

import org.gvsig.fmap.mapcontrol.MapControl;

import org.gvsig.fmap.mapcontrol.MapControlCreationListener;

import org.gvsig.fmap.mapcontrol.MapControlLocator;

import org.gvsig.fmap.mapcontrol.tools.Behavior.Behavior;

import org.gvsig.fmap.mapcontrol.tools.Behavior.MouseMovementBehavior;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.extensionpoint.ExtensionPoint;

import org.gvsig.tools.observer.Notification;

import org.gvsig.tools.observer.Observable;

import org.gvsig.tools.observer.Observer;

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

import org.gvsig.vectorediting.lib.spi.EditingProviderLocator;

import org.gvsig.vectorediting.lib.spi.EditingProviderManager;

import org.gvsig.vectorediting.swing.api.EditingContext;

import org.gvsig.vectorediting.swing.api.EditingSwingLocator;

import org.gvsig.vectorediting.swing.api.EditingSwingManager;

public class EditingExtension extends Extension implements Observer {

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

    public void execute(String actionCommand) {

        IView view = getActiveView();

        EditingSwingManager swingManager =

            EditingSwingLocator.getSwingManager();

        if (view != null) {

            FLyrVect layer = getActiveLayer(view);

            EditingContext editingContext =

                swingManager.getEditingContext(view.getMapControl());

            if ("start-editing".equals(actionCommand)) {

                if (canBeEdited(layer)) {

                    MapControl mapControl = view.getMapControl();

                    StatusBarListener sbl = new StatusBarListener(mapControl);

                    editingContext.beginEdition(layer,

                        new Behavior[] { new MouseMovementBehavior(sbl) });

                    editingContext.addObserver(this);

                    ApplicationLocator.getManager().refreshMenusAndToolBars();

                }

            } else if ("end-editing".equals(actionCommand)) {

                if ((layer != null) && layer.isEditing()) {

                    editingContext.endEdition(layer);

                    ApplicationLocator.getManager().refreshMenusAndToolBars();

                }

            }

        }

    }

    public void initialize() {

        registerIcons();

        // Disable default view panel console. Uses editing context console.

        DefaultViewPanel.setDisableConsole(true);

        // Adding TOC menu entry

        ExtensionPoint exPoint = ToolsLocator.getExtensionPointManager().add(

                "View_TocActions", "");

        exPoint.append(

            StartEditingTocMenuEntry.EXTENSION_POINT_NAME,

                "TOC popup menu to start vector layer's editing",

                new StartEditingTocMenuEntry());

        exPoint.append(

            EndEditingTocMenuEntry.EXTENSION_POINT_NAME,

                "TOC popup menu to end vector layer's editing",

                new EndEditingTocMenuEntry());

    }

    private void registerIcons() {

        IconThemeHelper.registerIcon("vectorediting", "vector-editing", this);

    }

    @Override

    public void postInitialize() {

        super.postInitialize();

        registerSymbols();

        MapControlLocator.getMapControlManager().addMapControlCreationListener(new MapControlCreationListener() {

            public MapControl mapControlCreated(MapControl mapControl) {

                EditingContext editingContext = EditingSwingLocator.getSwingManager().getEditingContext(mapControl);

                StatusBarListener sbl = new StatusBarListener(mapControl);

                editingContext.setDefaultBehaviors(new Behavior[] { new MouseMovementBehavior(sbl) });

                editingContext.addObserver(EditingExtension.this);

                ApplicationLocator.getManager().refreshMenusAndToolBars();

                return mapControl;

            }

        });

    }

    /**

     * Register all symbols in the plugin symbols folder in the providerManager.

     * The description of the symbols must be unique because the key used for registration is the proper description of the symbol.

     *

     */

    private void registerSymbols() {

        EditingProviderManager providerManager =

            EditingProviderLocator.getProviderManager();

        SymbolManager symbolManager = MapContextLocator.getSymbolManager();

        File pluginFolder = getPlugin().getPluginDirectory();

        String pathSeparator = System.getProperty("file.separator");

        String symbolsPath = pluginFolder.getAbsolutePath()+pathSeparator+"symbols"+pathSeparator+"editing";

        File symbolsFolder = new File(symbolsPath);

        ISymbol[] symbols = null;

        try {

            symbols = symbolManager.loadSymbols(symbolsFolder);

        } catch (SymbolException e) {

            logger.warn("No symbols loaded from "+symbolsPath, e);

        }

        if (symbols != null) {

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

                ISymbol symbol = symbols[i];

                String description = symbol.getDescription();

                providerManager.registerSymbol(description, symbol);

            }

        }

    }

    public boolean isEnabled() {

        return true;

    }

    public boolean isVisible() {

        return true;

    }

    @Override

    public boolean isVisible(String action) {

        IView view = getActiveView();

        FLyrVect activeLayer = getActiveLayer(view);

        if ("start-editing".equals(action)) {

            return ((view != null) && (activeLayer != null) && !activeLayer

                .isEditing());

        } else {

            return ((view != null) && (activeLayer != null) && activeLayer

                .isEditing());

        }

    }

    @Override

    public boolean isEnabled(String action) {

        IView vista = getActiveView();

        FLyrVect activeLayer = getActiveLayer(vista);

        if ("start-editing".equals(action)) {

            return (canBeEdited(activeLayer));

        } else if ("end-editing".equals(action)) {

            return activeLayer.isEditing();

        }

        return false;

    }

    @Override

    public boolean canQueryByAction() {

        return true;

    }

    private IView getActiveView() {

        ApplicationManager application = ApplicationLocator.getManager();

        IView view = (IView) application.getActiveComponent(ViewDocument.class);

        return view;

    }

    private boolean canBeEdited(FLyrVect layer) {

        if (layer != null && layer.isAvailable()) {

            boolean isWritable = layer.isWritable();

            boolean isNotTransformed =

                layer.getFeatureStore().getTransforms().isEmpty();

            return isWritable && isNotTransformed && !layer.isEditing();

        }

        return false;

    }

    private FLyrVect getActiveLayer(IView vista) {

        if (vista != null) {

            ViewDocument viewDocument = vista.getViewDocument();

            FLayer[] actives =

                viewDocument.getMapContext().getLayers().getActives();

            if ((actives.length == 1) && (actives[0] instanceof FLyrVect)) {

                return (FLyrVect) actives[0];

            }

        }

        return null;

    }

    public void update(Observable observable, Object notification) {

        if (notification instanceof Notification){

            ApplicationManager appManager =

                ApplicationLocator.getManager();

            Notification n = (Notification)notification;

            if (n.getType().equalsIgnoreCase(EditingContext.CHANGE_SELECTED_TOOL_NOTIFICATION)){

                String name = (String)n.getValue();

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

                PluginServices.getMainFrame().setSelectedTool(name);

                appManager.refreshMenusAndToolBars();

            } else if (n.getType().equalsIgnoreCase(EditingContext.REFRESH_TOOLS_NOTIFICATION)){

                appManager.refreshMenusAndToolBars();

            }

        }

    }

    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;

            }

        };

    }

}