Application: gvSIG desktop » gvSIG raster

package org.gvsig.raster.lib.buffer.impl;

import java.util.ArrayList;

import java.util.Iterator;

import java.util.List;

import org.cresques.cts.ICoordTrans;

import org.cresques.cts.IProjection;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

import org.gvsig.fmap.geom.operation.GeometryOperationException;

import org.gvsig.fmap.geom.operation.GeometryOperationNotSupportedException;

import org.gvsig.fmap.geom.primitive.Envelope;

import org.gvsig.fmap.geom.primitive.Point;

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

import org.gvsig.raster.lib.buffer.api.Band.BandByte;

import org.gvsig.raster.lib.buffer.api.Band.BandDouble;

import org.gvsig.raster.lib.buffer.api.Band.BandFloat;

import org.gvsig.raster.lib.buffer.api.Band.BandInt;

import org.gvsig.raster.lib.buffer.api.Band.BandShort;

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

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

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

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

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

import org.gvsig.raster.lib.buffer.api.exceptions.BandException;

import org.gvsig.raster.lib.buffer.api.statistics.Histogram;

import org.gvsig.raster.lib.buffer.api.statistics.Statistics;

import org.gvsig.raster.lib.buffer.impl.exceptions.CreateBufferException;

import org.gvsig.tools.task.SimpleTaskStatus;

/**

 * @author fdiaz

 *

 */

public class DefaultBuffer implements Buffer {

    protected static final Logger logger = LoggerFactory.getLogger(DefaultBuffer.class);

    private List<Band> bands;

    private int rows;

    private int columns;

    private FilterList filters;

    private IProjection projection;

    private BandPageManager bandPageManager;

    /**

     * @author fdiaz

     *

     */

    public class BandsIterator implements Iterator<Band> {

        private Band band;

        private int current;

        /**

         *

         */

        public BandsIterator() {

            this.current = 0;

            if( getBandCount()>0 ) {

                this.band = (Band) getBand(0);

            } else {

                this.band = null;

            }

        }

        @Override

        public boolean hasNext() {

            return this.current < getBandCount() ;

        }

        @Override

        public Band next() {

            this.current++;

            return this.band;

        }

        @Override

        public void remove() {

            throw new UnsupportedOperationException("Not supported yet.");

        }

    }

    /**

     * @param rows

     * @param columns

     * @param bandDataTypes

     * @param projection

     * @param bandPageManager

     * @throws CreateBufferException

     */

    public DefaultBuffer(int rows, int columns, int[] bandDataTypes, IProjection projection, BandPageManager bandPageManager) throws CreateBufferException {

        this.rows = rows;

        this.columns = columns;

        this.projection = projection;

        this.bandPageManager = bandPageManager;

        try {

            createBands(bandDataTypes);

        } catch (BandException e) {

            throw new CreateBufferException(e);

        }

    }

    private void createBands(int[] bandDataTypes) throws BandException {

        BufferManager bufferManager = RasterLocator.getBufferManager();

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

            bands.add(bufferManager.createBand(bandDataTypes[i], this.rows, this.columns, null, this.bandPageManager));

        }

    }

    /**

     * @param rows

     * @param columns

     * @param projection

     * @throws CreateBufferException

     */

    public DefaultBuffer(int rows, int columns, int[] bandDataTypes, IProjection projection) throws CreateBufferException {

        this(rows, columns, bandDataTypes, projection, null);

    }

    @Override

    public Histogram getHistogram(SimpleTaskStatus status) {

        // TODO Auto-generated method stub

        return null;

    }

    @Override

    public Statistics getStatistics(SimpleTaskStatus status) {

        // TODO Auto-generated method stub

        return null;

    }

    @Override

    public Iterator<Band> iterator() {

        return new BandsIterator();

    }

    @Override

    public void filter(FilterList filterList) {

        this.filters = filterList;

    }

    @Override

    public int getBandCount() {

        return bands.size();

    }

    @Override

    public Band[] getBands() {

        return (Band[]) bands.toArray();

    }

    @Override

    public int getColumns() {

        return this.columns;

    }

    @Override

    public int getRows() {

        return this.rows;

    }

    @Override

    public Envelope getEnvelope() {

        // TODO Auto-generated method stub

        //FIXME: ?De d?nde sacamos este envelope?

        return null;

    }

    @Override

    public IProjection getProjection() {

        return this.projection;

    }

    @Override

    public boolean isInside(int cellX, int cellY) {

        return (cellX>=0 && cellX < this.columns && cellY >= 0 && cellY < this.rows);

    }

    @Override

    public boolean isInside(Point point) {

        try {

            return getEnvelope().getGeometry().contains(point);

        } catch (GeometryOperationNotSupportedException | GeometryOperationException e) {

            logger.warn("It could not determine if the point is on the envelope", e);

            return false;

        }

    }

    @Override

    public void addBand(Band band) {

        this.bands.add(band);

    }

    @Override

    public void setBand(int pos, Band band) throws BandException {

        this.bands.get(pos).copyFrom(band);

    }

    @Override

    public void removeBand(int pos) {

        this.bands.remove(pos);

    }

    @Override

    public Band getBand(int pos) {

        return this.bands.get(pos);

    }

    @Override

    public BandByte getBandByte(int pos) {

        return (BandByte) this.bands.get(pos);

    }

    @Override

    public BandShort getBandShort(int pos) {

        return (BandShort) this.bands.get(pos);

    }

    @Override

    public BandInt getBandInt(int pos) {

        return (BandInt) this.bands.get(pos);

    }

    @Override

    public BandFloat getBandFloat(int pos) {

        return (BandFloat) this.bands.get(pos);

    }

    @Override

    public BandDouble getBandDouble(int pos) {

        return (BandDouble) this.bands.get(pos);

    }

    @Override

    public void switchBands(int[] positions) {

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

        if(positions.length != this.getBandCount()){

            return;

        }

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

            if(positions[i] >=  positions.length || positions[i] < 0) {

                return;

            }

            Integer position = new Integer(positions[i]);

            if(visited.contains(position)){

                return;

            }

            visited.add(position);

        }

        List<Band> auxBands = new ArrayList<Band>(bands.size());

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

            auxBands.set(i, bands.get(positions[i]));

        }

        bands = auxBands;

    }

    @Override

    public void switchBands(int pos1, int pos2) {

        Band auxBand = bands.get(pos1);

        bands.set(pos1, bands.get(pos2));

        bands.set(pos2, auxBand);

    }

    @Override

    public Buffer createInterpolated(int rows, int columns, int interpolationMode, SimpleTaskStatus status) {

        // TODO Auto-generated method stub

        return null;

    }

    @Override

    public Buffer convert(ICoordTrans ct, SimpleTaskStatus status) {

        // TODO Auto-generated method stub

        return null;

    }

    @Override

    public boolean isPaginated() {

        return bandPageManager!=null;

    }

}