#include "qdata.h"
#include "flex.h"

#define		MAXFILES	2048

typedef	struct
{
	int			x;
	int			y;
	int			w;
	int			h;
	int			cw;
	int			ch;
	int			rw;
	int			index;
	int			depth;
	int			col;
	int			baseline;
	char		name[128];
} Coords;

int				filenum;
int				valid;
Coords 			in[MAXFILES];
Coords			out;
char		  	outscript[256];
char		  	sourcedir[256];
char		  	outusage[256];
char		  	root[32];

int				destsize = 0;
byte		  	*pixels = NULL;				// Buffer to load image
long		  	*outpixels = NULL;			// Buffer to store combined textures
long		  	*usagemap = NULL;			// Buffer of usage map
void		  	*bmptemp = NULL;			// Buffer of usage map
byte		  	*map = NULL;

int				xcharsize;
int				ycharsize;
int				dosort = 0;
int				missed = 0;
int				overlap = 0;
int				nobaseline = 0;
int				percent;

//////////////////////////////////////////////////
// Setting the char based usage map				//
//////////////////////////////////////////////////

byte	TryPlace(Coords *coord)
{
	int		x, y;
	byte	entry = 0;
	byte	*mapitem;
	
	mapitem = map + (coord->x / xcharsize) + ((coord->y / ycharsize) * out.cw);

	for (y = 0; y < coord->ch; y++, mapitem += out.cw - coord->cw)
	{
		for (x = 0; x < coord->cw; x++)
		{
			if (entry |= *mapitem++ & 8)
			{
				return(entry);
			}
		}
	}
	return(entry);
}

void	SetMap(Coords *coord)
{
	int		x, y;
	byte	*mapitem;

	mapitem = map + (coord->x / xcharsize) + ((coord->y / ycharsize) * out.cw);

	for (y = 0; y < coord->ch; y++, mapitem += out.cw - coord->cw)
		for (x = 0; x < coord->cw; x++)
			*mapitem++ |= 8;
}

//////////////////////////////////////////////////
// Setting the pixel based usage map			//
//////////////////////////////////////////////////

void	CheckOverlap(Coords *coord)
{
	int			x;
	int			y;
	long		*dest;

	x = coord->x;
	y = coord->y;

	dest = (long *)(usagemap + x + (y * out.w));

	for (y = 0; y < coord->h; y++, dest += out.w - coord->w)
	{
		for (x = 0; x < coord->w; x++)
		{
			if (*dest++)
			{
				overlap++;
				return;
			}
		}
	}
}

void	SetUsageMap(Coords *coord)
{
	int			x;
	int			y;
	long		*dest;

	x = coord->x;
	y = coord->y;

	dest = (long *)(usagemap + x + (y * out.w));

	for (y = 0; y < coord->h; y++, dest += out.w - coord->w)
	{
		for (x = 0; x < coord->w; x++)
		{
			*dest++ = coord->col;
		}
	}
}

//////////////////////////////////////////////////
// Flips the BMP image to the correct way up	//
//////////////////////////////////////////////////

void	CopyLine(byte *dest, byte *src, int size)
{
	int		x;
	
	for (x = 0; x < size; x++)
		*dest++ = *src++;
}

/****************************************************/
/* Printing headers etc								*/
/****************************************************/

void RemoveLeading(char *name)
{
	int		i;
	char	temp[128];

	for(i = strlen(name) - 1; i > 0; i--)
	{
		if((name[i] == '\\') || (name[i] == '/'))
		{
			strcpy(temp, name + i + 1);
			strcpy(name, temp);
			return;
		}
	}
}

void RemoveExt(char *name)
{
	while ((*name != '.') && *name)
		name++;
	*name = 0;
}

/****************************************************/
/* Misc calcualtions								*/
/****************************************************/

int	TotalArea()
{
	int		i;
	int		total = 0;

	for (i = 0; i < (filenum + 2); i++)
		total += in[i].w * in[i].h;

	return(total);
}

/****************************************************/
/* Setup and checking of all info					*/
/****************************************************/

void	InitVars()
{
	filenum = 0;
	valid = 0;
   	dosort = 0;
   	missed = 0;
   	overlap = 0;
   	nobaseline = 0;
	
	memset(outscript, 0, sizeof(outscript));
	memset(outscript, 0, sizeof(sourcedir));
	memset(outscript, 0, sizeof(outusage));
	memset(outscript, 0, sizeof(root));

	memset(in, 0, sizeof(in));
	memset(&out, 0, sizeof(out));
}
void Cleanup()
{
	if (pixels)
		free(pixels);
	if (usagemap)
		free(usagemap);
	if (outpixels)
		free(outpixels);
	if (bmptemp)
		free(bmptemp);
	if (map)
		free(map);
}

typedef struct glxy_s
{
	float	xl, yt, xr, yb;
	int		w, h, baseline;
} glxy_t;

int	SaveScript(char *name)
{
	FILE		*fp;
	int			i, j;
	glxy_t		buff;
	
	if(fp = fopen(name, "wb"))
	{
		for (j = 0; j < filenum; j++)
		{
			for (i = 0; i < filenum; i++)
			{
				if (in[i].index == j)
				{
					if (in[i].depth)
					{
						buff.xl = (float)in[i].x / (float)out.w;
						buff.yt = (float)in[i].y / (float)out.h;
						buff.xr = ((float)in[i].w + (float)in[i].x) / (float)out.w;
						buff.yb = ((float)in[i].h + (float)in[i].y) / (float)out.h;
						buff.w = in[i].w;
						buff.h = in[i].h;
						buff.baseline = in[i].baseline;
					}
					else
					{
						memset(&buff, 0, sizeof(glxy_t));
					}
					fwrite(&buff, 1, sizeof(glxy_t), fp);
					i = filenum;
				}
			}
		}
		fclose(fp);
		return(true);
	}
	else
		return(false);
}

int		GetScriptInfo(char *name)
{
	FILE		*fp;
	char		buffer[256];
	char		tempbuff[256];
	char		delims[] = {" \t,\n"};

	printf("Opening script file %s.\n", name);

	if (fp = fopen(name, "r"))
	{
		while(fgets(buffer, 256, fp))
		{
			if (strncmp(buffer, "//", 2) && strncmp(buffer, "\n", 1))
			{
				strupr(buffer);
				strcpy(tempbuff, buffer);
				if (strcmp(strtok(tempbuff, delims), "OUTPUT") == 0)
				{
					strcpy(out.name, strtok(NULL, delims));
					strlwr(out.name);
				}

				strcpy(tempbuff, buffer);
				if (strcmp(strtok(tempbuff, delims), "SOURCEDIR") == 0)
				{
					strcpy(tempbuff, strtok(NULL, delims));
					strcpy(sourcedir, ExpandPathAndArchive(tempbuff));
				}

				strcpy(tempbuff, buffer);
				if (strcmp(strtok(tempbuff, delims), "DOSORT") == 0)
					dosort = 1;

				strcpy(tempbuff, buffer);
				if (strcmp(strtok(tempbuff, delims), "XCHARSIZE") == 0)
					xcharsize = strtol(strtok(NULL, delims), NULL, 0);

				strcpy(tempbuff, buffer);
				if (strcmp(strtok(tempbuff, delims), "YCHARSIZE") == 0)
					ycharsize = strtol(strtok(NULL, delims), NULL, 0);

				strcpy(tempbuff, buffer);
				if (strcmp(strtok(tempbuff, delims), "OUTSCRIPT") == 0)
				{
					strcpy(outscript, strtok(NULL, delims));
					strlwr(outscript);
				}

				strcpy(tempbuff, buffer);
				if (strcmp(strtok(tempbuff, delims), "OUTUSAGE") == 0)
					strcpy(outusage, strtok(NULL, delims));

				strcpy(tempbuff, buffer);
				if (strcmp(strtok(tempbuff, delims), "POS") == 0)
				{
					out.w = strtol(strtok(NULL, delims), NULL, 0);
			 		out.h = strtol(strtok(NULL, delims), NULL, 0);
				}

				strcpy(tempbuff, buffer);
				if (strcmp(strtok(tempbuff, delims), "FILE") == 0)
				{
					strcpy(in[filenum].name, strtok(NULL, delims));
					in[filenum].x = strtol(strtok(NULL, delims), NULL, 0);
					in[filenum].y = strtol(strtok(NULL, delims), NULL, 0);
					in[filenum].col = strtol(strtok(NULL, delims), NULL, 0);
					filenum++;
				}
			}
		}
		fclose(fp);
		return(true);
	}
	else
	{
		printf("ERROR : Could not open script file.\n");
		return(false);
	}
}

int	CheckVars()
{
	int		i;

	 if (out.name[0] == 0)
	 {
	 	printf("ERROR : No output name specified.\n");
		return(false);
	}
	if ((out.w <= 0) || (out.h <= 0))
	{
		printf("ERROR : Invalid VRAM coordinates.\n");
		return(false);
	}
	if (filenum == 0)
	{
		printf("ERROR : No input files specified.\n");
		return(false);
	}
	for (i = 0; i < filenum; i++)
		if (in[i].name[0] == 0)
		{
			printf("ERROR : Input filename invalid.\n");
			return(false);
		}
	return(true);
}

// Makes sure texture is totally within the output area

int	CheckCoords(Coords *coord)
{
	if ((coord->x + coord->w) > out.w)
		return(false);
	if ((coord->y + coord->h) > out.h)
		return(false);

	return(true);
}
// Gets the width, height, palette width and palette height of each BMP file

int		GetFileDimensions()
{
	int			i;
	int			width, height;
	char		name[128];

	for (i = 0; i < filenum; i++)
	{
		in[i].index = i;

		strcpy(name, sourcedir);
		strcat(name, in[i].name);
		printf("Getting file dimensions, file : %s        \r", in[i].name);
		if(FileExists(name))
		{
			LoadAnyImage(name, NULL, NULL, &width, &height);
			in[i].depth = 32;
			in[i].rw = width;
			in[i].w = width;					 	// makes it width in 
			in[i].h = height;
			in[i].cw = (in[i].w + (xcharsize - 1)) / xcharsize;
			in[i].ch = (in[i].h + (ycharsize - 1)) / ycharsize;

			if (!CheckCoords(&in[i]) && (in[i].x >= 0))
			{
			 	printf("Error : texture %s out of bounds.\n", in[i].name);
				return(false);
			}
			valid++;
		}
		else
		{
			in[i].depth = 0;
			in[i].x = -1;
			in[i].y = -1;
			in[i].w = 0;
			in[i].h = 0;
		}
	}
	printf("\n\n");
	return(true);
}

// Sorts files into order for optimal space finding
// Fixed position ones first, followed by the others in descending size
// The theory being that it is easier to find space for smaller textures.
// size = (width + height)
// For space finding it is easier to place a 32x32 than a 128x2

#define	WEIGHT	0x8000

void	Swap(Coords *a, Coords *b)
{
	Coords		c;

	c = *a;
	*a = *b;
	*b = c;
}

void	SortInNames()
{
	int		i, j;
	int		largest, largcount;
	int		size;

	printf("Sorting filenames by size.\n\n");

	for (j = 0; j < filenum; j++)
	{
		largest = -1;
		largcount = -1;

		for (i = j; i < filenum; i++)
		{
			if (in[i].depth)
			{
				size = in[i].w + in[i].h;
		
				if ((in[i].x < 0) && (size > largest))
				{
					largcount = i;
					largest = size;
				}
			}
		}
		if ((largcount >= 0) && (largcount != j))
			Swap(&in[j], &in[largcount]);
	}
}

int	SetVars(char *name)
{
	if (!GetScriptInfo(name))
		return(false);

	if (!CheckVars())
		return(false);

	destsize = out.w * out.h;

	out.cw = out.w / xcharsize;
	out.ch = out.h / ycharsize;

	if ((usagemap = (long *)SafeMalloc(destsize * 4, "")) == NULL)
		return(false);
	if ((outpixels = (long *)SafeMalloc(destsize * 4, "")) == NULL)
		return(false); 
	if ((bmptemp = (void *)SafeMalloc(destsize * 4, "")) == NULL)
		return(false);
	if ((map = (byte *)SafeMalloc(destsize / (xcharsize * ycharsize), "")) == NULL)
		return(false);

	if (GetFileDimensions() == false)
		return(false);

	if (dosort)
		SortInNames();

	return(true);
}
/****************************************************/
/* Actual copying routines							*/
/****************************************************/

int FindCoords(Coords *coord)
{
	int		tx, ty;

	if (coord->x >= 0)
	{	
		SetMap(coord);
		return(true);
	}
	else
	{
		for (ty = 0; ty < out.ch; ty++)
		{
			for (tx = 0; tx < out.cw; tx++)
			{
				coord->x = (tx * xcharsize);
				coord->y = (ty * ycharsize);
	
				if (CheckCoords(coord) && !TryPlace(coord))
				{
					SetMap(coord);
					return(true);
				}
			}
		}
	}
	coord->x = -1;
	coord->y = -1;

	return(false);
}

void CheckBaseline(int i)
{
	int		y;
	long	*pix;

	in[i].baseline = -1;
	pix = (long *)pixels;

	for(y = 0; y < in[i].h; y++, pix += in[i].w)
	{
		if((*pix & 0x00ffffff) == 0x00ff00ff)
		{
			in[i].baseline = y;
			break;
		}
	}
	pix = (long *)pixels;
	for(y = 0; y < in[i].w * in[i].h; y++, pix++)
	{
		if((*pix & 0x00ffffff) == 0x00ff00ff)
		{
			*pix = 0;
		}
	}

	if(in[i].baseline == -1)
	{
		printf("\nERROR : %s has no baseline\n", in[i].name);
		nobaseline++;
	}
}

void	CopyToMain32(Coords *coord)
{
	int			x;
	int			y;
	long		*source;
	long		*dest;

	x = coord->x;
	y = coord->y;

	source = (long *)pixels;
	dest = (long *)(outpixels + x + (y * out.w));

	for (y = 0; y < coord->h; y++, dest += out.w - coord->w)
	{
		for (x = 0; x < coord->w; x++)
		{
			*dest++ = *source++;
		}
	}
}

void CreateMain()
{
	int			i, count;
	int			width, height;
	char		name[128];
	
	for (i = 0, count = 0; i < filenum; i++)
	{
		if (in[i].depth)
		{
			printf("\rProcessing %d of %d (%d missed, %d overlapping, %d nobase)\r", count + 1, valid, missed, overlap, nobaseline);
			count++;
			if (!FindCoords(&in[i]))
				missed++;
			else
			{
				strcpy(name, sourcedir);
				strcat(name, in[i].name);
				LoadAnyImage(name, &pixels, NULL, &width, &height);
				CheckBaseline(i);
				CheckOverlap(&in[i]);
				CopyToMain32(&in[i]);
				SetUsageMap(&in[i]);
			}
		}
	}
}

void Cmd_TextureMix()
{
	miptex32_t		*qtex32;
	char			filename[1024];
	int				size;

	InitVars();

	GetScriptToken (false);

	strcpy(root, token);
	RemoveExt(root);
	RemoveLeading(root);

	strcpy(filename, ExpandPathAndArchive(token));
	if (SetVars(filename))
	{
		// Create combined texture
		percent = ((TotalArea() * 100) / (out.w * out.h));
		printf("Total area consumed : %d%%\n", percent);
		printf("Texture resolution  : %dx%d pixels.\n", xcharsize, ycharsize);
		CreateMain();

		// Save image as m32
		sprintf (filename, "%spics/misc/%s.m32", gamedir, out.name);
		qtex32 = CreateMip32((unsigned long *)outpixels, out.w, out.h, &size, false);

		qtex32->contents = 0;
		qtex32->value = 0;
		qtex32->scale_x = 1.0;
		qtex32->scale_y = 1.0;
		sprintf (qtex32->name, "misc/%s", out.name);

		printf ("\n\nwriting %s\n", filename);
		SaveFile (filename, (byte *)qtex32, size);
		free (qtex32);

		// Save out script file
		sprintf (filename, "%spics/misc/%s.fnt", gamedir, outscript);
		printf("Writing %s as script file\n", filename);
		if (!SaveScript(filename))
		{
			printf("Unable to save output script.\n");
		}
	}
	printf("Everythings groovy.\n");
	Cleanup();
}

// end