Application: gvSIG desktop » gvSIG raster

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

 */

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

import java.util.ArrayList;

import java.util.Arrays;

import java.util.List;

import javax.swing.table.AbstractTableModel;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

import org.gvsig.raster.lib.buffer.api.BufferLocator;

import org.gvsig.raster.lib.buffer.api.BufferManager;

import org.gvsig.raster.lib.legend.api.ColorInterpretation;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.dataTypes.CoercionException;

import org.gvsig.tools.i18n.I18nManager;

import org.gvsig.tools.locator.LocatorException;

/**

 * @author fdiaz

 *

 */

public class BandsTableModel extends AbstractTableModel {

    /**

     *

     */

    private static final long serialVersionUID = 5993288441174158255L;

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

    List<RasterStoreBand> rasterStoreBands;

    public static final List<String> RGBColorSpace = Arrays.asList(ColorInterpretation.RED_BAND, ColorInterpretation.GREEN_BAND, ColorInterpretation.BLUE_BAND, ColorInterpretation.ALPHA_BAND);

    public static final List<String> CMYKColorSpace = Arrays.asList(ColorInterpretation.CYAN_BAND, ColorInterpretation.MAGENTA_BAND, ColorInterpretation.YELLOW_BAND, ColorInterpretation.BLACK_BAND);

    public static final List<String> HSLColorSpace = Arrays.asList(ColorInterpretation.HUE_BAND, ColorInterpretation.SATURATION_BAND, ColorInterpretation.LIGHTNESS_BAND);

    public static final List<String> YCBCRColorSpace = Arrays.asList(ColorInterpretation.YCBCR_CB_BAND, ColorInterpretation.YCBCR_CR_BAND, ColorInterpretation.YCBCR_Y_BAND);

    public static final List<List<String>> ColorSpaces = Arrays.asList(RGBColorSpace, CMYKColorSpace, HSLColorSpace, YCBCRColorSpace);

    /**

     *

     */

    public BandsTableModel() {

        rasterStoreBands = new ArrayList<RasterStoreBand>();

    }

    @Override

    public int getRowCount() {

        return rasterStoreBands.size();

    }

    @Override

    public int getColumnCount() {

        return 7;

    }

    @Override

    public Object getValueAt(int rowIndex, int columnIndex) {

        RasterStoreBand rasterStoreBand = rasterStoreBands.get(rowIndex);

        switch (columnIndex) {

        case 0:

            return rasterStoreBand.getBandColorInterpretation();

        case 1:

            return rasterStoreBand.getNoDataNumber();

        case 2:

            return rasterStoreBand.getDataType();

        case 3:

            return rasterStoreBand.getStore().getName();

        case 4:

            return rasterStoreBand.getBand();

        case 5:

            return rasterStoreBand.getBandName();

        case 6:

            return rasterStoreBand.getStore().getFullName();

        }

        return null;

    }

    @Override

    public String getColumnName(int column) {

        I18nManager i18nManager = ToolsLocator.getI18nManager();

        switch (column) {

        case 0:

            return i18nManager.getTranslation("_color");

        case 1:

            return i18nManager.getTranslation("_nodata");

        case 2:

            return i18nManager.getTranslation("_data_type");

        case 3:

            return i18nManager.getTranslation("_store_name");

        case 4:

            return i18nManager.getTranslation("_band");

        case 5:

            return i18nManager.getTranslation("_band_name");

        case 6:

            return i18nManager.getTranslation("_store_full_name");

        }

        return null;

    }

    @Override

    public boolean isCellEditable(int rowIndex, int columnIndex) {

        if(columnIndex>1){

            return false;

        }

        return true;

    }

    @Override

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

        Object value = aValue;

        if(columnIndex==0){

            rasterStoreBands.get(rowIndex).setBandColorInterpretation((String)aValue);

            String colorInterpretation = ColorInterpretation.PALETTE_BAND;

            if(((String)aValue).equalsIgnoreCase(colorInterpretation)){

                setUndefinedAnyOtherBand(rowIndex);

            }

            colorInterpretation = ColorInterpretation.GRAY_BAND;

            if(((String)aValue).equalsIgnoreCase(colorInterpretation)){

                setUndefinedAnyOtherBand(rowIndex);

            }

            colorInterpretation = ColorInterpretation.RED_BAND;

            if(((String)aValue).equalsIgnoreCase(colorInterpretation)){

                setUndefinedAnyOtherBandWithSameColorInterpretation(rowIndex, colorInterpretation);

            }

            colorInterpretation = ColorInterpretation.GREEN_BAND;

            if(((String)aValue).equalsIgnoreCase(colorInterpretation)){

                setUndefinedAnyOtherBandWithSameColorInterpretation(rowIndex, colorInterpretation);

            }

            colorInterpretation = ColorInterpretation.BLUE_BAND;

            if(((String)aValue).equalsIgnoreCase(colorInterpretation)){

                setUndefinedAnyOtherBandWithSameColorInterpretation(rowIndex, colorInterpretation);

            }

            colorInterpretation = ColorInterpretation.ALPHA_BAND;

            if(((String)aValue).equalsIgnoreCase(colorInterpretation)){

                setUndefinedAnyOtherBandWithSameColorInterpretation(rowIndex, colorInterpretation);

            }

            colorInterpretation = ColorInterpretation.HUE_BAND;

            if(((String)aValue).equalsIgnoreCase(colorInterpretation)){

                setUndefinedAnyOtherBandWithSameColorInterpretation(rowIndex, colorInterpretation);

            }

            colorInterpretation = ColorInterpretation.SATURATION_BAND;

            if(((String)aValue).equalsIgnoreCase(colorInterpretation)){

                setUndefinedAnyOtherBandWithSameColorInterpretation(rowIndex, colorInterpretation);

            }

            colorInterpretation = ColorInterpretation.LIGHTNESS_BAND;

            if(((String)aValue).equalsIgnoreCase(colorInterpretation)){

                setUndefinedAnyOtherBandWithSameColorInterpretation(rowIndex, colorInterpretation);

            }

            colorInterpretation = ColorInterpretation.CYAN_BAND;

            if(((String)aValue).equalsIgnoreCase(colorInterpretation)){

                setUndefinedAnyOtherBandWithSameColorInterpretation(rowIndex, colorInterpretation);

            }

            colorInterpretation = ColorInterpretation.MAGENTA_BAND;

            if(((String)aValue).equalsIgnoreCase(colorInterpretation)){

                setUndefinedAnyOtherBandWithSameColorInterpretation(rowIndex, colorInterpretation);

            }

            colorInterpretation = ColorInterpretation.YELLOW_BAND;

            if(((String)aValue).equalsIgnoreCase(colorInterpretation)){

                setUndefinedAnyOtherBandWithSameColorInterpretation(rowIndex, colorInterpretation);

            }

            colorInterpretation = ColorInterpretation.BLACK_BAND;

            if(((String)aValue).equalsIgnoreCase(colorInterpretation)){

                setUndefinedAnyOtherBandWithSameColorInterpretation(rowIndex, colorInterpretation);

            }

            colorInterpretation = ColorInterpretation.YCBCR_Y_BAND;

            if(((String)aValue).equalsIgnoreCase(colorInterpretation)){

                setUndefinedAnyOtherBandWithSameColorInterpretation(rowIndex, colorInterpretation);

            }

            colorInterpretation = ColorInterpretation.YCBCR_CB_BAND;

            if(((String)aValue).equalsIgnoreCase(colorInterpretation)){

                setUndefinedAnyOtherBandWithSameColorInterpretation(rowIndex, colorInterpretation);

            }

            colorInterpretation = ColorInterpretation.YCBCR_CR_BAND;

            if(((String)aValue).equalsIgnoreCase(colorInterpretation)){

                setUndefinedAnyOtherBandWithSameColorInterpretation(rowIndex, colorInterpretation);

            }

            fireTableDataChanged();

        } else if(columnIndex==1){

//            Number noDataNumber = rasterStoreBands.get(rowIndex).getNoDataNumber();

//            value = noDataNumber;

            int dataType = rasterStoreBands.get(rowIndex).getDataType();

//            Object noDataValue = null;

            BufferManager bufferManager = BufferLocator.getBufferManager();

            try {

                value = bufferManager.coerce(dataType, aValue);

                rasterStoreBands.get(rowIndex).setNoDataNumber((Number)value);

            } catch (LocatorException | CoercionException e) {

                I18nManager i18nManager = ToolsLocator.getI18nManager();

                LOG.warn(i18nManager.getTranslation(

                    "_it_cant_be_reconized_XvalueX_as_a_XdatatypenameX.",

                    new String[]{aValue.toString(), bufferManager.getTypeName(dataType)}));

            }

//

//            if(noData==null){

//                value = bufferManager.createNoData((Number)noDataValue, null);

//                rasterStoreBands.get(rowIndex).setNoData((NoData)value);

//            } else {

//                noData.setValue((Number)noDataValue);

//                value = noData;

//            }

        }

        super.setValueAt(value, rowIndex, columnIndex);

    }

    /**

     * @param rowIndex

     */

    private void setUndefinedAnyOtherBand(int rowIndex) {

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

            if(i!=rowIndex){

                rasterStoreBands.get(i).setBandColorInterpretation(ColorInterpretation.UNDEFINED_BAND);

            }

        }

    }

    /**

     * Sets undefined the color interpretation of any other band with the same color interpretation

     * or with a color interpretation incompatible with the color space of the colorInterpretation parameter.

     * @param rowIndex

     * @param colorInterpretation

     */

    private void setUndefinedAnyOtherBandWithSameColorInterpretation(int rowIndex, String colorInterpretation) {

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

            if(i!=rowIndex){

                if(((String)getValueAt(i, 0)).equalsIgnoreCase(colorInterpretation)){

                    rasterStoreBands.get(i).setBandColorInterpretation(ColorInterpretation.UNDEFINED_BAND);

                }

                if(!areCompatible(colorInterpretation, ((String)getValueAt(i, 0)))){

                    rasterStoreBands.get(i).setBandColorInterpretation(ColorInterpretation.UNDEFINED_BAND);

                }

            }

        }

    }

    private boolean areCompatible(String colorInterpretation, String colorInterpretation2) {

        for(List<String> colorSpace : ColorSpaces) {

            if(colorSpace.contains(colorInterpretation)){

                if(colorSpace.contains(colorInterpretation2)){

                    return true;

                }

            }

        }

        return false;

    }

    /**

     * @param rasterStoreBand

     */

    public void add(RasterStoreBand rasterStoreBand) {

        rasterStoreBands.add(rasterStoreBand);

    }

    /**

     * @param selectedRow

     */

    public void removeElementAt(int selectedRow) {

        rasterStoreBands.remove(selectedRow);

    }

    public RasterStoreBand getElementAt(int selectedRow) {

        return rasterStoreBands.get(selectedRow);

    }

}