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.impl.store.properties;

import java.util.ArrayList;

import java.util.List;

import org.gvsig.fmap.dal.coverage.store.props.ColorInterpretation;

/**

 * Clase que contiene la interpretaci?n de color por banda. Inicialmente

 * es inicializada con los valores contenidos en el raster si los tiene. Despu?s 

 * estos valores pueden ser modificados.

 * 

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

 */

public class DataStoreColorInterpretation implements ColorInterpretation {

        /**

         * Interpretaci?n de color para cada banda

         */

        private String[]           colorInterpretation = null;

        /**

         * true si la imagen tiene una banda con el identificador de interpretaci?n de

         * color a Alpha

         */

        /**

         * Constructor vacio. 

         */

        public DataStoreColorInterpretation() {

                this.colorInterpretation = new String[0];

        }

        /**

         * Constructor que asigna los valores de interpretaci?n de color 

         */

        public DataStoreColorInterpretation(String[] colorInterp) {

                this.colorInterpretation = colorInterp;

        }

        public static DataStoreColorInterpretation createDefaultInterpretation(int nBands) {

                if(nBands <= 0)

                        return null;

                if(nBands == 1)

                        return new DataStoreColorInterpretation(new String[]{GRAY_BAND});

                if(nBands == 2)

                        return new DataStoreColorInterpretation(new String[]{GRAY_BAND, ALPHA_BAND});

                if(nBands == 3)

                        return new DataStoreColorInterpretation(new String[]{RED_BAND, GREEN_BAND, BLUE_BAND});

                return new DataStoreColorInterpretation(new String[]{RED_BAND, GREEN_BAND, BLUE_BAND, ALPHA_BAND});

        }

        public static DataStoreColorInterpretation createPaletteInterpretation() {

                return new DataStoreColorInterpretation(new String[]{PAL_BAND});

        }

        /**

         * Obtiene una interpretaci?n de color GRAY

         * @return DatasetColorInterpretation

         */

        public static DataStoreColorInterpretation createGrayInterpretation() {

                return new DataStoreColorInterpretation(new String[]{GRAY_BAND});

        }

        /**

         * Obtiene una interpretaci?n de color RGB

         * @return DatasetColorInterpretation

         */

        public static DataStoreColorInterpretation createARGBInterpretation() {

                return new DataStoreColorInterpretation(new String[]{RED_BAND, GREEN_BAND, BLUE_BAND, ALPHA_BAND});

        }

        /**

         * Obtiene una interpretaci?n de color RGB

         * @return DatasetColorInterpretation

         */

        public static DataStoreColorInterpretation createRGBInterpretation() {

                return new DataStoreColorInterpretation(new String[]{RED_BAND, GREEN_BAND, BLUE_BAND});

        }

        /**

         * Constructor que inicializa el n?mero de valores de la interpretaci?n de 

         * color. Implica asignar posteriormente los valores a las bandas.

         */

        public DataStoreColorInterpretation(int values) {

                colorInterpretation = new String[values];

        }

        /**

         * Inicializa el vector de cadenas que contendr?n el nombre de la interpretaci?n 

         * de color asignada a cada banda. Este valor es el devuelto por la imagen.

         * @param values N?mero de valores

         */

        public void initColorInterpretation(int values) {

                colorInterpretation = new String[values];

        }

        public boolean isRGB() {

                if(colorInterpretation != null) {

                        if (colorInterpretation.length == 3 && 

                                colorInterpretation[0] == RED_BAND && 

                                colorInterpretation[1] == BLUE_BAND && 

                                colorInterpretation[2] == BLUE_BAND)

                                return true;

                }

                return false;

        }

        public boolean isARGB() {

                if(colorInterpretation != null) {

                        if (colorInterpretation.length == 4 && 

                                colorInterpretation[0] == RED_BAND && 

                                colorInterpretation[1] == BLUE_BAND && 

                                colorInterpretation[2] == BLUE_BAND &&

                                colorInterpretation[3] == ALPHA_BAND)

                                return true;

                }

                return false;

        }

        public String[] getValues() {

                return colorInterpretation;

        }

        /**

         * Asigna los valores de la interpretaci?n de color

         * @return String[]

         */

        public void setValues(String[] colorInterp) {

                colorInterpretation = colorInterp;

        }

        public void setColorInterpValue(int band, String value) {

                try{

                        colorInterpretation[band] = value;

                }catch(ArrayIndexOutOfBoundsException ex) {

                        //No asignamos el elemento

                }

        }

        public int getBand(String id) {

                if(colorInterpretation != null) {

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

                                if(colorInterpretation[i] != null && colorInterpretation[i].equals(id))

                                        return i;

                }

                return -1;

        }

        /**

         * Obtiene la posici?n de las bandas que contienen el identificador pasado por par?metro 

         * o null si no tiene dicho identificador.

         * @return Lista con las posiciones de las bandas que contienen el identificador o null si no lo tiene.

         */

        public int[] getBands(String id) {

                if(colorInterpretation != null){

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

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

                                if(colorInterpretation[i].equals(id))

                                        array.add(new Integer(i));

                        int[] list = new int[array.size()];

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

                                list[i] = ((Integer)array.get(i)).intValue();

                        return list;

                }

                return null;

        }

        public boolean hasAlphaBand() {

                String[] values = getValues();

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

                        if(values[i]  != null && values[i].equals("Alpha"))

                                return true;

                }

                return false;

        }        

        public int length() {

                return colorInterpretation.length;

        }

        public String get(int i) {

                if (i >= colorInterpretation.length)

                        return null;

                return colorInterpretation[i];

        }

        public void addColorInterpretation(ColorInterpretation ci) {

                String[] newCI = new String[colorInterpretation.length + ci.length()];

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

                        newCI[i] = colorInterpretation[i];

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

                        newCI[colorInterpretation.length + i] = ci.get(i);

                }

                this.colorInterpretation = newCI;

        }

        public boolean isUndefined() {

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

                        if (colorInterpretation[i] != null) {

                                if (!colorInterpretation[i].equals(UNDEF_BAND) &&

                                                !colorInterpretation[i].equals(ALPHA_BAND))

                                        return false;

                        }

                }

                return true;

        }

        public ColorInterpretation cloneColorInterpretation() {

                DataStoreColorInterpretation ci = new DataStoreColorInterpretation();

                String[] l = new String[colorInterpretation.length];

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

                        l[i] = colorInterpretation[i];

                }

                ci.colorInterpretation = l;

                return ci;

        }

}