Application: gvSIG desktop

package org.gvsig.fmap.mapcontext.raster.swing.impl;

/* gvSIG. Desktop Geographic Information System.

 *

 * Copyright ? 2007-2017 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.

 */

import java.util.ArrayList;

import java.util.Collections;

import java.util.List;

import javax.swing.table.AbstractTableModel;

import org.gvsig.fmap.dal.raster.BandDescriptor;

import org.gvsig.fmap.mapcontext.raster.swing.SelectableBandDescriptorsTableModel;

import org.gvsig.raster.swing.buffer.SelectableBandsTableModel;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.i18n.I18nManager;

/**

 * @author fdiaz

 *

 */

public class DefaultSelectableBandDescriptorsTableModel extends AbstractTableModel implements SelectableBandDescriptorsTableModel {

    /**

     *

     */

    private static final long serialVersionUID = -3653794540611142562L;

    private List<SelectableBand> bands;

    private static final int COLUMNS = 5;

    private static final int COLUMN_SELECTED = 0;

    private static final int COLUMN_NUMBER = 1;

    private static final int COLUMN_BAND_STORE_NAME = 2;

    private static final int COLUMN_BAND_NUMBER_IN_STORE = 3;

    private static final int COLUMN_BAND_NAME = 4;

    /**

     * @param bandDescriptors

     */

    public DefaultSelectableBandDescriptorsTableModel(List<BandDescriptor> bandDescriptors) {

        bands = new ArrayList<DefaultSelectableBandDescriptorsTableModel.SelectableBand>();

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

            SelectableBand selectableBand = new SelectableBand(i, bandDescriptors.get(i), true);

            bands.add(selectableBand);

        }

    }

    /* (non-Javadoc)

     * @see javax.swing.table.TableModel#getRowCount()

     */

    @Override

    public int getRowCount() {

        return bands.size();

    }

    /* (non-Javadoc)

     * @see javax.swing.table.TableModel#getColumnCount()

     */

    @Override

    public int getColumnCount() {

        return COLUMNS;

    }

    /* (non-Javadoc)

     * @see javax.swing.table.TableModel#getValueAt(int, int)

     */

    @Override

    public Object getValueAt(int rowIndex, int columnIndex) {

        switch (columnIndex) {

        case COLUMN_SELECTED:

            return bands.get(rowIndex).isSelected();

        case COLUMN_NUMBER:

            return bands.get(rowIndex).getNumber();

        case COLUMN_BAND_STORE_NAME:

            return bands.get(rowIndex).getBandDescriptor().getStore().getName();

        case COLUMN_BAND_NUMBER_IN_STORE:

            return bands.get(rowIndex).getBandDescriptor().getBand();

        case COLUMN_BAND_NAME:

            return bands.get(rowIndex).getBandDescriptor().getName();

        }

        return null;

    }

    @Override

    public void setValueAt(Object aValue, int rowIndex, int columnIndex) {

        if (rowIndex < bands.size()) {

            if (columnIndex < COLUMNS) {

                switch (columnIndex) {

                case COLUMN_SELECTED:

                    if (aValue == null) {

                        bands.get(rowIndex).setSelected(false);

                        break;

                    }

                    if (!(aValue instanceof Boolean)) {

                        throw new IllegalArgumentException("The object '" + aValue.toString() + "' must be a Boolean.");

                    }

                    bands.get(rowIndex).setSelected((Boolean) aValue);

                    this.fireTableDataChanged();

                    break;

                default:

                    break;

                }

            }

        }

    }

    @Override

    public String getColumnName(int column) {

        I18nManager i18nManager = ToolsLocator.getI18nManager();

        switch (column) {

        case COLUMN_SELECTED:

            return i18nManager.getTranslation("_selected");

        case COLUMN_NUMBER:

            return i18nManager.getTranslation("_number");

        case COLUMN_BAND_STORE_NAME:

            return i18nManager.getTranslation("_store_name");

        case COLUMN_BAND_NUMBER_IN_STORE:

            return i18nManager.getTranslation("_band_number");

        case COLUMN_BAND_NAME:

            return i18nManager.getTranslation("_band_name");

        }

        return null;

    }

    public boolean isCellEditable(int row, int col) {

        if (col == COLUMN_SELECTED) {

            return true;

        }

        return false;

    }

    @Override

    public Class<?> getColumnClass(int col) {

        if (col == COLUMN_SELECTED) {

            return Boolean.class;

        }

        return super.getColumnClass(col);

    }

    /**

     * @param rowIndex

     */

    public void up(int rowIndex){

        if(rowIndex>0){

            SelectableBand aux = bands.get(rowIndex);

            bands.set(rowIndex, bands.get(rowIndex-1));

            bands.set(rowIndex-1, aux);

            this.fireTableDataChanged();

        }

    }

    /**

     * @param rowIndex

     */

    public void down(int rowIndex){

        if(rowIndex<(bands.size()-1)){

            SelectableBand aux = bands.get(rowIndex);

            bands.set(rowIndex, bands.get(rowIndex+1));

            bands.set(rowIndex+1, aux);

            this.fireTableDataChanged();

        }

    }

    /**

     * @return a list of selected bands.

     */

    public List<Integer> getSelectedBands() {

        List<Integer> selectedBands = new ArrayList<Integer>();

        for (SelectableBand selBand : this.bands) {

            if(selBand.isSelected()){

                selectedBands.add(selBand.getNumber());

            }

        }

        return Collections.unmodifiableList(selectedBands);

    }

    public void selectBand(int band) {

        this.bands.get(band).setSelected(true);

        this.fireTableDataChanged();

    }

    public void unselectBand(int band) {

        this.bands.get(band).setSelected(false);

        this.fireTableDataChanged();

    }

    public void selectAllBands() {

        for (SelectableBand selBand : this.bands) {

            selBand.setSelected(true);

            this.fireTableDataChanged();

        }

    }

    public void unselectAllBands() {

        for (SelectableBand selBand : this.bands) {

            selBand.setSelected(false);

            this.fireTableDataChanged();

        }

    }

    /**

     * @author fdiaz

     *

     */

    public static class SelectableBand {

        private boolean selected;

        private int number;

        private BandDescriptor bandDescriptor;

        /**

         * @param bandDescriptor

         * @param selected

         */

        public SelectableBand(int number, BandDescriptor bandDescriptor, boolean selected) {

            this.setNumber(number);

            this.setBandDescriptor(bandDescriptor);

            this.setSelected(selected);

        }

        /**

         * @return the number

         */

        public int getNumber() {

            return number;

        }

        /**

         * @param number the number to set

         */

        public void setNumber(int number) {

            this.number = number;

        }

        /**

         * @return the selected

         */

        public boolean isSelected() {

            return selected;

        }

        /**

         * @param selected the selected to set

         */

        public void setSelected(boolean selected) {

            this.selected = selected;

        }

        /**

         * @return the bandDescriptor

         */

        public BandDescriptor getBandDescriptor() {

            return bandDescriptor;

        }

        /**

         * @param bandDescriptor the bandDescriptor to set

         */

        public void setBandDescriptor(BandDescriptor bandDescriptor) {

            this.bandDescriptor = bandDescriptor;

        }

    }

}