| GIF_Reader.java |
package ij.plugin;
import ij.*;
import ij.process.*;
import ij.io.*;
import java.net.*;
import java.io.*;
import java.util.*;
import java.awt.*;
import java.awt.image.*;
/** This plugin opens GIFs and Animated GIFs. */
public class GIF_Reader extends ImagePlus implements PlugIn {
public void run(String arg) {
OpenDialog od = new OpenDialog("Open GIF...", arg);
String name = od.getFileName();
if (name==null)
return;
String dir = od.getDirectory();
GifDecoder d = new GifDecoder();
int status = d.read(dir+name);
int n = d.getFrameCount();
ImageStack stack = null;
if (n==1) {
Image img = Toolkit.getDefaultToolkit().createImage(dir+name);
setImage(img);
setTitle(name);
} else {
for (int i=0; i < n; i++) {
ImageProcessor frame = d.getFrame(i);
if (i==0)
stack = new ImageStack(frame.getWidth(), frame.getHeight());
int t = d.getDelay(i); // display duration of frame in milliseconds
stack.addSlice(null, frame);
}
if (stack==null)
return;
setStack(name, stack);
if (getType()==COLOR_RGB)
Opener.convertGrayJpegTo8Bits(this);
}
FileInfo fi = new FileInfo();
fi.fileFormat = fi.GIF_OR_JPG;
fi.fileName = name;
fi.directory = dir;
setFileInfo(fi);
}
}
/**
* Class GifDecoder - Decodes a GIF file into one or more frames.
* No copyright asserted on this code assembly.
*
* @author Kevin Weiner, FM Software; LZW decoder adapted from John Cristy's ImageMagick.
* @version 1.0 January 2001
*
* June 2001: Updated to work with ImageJ and JDK 1.1
*/
class GifDecoder {
/**
* File read status: No errors.
*/
public static final int STATUS_OK = 0;
/**
* File read status: Error decoding file (may be partially decoded)
*/
public static final int STATUS_FORMAT_ERROR = 1;
/**
* File read status: Unable to open source.
*/
public static final int STATUS_OPEN_ERROR = 2;
private BufferedInputStream in;
private int status;
private int width; // full image width
private int height; // full image height
private boolean gctFlag; // global color table used
private int gctSize; // size of global color table
private int loopCount; // iterations; 0 = repeat forever
private int[] gct; // global color table
private int[] lct; // local color table
private int[] act; // active color table
private int bgIndex; // background color index
private int bgColor; // background color
private int lastBgColor; // previous bg color
private int pixelAspect; // pixel aspect ratio
private boolean lctFlag; // local color table flag
private boolean interlace; // interlace flag
private int lctSize; // local color table size
private int ix, iy, iw, ih; // current image rectangle
private Rectangle lastRect; // last image rect
private ImageProcessor image; // current frame
private ImageProcessor lastImage; // previous frame
private byte[] block = new byte[256]; // current data block
private int blockSize = 0; // block size
// last graphic control extension info
private int dispose = 0; // 0=no action; 1=leave in place; 2=restore to bg; 3=restore to prev
private int lastDispose = 0;
private boolean transparency = false; // use transparent color
private int delay = 0; // delay in milliseconds
private int transIndex; // transparent color index
private static final int MaxStackSize = 4096; // max decoder pixel stack size
// LZW decoder working arrays
private short[] prefix;
private byte[] suffix;
private byte[] pixelStack;
private byte[] pixels;
private Vector frames; // frames read from current file
private int frameCount;
/**
* Gets display duration for specified frame.
*
* @param n int index of frame
* @return delay in milliseconds
*/
public int getDelay(int n) {
delay = -1;
if ((n >= 0) && (n < frameCount))
delay = ((GifFrame) frames.elementAt(n)).delay;
return delay;
}
/**
* Gets the image contents of frame n.
*
* @return ImageProcessor representation of frame, or null if n is invalid.
*/
public ImageProcessor getFrame(int n) {
ImageProcessor im = null;
if ((n >= 0) && (n < frameCount))
im = ((GifFrame) frames.elementAt(n)).image;
return im;
}
/**
* Gets the number of frames read from file.
* @return int frame count
*/
public int getFrameCount() {
return frameCount;
}
/**
* Gets the first (or only) image read.
*
* @return ImageProcessor containing first frame, or null if none.
*/
public ImageProcessor getImage() {
return getFrame(0);
}
/**
* Gets the "Netscape" iteration count, if any.
* A count of 0 means repeat indefinitiely.
*
* @return int iteration count if one was specified, else 1.
*/
public int getLoopCount() {
return loopCount;
}
/**
* Reads GIF image from stream
*
* @param BufferedInputStream containing GIF file.
* @return int read status code
*/
public int read(BufferedInputStream is) {
init();
if (is != null) {
in = is;
readHeader();
if (!err()) {
readContents();
if (frameCount < 0)
status = STATUS_FORMAT_ERROR;
}
} else {
status = STATUS_OPEN_ERROR;
}
try {
is.close();
} catch (IOException e) {}
return status;
}
/**
* Reads GIF file from specified source (file or URL string)
*
* @param name File source string
* @return int read status code
*/
public int read(String name) {
status = STATUS_OK;
try {
name = name.trim();
if (name.indexOf("://") > 0) {
URL url = new URL(name);
in = new BufferedInputStream(url.openStream());
} else {
in = new BufferedInputStream(new FileInputStream(name));
}
status = read(in);
} catch (IOException e) {
status = STATUS_OPEN_ERROR;
}
return status;
}
/**
* Decodes LZW image data into pixel array.
* Adapted from John Cristy's ImageMagick.
*/
private void decodeImageData() {
int NullCode = -1;
int npix = iw * ih;
int available, clear, code_mask, code_size, end_of_information, in_code,
old_code, bits, code, count, i, datum, data_size, first, top, bi, pi;
if ((pixels == null) || (pixels.length < npix))
pixels = new byte[npix]; // allocate new pixel array
if (prefix == null)
prefix = new short[MaxStackSize];
if (suffix == null)
suffix = new byte[MaxStackSize];
if (pixelStack == null)
pixelStack = new byte[MaxStackSize+1];
// Initialize GIF data stream decoder.
data_size = read();
clear = 1 << data_size;
end_of_information = clear + 1;
available = clear + 2;
old_code = NullCode;
code_size = data_size + 1;
code_mask = (1 << code_size) - 1;
for (code = 0; code < clear; code++) {
prefix[code] = 0;
suffix[code] = (byte) code;
}
// Decode GIF pixel stream.
datum = bits = count = first = top = pi = bi = 0;
for (i = 0; i < npix; ) {
if (top == 0) {
if (bits < code_size) {
// Load bytes until there are enough bits for a code.
if (count == 0) {
// Read a new data block.
count = readBlock();
if (count <= 0)
break;
bi = 0;
}
datum += (((int) block[bi]) & 0xff) << bits;
bits += 8;
bi++;
count--;
continue;
}
// Get the next code.
code = datum & code_mask;
datum >>= code_size;
bits -= code_size;
// Interpret the code
if ((code > available) || (code == end_of_information))
break;
if (code == clear) {
// Reset decoder.
code_size = data_size + 1;
code_mask = (1 << code_size) - 1;
available = clear + 2;
old_code = NullCode;
continue;
}
if (old_code == NullCode) {
pixelStack[top++] = suffix[code];
old_code = code;
first = code;
continue;
}
in_code = code;
if (code == available) {
pixelStack[top++] = (byte) first;
code = old_code;
}
while (code > clear) {
pixelStack[top++] = suffix[code];
code = prefix[code];
}
first = ((int) suffix[code]) & 0xff;
// Add a new string to the string table,
if (available >= MaxStackSize)
break;
pixelStack[top++] = (byte) first;
prefix[available] = (short) old_code;
suffix[available] = (byte) first;
available++;
if (((available & code_mask) == 0) && (available < MaxStackSize)) {
code_size++;
code_mask += available;
}
old_code = in_code;
}
// Pop a pixel off the pixel stack.
top--;
pixels[pi++] = pixelStack[top];
i++;
}
for (i = pi; i < npix; i++)
pixels[i] = 0; // clear missing pixels
}
/**
* Returns true if an error was encountered during reading/decoding
*/
private boolean err() {
return status != STATUS_OK;
}
/**
* Initializes or re-initializes reader
*/
private void init() {
status = STATUS_OK;
frameCount = 0;
frames = new Vector();
gct = null;
lct = null;
}
/**
* Reads a single byte from the input stream.
*/
private int read() {
int curByte = 0;
try {
curByte = in.read();
} catch (IOException e) {
status = STATUS_FORMAT_ERROR;
}
return curByte;
}
/**
* Reads next variable length block from input.
*
* @return int number of bytes stored in "buffer"
*/
private int readBlock() {
blockSize = read();
int n = 0;
int count;
if (blockSize > 0) {
try {
while (n<blockSize) {
count = in.read(block, n, blockSize-n);
if (count==-1)
break;
n += count;
}
} catch (IOException e) {}
if (n < blockSize)
status = STATUS_FORMAT_ERROR;
}
return n;
}
/**
* Reads color table as 256 RGB integer values
*
* @param ncolors int number of colors to read
* @return int array containing 3*ncolors color components
*/
private int[] readColorTable(int ncolors) {
int nbytes = 3*ncolors;
int[] tab = null;
byte[] c = new byte[nbytes];
int n = 0;
try {
n = in.read(c);
} catch (IOException e) {}
if (n < nbytes)
status = STATUS_FORMAT_ERROR;
else {
tab = new int[256]; // max size to avoid bounds checks
int i = 0;
int j = 0;
while (i < ncolors) {
int r = ((int) c[j++]) & 0xff;
int g = ((int) c[j++]) & 0xff;
int b = ((int) c[j++]) & 0xff;
tab[i++] = 0xff000000 | (r << 16) | (g << 8) | b;
}
}
return tab;
}
/**
* Main file parser. Reads GIF content blocks.
*/
private void readContents() {
// read GIF file content blocks
boolean done = false;
while (!(done || err())) {
int code = read();
switch (code) {
case 0x2C: // image separator
readImage();
break;
case 0x21: // extension
code = read();
switch (code) {
case 0xf9: // graphics control extension
readGraphicControlExt();
break;
case 0xff: // application extension
readBlock();
String app = "";
for (int i = 0; i < 11; i++)
app += (char) block[i];
if (app.equals("NETSCAPE2.0"))
readNetscapeExt();
else
skip(); // don't care
break;
default: // uninteresting extension
skip();
}
break;
case 0x3b: // terminator
done = true;
break;
default:
status = STATUS_FORMAT_ERROR;
}
}
}
/**
* Reads Graphics Control Extension values
*/
private void readGraphicControlExt() {
read(); // block size
int packed = read(); // packed fields
dispose = (packed & 0x1c) >> 1;
transparency = (packed & 1) != 0;
delay = readShort() * 10; // delay in milliseconds
transIndex = read(); // transparent color index
read(); // block terminator
}
/**
* Reads GIF file header information.
*/
private void readHeader() {
String id = "";
for (int i = 0; i < 6; i++)
id += (char) read();
if (!id.startsWith("GIF")) {
status = STATUS_FORMAT_ERROR;
return;
}
readLSD();
if (gctFlag && !err()) {
gct = readColorTable(gctSize);
bgColor = gct[bgIndex];
}
}
/**
* Reads next frame image
*/
private void readImage() {
ix = readShort(); // (sub)image position & size
iy = readShort();
iw = readShort();
ih = readShort();
int packed = read();
lctFlag = (packed & 0x80) != 0; // 1 - local color table flag
interlace = (packed & 0x40) != 0; // 2 - interlace flag
// 3 - sort flag
// 4-5 - reserved
lctSize = 2 << (packed & 7); // 6-8 - local color table size
if (lctFlag) {
lct = readColorTable(lctSize); // read table
act = lct; // make local table active
} else {
act = gct; // make global table active
if (bgIndex == transIndex)
bgColor = 0;
}
int save = 0;
if (transparency) {
save = act[transIndex];
act[transIndex] = 0; // set transparent color if specified
}
if (act == null) {
status = STATUS_FORMAT_ERROR; // no color table defined
}
if (err()) return;
decodeImageData(); // decode pixel data
skip();
if (err()) return;
frameCount++;
// create new image to receive frame data
image = new ColorProcessor(width, height);
setPixels(); // transfer pixel data to image
frames.addElement(new GifFrame(image, delay)); // add image to frame list
if (transparency)
act[transIndex] = save;
resetFrame();
}
/**
* Reads Logical Screen Descriptor
*/
private void readLSD() {
// logical screen size
width = readShort();
height = readShort();
// packed fields
int packed = read();
gctFlag = (packed & 0x80) != 0; // 1 : global color table flag
// 2-4 : color resolution
// 5 : gct sort flag
gctSize = 2 << (packed & 7); // 6-8 : gct size
bgIndex = read(); // background color index
pixelAspect = read(); // pixel aspect ratio
}
/**
* Reads Netscape extenstion to obtain iteration count
*/
private void readNetscapeExt() {
do {
readBlock();
if (block[0] == 0x03) {
// loop count sub-block
int b1 = ((int) block[1]) & 0xff;
int b2 = ((int) block[2]) & 0xff;
loopCount = (b2 << 8) | b1;
}
} while ((blockSize > 0) && !err());
}
/**
* Reads next 16-bit value, LSB first
*/
private int readShort() {
// read 16-bit value, LSB first
return read() | (read() << 8);
}
/**
* Resets frame state for reading next image.
*/
private void resetFrame() {
lastDispose = dispose;
lastRect = new Rectangle(ix, iy, iw, ih);
lastImage = image;
lastBgColor = bgColor;
int dispose = 0;
boolean transparency = false;
int delay = 0;
lct = null;
}
/**
* Creates new frame image from current data (and previous
* frames as specified by their disposition codes).
*/
private void setPixels() {
// expose destination image's pixels as int array
int[] dest = (int[]) image.getPixels();
// fill in starting image contents based on last image's dispose code (if any)
if (lastDispose > 0) {
if (lastDispose == 3) { // use image before last
int n = frameCount - 2;
if (n > 0)
lastImage = getFrame(n-1);
else
lastImage = null;
}
if (lastImage != null) {
int[] prev = (int[]) lastImage.getPixels();
System.arraycopy(prev, 0, dest, 0, width*height); // copy pixels
if ((lastDispose == 2) && (lastBgColor != 0)) {
// fill last image rect area with background color
image.setColor(new Color(lastBgColor));
image.setRoi(lastRect);
image.fill();
}
}
}
// copy each source line to the appropriate place in the destination
int pass = 1;
int inc = 8;
int iline = 0;
for (int i = 0; i < ih; i++) {
int line = i;
if (interlace) {
if (iline >= ih) {
pass++;
switch (pass) {
case 2:
iline = 4;
break;
case 3:
iline = 2;
inc = 4;
break;
case 4:
iline = 1;
inc = 2;
}
}
line = iline;
iline += inc;
}
line += iy;
if (line < height) {
int k = line * width;
int dx = k + ix; // start of line in dest
int dlim = dx + iw; // end of dest line
if ((k + width) < dlim)
dlim = k + width; // past dest edge
int sx = i * iw; // start of line in source
while (dx < dlim) {
// map color and insert in destination
int index = ((int) pixels[sx++]) & 0xff;
dest[dx++] = act[index];
}
}
}
}
/**
* Skips variable length blocks up to and including
* next zero length block.
*/
private void skip() {
do {
readBlock();
} while ((blockSize > 0) && !err());
}
}
class GifFrame {
public GifFrame(ImageProcessor im, int del) {
image = im;
delay = del;
}
public ImageProcessor image;
public int delay;
}