/*
Copyright (C) 1996-1997 Id Software, Inc.

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., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.

	$Id: wad.c,v 1.20 2007-10-04 14:56:54 dkure Exp $
*/
// wad.c

#include "quakedef.h"
#include "gl_model.h"
#include "gl_local.h"
#include "wad.h"
#include "crc.h"


static int			wad_numlumps;
static int			wad_filesize;
static lumpinfo_t	*wad_lumps;
static byte			*wad_base = NULL;


static void W_InsertOcranaLeds (byte *data);

void SwapPic (qpic_t *pic) {
	pic->width = LittleLong(pic->width);
	pic->height = LittleLong(pic->height);	
}

/*
Lowercases name and pads with spaces and a terminating 0 to the length of lumpinfo_t->name.
Used so lumpname lookups can proceed rapidly by comparing 4 chars at a time
Space padding is so names can be printed nicely in tables. Can safely be performed in place.
*/
void W_CleanupName (char *in, char *out) {
	int i, c;

	for (i = 0; i < 16; i++) {
		c = in[i];
		if (!c)
			break;

		if (c >= 'A' && c <= 'Z')
			c += ('a' - 'A');
		out[i] = c;
	}

	for ( ; i < 16; i++)
		out[i] = 0;
}

void W_FreeWadFile (void)
{
	Q_free (wad_base);
	wad_base  = NULL;
	wad_lumps = NULL;
	wad_numlumps = 0;
	wad_filesize = 0;
}

void W_LoadWadFile (char *filename) {
	lumpinfo_t *lump_p;
	wadinfo_t *header;
	unsigned i;
	int infotableofs;

	// only one .wad can be loaded at a time
	W_FreeWadFile ();

	wad_base = FS_LoadHeapFile (filename, &wad_filesize);

	if (!wad_base) {
		if (!strcmp(filename, "gfx.wad"))
			Sys_Error("Couldn't load gfx.wad.\n"
					  "This usually happens when you don't have original Quake 1 files in id1 subdirectory.\n"
					  "Ensure you have pak0.pak and pak1.pak in subdirectory id1.");
		else
			Sys_Error ("W_LoadWadFile: couldn't load %s", filename);
	}
	header = (wadinfo_t *)wad_base;

	if (memcmp(header->identification, "WAD2", 4))
		Sys_Error ("Wad file %s doesn't have WAD2 id\n",filename);

	wad_numlumps = LittleLong(header->numlumps);
	infotableofs = LittleLong(header->infotableofs);
	wad_lumps = (lumpinfo_t *)(wad_base + infotableofs);

	if (infotableofs + wad_numlumps * sizeof(lump_t) > wad_filesize)
		Sys_Error ("Wad lump table exceeds file size");
	
	for (i = 0, lump_p = wad_lumps; i < wad_numlumps; i++,lump_p++) {
		lump_p->filepos = LittleLong(lump_p->filepos);
		lump_p->size = LittleLong(lump_p->size);

		W_CleanupName (lump_p->name, lump_p->name);

		if (lump_p->filepos < sizeof(wadinfo_t) || lump_p->filepos + LittleLong(lump_p->disksize) > wad_filesize)
			Sys_Error ("Wad lump %s exceeds file size", lump_p->name);

		if (lump_p->type == TYP_QPIC)
			SwapPic ( (qpic_t *)(wad_base + lump_p->filepos));
	}
}

lumpinfo_t *W_GetLumpinfo (char *name) {
	int i;
	lumpinfo_t	*lump_p;
	char clean[16];

	W_CleanupName (name, clean);
	for (lump_p = wad_lumps, i = 0; i < wad_numlumps; i++,lump_p++) {
		if (!strcmp(clean, lump_p->name))
			return lump_p;
	}

	Sys_Error ("W_GetLumpinfo: %s not found", name);
	return NULL;
}

void *W_GetLumpName (char *name) {
	lumpinfo_t *lump;

	lump = W_GetLumpinfo (name);

    // Make sure we have a lump. 
	if(!lump)
      return NULL;

	// If we got the conchars lump, check if the CRC is the same
	// as in the original gfx.wad, and if so, insert leds into it.
	if (!strcmp(name, "conchars")) 
	{
		if (lump->filepos + lump->size < wad_filesize
			&& CRC_Block (wad_base + lump->filepos, lump->size) == 798)
		{
			W_InsertOcranaLeds (wad_base + lump->filepos); 
		}
	}

	return (void *) (wad_base + lump->filepos);
}

void *W_GetLumpNum (int num) {
	lumpinfo_t *lump;
	
	if (num < 0 || num > wad_numlumps)
		Sys_Error ("W_GetLumpNum: bad number: %i", num);

	lump = wad_lumps + num;

	return (void *) (wad_base + lump->filepos);
}

static byte ocrana_leds[4][8][8] = {
    {
	// green
	{0x00,0x38,0x3b,0x3b,0x3b,0x3b,0x35,0x00},
	{0x38,0x3b,0x3d,0x3f,0x3f,0x3d,0x38,0x35},
	{0x3b,0x3d,0xfe,0x3f,0x3f,0x3f,0x3b,0x35},
	{0x3b,0x3f,0x3f,0x3f,0x3f,0x3f,0x3b,0x35},
	{0x3b,0x3f,0x3f,0x3f,0x3f,0x3d,0x3b,0x35},
	{0x3b,0x3d,0x3f,0x3f,0x3d,0x3b,0x38,0x35},
	{0x35,0x38,0x3b,0x3b,0x3b,0x38,0x35,0x35},
	{0x00,0x35,0x35,0x35,0x35,0x35,0x35,0x00}
    },
    {
	// red
	{0x00,0xf8,0xf9,0xf9,0xf9,0xf9,0x4c,0x00},
	{0xf8,0xf9,0xfa,0xfb,0xfb,0xfa,0xf8,0x4c},
	{0xf9,0xfa,0xfe,0xfb,0xfb,0xfb,0xf9,0x4c},
	{0xf9,0xfb,0xfb,0xfb,0xfb,0xfb,0xf9,0x4c},
	{0xf9,0xfb,0xfb,0xfb,0xfb,0xfa,0xf9,0x4c},
	{0xf9,0xfa,0xfb,0xfb,0xfa,0xf9,0xf8,0x4c},
	{0x4c,0xf8,0xf9,0xf9,0xf9,0xf8,0x4c,0x4c},
	{0x00,0x4c,0x4c,0x4c,0x4c,0x4c,0x4c,0x00}
    },
    {
	// yellow
	{0x00,0xc8,0xc5,0xc5,0xc5,0xc5,0xcb,0x00},
	{0xc8,0xc5,0xc2,0x6f,0x6f,0xc2,0xc8,0xcb},
	{0xc5,0xc2,0xfe,0x6f,0x6f,0x6f,0xc5,0xcb},
	{0xc5,0x6f,0x6f,0x6f,0x6f,0x6f,0xc5,0xcb},
	{0xc5,0x6f,0x6f,0x6f,0x6f,0xc2,0xc5,0xcb},
	{0xc5,0xc2,0x6f,0x6f,0xc2,0xc5,0xc8,0xcb},
	{0xcb,0xc8,0xc5,0xc5,0xc5,0xc8,0xcb,0xcb},
	{0x00,0xcb,0xcb,0xcb,0xcb,0xcb,0xcb,0x00}
    },
    {
	// blue
	{0x00,0xd8,0xd5,0xd5,0xd5,0xd5,0xdc,0x00},
	{0xd8,0xd5,0xd2,0xd0,0xd0,0xd2,0xd8,0xdc},
	{0xd5,0xd2,0xfe,0xd0,0xd0,0xd0,0xd5,0xdc},
	{0xd5,0xd0,0xd0,0xd0,0xd0,0xd0,0xd5,0xdc},
	{0xd5,0xd0,0xd0,0xd0,0xd0,0xd2,0xd5,0xdc},
	{0xd5,0xd2,0xd0,0xd0,0xd2,0xd5,0xd8,0xdc},
	{0xdc,0xd8,0xd5,0xd5,0xd5,0xd8,0xdc,0xdc},
	{0x00,0xdc,0xdc,0xdc,0xdc,0xdc,0xdc,0x00}
    }
};

static void W_InsertOcranaLeds (byte *data) {
	byte *leddata;
	int i, x, y;

	for (i = 0; i < 4; i++) {
		leddata = data + (0x80 >> 4 << 10) + (0x06 << 3) + (i << 3);

		for (y = 0; y < 8; y++) {
			for (x = 0; x < 8; x++)
				*leddata++ = ocrana_leds[i][y][x];
			leddata += 128 - 8;
		}
	}
}

/*
=============================================================================
WAD3 Texture Loading for BSP 3.0 Support
=============================================================================
*/

#define TEXWAD_MAXIMAGES 16384
typedef struct {
	char name[MAX_QPATH];
	vfsfile_t *file;
	int position;
	int size;
} texwadlump_t;

static texwadlump_t texwadlump[TEXWAD_MAXIMAGES];
static int wad3_numlumps = 0;

void WAD3_LoadWadFile (char *filename)
{
	lumpinfo_t *lumps, *lump_p;
	wadinfo_t header;
	int i, j, infotableofs, numlumps, lowmark;
	vfsfile_t *file;
	vfserrno_t err;

	if (wad3_numlumps == TEXWAD_MAXIMAGES)
		return;

	if (!(file = FS_OpenVFS(va("textures/halflife/%s", filename), "rb", FS_ANY)))
		if (!(file = FS_OpenVFS(va("textures/wad3/%s", filename), "rb", FS_ANY)))
			if (!(file = FS_OpenVFS(va("textures/%s", filename), "rb", FS_ANY)))
				if (!(file = FS_OpenVFS(filename, "rb", FS_ANY)))
				{
					Com_Printf ("WAD3_LoadWadFile: couldn't load halflife wad \"%s\"", filename);
					return;
				}
	if (VFS_READ(file, &header, sizeof(wadinfo_t), &err) != sizeof(wadinfo_t)) {
		Com_Printf ("WAD3_LoadWadFile: unable to read wad header\n");
		return;
	}

	if (memcmp(header.identification, "WAD3", 4)) {
		Com_Printf ("WAD3_LoadWadFile: Wad file %s doesn't have WAD3 id\n",filename);
		return;
	}

	numlumps = LittleLong(header.numlumps);
	if (numlumps < 1 || numlumps > TEXWAD_MAXIMAGES) {
		Com_Printf ("WAD3_LoadWadFile: invalid number of lumps (%i)\n", numlumps);
		return;
	}

	infotableofs = LittleLong(header.infotableofs);
	if (VFS_SEEK(file, infotableofs, SEEK_SET)) {
		Com_Printf ("WAD3_LoadWadFile: unable to seek to lump table\n");
		return;
	}

	lowmark = Hunk_LowMark();
	if (!(lumps = Hunk_Alloc(sizeof(lumpinfo_t) * numlumps))) {
		Com_Printf ("WAD3_LoadWadFile: unable to allocate temporary memory for lump table\n");
		return;
	}

	if (((size_t) VFS_READ(file, lumps, sizeof(lumpinfo_t)*numlumps, &err)) != sizeof(lumpinfo_t) * numlumps) {
		Com_Printf ("WAD3_LoadWadFile: unable to read lump table\n");
		Hunk_FreeToLowMark(lowmark);
		return;
	}

	for (i = 0, lump_p = lumps; i < numlumps; i++,lump_p++) {
		W_CleanupName (lump_p->name, lump_p->name);
		for (j = 0; j < wad3_numlumps; j++) {
			if (!strcmp(lump_p->name, texwadlump[j].name))
				goto skip_duplicate;
		}
		strlcpy (texwadlump[j].name, lump_p->name, sizeof(texwadlump[j].name));
		texwadlump[j].file = file;
		texwadlump[j].position = LittleLong(lump_p->filepos);
		texwadlump[j].size = LittleLong(lump_p->disksize);
		wad3_numlumps++;
		if (wad3_numlumps == TEXWAD_MAXIMAGES)
			break;
skip_duplicate:;
	}

	Hunk_FreeToLowMark(lowmark);
	//leaves the file open
}


//converts paletted to rgba
byte *ConvertWad3ToRGBA(int width, int height, byte *in, qbool alpha)
{
	byte *data, *pal;
	int i, p, image_size;

	image_size = width * height;
	data = Q_malloc (image_size * 4);

	pal = (byte *) in + ((image_size * 85) / 64) + 2;

	for (i = 0; i < image_size; i++) {
		p = *in++;
		if (p == 255 && alpha) {
			((int *) data)[i] = 0;
		} else {
			p *= 3;
			data[i * 4 + 0] = pal[p];
			data[i * 4 + 1] = pal[p + 1];
			data[i * 4 + 2] = pal[p + 2];
			data[i * 4 + 3] = 255;
		}
	}
	return data;
}

byte *WAD3_LoadTexture (texture_t *tx)
{
	int i, j = 0;
	miptex_t *mt;
	byte *data;
	vfsfile_t *file;
	vfserrno_t err;

	if (tx->offsets[0])
		return ConvertWad3ToRGBA(tx->width, tx->height, (byte *)(tx + 1), (tx->name[0] == '{'));

	for (i = 0; i < wad3_numlumps; i++) {
		if (strcasecmp(tx->name, texwadlump[i].name))
			continue;
		
		file = texwadlump[i].file;
		if (VFS_SEEK(file, texwadlump[i].position, SEEK_SET)) {
			Com_Printf("WAD3_LoadTexture: corrupt WAD3 file\n");
			return NULL;
		}
		mt = Q_malloc(texwadlump[i].size);
		if (VFS_READ(file, mt, texwadlump[i].size, &err) < texwadlump[i].size) {
			Com_Printf("WAD3_LoadTexture: corrupt WAD3 file\n");
			Q_free (mt);
			return NULL;
		}
		mt->width = LittleLong(mt->width);
		mt->height = LittleLong(mt->height);
		if (mt->width != tx->width || mt->height != tx->height) {
			Q_free (mt);
			return NULL;
		}
		for (j = 0;j < MIPLEVELS;j++)
			mt->offsets[j] = LittleLong(mt->offsets[j]);
		data = ConvertWad3ToRGBA(mt->width, mt->height, (byte *)(mt + 1), (tx->name[0] == '{'));
		Q_free (mt);
		return data;
	}
	return NULL;
}