Application: gvSIG desktop » gvSIG legends: base legends

package org.gvsig.legend.urbanhorizontalsignage.app;

import javax.swing.JOptionPane;

import org.gvsig.andami.IconThemeHelper;

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

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

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.mapcontext.MapContext;

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

import org.gvsig.legend.urbanhorizontalsignage.lib.api.UrbanHorizontalSignageLegend;

import org.gvsig.legend.urbanhorizontalsignage.lib.api.UrbanHorizontalSignageLocator;

import org.gvsig.legend.urbanhorizontalsignage.lib.api.UrbanHorizontalSignageManager;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.dispose.DisposeUtils;

import org.gvsig.tools.i18n.I18nManager;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

public class UrbanHorizontalSignageExtension extends Extension {

    private static final Logger LOGGER = LoggerFactory.getLogger(UrbanHorizontalSignageExtension.class);

    @Override

    public void initialize() {

        IconThemeHelper.registerIcon("legend", "legend-overview-vectorial-urbanhorizontalsignage", this);

        IconThemeHelper.registerIcon("action", "layer-modify-clipboard-copy-as-polygons", this);

    }

    @Override

    public void execute(String actionCommand) {

        if (actionCommand.compareToIgnoreCase("layer-modify-clipboard-copy-as-polygons") != 0) {

            return;

        }

        DataManager dataManager = DALLocator.getDataManager();

        ApplicationManager application = ApplicationLocator.getManager();

        ViewDocument viewdoc = (ViewDocument) application.getActiveDocument(ViewManager.TYPENAME);

        if (viewdoc == null) {

            return;

        }

        MapContext mapContext = viewdoc.getMapContext();

        FLyrVect layer = mapContext.getLayers().getFirstActiveVectorLayer();

        if (layer == null) {

            return;

        }

        FeatureStore targetStore = null;

        FeatureSet featureSet = null;

        try {

            UrbanHorizontalSignageManager uhsManager = UrbanHorizontalSignageLocator.getUrbanHorizontalSignageManager();

            //TODO: Presentar formulario para pedir el targetLenghtFieldName

            UrbanHorizontalSignageLegend legend = (UrbanHorizontalSignageLegend) layer.getLegend();

            FeatureSelection selection = layer.getFeatureStore().getFeatureSelection();

            targetStore = uhsManager.convertLinesToPolygons(selection, legend.getConfig(), null);

            featureSet = targetStore.getFeatureSet();

            I18nManager i18n = ToolsLocator.getI18nManager();

            int n = application.confirmDialog(

                    "_Number_of_geometries_to_be_copied_to_the_clipboard" + " " + featureSet.getSize() + "\n" + "_Do_you_want_to_continue",

                    i18n.getTranslation("_Copy_selected_features_as_polygons_to_clipboard"),

                    JOptionPane.YES_NO_OPTION,

                    JOptionPane.QUESTION_MESSAGE);

            if (n == JOptionPane.YES_OPTION) {

                dataManager.putFeaturesInClipboard(featureSet);

            }

        } catch (Exception e) {

            LOGGER.warn("Can't get the selection as JSON.", e);

        } finally {

            DisposeUtils.disposeQuietly(targetStore);

            DisposeUtils.disposeQuietly(featureSet);

        }

    }

    @Override

    public boolean isEnabled() {

        // it's enabled if there is exactly one vector layer in the active view

        // and it has a selection

        ApplicationManager application = ApplicationLocator.getManager();

        ViewDocument viewdoc = (ViewDocument) application.getActiveDocument(ViewManager.TYPENAME);

        if (viewdoc == null) {

            return false;

        }

        MapContext mapContext = viewdoc.getMapContext();

        FLyrVect layer = mapContext.getLayers().getFirstActiveVectorLayer();

        if (layer == null) {

            return false;

        }

        if (!layer.isAvailable()) {

            // This can happen when opening a persisted project

            // and there is a "slow" layer (GeoDB)

            return false;

        }

        try {

            FeatureStore store = layer.getFeatureStore();

            FeatureSelection selection = store.getFeatureSelection();

            if (selection.isEmpty()) {

                return false;

            }

            return (layer.getLegend() instanceof UrbanHorizontalSignageLegend);

//            return !selection.isEmpty();

        } catch (Exception ex) {

            LOGGER.warn("Can't get selection from layer '" + layer.getName() + "'.", ex);

            return false;

        }

    }

    @Override

    public boolean isVisible() {

        ApplicationManager application = ApplicationLocator.getManager();

        return application.getActiveDocument(ViewManager.TYPENAME) != null;

    }

}