Application: gvSIG desktop » Library: org.gvsig.tools

/**

 * gvSIG. Desktop Geographic Information System.

 *

 * Copyright (C) 2007-2013 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 3

 * 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.tools.swing.impl.bufferedImage;

import java.awt.Point;

import java.awt.image.BufferedImage;

import java.awt.image.ColorModel;

import java.awt.image.DataBuffer;

import java.awt.image.SampleModel;

import java.io.File;

import java.io.IOException;

import java.io.RandomAccessFile;

import java.nio.Buffer;

import java.nio.ByteBuffer;

import java.nio.IntBuffer;

import java.nio.MappedByteBuffer;

import java.nio.ShortBuffer;

import java.nio.channels.FileChannel;

import org.gvsig.tools.ToolsLocator;

/*

 * Based on portions of code from TwelveMonkeys ImageIO 3.3.2

 * under BSD 3-Clause License.

 * Copyright (c) 2008-2015, Harald Kuhr

 * https://github.com/haraldk/TwelveMonkeys

 */

public class VirtualBufferedImageHelper {

    public static abstract class VirtualDataBuffer extends DataBuffer {

        protected final Buffer buffer;

        protected VirtualDataBuffer(final int type, final int size, final int numBanks) throws IOException {

            super(type, size, numBanks);

            if( size < 0 ) {

                throw new IllegalArgumentException("size can't be less than zero");

            }

            int componentSize = DataBuffer.getDataTypeSize(type) / 8;

            File tempFile = ToolsLocator.getFoldersManager().getUniqueTemporaryFile("BufferedImage.dat");

            try {

                RandomAccessFile raf = new RandomAccessFile(tempFile, "rw");

                long length = ((long) size) * componentSize * numBanks;

                raf.setLength(length);

                try (FileChannel channel = raf.getChannel()) {

                    MappedByteBuffer byteBuffer = channel.map(FileChannel.MapMode.READ_WRITE, 0, length);

                    switch( type ) {

                    case DataBuffer.TYPE_BYTE:

                        buffer = byteBuffer;

                        break;

                    case DataBuffer.TYPE_USHORT:

                        buffer = byteBuffer.asShortBuffer();

                        break;

                    case DataBuffer.TYPE_INT:

                        buffer = byteBuffer.asIntBuffer();

                        break;

                    default:

                        throw new IllegalArgumentException("Unsupported data type: " + type);

                    }

                }

            } finally {

                tempFile.deleteOnExit();

            }

        }

    }

    public static class DataBufferByte extends VirtualDataBuffer {

        public DataBufferByte(int size, int numBanks) throws IOException {

            super(DataBuffer.TYPE_BYTE, size, numBanks);

        }

        @Override

        public int getElem(int bank, int i) {

            return ((ByteBuffer) buffer).get(bank * size + i) & 0xff;

        }

        @Override

        public void setElem(int bank, int i, int val) {

            ((ByteBuffer) buffer).put(bank * size + i, (byte) val);

        }

    }

    private static class DataBufferUShort extends VirtualDataBuffer {

        public DataBufferUShort(int size, int numBanks) throws IOException {

            super(DataBuffer.TYPE_USHORT, size, numBanks);

        }

        @Override

        public int getElem(int bank, int i) {

            return ((ShortBuffer) buffer).get(bank * size + i) & 0xffff;

        }

        @Override

        public void setElem(int bank, int i, int val) {

            ((ShortBuffer) buffer).put(bank * size + i, (short) val);

        }

    }

    private static class DataBufferInt extends VirtualDataBuffer {

        public DataBufferInt(int size, int numBanks) throws IOException {

            super(DataBuffer.TYPE_INT, size, numBanks);

        }

        @Override

        public int getElem(int bank, int i) {

            return ((IntBuffer) buffer).get(bank * size + i);

        }

        @Override

        public void setElem(int bank, int i, int val) {

            ((IntBuffer) buffer).put(bank * size + i, val);

        }

    }

    private static class WritableRaster extends java.awt.image.WritableRaster {

        public WritableRaster(final SampleModel model, final DataBuffer buffer) {

            super(model, buffer, new Point()); // Este constructor es protegido.

        }

    }

    public static class IORuntimeException extends RuntimeException {

        private static final long serialVersionUID = -4461981890554048338L;

        public IORuntimeException(IOException ex) {

            super(ex.getMessage(), ex);

        }

    }

    public static DataBuffer createVirtualDataBuffer(final int type, final int size, final int numBanks) throws IOException {

        switch( type ) {

        case DataBuffer.TYPE_BYTE:

            return new DataBufferByte(size, numBanks);

        case DataBuffer.TYPE_USHORT:

            return new DataBufferUShort(size, numBanks);

        case DataBuffer.TYPE_INT:

            return new DataBufferInt(size, numBanks);

        default:

            throw new IllegalArgumentException("Unsupported data type: " + type);

        }

    }

    public static BufferedImage createVirtualBufferedImage(int width, int height, int type) {

        BufferedImage temp = new BufferedImage(1, 1, type);

        SampleModel sampleModel = temp.getSampleModel().createCompatibleSampleModel(width, height);

        ColorModel colorModel = temp.getColorModel();

        BufferedImage x = createVirtualBufferedImage(width, height, sampleModel, colorModel);

        return x;

    }

    public static BufferedImage createVirtualBufferedImage(int width, int height, SampleModel sampleModel, ColorModel colorModel) throws IORuntimeException {

        DataBuffer buffer;

        try {

            buffer = createVirtualDataBuffer(sampleModel.getTransferType(), width * height * sampleModel.getNumDataElements(), 1);

        } catch (IOException ex) {

            throw new IORuntimeException(ex);

        }

        WritableRaster raster = new WritableRaster(sampleModel, buffer);

        BufferedImage x = new BufferedImage(colorModel, raster, colorModel.isAlphaPremultiplied(), null);

//        try {

//            Field field = x.getClass().getDeclaredField("imageType");

//            field.setAccessible(true);

//            field.setInt(x, type);

//            field.setAccessible(false);

//        } catch (NoSuchFieldException | SecurityException | IllegalArgumentException | IllegalAccessException e) {

//            throw new RuntimeException("Can't set the image type to the BufferedImage.", e);

//        }

        return x;

    }

}