/*
   rdesktop: A Remote Desktop Protocol client.
   Bitmap decompression routines
   Copyright (C) Matthew Chapman 1999-2000
   
   This program is free software; you can redistribute it and/or modify
   it under the terms of the GNU General Public License as published by
   the Free Software Foundation; either version 2 of the License, or
   (at your option) any later version.
   
   This program is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   GNU General Public License for more details.
   
   You should have received a copy of the GNU General Public License
   along with this program; if not, write to the Free Software
   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
*/

#include "rdesktop.h"

#define CVAL(p)   (*(p++))
#define SVAL(p)   ((*((p++) + 1) << 8) | CVAL(p))

#define REPEAT(statement) { while ((count > 0) && (x < width)) { statement; count--; x++; } }
#define MASK_UPDATE() { mixmask <<= 1; if (mixmask == 0) { mask = CVAL(input); mixmask = 1; } }

BOOL bitmap_decompress(unsigned char *output, int width, int height,
		       unsigned char *input, int size)
{
	unsigned char *end = input + size;
	unsigned char *prevline = NULL, *line = NULL;
	int opcode, count, offset, isfillormix, x = width;
	int lastopcode = -1, insertmix = False;
	uint8 code, colour1 = 0, colour2 = 0;
	uint8 mixmask, mask = 0, mix = 0xff;

	while (input < end)
	{
		code = CVAL(input);
		opcode = code >> 4;

		/* Handle different opcode forms */
		switch (opcode)
		{
			case 0xc:
			case 0xd:
			case 0xe:
				opcode -= 6;
				count = code & 0xf;
				offset = 16;
				break;

			case 0xf:
				opcode = code & 0xf;
				count = (opcode < 13) ? SVAL(input) : 1;
				offset = 0;
				break;

			default:
				opcode >>= 1;
				count = code & 0x1f;
				offset = 32;
				break;
		}

		/* Handle strange cases for counts */
		if (offset != 0)
		{
			isfillormix = ((opcode == 2) || (opcode == 7));

			if (count == 0)
			{
				if (isfillormix)
					count = CVAL(input) + 1;
				else
					count = CVAL(input) + offset;
			}
			else if (isfillormix)
			{
				count <<= 3;
			}
		}

		/* Read preliminary data */
		mixmask = 0;
		switch (opcode)
		{
			case 0: /* Fill */
				if ((lastopcode == opcode) && (x != width))
					insertmix = True;
				break;
			case 8: /* Bicolour */
				colour1 = CVAL(input);
			case 3: /* Colour */
				colour2 = CVAL(input);
				break;
			case 6: /* SetMix/Mix */
			case 7: /* SetMix/FillOrMix */
				mix = CVAL(input);
				opcode -= 5;
				break;
		}
		lastopcode = opcode;

		/* Output body */
		while (count > 0)
		{
			if (x >= width)
			{
				if (height <= 0)
					return False;

				x = 0;
				height--;

				prevline = line;
				line = output + height * width;
			}

			switch (opcode)
			{
				case 0: /* Fill */
					if (insertmix)
					{
						if (prevline == NULL)
							line[x] = mix;
						else
							line[x] = prevline[x] ^ mix;

						insertmix = False;
						count--;
						x++;
					}

					if (prevline == NULL)
						REPEAT(line[x] = 0)
					else
						REPEAT(line[x] = prevline[x])
					break;

				case 1: /* Mix */
					if (prevline == NULL)
						REPEAT(line[x] = mix)
					else
						REPEAT(line[x] = prevline[x] ^ mix)
					break;

				case 2: /* Fill or Mix */
					if (prevline == NULL)
					    REPEAT( 
						   MASK_UPDATE();

						   if (mask & mixmask)
							line[x] = mix;
						   else
							line[x] = 0;
					    )
					else
					    REPEAT(
						   MASK_UPDATE();

						   if (mask & mixmask)
							line[x] = prevline[x] ^ mix;
						   else
							line[x] = prevline[x];
					    )
					break;

				case 3: /* Colour */
					REPEAT(line[x] = colour2)
					break;

				case 4: /* Copy */
					REPEAT(line[x] = CVAL(input))
					break;

				case 8: /* Bicolour */
					REPEAT(line[x] = colour1; line[++x] = colour2)
					break;

				case 13: /* White */
					REPEAT(line[x] = 0xff)
					break;

				case 14: /* Black */
					REPEAT(line[x] = 0x00)
					break;

				default:
					NOTIMP("bitmap opcode 0x%x\n", opcode);
					return False;
			}
		}
	}

	return True;
}