Application: gvSIG desktop » gvSIG Linear Reference System

/* gvSIG. Desktop Geographic Information System.

 *

 * Copyright ? 2007-2015 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.lrs.app.generatedynamicsegmentation;

import java.awt.event.ActionEvent;

import java.awt.event.ActionListener;

import java.util.ArrayList;

import java.util.List;

import javax.swing.JOptionPane;

import org.apache.commons.lang3.StringUtils;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

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

import org.gvsig.app.project.documents.table.TableDocument;

import org.gvsig.app.project.documents.table.TableManager;

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

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

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

import org.gvsig.fmap.dal.store.shp.SHPNewStoreParameters;

import org.gvsig.fmap.mapcontext.MapContextLocator;

import org.gvsig.fmap.mapcontext.MapContextManager;

import org.gvsig.fmap.mapcontext.exceptions.LoadLayerException;

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

import org.gvsig.fmap.mapcontext.layers.operations.LayerCollection;

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

import org.gvsig.lrs.lib.api.LrsAlgorithm;

import org.gvsig.lrs.lib.api.LrsAlgorithmsLocator;

import org.gvsig.lrs.lib.api.LrsAlgorithmsManager;

import org.gvsig.lrs.lib.api.LrsGenerateDynamicSegmentationAlgorithmParams;

import org.gvsig.lrs.lib.api.exceptions.LrsGettingParametersException;

import org.gvsig.lrs.lib.api.exceptions.LrsNeededParameterException;

import org.gvsig.lrs.swing.api.JLrsAlgorithmParams;

import org.gvsig.lrs.swing.api.JLrsProgressDialog;

import org.gvsig.lrs.swing.api.LrsAlgorithmsSwingLocator;

import org.gvsig.lrs.swing.api.LrsAlgorithmsSwingManager;

import org.gvsig.lrs.swing.impl.JLrsGenerateDynamicSegmentationParamsController;

import org.gvsig.lrs.swing.impl.JLrsUtils;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.i18n.I18nManager;

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

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

import org.gvsig.tools.task.SimpleTaskStatus;

/**

 * @author dmartinez

 *

 */

public class GenerateDynamicSegmentationExtension extends Extension {

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

    /* (non-Javadoc)

     * @see org.gvsig.andami.plugins.IExtension#execute(java.lang.String)

     */

    public void execute(String actionCommand) {

        if (StringUtils.equalsIgnoreCase(actionCommand, "generate-dynamic-segmentation")) {

            IView view = getActiveView();

            FLayers layers = view.getMapControl().getMapContext().getLayers();

            List<FeatureStore> tables=getTables();

            final WindowManager winManager = ToolsSwingLocator.getWindowManager();

            final LrsAlgorithmsSwingManager manager = LrsAlgorithmsSwingLocator.getLrsAlgorithmsSwingManager();

            final I18nManager i18nManager = ToolsLocator.getI18nManager();

            JLrsGenerateDynamicSegmentationParamsController panel;

            try {

                panel = (JLrsGenerateDynamicSegmentationParamsController)manager.createJLrsGenerateDynamicSegmentationAlgorithmParameters(layers,tables, null);

            } catch (LrsNeededParameterException e2) {

                logger.error("Error creating panel", e2);

                JOptionPane.showMessageDialog(

                    null,

                    new StringBuilder().append(i18nManager.getTranslation("error_creating_panel"))

                        .append(":").append(e2.getLocalizedMessage()),

                    i18nManager.getTranslation("error"), JOptionPane.ERROR_MESSAGE);

                return;

            }

            panel.setVisibleAceptCancel(true);

            panel.addActionListener(new ActionListener() {

                public void actionPerformed(ActionEvent e) {

                    JLrsAlgorithmParams panel = (JLrsAlgorithmParams) e.getSource();

                    if (panel.isCanceled()) {

                        return;

                    }

                    LrsGenerateDynamicSegmentationAlgorithmParams params = null;

                    try {

                        params = (LrsGenerateDynamicSegmentationAlgorithmParams)panel.getParams();

                    } catch (LrsGettingParametersException e1) {

                        logger.error("Error getting parameters", e1);

                        JOptionPane.showMessageDialog(

                            null,

                            new StringBuilder().append(i18nManager.getTranslation("error_getting_parameters"))

                                .append(":").append(e1.getLocalizedMessage()),

                            i18nManager.getTranslation("error"), JOptionPane.ERROR_MESSAGE);

                        return;

                    } catch (LrsNeededParameterException e1) {

                        logger.warn("Error getting parameters", e1);

                        JOptionPane.showMessageDialog(

                            null,

                            i18nManager.getTranslation(e1.getMessage()),

                            i18nManager.getTranslation("warning"), JOptionPane.WARNING_MESSAGE);

                        return;

                    }

                    LrsAlgorithmsManager algorithmsManager = LrsAlgorithmsLocator.getLrsAlgorithmsManager();

                    final LrsAlgorithm algorithm = algorithmsManager.createLrsAlgorithm(params);

                    final SHPNewStoreParameters shpParams=(SHPNewStoreParameters)params.getNewFeatureStoreParameters();

                    final SimpleTaskStatus taskStatus =

                        ToolsLocator.getTaskStatusManager().createDefaultSimpleTaskStatus(i18nManager.getTranslation("progress"));

                    final JLrsProgressDialog progressDialog = manager.createJLrsProgressDialog(taskStatus);

                    winManager.showWindow(progressDialog.asJComponent(), i18nManager.getTranslation("generate_dynamic_segmentation"),

                        WindowManager.MODE.WINDOW);

                    Thread task = new Thread(new Runnable() {

                        public void run() {

                            try {

                                algorithm.execute(taskStatus);

                                int resultLoadLayer =JOptionPane.showConfirmDialog(

                                    null,

                                    i18nManager.getTranslation("load_layer_question"),

                                    i18nManager.getTranslation("load_layer_question_title"), JOptionPane.YES_NO_OPTION);

                                if (resultLoadLayer==JOptionPane.YES_OPTION){

                                    loadLayer(shpParams);

                                }

                            } catch (Exception e) {

                                logger.error("Error generating dynamic segmentation ", e);

                                String message = e.getMessage();

                                if (e.getCause() != null) {

                                    message = e.getCause().getMessage();

                                }

                                JOptionPane.showMessageDialog(

                                    progressDialog.asJComponent(),

                                    message,

                                    i18nManager.getTranslation("generate_dynamic_segmentation"),

                                    JOptionPane.ERROR_MESSAGE

                                );

                            }

                        }

                    });

                    task.start();

                }

            });

            winManager.showWindow(panel.asJComponent(), i18nManager.getTranslation("generate_dynamic_segmentation"),

                WindowManager.MODE.WINDOW);

        }

    }

    /* (non-Javadoc)

     * @see org.gvsig.andami.plugins.IExtension#initialize()

     */

    public void initialize() {

        registerIcons();

    }

    /* (non-Javadoc)

     * @see org.gvsig.andami.plugins.IExtension#isEnabled()

     */

    public boolean isEnabled() {

        IView view = getActiveView();

        if (view!=null){

            ViewDocument viewDocument = view.getViewDocument();

            LayerCollection layers = viewDocument.getMapContext().getLayers();

            if (JLrsUtils.validateMLayerWithNonGeomField(layers)){

                List<FeatureStore> tables=getTables();

                if (tables!=null&&tables.size()>0){

                    return JLrsUtils.validateStoreWithFieldsAndOneNumeric(tables);

                }

            }

        }

        return false;

    }

    /* (non-Javadoc)

     * @see org.gvsig.andami.plugins.IExtension#isVisible()

     */

    public boolean isVisible() {

        return true;

    }

    private void registerIcons() {

        IconThemeHelper.registerIcon("lrs", "pk_yellow", this);

    }

    private List<FeatureStore> getTables(){

        ApplicationManager application = ApplicationLocator.getManager();

        List<Document> documents=application.getCurrentProject().getDocuments(TableManager.TYPENAME);

        List<FeatureStore> tables=new ArrayList<FeatureStore>();

        if (documents!=null){

            for (int i = 0; i < documents.size(); i++) {

                TableDocument tableDocument=(TableDocument)documents.get(i);

                tables.add(tableDocument.getStore());

            }

        }

        return tables;

    }

    private IView getActiveView() {

        ApplicationManager application = ApplicationLocator.getManager();

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

        return view;

    }

    private void loadLayer(SHPNewStoreParameters shpParams){

        if (shpParams!=null ){

            String name=shpParams.getFile().getName();

            if (name.contains(".shp")){

                name=name.substring(0,name.lastIndexOf(".shp"));

            }

            MapContextManager mapContextManager =MapContextLocator.getMapContextManager();

            FLyrVect newLayer=null;

            try {

                newLayer = (FLyrVect) mapContextManager.createLayer(name,shpParams);

            } catch (LoadLayerException e) {

                logger.warn("Couldn't load layer",e);

            }

            getActiveView().getMapControl().getMapContext().getLayers().addLayer(newLayer);

        }

    }

}