Application: gvSIG desktop » gvSIG geoprocess

/**

 * gvSIG. Desktop Geographic Information System.

 *

 * Copyright (C) 2007-2012 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.geoprocess.algorithm.merge;

import java.awt.Dimension;

import java.awt.GridBagConstraints;

import java.awt.GridBagLayout;

import java.awt.Insets;

import java.awt.event.ActionEvent;

import java.awt.event.ActionListener;

import java.util.ArrayList;

import javax.swing.JCheckBox;

import javax.swing.JComboBox;

import javax.swing.JLabel;

import javax.swing.JPanel;

import org.gvsig.geoprocess.lib.api.GeoProcessLocator;

import org.gvsig.geoprocess.sextante.gui.algorithm.AlgorithmOutputPanel;

import org.gvsig.gui.beans.table.TableContainer;

import org.gvsig.gui.beans.table.exceptions.NotInitializeException;

import es.unex.sextante.core.GeoAlgorithm;

import es.unex.sextante.core.ObjectAndDescription;

import es.unex.sextante.core.OutputObjectsSet;

import es.unex.sextante.core.ParametersSet;

import es.unex.sextante.core.Sextante;

import es.unex.sextante.dataObjects.IVectorLayer;

import es.unex.sextante.gui.algorithm.GeoAlgorithmParametersPanel;

import es.unex.sextante.gui.algorithm.OutputChannelSelectionPanel;

import es.unex.sextante.gui.core.SextanteGUI;

import es.unex.sextante.outputs.Output;

/**

 * Panel for merge algorithm

 *

 * @author <a href="mailto:nachobrodin@gmail.com">Nacho Brodin</a>

 */

public class MergeParametersPanel extends GeoAlgorithmParametersPanel implements ActionListener {

    private static final long serialVersionUID = 1L;

    private GeoAlgorithm m_Algorithm = null;

    private JComboBox fieldsCombo = null;

    private JCheckBox allLayers = null;

    //private AlgorithmOutputPanel             output             = null;

    private final String[] columnNames = {GeoProcessLocator.getGeoProcessManager().getTranslation("Selected"),

        GeoProcessLocator.getGeoProcessManager().getTranslation("Layer")};

    private final int[] columnWidths = {35, 334};

    private TableContainer table = null;

    private ObjectAndDescription[] layerList = null;

    private AlgorithmOutputPanel algorithmOutputPanel = null;

    private OutputChannelSelectionPanel outputChannelSelectionPanelPol;

    private OutputChannelSelectionPanel outputChannelSelectionPanelLine;

    private OutputChannelSelectionPanel outputChannelSelectionPanelPoint;

    private JPanel outputPanel;

    public MergeParametersPanel() {

        super();

    }

    public void init(GeoAlgorithm algorithm) {

        m_Algorithm = algorithm;

        initGUI();

    }

    private void initGUI() {

        GridBagLayout gbl = new GridBagLayout();

        this.setLayout(gbl);

        GridBagConstraints gbc = new GridBagConstraints();

        gbc.fill = GridBagConstraints.HORIZONTAL;

        gbc.weightx = 1.0;

        gbc.gridx = 0;

        gbc.gridy = 0;

        gbc.insets = new Insets(5, 5, 8, 5);

        this.add(getAllLayersCheck(), gbc);

        gbc.gridy = 1;

        gbc.fill = GridBagConstraints.BOTH;

        gbc.insets = new Insets(0, 0, 12, 0);

        gbc.weighty = 1.0;

        this.add(getCheckBoxTable(), gbc);

        gbc.gridy = 2;

        gbc.fill = GridBagConstraints.HORIZONTAL;

        gbc.insets = new Insets(5, 5, 8, 5);

        gbc.weightx = 1.0;

        gbc.weighty = 0;

        this.add(getFieldsComboPanel(GeoProcessLocator.getGeoProcessManager().getTranslation("use_fields_from_layer"), getFieldsCombo()), gbc);

        gbc.gridy = 3;

        this.add(getOutputChannelSelectionPanel(), gbc);

        initTable();

    }

    /**

     * Gets the output panel (SEXTANTE)

     *

     * @return

     */

    private JPanel getOutputChannelSelectionPanel() {

        if (outputPanel == null) {

            try {

                outputPanel = new JPanel();

                outputPanel.setLayout(new GridBagLayout());

                final OutputObjectsSet ooSet = m_Algorithm.getOutputObjects();

                final Output out_pol = ooSet.getOutput(MergeAlgorithm.RESULT_POL);

                final Output out_line = ooSet.getOutput(MergeAlgorithm.RESULT_LINE);

                final Output out_point = ooSet.getOutput(MergeAlgorithm.RESULT_POINT);

                outputChannelSelectionPanelPol = new OutputChannelSelectionPanel(out_pol, m_Algorithm.getParameters());

                outputChannelSelectionPanelLine = new OutputChannelSelectionPanel(out_line, m_Algorithm.getParameters());

                outputChannelSelectionPanelPoint = new OutputChannelSelectionPanel(out_point, m_Algorithm.getParameters());

                String label = GeoProcessLocator.getGeoProcessManager().getTranslation("Merge");

                GridBagConstraints gbc = new GridBagConstraints();

                gbc.fill = GridBagConstraints.HORIZONTAL;

                gbc.insets = new Insets(0, 0, 4, 0);

                gbc.weightx = 1;

                gbc.gridx = 0;

                gbc.gridy = 0;

                outputPanel.add(new JLabel(" " + label + " ["

                        + GeoProcessLocator.getGeoProcessManager().getTranslation("Polygon") + "]               "), gbc);

                gbc.gridx = 0;

                gbc.gridy = 1;

                outputPanel.add(new JLabel(" " + label + " ["

                        + GeoProcessLocator.getGeoProcessManager().getTranslation("Line") + "]               "), gbc);

                gbc.gridx = 0;

                gbc.gridy = 2;

                outputPanel.add(new JLabel(" " + label + " ["

                        + GeoProcessLocator.getGeoProcessManager().getTranslation("Point") + "]               "), gbc);

                gbc.fill = GridBagConstraints.HORIZONTAL;

                gbc.weightx = 1;

                gbc.gridx = 1;

                gbc.gridy = 0;

                outputPanel.add(outputChannelSelectionPanelPol, gbc);

                gbc.gridx = 1;

                gbc.gridy = 1;

                outputPanel.add(outputChannelSelectionPanelLine, gbc);

                gbc.gridx = 1;

                gbc.gridy = 2;

                outputPanel.add(outputChannelSelectionPanelPoint, gbc);

            } catch (final Exception e) {

                Sextante.addErrorToLog(e);

            }

        }

        return outputPanel;

    }

    /**

     * Gets the output panel (DAL)

     *

     * @return

     */

    @SuppressWarnings("unused")

    private AlgorithmOutputPanel getAlgorithmOutputPanel() {

        if (algorithmOutputPanel == null) {

            algorithmOutputPanel = new AlgorithmOutputPanel();

        }

        return algorithmOutputPanel;

    }

    /**

     * Gets a new JPanel with the text and JComboBox

     *

     * @param text

     * @param combo

     * @return

     */

    public JPanel getFieldsComboPanel(String text, JComboBox combo) {

        JPanel panel = new JPanel();

        GridBagLayout gbl = new GridBagLayout();

        panel.setLayout(gbl);

        GridBagConstraints gbc = new GridBagConstraints();

        gbc.fill = GridBagConstraints.HORIZONTAL;

        gbc.weightx = 1.0;

        gbc.gridx = 0;

        gbc.insets = new Insets(0, 0, 4, 0);

        JLabel label = new JLabel(text);

//                label.setPreferredSize(new Dimension(80, 18));

        panel.add(label, gbc);

        gbc.fill = GridBagConstraints.HORIZONTAL;

        gbc.weightx = 1.0;

        gbc.gridx = 1;

//                gbc.anchor = GridBagConstraints.EAST;

        gbc.insets = new Insets(0, 0, 4, 0);

        panel.add(combo, gbc);

        return panel;

    }

    /**

     * Gets a ComboBox

     *

     * @return

     */

    public JComboBox getFieldsCombo() {

        if (fieldsCombo == null) {

            fieldsCombo = new JComboBox();

//                        fieldsCombo.setPreferredSize(new Dimension(0, 18));

            ObjectAndDescription[] fieldList = getLayerList();

            fieldsCombo.removeAllItems();

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

                fieldsCombo.addItem(fieldList[i]);

            }

        }

        return fieldsCombo;

    }

    /**

     * Gets a CheckBox

     *

     * @return

     */

    public JCheckBox getAllLayersCheck() {

        if (allLayers == null) {

            allLayers = new JCheckBox(GeoProcessLocator.getGeoProcessManager().getTranslation("select_all_layers"));

            allLayers.addActionListener(this);

        }

        return allLayers;

    }

    /**

     * Gets the summary table

     *

     * @return TableContainer

     */

    public TableContainer getCheckBoxTable() {

        if (table == null) {

            table = new TableContainer(columnNames, columnWidths, null);

            table.setModel("CheckBoxModel");

            table.setControlVisible(false);

            table.initialize();

        }

        return table;

    }

    //------------------------------------------------------------

    /*

         * (non-Javadoc)

         * @see java.awt.event.ActionListener#actionPerformed(java.awt.event.ActionEvent)

     */

    public void actionPerformed(ActionEvent e) {

        if (e.getSource() == getAllLayersCheck()) {

            //Selecci?n de todas las capas de la tabla

            try {

                for (int i = 0; i < getCheckBoxTable().getRowCount(); i++) {

                    if (getAllLayersCheck().isSelected()) {

                        getCheckBoxTable().getModel().setValueAt(new Boolean(true), i, 0);

                    } else {

                        getCheckBoxTable().getModel().setValueAt(new Boolean(false), i, 0);

                    }

                }

            } catch (NotInitializeException e1) {

                Sextante.addErrorToLog(e1);

            }

        }

    }

    /**

     * Adds to the table one entry for each field

     */

    private void initTable() {

        try {

            getCheckBoxTable().removeAllRows();

            ObjectAndDescription[] layerList = getLayerList();

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

                addTableRow(layerList[i].getDescription());

            }

        } catch (NotInitializeException e) {

            Sextante.addErrorToLog(e);

        }

    }

    /**

     * Adds one entry to the table

     *

     * @param layerName Layer name

     * @throws NotInitializeException

     */

    private void addTableRow(String layerName) throws NotInitializeException {

        Object[] row = {new Boolean(false),

            layerName};

        getCheckBoxTable().addRow(row);

    }

    @Override

    public void assignParameters() {

        try {

            ParametersSet params = m_Algorithm.getParameters();

            params.getParameter(MergeAlgorithm.FIELDLAYER).setParameterValue(getSelectedVectorLayer());

            params.getParameter(MergeAlgorithm.LAYERS).setParameterValue(getSelectedLayerList());

            OutputObjectsSet ooSet = m_Algorithm.getOutputObjects();

            Output outPol = ooSet.getOutput(MergeAlgorithm.RESULT_POL);

            Output outLine = ooSet.getOutput(MergeAlgorithm.RESULT_LINE);

            Output outPoint = ooSet.getOutput(MergeAlgorithm.RESULT_POINT);

            //Reponer estas l?neas para cambiar el panel de salida y comentar la siguiente

            //AlgorithmOutputPanel fsp = getAlgorithmOutputPanel();

            //out.setOutputChannel(new CompositeSourceOutputChannel(fsp.getOutputParameters()));

            outPol.setOutputChannel(outputChannelSelectionPanelPol.getOutputChannel());

            outLine.setOutputChannel(outputChannelSelectionPanelLine.getOutputChannel());

            outPoint.setOutputChannel(outputChannelSelectionPanelPoint.getOutputChannel());

        } catch (Exception e) {

            Sextante.addErrorToLog(e);

        }

    }

    @Override

    public void setOutputValue(String arg0, String arg1) {

    }

    @Override

    public void setParameterValue(String arg0, String arg1) {

    }

    /**

     * Gets the input layer list

     *

     * @return

     */

    private ObjectAndDescription[] getLayerList() {

        if (layerList == null) {

            IVectorLayer[] layers = SextanteGUI.getInputFactory()

                    .getVectorLayers(IVectorLayer.SHAPE_TYPE_WRONG);

            layerList = new ObjectAndDescription[layers.length];

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

                layerList[i] = new ObjectAndDescription(layers[i].getName(), layers[i]);

            }

        }

        return layerList;

    }

    /**

     * Gets the list of selected layers

     *

     * @return

     */

    private ArrayList<IVectorLayer> getSelectedLayerList() {

        ObjectAndDescription[] layerList = getLayerList();

        ArrayList<IVectorLayer> vLayer = new ArrayList<IVectorLayer>();

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

            Boolean check = (Boolean) getCheckBoxTable().getModel().getValueAt(i, 0);

            if (check.booleanValue()) {

                vLayer.add((IVectorLayer) layerList[i].getObject());

            }

        }

        return vLayer;

    }

    /**

     * Gets the selected vector layer in the JComboBox

     *

     * @return

     */

    private IVectorLayer getSelectedVectorLayer() {

        if (getFieldsCombo().getSelectedItem() != null) {

            return (IVectorLayer) ((ObjectAndDescription) getFieldsCombo().getSelectedItem()).getObject();

        }

        return null;

    }

}