org.jaitools.media.jai.rangelookup.RangeLookupOpImage Maven / Gradle / Ivy
/*
* Copyright (c) 2009, Michael Bedward. 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.
*
* 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 HOLDER 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 org.jaitools.media.jai.rangelookup;
import java.awt.Rectangle;
import java.awt.image.DataBuffer;
import java.awt.image.Raster;
import java.awt.image.RenderedImage;
import java.awt.image.WritableRaster;
import java.util.Map;
import javax.media.jai.AreaOpImage;
import javax.media.jai.ImageLayout;
import javax.media.jai.PointOpImage;
import javax.media.jai.RasterAccessor;
import javax.media.jai.RasterFormatTag;
/**
* This is a variation on the JAI {@linkplain javax.media.jai.LookupDescriptor}.
* It works with a {@linkplain RangeLookupTable} object in which each entry maps
* a source image value range to a destination image value.
*
* @see RangeLookupDescriptor
*
* @author Michael Bedward
* @since 1.0
* @version $Id$
*/
public class RangeLookupOpImage extends PointOpImage {
private RangeLookupTable table;
/* Source image variables */
private int[] srcBandOffsets;
private int srcPixelStride;
private int srcScanlineStride;
/* Destination image variables */
private int destWidth;
private int destHeight;
private int destBands;
private int[] dstBandOffsets;
private int dstPixelStride;
private int dstScanlineStride;
/**
* Constructor
* @param source a RenderedImage.
* @param config configurable attributes of the image (see {@link AreaOpImage})
* @param layout an ImageLayout optionally containing the tile grid layout,
* SampleModel, and ColorModel, or null.
* @param table an instance of RangeLookupTable that defines the mappings from source
* image value ranges to destination image values
*
* @see RangeLookupDescriptor
*/
public RangeLookupOpImage(RenderedImage source,
Map config,
ImageLayout layout,
RangeLookupTable table) {
super(source, layout, config, true);
this.table = table;
}
/**
* Do lookups for the specified destination rectangle
*
* @param sources an array of source Rasters
* @param dest a WritableRaster tile containing the area to be computed.
* @param destRect the rectangle within dest to be processed.
*/
@Override
protected void computeRect(Raster[] sources, WritableRaster dest, Rectangle destRect) {
RasterFormatTag[] formatTags = getFormatTags();
Raster source = sources[0];
Rectangle srcRect = mapDestRect(destRect, 0);
RasterAccessor srcAcc =
new RasterAccessor(source, srcRect,
formatTags[0], getSourceImage(0).getColorModel());
RasterAccessor destAcc =
new RasterAccessor(dest, destRect,
formatTags[1], getColorModel());
doLookup(srcAcc, destAcc);
}
private void doLookup(RasterAccessor srcAcc, RasterAccessor destAcc) {
destWidth = destAcc.getWidth();
destHeight = destAcc.getHeight();
destBands = destAcc.getNumBands();
dstBandOffsets = destAcc.getBandOffsets();
dstPixelStride = destAcc.getPixelStride();
dstScanlineStride = destAcc.getScanlineStride();
srcBandOffsets = srcAcc.getBandOffsets();
srcPixelStride = srcAcc.getPixelStride();
srcScanlineStride = srcAcc.getScanlineStride();
switch (destAcc.getDataType()) {
case DataBuffer.TYPE_BYTE:
lookupAsByteData(srcAcc, destAcc);
break;
case DataBuffer.TYPE_INT:
lookupAsIntData(srcAcc, destAcc);
break;
case DataBuffer.TYPE_SHORT:
lookupAsShortData(srcAcc, destAcc);
break;
case DataBuffer.TYPE_USHORT:
lookupAsUShortData(srcAcc, destAcc);
break;
case DataBuffer.TYPE_FLOAT:
lookupAsFloatData(srcAcc, destAcc);
break;
case DataBuffer.TYPE_DOUBLE:
lookupAsDoubleData(srcAcc, destAcc);
break;
}
if (destAcc.isDataCopy()) {
destAcc.clampDataArrays();
destAcc.copyDataToRaster();
}
}
private void lookupAsByteData(RasterAccessor srcAcc, RasterAccessor destAcc) {
byte srcData[][] = srcAcc.getByteDataArrays();
byte destData[][] = destAcc.getByteDataArrays();
for (int k = 0; k < destBands; k++) {
int destY = destAcc.getY();
byte destBandData[] = destData[k];
byte srcBandData[] = srcData[k];
int srcScanlineOffset = srcBandOffsets[k];
int dstScanlineOffset = dstBandOffsets[k];
for (int j = 0; j < destHeight; j++, destY++) {
int destX = destAcc.getX();
int srcPixelOffset = srcScanlineOffset;
int dstPixelOffset = dstScanlineOffset;
for (int i = 0; i < destWidth; i++, destX++) {
byte val = (byte) (srcBandData[srcPixelOffset] & 0xff);
destBandData[dstPixelOffset] = table.getDestValue(val).byteValue();
srcPixelOffset += srcPixelStride;
dstPixelOffset += dstPixelStride;
}
srcScanlineOffset += srcScanlineStride;
dstScanlineOffset += dstScanlineStride;
}
}
}
private void lookupAsShortData(RasterAccessor srcAcc, RasterAccessor destAcc) {
short srcData[][] = srcAcc.getShortDataArrays();
short destData[][] = destAcc.getShortDataArrays();
for (int k = 0; k < destBands; k++) {
int destY = destAcc.getY();
short destBandData[] = destData[k];
short srcBandData[] = srcData[k];
int srcScanlineOffset = srcBandOffsets[k];
int dstScanlineOffset = dstBandOffsets[k];
for (int j = 0; j < destHeight; j++, destY++) {
int destX = destAcc.getX();
int srcPixelOffset = srcScanlineOffset;
int dstPixelOffset = dstScanlineOffset;
for (int i = 0; i < destWidth; i++, destX++) {
short val = srcBandData[srcPixelOffset];
destBandData[dstPixelOffset] = table.getDestValue(val).shortValue();
srcPixelOffset += srcPixelStride;
dstPixelOffset += dstPixelStride;
}
srcScanlineOffset += srcScanlineStride;
dstScanlineOffset += dstScanlineStride;
}
}
}
private void lookupAsUShortData(RasterAccessor srcAcc, RasterAccessor destAcc) {
short srcData[][] = srcAcc.getShortDataArrays();
short destData[][] = destAcc.getShortDataArrays();
for (int k = 0; k < destBands; k++) {
int destY = destAcc.getY();
short destBandData[] = destData[k];
short srcBandData[] = srcData[k];
int srcScanlineOffset = srcBandOffsets[k];
int dstScanlineOffset = dstBandOffsets[k];
for (int j = 0; j < destHeight; j++, destY++) {
int destX = destAcc.getX();
int srcPixelOffset = srcScanlineOffset;
int dstPixelOffset = dstScanlineOffset;
for (int i = 0; i < destWidth; i++, destX++) {
int val = srcBandData[srcPixelOffset] & 0xffff;
destBandData[dstPixelOffset] = table.getDestValue(val).shortValue();
srcPixelOffset += srcPixelStride;
dstPixelOffset += dstPixelStride;
}
srcScanlineOffset += srcScanlineStride;
dstScanlineOffset += dstScanlineStride;
}
}
}
private void lookupAsIntData(RasterAccessor srcAcc, RasterAccessor destAcc) {
int srcData[][] = srcAcc.getIntDataArrays();
int destData[][] = destAcc.getIntDataArrays();
for (int k = 0; k < destBands; k++) {
int destY = destAcc.getY();
int destBandData[] = destData[k];
int srcBandData[] = srcData[k];
int srcScanlineOffset = srcBandOffsets[k];
int dstScanlineOffset = dstBandOffsets[k];
for (int j = 0; j < destHeight; j++, destY++) {
int destX = destAcc.getX();
int srcPixelOffset = srcScanlineOffset;
int dstPixelOffset = dstScanlineOffset;
for (int i = 0; i < destWidth; i++, destX++) {
int val = srcBandData[srcPixelOffset];
destBandData[dstPixelOffset] = table.getDestValue(val).intValue();
srcPixelOffset += srcPixelStride;
dstPixelOffset += dstPixelStride;
}
srcScanlineOffset += srcScanlineStride;
dstScanlineOffset += dstScanlineStride;
}
}
}
private void lookupAsFloatData(RasterAccessor srcAcc, RasterAccessor destAcc) {
float srcData[][] = srcAcc.getFloatDataArrays();
float destData[][] = destAcc.getFloatDataArrays();
for (int k = 0; k < destBands; k++) {
int destY = destAcc.getY();
float destBandData[] = destData[k];
float srcBandData[] = srcData[k];
int srcScanlineOffset = srcBandOffsets[k];
int dstScanlineOffset = dstBandOffsets[k];
for (int j = 0; j < destHeight; j++, destY++) {
int destX = destAcc.getX();
int srcPixelOffset = srcScanlineOffset;
int dstPixelOffset = dstScanlineOffset;
for (int i = 0; i < destWidth; i++, destX++) {
float val = srcBandData[srcPixelOffset];
destBandData[dstPixelOffset] = table.getDestValue(val).floatValue();
srcPixelOffset += srcPixelStride;
dstPixelOffset += dstPixelStride;
}
srcScanlineOffset += srcScanlineStride;
dstScanlineOffset += dstScanlineStride;
}
}
}
private void lookupAsDoubleData(RasterAccessor srcAcc, RasterAccessor destAcc) {
double srcData[][] = srcAcc.getDoubleDataArrays();
double destData[][] = destAcc.getDoubleDataArrays();
for (int k = 0; k < destBands; k++) {
int destY = destAcc.getY();
double destBandData[] = destData[k];
double srcBandData[] = srcData[k];
int srcScanlineOffset = srcBandOffsets[k];
int dstScanlineOffset = dstBandOffsets[k];
for (int j = 0; j < destHeight; j++, destY++) {
int destX = destAcc.getX();
int srcPixelOffset = srcScanlineOffset;
int dstPixelOffset = dstScanlineOffset;
for (int i = 0; i < destWidth; i++, destX++) {
double val = srcBandData[srcPixelOffset];
destBandData[dstPixelOffset] = table.getDestValue(val).doubleValue();
srcPixelOffset += srcPixelStride;
dstPixelOffset += dstPixelStride;
}
srcScanlineOffset += srcScanlineStride;
dstScanlineOffset += dstScanlineStride;
}
}
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy