/*
* Copyright (c) 2009-2012 jMonkeyEngine
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are
* met:
*
* * Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
*
* * Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
*
* * Neither the name of 'jMonkeyEngine' nor the names of its contributors
* may be used to endorse or promote products derived from this software
* without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
* TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
* PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
* CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
* EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
* PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
* PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
* LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
* NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/
package com.jme3.texture.image;
import com.jme3.math.ColorRGBA;
import com.jme3.math.FastMath;
import com.jme3.texture.Image;
import java.nio.ByteBuffer;
public class MipMapImageRaster extends ImageRaster {
private final int[] components = new int[4];
private ByteBuffer buffer;
private final Image image;
private final ImageCodec codec;
private int width[];
private int height[];
private final byte[] temp;
private int slice;
private int mipLevel;
private int[] offsets;
private void rangeCheck(int x, int y) {
if (x < 0 || y < 0 || x >= width[mipLevel] || y >= height[mipLevel]) {
throw new IllegalArgumentException("x and y must be inside the image dimensions");
}
}
public MipMapImageRaster(Image image, int slice) {
this.image = image;
this.slice = slice;
this.buffer = image.getData(slice);
this.codec = ImageCodec.lookup(image.getFormat());
if (image.hasMipmaps()) {
int nbMipMap = image.getMipMapSizes().length;
this.width = new int[nbMipMap];
this.height = new int[nbMipMap];
this.offsets = new int[nbMipMap];
for (int i = 0; i < nbMipMap; i++) {
width[i] = Math.max(1, image.getWidth() >> i);
height[i] = Math.max(1, image.getHeight() >> i);
if (i > 0) {
offsets[i] = image.getMipMapSizes()[i - 1] + offsets[i - 1];
}
}
} else {
throw new IllegalArgumentException("Image must have MipMapSizes initialized.");
}
if (codec instanceof ByteAlignedImageCodec || codec instanceof ByteOffsetImageCodec) {
this.temp = new byte[codec.bpp];
} else {
this.temp = null;
}
}
public void setSlice(int slice) {
this.slice = slice;
this.buffer = image.getData(slice);
}
public void setMipLevel(int mipLevel) {
if (mipLevel >= image.getMipMapSizes().length || mipLevel < 0) {
throw new IllegalArgumentException("Mip level must be between 0 and " + image.getMipMapSizes().length);
}
this.mipLevel = mipLevel;
}
@Override
public void setPixel(int x, int y, ColorRGBA color) {
rangeCheck(x, y);
// Check flags for grayscale
if (codec.isGray) {
float gray = color.r * 0.27f + color.g * 0.67f + color.b * 0.06f;
color = new ColorRGBA(gray, gray, gray, color.a);
}
switch (codec.type) {
case ImageCodec.FLAG_F16:
components[0] = (int) FastMath.convertFloatToHalf(color.a);
components[1] = (int) FastMath.convertFloatToHalf(color.r);
components[2] = (int) FastMath.convertFloatToHalf(color.g);
components[3] = (int) FastMath.convertFloatToHalf(color.b);
break;
case ImageCodec.FLAG_F32:
components[0] = (int) Float.floatToIntBits(color.a);
components[1] = (int) Float.floatToIntBits(color.r);
components[2] = (int) Float.floatToIntBits(color.g);
components[3] = (int) Float.floatToIntBits(color.b);
break;
case 0:
// Convert color to bits by multiplying by size
components[0] = Math.min((int) (color.a * codec.maxAlpha + 0.5f), codec.maxAlpha);
components[1] = Math.min((int) (color.r * codec.maxRed + 0.5f), codec.maxRed);
components[2] = Math.min((int) (color.g * codec.maxGreen + 0.5f), codec.maxGreen);
components[3] = Math.min((int) (color.b * codec.maxBlue + 0.5f), codec.maxBlue);
break;
}
codec.writeComponents(getBuffer(), x, y, width[mipLevel], offsets[mipLevel], components, temp);
image.setUpdateNeeded();
}
private ByteBuffer getBuffer() {
if (buffer == null) {
this.buffer = image.getData(slice);
}
return buffer;
}
@Override
public ColorRGBA getPixel(int x, int y, ColorRGBA store) {
rangeCheck(x, y);
codec.readComponents(getBuffer(), x, y, width[mipLevel], offsets[mipLevel], components, temp);
if (store == null) {
store = new ColorRGBA();
}
switch (codec.type) {
case ImageCodec.FLAG_F16:
store.set(FastMath.convertHalfToFloat((short) components[1]),
FastMath.convertHalfToFloat((short) components[2]),
FastMath.convertHalfToFloat((short) components[3]),
FastMath.convertHalfToFloat((short) components[0]));
break;
case ImageCodec.FLAG_F32:
store.set(Float.intBitsToFloat((int) components[1]),
Float.intBitsToFloat((int) components[2]),
Float.intBitsToFloat((int) components[3]),
Float.intBitsToFloat((int) components[0]));
break;
case 0:
// Convert to float and divide by bitsize to get into range 0.0 - 1.0.
store.set((float) components[1] / codec.maxRed,
(float) components[2] / codec.maxGreen,
(float) components[3] / codec.maxBlue,
(float) components[0] / codec.maxAlpha);
break;
}
if (codec.isGray) {
store.g = store.b = store.r;
} else {
if (codec.maxRed == 0) {
store.r = 1;
}
if (codec.maxGreen == 0) {
store.g = 1;
}
if (codec.maxBlue == 0) {
store.b = 1;
}
if (codec.maxAlpha == 0) {
store.a = 1;
}
}
return store;
}
@Override
public int getWidth() {
return width[mipLevel];
}
@Override
public int getHeight() {
return height[mipLevel];
}
}