Application: gvSIG desktop » gvSIG raster

/* gvSIG. Geographic Information System of the Valencian Government

*

* Copyright (C) 2007-2008 Infrastructures and Transports Department

* of the Valencian Government (CIT)

* 

* 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.

* 

*/

package org.gvsig.raster.tasseledcap.swing.impl.main;

import java.awt.BorderLayout;

import java.awt.GridBagConstraints;

import java.awt.GridBagLayout;

import java.io.File;

import java.util.EventListener;

import java.util.List;

import javax.swing.BorderFactory;

import javax.swing.ButtonGroup;

import javax.swing.JCheckBox;

import javax.swing.JComponent;

import javax.swing.JLabel;

import javax.swing.JPanel;

import javax.swing.JRadioButton;

import org.gvsig.fmap.dal.coverage.store.RasterDataStore;

import org.gvsig.i18n.Messages;

import org.gvsig.raster.algorithm.BasicAPISwingPanel;

import org.gvsig.raster.algorithm.RasterBaseAlgorithmLibrary;

import org.gvsig.raster.algorithm.process.DataProcess;

import org.gvsig.raster.algorithm.process.ProcessException;

import org.gvsig.raster.swing.RasterSwingLocator;

import org.gvsig.raster.swing.RasterSwingManager;

import org.gvsig.raster.swing.newlayer.CreateNewLayerPanel;

import org.gvsig.raster.swing.pagedtable.ModelLoader;

import org.gvsig.raster.swing.pagedtable.PagedTable;

import org.gvsig.raster.tasseledcap.algorithm.TasseledCapAlgorithmLibrary;

/**

 * @author Nacho Brodin (nachobrodin@gmail.com)

 */

public class TasseledCapMainPanelImpl extends JPanel implements BasicAPISwingPanel {

        private static final long               serialVersionUID     = 1L;

        private String                          layerName            = null;

        private CreateNewLayerPanel             newLayerPanel        = null;

        private Object                          inputStore           = null;

        private PagedTable                      table                = null;

        private JCheckBox                       roisCheck            = null;

        private int                             bandList             = 0;

        private String                          roisEPSG             = null;

        private JRadioButton                    landsatMS            = null;

        private JRadioButton                    landsatTM            = null;

        private JRadioButton                    landsatETM           = null;

        public TasseledCapMainPanelImpl(Object inputStore, String layerName, int bandList, String roisEPSG) {

                this.inputStore = inputStore;

                this.layerName = layerName;

                this.bandList = bandList;

                this.roisEPSG = roisEPSG;

                init();

        }

        private void init() {

                setLayout(new GridBagLayout());

                GridBagConstraints gridBagConstraints1 = new GridBagConstraints();

                gridBagConstraints1.fill = java.awt.GridBagConstraints.HORIZONTAL;

                gridBagConstraints1.weightx = 1;

                gridBagConstraints1.insets = new java.awt.Insets(0, 0, 0, 0);

                gridBagConstraints1.gridx = 0;

                gridBagConstraints1.gridy = 0;

                add(getLayerPanel(), gridBagConstraints1);

                gridBagConstraints1.gridy = 2;

                add((JComponent)getOptionsPanel(), gridBagConstraints1);

                gridBagConstraints1.gridy = 3;

                add((JComponent)getCreateNewLayerPanel(), gridBagConstraints1);

                if(roisEPSG != null) {

                        gridBagConstraints1.gridy = 4;

                        add(getROIsCheck(), gridBagConstraints1);

                }

                gridBagConstraints1.fill = java.awt.GridBagConstraints.BOTH;

                gridBagConstraints1.gridy = 1;

                gridBagConstraints1.weighty = 1;

                add(getInputsPanel(), gridBagConstraints1);

        }

        private JPanel getLayerPanel() {

                JPanel p = new JPanel();

                p.setBorder(BorderFactory.createTitledBorder(Messages.getText("layer")));

                p.setLayout(new GridBagLayout());

                GridBagConstraints gridBagConstraints1 = new GridBagConstraints();

                gridBagConstraints1.fill = java.awt.GridBagConstraints.HORIZONTAL;

                gridBagConstraints1.weightx = 1;

                gridBagConstraints1.insets = new java.awt.Insets(0, 0, 2, 0);

                gridBagConstraints1.gridx = 0;

                gridBagConstraints1.gridy = 0;

                p.add(new JLabel(layerName), gridBagConstraints1);

                return p;

        }

        private JPanel getInputsPanel() {

                JPanel p = new JPanel();

                p.setBorder(BorderFactory.createTitledBorder(Messages.getText("bands")));

                p.setLayout(new BorderLayout());

                GridBagConstraints gridBagConstraints1 = new GridBagConstraints();

                gridBagConstraints1.fill = java.awt.GridBagConstraints.BOTH;

                gridBagConstraints1.weightx = 1;

                gridBagConstraints1.weighty = 1;

                gridBagConstraints1.insets = new java.awt.Insets(2, 2, 2, 2);

                gridBagConstraints1.gridx = 0;

                gridBagConstraints1.gridy = 0;

                p.add(getPagedTable().getComponent(), BorderLayout.CENTER);

                return p;

        }

        public JComponent getComponent() {

                return this;

        }

        public PagedTable getPagedTable() {

                if(table == null) {

                        RasterSwingManager manager = RasterSwingLocator.getSwingManager();

                        String[] columnNames = {

                                        "", 

                                        Messages.getText("bands")};

                        int[] columnSizes = {30, -1};

                        BandTableModel model = new BandTableModel(columnNames);

                        ModelLoader loader = manager.createModelLoader(model);

                        CheckBoxColumnRenderer render = new CheckBoxColumnRenderer(null);

                        CheckBoxColumnEditor editor = new CheckBoxColumnEditor();

                        loader.setRenderForColumn(0, render);

                        loader.setCellEditorForColumn(0, editor);

                        loader.setColumnNames(columnNames);

                        loader.setColumnWidths(columnSizes);

                        table = manager.createPagedTable(loader);

                        table.showControllerTable(false);

                        table.showMoveRowsControls(false);

                }

                return table;

        }

        private JPanel getOptionsPanel() {

                JPanel p = new JPanel();

                ButtonGroup g = new ButtonGroup();

                g.add(getLandsatETMRadioButton());

                g.add(getLandsatTMRadioButton());

                g.add(getLandsatMSRadioButton());

                p.setBorder(BorderFactory.createTitledBorder(Messages.getText("options")));

                p.setLayout(new GridBagLayout());

                GridBagConstraints gridBagConstraints1 = new GridBagConstraints();

                gridBagConstraints1.fill = java.awt.GridBagConstraints.HORIZONTAL;

                gridBagConstraints1.weightx = 1;

                gridBagConstraints1.insets = new java.awt.Insets(0, 0, 2, 0);

                gridBagConstraints1.gridx = 0;

                gridBagConstraints1.gridy = 0;

                p.add(getLandsatMSRadioButton(), gridBagConstraints1);

                gridBagConstraints1.gridy = 1;

                p.add(getLandsatTMRadioButton(), gridBagConstraints1);

                gridBagConstraints1.gridy = 2;

                p.add(getLandsatETMRadioButton(), gridBagConstraints1);

                return p;

        }

        private JRadioButton getLandsatMSRadioButton() {

                if(landsatMS == null) {

                        landsatMS = new JRadioButton("Landsat MS");

                        landsatMS.setSelected(true);

                }

                return landsatMS;

        }

        private JRadioButton getLandsatTMRadioButton() {

                if(landsatTM == null) {

                        landsatTM = new JRadioButton("Landsat TM");

                }

                return landsatTM;

        }

        private JRadioButton getLandsatETMRadioButton() {

                if(landsatETM == null) {

                        landsatETM = new JRadioButton("Landsat ETM");

                }

                return landsatETM;

        }

        public DataProcess getResult() throws ProcessException {

                DataProcess task = RasterBaseAlgorithmLibrary.getManager().createRasterTask(TasseledCapAlgorithmLibrary.TASSELEDCAP_PROCESS_LABEL);

                List<String> params = task.getRasterTaskInputParameters(TasseledCapAlgorithmLibrary.TASSELEDCAP_PROCESS_LABEL);

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

                        String paramName = params.get(i);

                        Class<?> paramType = task.getParameterTypeByProcess(TasseledCapAlgorithmLibrary.TASSELEDCAP_PROCESS_LABEL, paramName);

                        if(paramType == RasterDataStore.class) {

                                task.addParam(paramName, (RasterDataStore)inputStore);

                        }

                        if(paramName.equals("PATH")) {

                                String filename = newLayerPanel.getDirectorySelected() + File.separator + newLayerPanel.getFileSelected();

                                if(!filename.endsWith(".tif"))

                                        filename += ".tif";

                                task.addParam(paramName, filename);

                        }

                        if(paramType == Boolean[].class) {

                                boolean[] bands = new boolean[getPagedTable().getRowCount()];

                                for (int j = 0; j < getPagedTable().getRowCount(); j++) {

                                        Object obj = getPagedTable().getValueAt(j, 0);

                                        if(obj instanceof Boolean)

                                                bands[j] = ((Boolean)obj).booleanValue();

                                }

                                task.addParam(paramName, bands);

                        }

                        if(getROIsCheck().isSelected()) {

                                if(paramName.equals("ROI_EPSG")) {

                                        task.addParam(paramName, roisEPSG);

                                }

                        }

                        if(paramType == Integer.class) {

                                int type = getSelectedLandsatType();

                                task.addParam(paramName, type);

                        }

                }

                return task;

        }

        private int getSelectedLandsatType() {

                if(getLandsatMSRadioButton().isSelected())

                        return 0;

                if(getLandsatTMRadioButton().isSelected())

                        return 1;

                if(getLandsatETMRadioButton().isSelected())

                        return 2;

                return -1;

        }

        public JCheckBox getROIsCheck() {

                if(roisCheck == null) {

                        roisCheck = new JCheckBox(Messages.getText("use_rois"));

                }

                return roisCheck;

        }

        public CreateNewLayerPanel getCreateNewLayerPanel() {

                if(newLayerPanel == null) {

                        newLayerPanel = RasterSwingLocator.getSwingManager().createNewLayerPanel();

                }

                return newLayerPanel;

        }

        public void initialize() {

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

                        getPagedTable().addRow(new Object[]{true, "B" + i});                        

                }                        

        }

        public int getComponentIDByObject(Object obj) {

                return -1;

        }

        public void addListener(EventListener listener) {

        }

}