Application: gvSIG desktop » gvSIG raster

package org.gvsig.raster.impl.provider.fusion;

import java.awt.geom.Point2D;

import java.awt.geom.Rectangle2D;

import java.util.ArrayList;

import java.util.List;

import org.gvsig.fmap.dal.coverage.dataset.Buffer;

import org.gvsig.raster.impl.datastruct.ExtentImpl;

public class FirstFusionMethod extends AbstractFusionMethod {

        protected SubSquareList     subSquareList    = null;

        private int                 bandCount        = 0;

        private int[]               lastPixelCoords  = new int[]{-1, -1};

        private byte[]              dataByte         = null;

        private short[]             dataShort        = null;

        private int[]               dataInt          = null;

        private float[]             dataFloat        = null;

        private double[]            dataDouble       = null;

        public class SubSquareList extends ArrayList<SubSquare> {

                private static final long serialVersionUID = 1L;

                private int               bandCount        = 0;

                public SubSquareList(int bandCount) {

                        this.bandCount = bandCount;

                }

                public byte[] getByteValue(int row, int col) {

                        byte[] value = new byte[bandCount];

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

                                Point2D coord = get(i).getLocalCoords(row, col);

                                if(coord != null) {

                                        get(i).getBuffer().getElemByte((int)coord.getY(), (int)coord.getX(), value);

                                        return value;

                                }

                        }

                        return null;

                }

                public float[] getFloatValue(int row, int col) {

                        float[] value = new float[bandCount];

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

                                Point2D coord = get(i).getLocalCoords(row, col);

                                if(coord != null) {

                                        get(i).getBuffer().getElemFloat((int)coord.getY(), (int)coord.getX(), value);

                                        return value;

                                }

                        }

                        return null;

                }

        }

        public FirstFusionMethod(

                        List<Buffer> bufferList, 

                        PixelSquareStructure pxSquare) {

                super(pxSquare);

                switch (pxSquare.dataType) {

                case Buffer.TYPE_BYTE:

                        dataByte = new byte[pxSquare.bandCount];

                        break;

                case Buffer.TYPE_SHORT:

                        dataShort = new short[pxSquare.bandCount];

                        break;

                case Buffer.TYPE_INT:

                        dataInt = new int[pxSquare.bandCount];

                        break;

                case Buffer.TYPE_FLOAT:

                        dataFloat = new float[pxSquare.bandCount];

                        break;

                case Buffer.TYPE_DOUBLE:

                        dataDouble = new double[pxSquare.bandCount];

                        break;

                }

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

                        bandCount = bufferList.get(i).getBandCount() > bandCount ? bufferList.get(i).getBandCount() : bandCount; 

                }

                subSquareList = new SubSquareList(bandCount);

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

                        SubSquare square = new SubSquare(

                                        pxSquare.bbox, 

                                        new Rectangle2D.Double(0, 0, pxSquare.width, pxSquare.height),

                                        new ExtentImpl(bufferList.get(i).getDataExtent()),

                                        new Rectangle2D.Double(0, 0, bufferList.get(i).getWidth(), bufferList.get(i).getHeight()),

                                        bufferList.get(i));

                        subSquareList.add(square);

                }

        }

        public Byte getByteValue(int row, int col, int band) {

                if(row != lastPixelCoords[0] || col != lastPixelCoords[1]) {

                        //Lee todas las bandas de una y devuelve la que corresponda al par?metro band

                        dataByte = subSquareList.getByteValue(row, col);

                        lastPixelCoords[0] = row;

                        lastPixelCoords[1] = col;

                }

                if(dataByte != null)

                        return dataByte[band];

                return null;

        }

        public Float getFloatValue(int row, int col, int band) {

                if(row == lastPixelCoords[0] && col == lastPixelCoords[1]) {

                        dataFloat = subSquareList.getFloatValue(row, col);

                        lastPixelCoords[0] = row;

                        lastPixelCoords[1] = col;

                }

                if(dataFloat != null)

                        return dataFloat[band];

                return null;

        }

        public int getBandCount() {

                return bandCount;

        }

}