/**
* Returns a new RandomAccessInputStream around the raw handle underlying the provided stream,
* instead of using a zip handle.
*
* <p>NB: closes the provided stream.
*/
public static RandomAccessInputStream getRawStream(
final LocationService locationService, final RandomAccessInputStream stream)
throws IOException {
// NB: We need a raw handle on the ZIP data itself, not a ZipHandle.
final String id = stream.getFileName();
final IRandomAccess rawHandle = locationService.getHandle(id, false, false);
return new RandomAccessInputStream(locationService.getContext(), rawHandle, id);
}
/**
* Extracts the String id of the provided stream.
*
* @param locationService - The location service to use for mapping files
* @param stream - Stream, built around a .zip file, to extract the actual id from
* @param mappedFiles - Optional param. If provided, all discovered entries in the underlying
* archive will be added to this list.
* @return An id of the base entry in the .zip
* @throws IOException
*/
public static String unzipId(
final LocationService locationService,
final RandomAccessInputStream stream,
final List<String> mappedFiles)
throws IOException {
final ZipInputStream zip = new ZipInputStream(stream);
ZipEntry ze = null;
while (true) {
ze = zip.getNextEntry();
if (ze == null) break;
final ZipHandle handle =
new ZipHandle(locationService.getContext(), stream.getFileName(), ze);
locationService.mapFile(ze.getName(), handle);
if (mappedFiles != null) mappedFiles.add(ze.getName());
}
final ZipHandle base = new ZipHandle(locationService.getContext(), stream.getFileName());
final String id = base.getEntryName();
base.close();
return id;
}
/**
* The CodecOptions parameter should have the following fields set: {@link CodecOptions#width
* width} {@link CodecOptions#height height}
*
* @see Codec#decompress(RandomAccessInputStream, CodecOptions)
*/
@Override
public byte[] decompress(final RandomAccessInputStream in, CodecOptions options)
throws FormatException, IOException {
if (in == null) throw new IllegalArgumentException("No data to decompress.");
if (options == null) options = CodecOptions.getDefaultOptions();
in.skipBytes(8);
final int plane = options.width * options.height;
stride = options.width;
final int rowInc = stride - 4;
short opcode;
int nBlocks;
int colorA = 0, colorB;
final int[] color4 = new int[4];
int index, idx;
int ta, tb;
int blockPtr = 0;
rowPtr = pixelPtr = 0;
int pixelX, pixelY;
final int[] pixels = new int[plane];
final byte[] rtn = new byte[plane * 3];
while (in.read() != (byte) 0xe1) {
/* Read to block metadata */
}
in.skipBytes(3);
// totalBlocks = ((options.width + 3) / 4) * ((options.height + 3) / 4);
while (in.getFilePointer() + 2 < in.length()) {
opcode = in.readByte();
nBlocks = (opcode & 0x1f) + 1;
if ((opcode & 0x80) == 0) {
if (in.getFilePointer() >= in.length()) break;
colorA = (opcode << 8) | in.read();
opcode = 0;
if (in.getFilePointer() >= in.length()) break;
if ((in.read() & 0x80) != 0) {
opcode = 0x20;
nBlocks = 1;
}
in.seek(in.getFilePointer() - 1);
}
switch (opcode & 0xe0) {
case 0x80:
while (nBlocks-- > 0) {
updateBlock(options.width);
}
break;
case 0xa0:
if (in.getFilePointer() + 2 >= in.length()) break;
colorA = in.readShort();
while (nBlocks-- > 0) {
blockPtr = rowPtr + pixelPtr;
for (pixelY = 0; pixelY < 4; pixelY++) {
for (pixelX = 0; pixelX < 4; pixelX++) {
if (blockPtr >= pixels.length) break;
pixels[blockPtr] = colorA;
final short s = (short) (pixels[blockPtr] & 0x7fff);
unpack(s, rtn, blockPtr, pixels.length);
blockPtr++;
}
blockPtr += rowInc;
}
updateBlock(options.width);
}
break;
case 0xc0:
case 0x20:
if (in.getFilePointer() + 2 >= in.length()) break;
if ((opcode & 0xe0) == 0xc0) {
colorA = in.readShort();
}
colorB = in.readShort();
color4[0] = colorB;
color4[1] = 0;
color4[2] = 0;
color4[3] = colorA;
ta = (colorA >> 10) & 0x1f;
tb = (colorB >> 10) & 0x1f;
color4[1] |= ((11 * ta + 21 * tb) >> 5) << 10;
color4[2] |= ((21 * ta + 11 * tb) >> 5) << 10;
ta = (colorA >> 5) & 0x1f;
tb = (colorB >> 5) & 0x1f;
color4[1] |= ((11 * ta + 21 * tb) >> 5) << 5;
color4[2] |= ((21 * ta + 11 * tb) >> 5) << 5;
ta = colorA & 0x1f;
tb = colorB & 0x1f;
color4[1] |= (11 * ta + 21 * tb) >> 5;
color4[2] |= (21 * ta + 11 * tb) >> 5;
while (nBlocks-- > 0) {
blockPtr = rowPtr + pixelPtr;
for (pixelY = 0; pixelY < 4; pixelY++) {
if (in.getFilePointer() >= in.length()) break;
index = in.read();
for (pixelX = 0; pixelX < 4; pixelX++) {
idx = (index >> (2 * (3 - pixelX))) & 3;
if (blockPtr >= pixels.length) break;
pixels[blockPtr] = color4[idx];
final short s = (short) (pixels[blockPtr] & 0x7fff);
unpack(s, rtn, blockPtr, pixels.length);
blockPtr++;
}
blockPtr += rowInc;
}
updateBlock(options.width);
}
break;
case 0x00:
blockPtr = rowPtr + pixelPtr;
for (pixelY = 0; pixelY < 4; pixelY++) {
for (pixelX = 0; pixelX < 4; pixelX++) {
if ((pixelY != 0) || (pixelX != 0)) {
if (in.getFilePointer() + 2 >= in.length()) break;
colorA = in.readShort();
}
if (blockPtr >= pixels.length) break;
pixels[blockPtr] = colorA;
final short s = (short) (pixels[blockPtr] & 0x7fff);
unpack(s, rtn, blockPtr, pixels.length);
blockPtr++;
}
blockPtr += rowInc;
}
updateBlock(options.width);
break;
}
}
return rtn;
}
/**
* The CodecOptions parameter should have the following fields set: {@link
* CodecOptions#interleaved interleaved} {@link CodecOptions#littleEndian littleEndian}
*
* @see Codec#decompress(RandomAccessInputStream, CodecOptions)
*/
@Override
public byte[] decompress(final RandomAccessInputStream in, CodecOptions options)
throws FormatException, IOException {
BufferedImage b;
final long fp = in.getFilePointer();
try {
try {
while (in.read() != (byte) 0xff || in.read() != (byte) 0xd8) {
/* Read to data. */
}
in.seek(in.getFilePointer() - 2);
} catch (final EOFException e) {
in.seek(fp);
}
b = ImageIO.read(new BufferedInputStream(new DataInputStream(in), 8192));
} catch (final IOException exc) {
// probably a lossless JPEG; delegate to LosslessJPEGCodec
in.seek(fp);
return new LosslessJPEGCodec().decompress(in, options);
}
if (options == null) options = CodecOptions.getDefaultOptions();
final byte[][] buf = AWTImageTools.getPixelBytes(b, options.littleEndian);
// correct for YCbCr encoding, if necessary
if (options.ycbcr && buf.length == 3) {
final int nBytes = buf[0].length / (b.getWidth() * b.getHeight());
final int mask = (int) (Math.pow(2, nBytes * 8) - 1);
for (int i = 0; i < buf[0].length; i += nBytes) {
final int y = DataTools.bytesToInt(buf[0], i, nBytes, options.littleEndian);
int cb = DataTools.bytesToInt(buf[1], i, nBytes, options.littleEndian);
int cr = DataTools.bytesToInt(buf[2], i, nBytes, options.littleEndian);
cb = Math.max(0, cb - 128);
cr = Math.max(0, cr - 128);
final int red = (int) (y + 1.402 * cr) & mask;
final int green = (int) (y - 0.34414 * cb - 0.71414 * cr) & mask;
final int blue = (int) (y + 1.772 * cb) & mask;
DataTools.unpackBytes(red, buf[0], i, nBytes, options.littleEndian);
DataTools.unpackBytes(green, buf[1], i, nBytes, options.littleEndian);
DataTools.unpackBytes(blue, buf[2], i, nBytes, options.littleEndian);
}
}
byte[] rtn = new byte[buf.length * buf[0].length];
if (buf.length == 1) rtn = buf[0];
else {
if (options.interleaved) {
int next = 0;
for (int i = 0; i < buf[0].length; i++) {
for (int j = 0; j < buf.length; j++) {
rtn[next++] = buf[j][i];
}
}
} else {
for (int i = 0; i < buf.length; i++) {
System.arraycopy(buf[i], 0, rtn, i * buf[0].length, buf[i].length);
}
}
}
return rtn;
}