Application: gvSIG desktop

package org.gvsig.app.extension.geometriespalette;

import java.awt.Dimension;

import javax.swing.SwingUtilities;

import org.apache.commons.lang3.StringUtils;

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.addlayer.AddLayerDialog;

import static org.gvsig.app.extension.AddLayer.createAddLayerDialog;

import org.gvsig.app.gui.WizardPanel;

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

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

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

import org.gvsig.fmap.crs.CRSFactory;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

import org.gvsig.fmap.dal.DataStore;

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

import org.gvsig.fmap.mapcontext.MapContext;

import org.gvsig.fmap.mapcontext.MapContextLocator;

import org.gvsig.fmap.mapcontext.MapContextManager;

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

import org.gvsig.fmap.mapcontrol.MapControl;

import org.gvsig.fmap.mapcontrol.MapControlLocator;

import org.gvsig.fmap.mapcontrol.MapControlManager;

import org.gvsig.tools.arguments.Arguments;

import org.gvsig.tools.dispose.DisposeUtils;

import org.gvsig.tools.swing.api.ToolsSwingLocator;

import org.gvsig.tools.swing.api.ToolsSwingUtils;

import org.gvsig.tools.swing.api.windowmanager.WindowManager;

/**

 *

 * @author gvSIG Team

 */

@SuppressWarnings("UseSpecificCatch")

public class GeometriesPaletteExtension extends Extension {

    @Override

    public void initialize() {

    }

    @Override

    public void execute(String actionCommand) {

        if (StringUtils.equalsIgnoreCase("tools-open-geometriespalette", actionCommand)) {

            doOpenGeometriesPalette();

        }

    }

    @Override

    public void execute(String command, Object[] args) {

        Arguments arguments = Arguments.create(args);

        if (StringUtils.equalsIgnoreCase("tools-open-geometriespalette", command)) {

            String labelattr = (String) arguments.get("label", String.class, null);

            String tableName = (String) arguments.get("tableName", String.class, null);

            String title = (String) arguments.get("title", String.class, null);

            String filter = (String) arguments.get("filter", String.class, null);

            if( StringUtils.isBlank(tableName) ) {

                doOpenGeometriesPalette();

                return;

            }

            doOpenGeometriesPalette(tableName, labelattr, title, filter);

        }

    }

    @SuppressWarnings("SleepWhileInLoop")

    private void doOpenGeometriesPalette() {

        AddLayerDialog addLayerDialog = null;

        final WindowManager windowManager = ToolsSwingLocator.getWindowManager();

        MapContextManager mapContextManager = MapContextLocator.getMapContextManager();

        MapControlManager mapControlManager = MapControlLocator.getMapControlManager();

        final MapContext mapContext = mapContextManager.createMapContext();

        MapControl mapControl = null;

        try {

            ApplicationManager application = ApplicationLocator.getApplicationManager();

            ViewDocument view = (ViewDocument) application.getCurrentProject().getFirstDocument(ViewDocument.class);

            if( view == null || view.getProjection()==null ) {

                mapContext.setProjection(CRSFactory.getCRS("EPSG:4326"));

            } else {

                mapContext.setProjection(view.getProjection());

            }

            mapControl = mapControlManager.createJMapControlPanel(mapContext);

            addLayerDialog = createAddLayerDialog(mapControl, mapContext, null);

            Dimension screensize = ToolsSwingUtils.getCurrentScreenSize();            

            ToolsSwingUtils.ensureHeightWitdh(

                    addLayerDialog, 

                    3*(screensize.height/4), 

                    2*(screensize.width/4), 

                    7*(screensize.height/8), 

                    3*(screensize.width/4)

            );

            windowManager.showWindow(

                    addLayerDialog,

                    "Select palette layer",

                    WindowManager.MODE.DIALOG

            );

            if (addLayerDialog.isAccepted()) {

                if (addLayerDialog.getSelectedTab() instanceof WizardPanel) {

                    WizardPanel panel = (WizardPanel) addLayerDialog.getSelectedTab();

                    panel.executeWizard();

                    // Las capas se cargan en segundo plano, asi que si tratamos de 

                    // acceder a ellas ya mismo aun no estan cargadas.

                    // No es muy ortodox pero nos esperaremos un poco.

                    Thread task = new Thread(() -> {

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

                            try { Thread.sleep(1000); } catch (Exception ex) { }

                            if (!mapContext.getLayers().isEmpty()) {

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

                                SwingUtilities.invokeLater(() -> {

                                    GeometriesPalettePanel.showPalette(layer.getFeatureStore(), null, layer.getName(), null);

                                });

                                return;

                            }

                        }

                        // TODO: Avisar que no se ha cargado ninguna capa ?

                    }, "GeometriesPaleteLoader");

                    task.start();

                }

            }

        } catch (Exception ex) {

            logger.warn("Can't open palete from layer",ex);

        } finally {

            DisposeUtils.disposeQuietly(addLayerDialog);

            DisposeUtils.disposeQuietly(mapControl);

            DisposeUtils.disposeQuietly(mapContext);

        }

    }

    private void doOpenGeometriesPalette(String tableName, String labelattr, String title, String filter) {

        DataManager dataManager = DALLocator.getDataManager();

        DataStore store = dataManager.getStoresRepository().getStore(tableName);

        if( !(store instanceof FeatureStore) ) {

            logger.warn("The table name '"+tableName+"' is not valid.");

            return;

        }

        FeatureStore featureStore = (FeatureStore) store;

        GeometriesPalettePanel.showPalette(featureStore, labelattr, title, filter);

    }

    @Override

    public boolean isEnabled() {

        return true;

    }

    @Override

    public boolean isVisible() {

        return true;

    }

}