Application: gvSIG desktop

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

import java.awt.Color;

import java.awt.image.BufferedImage;

import java.util.ArrayList;

import java.util.List;

import org.cresques.cts.IProjection;

import org.gvsig.fmap.geom.Geometry;

import org.gvsig.fmap.geom.GeometryLocator;

import org.gvsig.fmap.geom.GeometryManager;

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

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

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

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

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

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

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

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

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

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

import org.gvsig.tools.locator.LocatorException;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

/**

 * @author fdiaz

 *

 */

public class DefaultBuffer extends AbstractBuffer {

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

    /**

     * Default constructor of buffer.

     *

     * @param rows

     *            Rows of buffer and buffer bands.

     * @param columns

     *            Columns of buffer bands.

     * @param bandDataTypes

     *            Type of buffer bands. The band types also indicates the number

     *            of buffer bands.

     * @param bandNoData

     *            Band noData. If there are bans without NoData, bands will have

     *            undefined NoData value.

     * @param projection

     *            Projection of buffer

     * @param envelope

     *            Envelope of buffer

     * @param pageManagers

     *            Page manager to paginate this buffer. If page manager is null,

     *            buffer will be loaded in memory.

     * @throws CreateBufferException

     *             If there are some problems creating buffer.

     */

    public DefaultBuffer(int rows, int columns, int[] bandDataTypes, NoData[] bandNoData,

        IProjection projection, Envelope envelope, List<PageManager> pageManagers)

        throws CreateBufferException {

        if (rows < 1) {

            try {

                throw new IllegalArgumentException("Can't create buffer with rows = " + rows);

            } catch (IllegalArgumentException e) {

                logger.warn(e.getMessage(), e);

                throw e;

            }

        }

        if (columns < 1) {

            try {

                throw new IllegalArgumentException("Can't create buffer with columns = " + columns);

            } catch (IllegalArgumentException e) {

                logger.warn(e.getMessage(), e);

                throw e;

            }

        }

        this.bands = new ArrayList<Band>();

        this.projection = projection;

        try {

            if (envelope == null) {

                GeometryManager geometryManager = GeometryLocator.getGeometryManager();

                envelope =

                    geometryManager.createEnvelope(0, 0, columns, rows,

                        Geometry.SUBTYPES.GEOM2D);

            }

            this.dimensions = BufferLocator.getBufferManager().createBufferDimensions(rows, columns, envelope);

            if (bandDataTypes != null) {

                createBands(bandDataTypes, bandNoData, pageManagers);

            }

        } catch (Exception e) {

            throw new CreateBufferException(e);

        }

    }

    private void createBands(int[] bandDataTypes, NoData[] bandNoData,

        List<PageManager> pageManagers) throws BandException {

        BufferManager bufferManager = BufferLocator.getBufferManager();

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

            NoData noDataBand = null;

            if (bandNoData != null && i < bandNoData.length) {

                noDataBand = bandNoData[i];

            }

            PageManager pageManager = null;

            if (pageManagers != null && i < pageManagers.size()) {

                pageManager = pageManagers.get(i);

            }

            Band band =

                bufferManager.createBand(bandDataTypes[i], this.getDimensions().getRows(), this.getDimensions().getColumns(), noDataBand,

                    pageManager);

            bands.add(band);

            band.addObserver(this);

        }

    }

    @Override

    public Object clone() throws CloneNotSupportedException {

        Buffer cloned;

        try {

            cloned =

                BufferLocator.getBufferManager().createBuffer(this.getRows(), this.getColumns(),

                    this.getBandTypes(), this.getBandNoData(), this.getProjection(), this.dimensions.getEnvelope());

        } catch (LocatorException | BufferException e) {

            throw new CloneNotSupportedException("Can't clone buffer.");

        }

        for (int i = 0; i < this.getBandCount(); i++) {

            cloned.addBand((Band) this.getBand(i).clone());

        }

        return cloned;

    }

    public BufferedImage getBufferedImage() {

        int imageType;

        int[] bandTypes = getBandTypes();

        if(bandTypes.length<3 ||bandTypes.length>4){

            return null;

        }

        int type = bandTypes[0];

        if(type == BufferManager.TYPE_BYTE){

            if(bandTypes.length==3){

                imageType = BufferedImage.TYPE_3BYTE_BGR;

            } else {

                imageType = BufferedImage.TYPE_4BYTE_ABGR;

            }

//        } else if(type == BufferManager.TYPE_INT) {

//            if(bandTypes.length==3){

//                imageType = BufferedImage.TYPE_INT_RGB;

//            } else {

//                imageType = BufferedImage.TYPE_INT_ARGB;

//            }

        } else {

            return null;

        }

        BufferedImage img = new BufferedImage(getColumns(), getRows(), imageType);

        for (int row=0; row<this.getRows(); row++){

            for (int column=0; column<this.getColumns(); column++){

                Color color;

                if(getBandCount()==4){

                    color = new Color(

                        ((Number)getBand(0).get(row, column)).intValue(),

                        ((Number)getBand(1).get(row, column)).intValue(),

                        ((Number)getBand(2).get(row, column)).intValue(),

                        ((Number)getBand(3).get(row, column)).intValue()

                    );

                } else {

                    color = new Color(

                        ((Number)getBand(0).get(row, column)).intValue(),

                        ((Number)getBand(1).get(row, column)).intValue(),

                        ((Number)getBand(2).get(row, column)).intValue()

                    );

                }

                img.setRGB(row, column, color.getRGB());

            }

        }

        return img;

    }

}