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;
	}