Application: gvSIG desktop » gvSIG Linear Reference System

/*

 * Copyright 2015 DiSiD Technologies S.L.L. All rights reserved.

 *

 * Project  : DiSiD org.gvsig.lrs.swing.impl

 * SVN Id   : $Id$

 */

package org.gvsig.lrs.swing.impl;

import java.awt.event.ActionEvent;

import java.awt.event.ActionListener;

import java.awt.event.ItemEvent;

import java.awt.event.ItemListener;

import java.io.File;

import java.util.List;

import java.util.Locale;

import javax.swing.ComboBoxModel;

import javax.swing.DefaultComboBoxModel;

import javax.swing.JComponent;

import javax.swing.JOptionPane;

import javax.swing.ListCellRenderer;

import org.apache.commons.io.FileUtils;

import org.apache.commons.lang3.StringUtils;

import org.cresques.cts.IProjection;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

import org.gvsig.fmap.dal.DataStoreParameters;

import org.gvsig.fmap.dal.exception.DataException;

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

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

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

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

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

import org.gvsig.fmap.geom.DataTypes;

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

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

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

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

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

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

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

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

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

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

import org.gvsig.tools.i18n.I18nManager;

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

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

import org.gvsig.tools.swing.api.threadsafedialogs.ThreadSafeDialogsManager;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

public class JLrsGenerateDynamicSegmentationParamsController extends JLrsGenerateDynamicSegmentationParamsView  implements JLrsAlgorithmParams{

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

    private boolean canceled;

    private ActionListenerSupport listeners = ToolsSwingLocator.getToolsSwingManager().createActionListenerSupport();

    private LayerCollection layers;

    /**

     * @param layers

     * @throws LrsNeededParameterException

     *

     */

    public JLrsGenerateDynamicSegmentationParamsController(LayerCollection layers, List<FeatureStore> stores , LrsAlgorithmParams params) throws LrsNeededParameterException {

        super();

        this.layers = layers;

        initComponents();

        translate();

        setParams(params);

    }

    public void addActionListener(ActionListener listener) {

        listeners.addActionListener(listener);

    }

    public void removeActionListener(ActionListener listener) {

        listeners.removeActionListener(listener);

    }

    private void translate() {

        I18nManager i18nManager = ToolsLocator.getI18nManager();

        lblInputLayer.setText(i18nManager.getTranslation(lblInputLayer.getText()));

        lblIdRouteField.setText(i18nManager.getTranslation(lblIdRouteField.getText()));

        lblTable.setText(i18nManager.getTranslation(lblTable.getText()));

        lblTableIdRouteField.setText(i18nManager.getTranslation(lblTableIdRouteField.getText()));

        lblLandmarkField.setText(i18nManager.getTranslation(lblLandmarkField.getText()));

        lblFinalLandmarkField.setText(i18nManager.getTranslation(lblFinalLandmarkField.getText()));

        lblValueField.setText(i18nManager.getTranslation(lblValueField.getText()));

        lblOutputFile.setText(i18nManager.getTranslation(lblOutputFile.getText()));

        btnBrowser.setText("\u2026");

        btnAccept.setText(i18nManager.getTranslation(btnAccept.getText()));

        btnCancel.setText(i18nManager.getTranslation(btnCancel.getText()));

    }

    /**

     *

     */

    public void setLocate(Locale locale) {

        Locale l = super.getLocale();

        if (!l.equals(locale)) {

            translate();

        }

        super.setLocale(locale);

    }

    /**

     * @throws LrsNeededParameterException

     */

    private void initComponents() throws LrsNeededParameterException {

        //TODO table como DataStore?

        if (layers==null||layers.getLayersCount()<1){

            //TODO Comprobar vectorial tambien

            throw new LrsNeededParameterException("need_layers", null);

        }

        // Modelos de los combos

        FLayersComboBoxModel inputLayerModel = new FLayersComboBoxModel(this.layers);

        FeatureAttributeDescriptorsComboBoxModel idRouteFieldComboBoxModel =

            new FeatureAttributeDescriptorsComboBoxModel();

        FLayersComboBoxModel tableLayerModel = new FLayersComboBoxModel(this.layers);

        FeatureAttributeDescriptorsComboBoxModel tableIdRouteFieldComboBoxModel =

            new FeatureAttributeDescriptorsComboBoxModel();

        FeatureAttributeDescriptorsComboBoxModel landmarkComboBoxModel =

            new FeatureAttributeDescriptorsComboBoxModel();

        FeatureAttributeDescriptorsComboBoxModel finalLandmarkComboBoxModel =

            new FeatureAttributeDescriptorsComboBoxModel();

        FeatureAttributeDescriptorsComboBoxModel valueComboBoxModel =

            new FeatureAttributeDescriptorsComboBoxModel();

        // Renderer para los combos de attributeDescriptors

        ListCellRenderer featureAttributeDescriptorCellRenderer = new FLayersAttributeDescriptorCellRenderer();

        // asignaci?n de renderers y modelos

        cmbIdRouteField.setRenderer(featureAttributeDescriptorCellRenderer);

        cmbTableIdRouteField.setRenderer(featureAttributeDescriptorCellRenderer);

        cmbLandmarkField.setRenderer(featureAttributeDescriptorCellRenderer);

        cmbFinalLandmarkField.setRenderer(featureAttributeDescriptorCellRenderer);

        cmbValueField.setRenderer(featureAttributeDescriptorCellRenderer);

        cmbIdRouteField.setModel(idRouteFieldComboBoxModel);

        cmbTableIdRouteField.setModel(tableIdRouteFieldComboBoxModel);

        cmbLandmarkField.setModel(landmarkComboBoxModel);

        cmbFinalLandmarkField.setModel(finalLandmarkComboBoxModel);

        cmbValueField.setModel(valueComboBoxModel);

        // renderer para el combo de capas

        ListCellRenderer layerNameCellRenderer=new FLayersLayerNameCellRenderer();

        cmbInputLayer.setRenderer(layerNameCellRenderer);

        cmbTable.setRenderer(layerNameCellRenderer);

        // asignaci?n modelo al combo de capas

        cmbInputLayer.setModel(inputLayerModel);

        cmbTable.setModel(tableLayerModel);

        // listener del combo de capas para que actualice el resto de

        // componentes

        cmbInputLayer.addItemListener(new ItemListener() {

            public void itemStateChanged(ItemEvent e) {

                Object item = e.getItem();

                if (item instanceof FLyrVect) {

                    FLyrVect layer = (FLyrVect) item;

                    updateLayerSource(layer);

                }

            }

        });

        cmbTable.addItemListener(new ItemListener() {

            public void itemStateChanged(ItemEvent e) {

                Object item = e.getItem();

                if (item instanceof FLyrVect) {

                    FLyrVect layer = (FLyrVect) item;

                    updateLayerTable(layer);

                }

            }

        });

        this.btnBrowser.addActionListener(new ActionListener() {

            public void actionPerformed(ActionEvent e) {

                doSelectOuputFile();

            }

        });

        this.btnAccept.addActionListener(new ActionListener() {

            public void actionPerformed(ActionEvent e) {

                canceled = false;

                doClose();

                listeners.fireActionEvent(new ActionEvent(JLrsGenerateDynamicSegmentationParamsController.this, 0, "accept"));

            }

        });

        this.btnCancel.addActionListener(new ActionListener() {

            public void actionPerformed(ActionEvent e) {

                canceled = true;

                doClose();

                listeners.fireActionEvent(new ActionEvent(JLrsGenerateDynamicSegmentationParamsController.this, 0, "cancel"));

            }

        });

        setVisibleAceptCancel(false);

        canceled = false;

    }

    private void doSelectOuputFile() {

        final I18nManager i18nManager = ToolsLocator.getI18nManager();

        ThreadSafeDialogsManager manager = ToolsSwingLocator.getThreadSafeDialogsManager();

        File[] files = manager.showSaveFileDialog(i18nManager.getTranslation("output_file"), null);

        if (files != null && files.length > 0) {

            File file = files[0];

            String absolutePath = file.getAbsolutePath();

            if(!StringUtils.endsWithIgnoreCase(absolutePath, ".shp")){

                absolutePath = new StringBuilder().append(absolutePath).append(".shp").toString();

            }

            File outputFile = FileUtils.getFile(absolutePath);

            if(outputFile.exists()) {

                int resp = JOptionPane.showOptionDialog(

                    this.asJComponent(),

                    i18nManager.getTranslation("file_already_exists_do_you_want_overwrite_it"),

                    i18nManager.getTranslation("output_file"),

                    JOptionPane.OK_CANCEL_OPTION,

                    JOptionPane.QUESTION_MESSAGE,

                    null,

                    null,

                    JOptionPane.OK_OPTION);

                if (resp == JOptionPane.OK_OPTION){

                    this.txtOutputFile.setText(absolutePath);

                } else {

                    doSelectOuputFile();

                }

            } else {

                this.txtOutputFile.setText(absolutePath);

            }

        }

    }

    private void doClose() {

        this.setVisible(false);

    }

    public boolean isCanceled() {

        return this.canceled;

    }

    public void setVisibleAceptCancel(boolean visible) {

        this.btnAccept.setVisible(visible);

        this.btnCancel.setVisible(visible);

    }

    public boolean getVisibleAceptCancel() {

        return this.btnAccept.isVisible();

    }

    public JComponent asJComponent() {

        return this;

    }

    public LrsAlgorithmParams getParams() throws LrsGettingParametersException, LrsNeededParameterException {

//        FeatureAttributeDescriptor idRouteField = (FeatureAttributeDescriptor) this.cmbIdRouteField.getSelectedItem();

//        if (idRouteField == null) {

//            throw new LrsNeededParameterException("need_to_select_a_route_identifier_field", null);

//        }

//        String outputLayerPath = this.txtOutputFile.getText();

//        if(StringUtils.isEmpty(outputLayerPath)){

//            throw new LrsNeededParameterException("need_to_fill_the_target_path", null);

//        }

//

        LrsAlgorithmsManager manager = LrsAlgorithmsLocator.getLrsAlgorithmsManager();

        LrsCalibrateRouteAlgorithmParams params = manager.createLrsCalibrateRouteAlgorithmParams();

//        // Recorrer todo del interfaz y rellenar los parametros

//        FeatureStore sourceFeatureStore = ((FLyrVect) this.cmbInputLayer.getSelectedItem()).getFeatureStore();

//        params.setSourceFeatureStore(sourceFeatureStore);

//        params.setIdRouteField((FeatureAttributeDescriptor) this.cmbIdRouteField.getSelectedItem());

//        FeatureStore calibratePointFeatureStore = ((FLyrVect) this.cmbCalibratePointLayer.getSelectedItem()).getFeatureStore();

//        params.setCalibratePointFeatureStore(calibratePointFeatureStore);

//        params.setCalibratePointIdRouteField((FeatureAttributeDescriptor) this.cmbCalibrationPointIdRouteField.getSelectedItem());

//        params.setFromMeasureField((FeatureAttributeDescriptor) this.cmbFromMeasureField.getSelectedItem());

//

//        SHPNewStoreParameters sHPNewStoreParameters=new SHPNewStoreParameters();

//        sHPNewStoreParameters.setFile(outputLayerPath);

//        params.setNewFeatureStoreParameters(sHPNewStoreParameters);

        // Crear un SHP a partir de la ruta con el atributo de indentificadores

        // de rutas y pasar el featureStore

//        DataManager dataManager = DALLocator.getDataManager();

//        FeatureType featureType = dataManager.createFeatureType();

//        EditableFeatureType editableFeatureType;

//        if (featureType instanceof EditableFeatureType) {

//            editableFeatureType = (EditableFeatureType) featureType;

//        } else {

//            editableFeatureType = featureType.getEditable();

//        }

//        editableFeatureType.add(idRouteField.getName(), idRouteField.getType(), idRouteField.getSize());

//        editableFeatureType.add("GEOMETRY", DataTypes.GEOMETRY);

//        editableFeatureType.setDefaultGeometryAttributeName("GEOMETRY");

//        EditableFeatureAttributeDescriptor geomAttributeDescriptor =

//            editableFeatureType.getEditableAttributeDescriptor("GEOMETRY");

//        // FIXME: Cambiar por GEOM2DM;

//        try {

//            geomAttributeDescriptor.setGeometryType(GeometryLocator.getGeometryManager().getGeometryType(

//                TYPES.MULTICURVE, SUBTYPES.GEOM3D));

//        } catch (GeometryException e2) {

//            throw new LrsGettingParametersException("Error setting geometry type in attribute descriptor", e2);

//        } catch (LocatorException e2) {

//            throw new LrsGettingParametersException("Error getting geometry manager", e2);

//        }

//

//        FeatureType sourceFeatureType;

//        try {

//            sourceFeatureType = sourceFeatureStore.getDefaultFeatureType();

//        } catch (DataException e1) {

//            throw new LrsGettingParametersException("Error getting feature type from source feature store", e1);

//        }

//        IProjection sourceProjection = sourceFeatureType.getDefaultSRS();

//        try {

//            manager.createNewShape(editableFeatureType, TYPES.MULTICURVE, outputLayerPath, sourceProjection);

//        } catch (Exception e1) {

//            throw new LrsGettingParametersException(new StringBuilder().append("Error creating a new shape in ")

//                .append(outputLayerPath).toString(), e1);

//        }

//        FeatureStore outputFeatoreStore;

//        try {

//            outputFeatoreStore = getFeatureStore(outputLayerPath, sourceProjection);

//        } catch (Exception e) {

//            throw new LrsGettingParametersException(new StringBuilder().append("Error getting feature store from ")

//                .append(outputLayerPath).toString(), e);

//        }

//        params.setOutputFeatureStore(outputFeatoreStore);

        // FIN DE Crear un SHP a partir de la ruta y de y pasar el featureStore

//        LrsMeasureCalculationMethods measureCalculationMethods = (LrsMeasureCalculationMethods) this.cmbMeasureCalcMethod.getSelectedItem();

//        params.setMeasureCalculationMethods(measureCalculationMethods);

//        DistanceUnits measureUnits = (DistanceUnits) this.cmbMeasureUnits.getSelectedItem();

//        params.setMeasureUnits(measureUnits);

//

//

//        Double searchRadius = (Double) this.txtSearchRadius.getValue();

//        params.setSearchRadius(searchRadius);

//

//        params.setInterpolateBetweenCalibrationPoints(this.chkInterpolateBetweenCalibrationPoints.isSelected());

//        params.setExtrapolateBeforeCalibrationPoints(this.chkExtrapolateBeforeCalibrationPoints.isSelected());

//        params.setExtrapolateAfterCalibrationPoints(this.chkExtrapolateAfterCalibrationPoints.isSelected());

//        params.setIgnoreSpatialGaps(this.chkIgnoreSpatialGaps.isSelected());

//        params.setIncludeAll(this.chkIncludeAll.isSelected());

        return params;

    }

    public void setParams(LrsAlgorithmParams params) {

//        if (params == null) {

//            clear();

//            return;

//        }

//        // Rellena los campos con los valores de params

//        LrsCalibrateRouteAlgorithmParams parameters = (LrsCalibrateRouteAlgorithmParams) params;

//

//        this.cmbMeasureCalcMethod.setSelectedItem(parameters.getMeasureCalculationMethods());

//        this.cmbMeasureUnits.setSelectedItem(parameters.getMeasureUnits());

//

//        // FIXME: rellenar ?C?mo?

//        FeatureStore sourceFeatureStore = parameters.getSourceFeatureStore();

//        if(layers!=null){

//            FLayer layer = layers.getLayer(sourceFeatureStore.getName());

//            this.cmbInputLayer.setSelectedItem(layer);

//        }

//        FeatureStore calibratePointFeaturesStore=parameters.getCalibratePointFeatureStore();

//        if(layers!=null){

//            FLayer layer = layers.getLayer(calibratePointFeaturesStore.getName());

//            this.cmbCalibratePointLayer.setSelectedItem(layer);

//        }

//        NewFeatureStoreParameters newFeatureStoreParameters = parameters.getNewFeatureStoreParameters();

//        if(newFeatureStoreParameters instanceof SHPNewStoreParameters){

//            this.txtOutputFile.setText(((SHPNewStoreParameters)newFeatureStoreParameters).getSHPFile().getAbsolutePath());

//        }

//

//        //FIN DEL FIXME

//

//        this.cmbIdRouteField.setSelectedItem(parameters.getIdRouteField());

//        this.cmbCalibrationPointIdRouteField.setSelectedItem(parameters.getCalibratePointIdRouteField());

//        this.cmbFromMeasureField.setSelectedItem(parameters.getFromMeasureField());

//

//        this.txtSearchRadius.setValue(parameters.getSearchRadius());

//        this.chkInterpolateBetweenCalibrationPoints.setSelected(parameters.interpolateBetweenCalibrationPoints());

//        this.chkExtrapolateBeforeCalibrationPoints.setSelected(parameters.extrapolateBeforeCalibrationPoints());

//        this.chkExtrapolateAfterCalibrationPoints.setSelected(parameters.extrapolateAfterCalibrationPoints());

//        this.chkIgnoreSpatialGaps.setSelected(parameters.ignoreSpatialGaps());

//        this.chkIncludeAll.setSelected(parameters.includeAll());

    }

    public void clear() {

//        this.txtOutputFile.setText("");

//        this.txtSearchRadius.setValue(0.0);

//

//        this.cmbInputLayer.setSelectedIndex(0);

//        Object inputLayerSelected = this.cmbInputLayer.getSelectedItem();

//        if (inputLayerSelected != null) {

//            this.updateLayerSource((FLyrVect) inputLayerSelected);

//        }

//        this.cmbCalibratePointLayer.setSelectedIndex(0);

//        Object calibratePointLayerSelected = this.cmbCalibratePointLayer.getSelectedItem();

//        if (calibratePointLayerSelected != null) {

//            this.updateLayerCalibratePoint((FLyrVect) calibratePointLayerSelected);

//        }

//        this.cmbIdRouteField.setSelectedIndex(0);

//        this.cmbCalibrationPointIdRouteField.setSelectedIndex(0);

//        this.cmbFromMeasureField.setSelectedIndex(0);

//        this.cmbMeasureCalcMethod.setSelectedIndex(0);

//        this.cmbMeasureUnits.setSelectedIndex(0);

//        this.chkInterpolateBetweenCalibrationPoints.setSelected(false);

//        this.chkExtrapolateBeforeCalibrationPoints.setSelected(false);

//        this.chkExtrapolateAfterCalibrationPoints.setSelected(false);

//        this.chkIgnoreSpatialGaps.setSelected(false);

//        this.chkIncludeAll.setSelected(false);

    }

    private void updateLayerSource(FLyrVect layer) {

        logger.info(new StringBuilder().append("Setting layer ").append(layer.getName()).toString());

        FeatureStore featureStore = layer.getFeatureStore();

        try {

            FeatureType type = featureStore.getDefaultFeatureType();

            // Se renuevan los combos de attributeDescriptors

            FeatureAttributeDescriptorsComboBoxModel idRouteFieldComboBoxModel =

                new FeatureAttributeDescriptorsComboBoxModel();

            this.cmbIdRouteField.setModel(idRouteFieldComboBoxModel);

            FeatureAttributeDescriptor[] attributeDescriptors = type.getAttributeDescriptors();

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

                FeatureAttributeDescriptor featureAttributeDescriptor = attributeDescriptors[i];

                if (featureAttributeDescriptor.getDataType().getType() != DataTypes.GEOMETRY) {

                    logger.info(new StringBuilder().append("Adding field ")

                        .append(featureAttributeDescriptor.getName()).toString());

                    JLrsGenerateDynamicSegmentationParamsController.this.cmbIdRouteField.addItem(featureAttributeDescriptor);

                }

            }

            this.cmbIdRouteField.invalidate();

        } catch (DataException e1) {

            logger.warn(

                new StringBuilder().append("Error getting default feature type from layer ").append(layer.getName())

                    .toString(), e1);

        }

    }

    private void updateLayerTable(FLyrVect layer) {

//        logger.info(new StringBuilder().append("Setting layer ").append(layer.getName()).toString());

//        FeatureStore featureStore = layer.getFeatureStore();

//        try {

//            FeatureType type = featureStore.getDefaultFeatureType();

//            // Se renuevan los combos de attributeDescriptors

//            FeatureAttributeDescriptorsComboBoxModel calibrationPointIdRouteFieldComboBoxModel =

//                new FeatureAttributeDescriptorsComboBoxModel();

//            FeatureAttributeDescriptorsComboBoxModel fromMeasureFieldComboBoxModel =

//                new FeatureAttributeDescriptorsComboBoxModel();

//

//            this.cmbCalibrationPointIdRouteField.setModel(calibrationPointIdRouteFieldComboBoxModel);

//            this.cmbFromMeasureField.setModel(fromMeasureFieldComboBoxModel);

//

//            FeatureAttributeDescriptor[] attributeDescriptors = type.getAttributeDescriptors();

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

//                FeatureAttributeDescriptor featureAttributeDescriptor = attributeDescriptors[i];

//                if (featureAttributeDescriptor.getDataType().getType() != DataTypes.GEOMETRY) {

//                    logger.info(new StringBuilder().append("Adding field ")

//                        .append(featureAttributeDescriptor.getName()).toString());

//

//                    JLrsGenerateDynamicSegmentationParamsController.this.cmbCalibrationPointIdRouteField.addItem(featureAttributeDescriptor);

//                    if (featureAttributeDescriptor.getDataType().isNumeric()) {

//                        JLrsGenerateDynamicSegmentationParamsController.this.cmbFromMeasureField.addItem(featureAttributeDescriptor);

//                    }

//                }

//            }

//            this.cmbCalibrationPointIdRouteField.invalidate();

//            this.cmbFromMeasureField.invalidate();

//

//        } catch (DataException e1) {

//            logger.warn(

//                new StringBuilder().append("Error getting default feature type from layer ").append(layer.getName())

//                    .toString(), e1);

//        }

    }

    protected FeatureStore getFeatureStore(String outputLayerPath, IProjection projection) throws Exception {

        DataManager dataManager = DALLocator.getDataManager();

        DataStoreParameters dataStoreParams = dataManager.createStoreParameters("Shape");

        dataStoreParams.setDynValue("shpfile", outputLayerPath);

        dataStoreParams.setDynValue("CRS", projection);

        dataStoreParams.setDynValue("useNullGeometry", false);

        dataStoreParams.validate();

        return (FeatureStore) dataManager.openStore("Shape", dataStoreParams);

    }

}