Application: gvSIG desktop » gvSIG 3D

package org.gvsig.view3d.vector.swing.impl;

import java.awt.event.ActionEvent;

import java.awt.event.ActionListener;

import java.util.Arrays;

import java.util.Collections;

import java.util.List;

import javax.swing.JComponent;

import org.apache.commons.lang3.StringUtils;

import org.opengis.metadata.Datatype;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

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.mapcontext.layers.FLayer;

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

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.dataTypes.CoercionException;

import org.gvsig.tools.dataTypes.DataTypes;

import org.gvsig.tools.dataTypes.DataTypesManager;

import org.gvsig.tools.dataTypes.DataTypesManager.Coercion;

import org.gvsig.tools.i18n.I18nManager;

import org.gvsig.view3d.lib.api.loader.LoaderParameters;

import org.gvsig.view3d.swing.api.exception.InvalidLoaderParametersException;

import org.gvsig.view3d.vector.lib.api.VectorElevationMode;

import org.gvsig.view3d.vector.lib.api.VectorExtrusionLoaderParameters;

import org.gvsig.view3d.vector.swing.api.JVectorExtrusionLoaderParameters;

public class JVectorExtrusionLoaderParametersController extends JVectorExtrusionLoaderParametersView

    implements JVectorExtrusionLoaderParameters {

    private static final long serialVersionUID = 26999444514713602L;

    private static final Logger LOG = LoggerFactory.getLogger(JVectorExtrusionLoaderParametersController.class);

    private JColorChooserButton colorChooserButton;

    private VectorExtrusionLoaderParameters parameters;

    public JVectorExtrusionLoaderParametersController(VectorExtrusionLoaderParameters parameters) {

        super();

        this.parameters = parameters;

        translate();

        initComponents();

    }

    private void translate() {

        I18nManager i18nManager = ToolsLocator.getI18nManager();

        this.elevationModeText.setText(i18nManager.getTranslation(this.elevationModeText.getText()));

        this.defaultColorText.setText(i18nManager.getTranslation(this.defaultColorText.getText()));

        this.constantHeightText.setText(i18nManager.getTranslation(this.constantHeightText.getText()));

        this.heightFieldText.setText(i18nManager.getTranslation(this.heightFieldText.getText()));

        this.verticalExaggerationText.setText(i18nManager.getTranslation(this.verticalExaggerationText.getText()));

    }

    @SuppressWarnings("unchecked")

    private void initComponents() {

        List<VectorElevationMode> elevationModes = Arrays.asList(VectorElevationMode.values());

        // Elevation mode

        this.elevationMode.setRenderer(new ElevationModeComboBoxRenderer());

        for (VectorElevationMode mode : elevationModes) {

            this.elevationMode.addItem(mode);

        }

        if(parameters.getVectorElevationMode() != null) {

            this.elevationMode.setSelectedItem(parameters.getVectorElevationMode());

        }

        // Default Color

        this.colorChooserButton = new JColorChooserButton(parameters.getDefaultColor());

        this.defaultColorPanel.add(colorChooserButton);

        DataTypesManager dataTypesManager = ToolsLocator.getDataTypesManager();

        Coercion stringCoercion = dataTypesManager.getCoercion(DataTypes.STRING);

        // Constant height

        try {

            this.constantHeight.setText((String) stringCoercion.coerce(parameters.getConstantHeight()));

        } catch (CoercionException e) {

            LOG.error(String.format("Can not parse constant height value %d to string", parameters.getConstantHeight()),

                e);

            this.constantHeight.setText("");

        }

        // Vertical exaggeration

        try {

            String verticalExaggeration = (String) stringCoercion.coerce(parameters.getVerticalExaggeration());

            if(StringUtils.isBlank(verticalExaggeration)) {

                this.verticalExaggeration.setText("1.0");

            } else {

                this.verticalExaggeration.setText(verticalExaggeration);

            }

        } catch (CoercionException e) {

            LOG.error(String.format("Can not parse vertical exaggeration value %d to string",

                parameters.getVerticalExaggeration()), e);

            this.verticalExaggeration.setText("");

        }

        // Height field

        heightField.addItem("");

        FLayer layer = this.parameters.getLayer();

        if (layer instanceof FLyrVect) {

            FLyrVect vectorLayer = (FLyrVect) layer;

            FeatureStore featureStore = vectorLayer.getFeatureStore();

            FeatureType featureType = null;

            try {

                featureType = featureStore.getDefaultFeatureType();

            } catch (DataException e) {

                LOG.error(String.format("Can not get feature type of %1s", vectorLayer.getName()), e);

                return;

            }

            FeatureAttributeDescriptor[] attributeDescriptors = featureType.getAttributeDescriptors();

            for (FeatureAttributeDescriptor attDesc : attributeDescriptors) {

                if(attDesc.getDataType().isNumeric()) {

                    heightField.addItem(attDesc.getName());

                }

            }

            if(StringUtils.isNotBlank(this.parameters.getHeightField()) &&

                featureType.getAttributeDescriptor(this.parameters.getHeightField()) != null){

                heightField.setSelectedItem(this.parameters.getHeightField());

                constantHeightText.setEnabled(false);

                constantHeight.setEnabled(false);

            }

        }

        // Initialize change listeners

        this.heightField.addActionListener(new ActionListener() {

            @Override

            public void actionPerformed(ActionEvent e) {

                String selectedValue = (String) heightField.getSelectedItem();

                if(StringUtils.isBlank(selectedValue)) {

                    constantHeightText.setEnabled(true);

                    constantHeight.setEnabled(true);

                } else {

                    constantHeightText.setEnabled(false);

                    constantHeight.setEnabled(false);

                }

            }

        });

    }

    @Override

    public LoaderParameters getLoaderParameters() throws InvalidLoaderParametersException {

        DataTypesManager dataTypesManager = ToolsLocator.getDataTypesManager();

        Coercion doubleCoercion = dataTypesManager.getCoercion(DataTypes.DOUBLE);

        this.parameters.setVectorElevationMode((VectorElevationMode) this.elevationMode.getSelectedItem());

        this.parameters.setDefaultColor(this.colorChooserButton.getSelectedColor());

        if(StringUtils.isNotBlank((String) this.heightField.getSelectedItem())) {

            this.parameters.setHeightField((String) this.heightField.getSelectedItem());

        } else {

            try {

                this.parameters.setConstantHeight((double) doubleCoercion.coerce(this.constantHeight.getText()));

            } catch (CoercionException e) {

                LOG.error(String.format("Can not parse constant height value %s to double", this.constantHeight.getText()),

                    e);

                throw new InvalidLoaderParametersException("Invalid constant height value", "constant_height_has_to_be_a_number",

                    e);

            }

        }

        try {

            this.parameters.setVerticalExaggeration((double) doubleCoercion.coerce(this.verticalExaggeration.getText()));

        } catch (CoercionException e) {

            LOG.error(String.format("Can not parse vertical exaggeration value %s to double",

                this.verticalExaggeration.getText()), e);

            throw new InvalidLoaderParametersException("Invalid vertical exaggeration value",

                "vertical_exaggeration_has_to_be_a_number", e);

        }

        return this.parameters;

    }

    @Override

    public JComponent asJComponent() {

        return this;

    }

}