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 org.slf4j.Logger;

import org.slf4j.LoggerFactory;

import org.gvsig.andami.IconThemeHelper;

import org.gvsig.andami.PluginServices;

import org.gvsig.andami.plugins.Extension;

import org.gvsig.app.ApplicationLocator;

import org.gvsig.app.ApplicationManager;

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.toolListeners.StatusBarListener;

import org.gvsig.fmap.mapcontext.MapContextLocator;

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

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.tools.Behavior.Behavior;

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

import org.gvsig.tools.observer.Notification;

import org.gvsig.tools.observer.Observable;

import org.gvsig.tools.observer.Observer;

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);

    }

    private void registerIcons() {

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

    }

    @Override

    public void postInitialize() {

        super.postInitialize();

        registerSymbols();

    }

    /**

     * 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();

                PluginServices.getMainFrame().setSelectedTool(name);

                appManager.refreshMenusAndToolBars();

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

                appManager.refreshMenusAndToolBars();

            }

        }

    }

}