#include <ultra64.h>
#include "constants.h"
#include "game/tex.h"
#include "game/texdecompress.h"
#include "bss.h"
#include "lib/rzip.h"
#include "lib/dma.h"
#include "lib/main.h"
#include "lib/memp.h"
#include "data.h"
#include "textureconfig.h"
#include "types.h"
#ifndef PLATFORM_N64
#include "video.h"
#endif

void texSetBitstring(u8 *bitstring)
{
	g_TexBitstring = bitstring;
	g_TexAccumValue = 0;
	g_TexAccumNumBits = 0;
}

s32 texReadBits(s32 wantnumbits)
{
	while (g_TexAccumNumBits < wantnumbits) {
		g_TexAccumValue = g_TexAccumValue << 8 | *g_TexBitstring;
		g_TexBitstring++;
		g_TexAccumNumBits += 8;
	}

	g_TexAccumNumBits -= wantnumbits;

	return (g_TexAccumValue >> g_TexAccumNumBits) & ((1 << wantnumbits) - 1);
}

extern u8 *g_TextureConfigSegment;
extern u32 g_TexBase;
extern Gfx *g_TexGdl3;
extern struct textureconfig *g_TexRedLinesConfigs;
extern struct textureconfig *g_TexGroup11Configs;

extern u8 EXT_SEG _textureconfigSegmentRomStart;
extern u8 EXT_SEG _textureconfigSegmentStart;
extern u8 EXT_SEG _textureconfigSegmentEnd;

void texReset(void)
{
	s32 stage;
#ifdef PLATFORM_N64
	u32 len = &_textureconfigSegmentEnd - &_textureconfigSegmentStart;
#endif
	s32 i;

#ifdef PLATFORM_N64
	g_TextureConfigSegment = mempAlloc(len, MEMPOOL_STAGE);
	dmaExec(g_TextureConfigSegment, (romptr_t)&_textureconfigSegmentRomStart, len);
	g_TexBase = (uintptr_t)g_TextureConfigSegment - ROM_SIZE * 1024 * 1024;

	g_TexGdl1 = (Gfx *)(g_TexBase + (uintptr_t)g_TcGdl1);
	g_TexGdl2 = (Gfx *)(g_TexBase + (uintptr_t)g_TcGdl2);
	g_TexGdl3 = (Gfx *)(g_TexBase + (uintptr_t)g_TcGdl3);
	g_ExplosionTexturePairs = (void *)(g_TexBase + (uintptr_t)g_TcExplosionTexturePairs);
	g_TexWallhitConfigs = (struct textureconfig *)(g_TexBase + (uintptr_t)g_TcWallhitConfigs);
	g_TexBeamConfigs = (struct textureconfig *)(g_TexBase + (uintptr_t)g_TcBeamConfigs);
	g_TexLaserConfigs = (struct textureconfig *)(g_TexBase + (uintptr_t)g_TcLaserConfigs);
	g_TexGroup03Configs = (struct textureconfig *)(g_TexBase + (uintptr_t)g_TcGroup03Configs);
	g_TexGeCrosshairConfigs = (struct textureconfig *)(g_TexBase + (uintptr_t)g_TcGeCrosshairConfigs);
	g_TexRedLinesConfigs = (struct textureconfig *)(g_TexBase + (uintptr_t)g_TcRedLineConfigs);
	g_TexShadowConfigs = (struct textureconfig *)(g_TexBase + (uintptr_t)g_TcShadowConfigs);
	g_TexShieldConfigs = (struct textureconfig *)(g_TexBase + (uintptr_t)g_TcShieldConfigs);
	g_TexShardConfigs = (struct textureconfig *)(g_TexBase + (uintptr_t)g_TcShardConfigs);
	g_TexScreenConfigs = (struct textureconfig *)(g_TexBase + (uintptr_t)g_TcScreenConfigs);
	g_TexSkyWaterConfigs = (struct textureconfig *)(g_TexBase + (uintptr_t)g_TcSkyWaterConfigs);
	g_TexGroup11Configs = (struct textureconfig *)(g_TexBase + (uintptr_t)g_TcGroup11Configs);
	g_TexLightGlareConfigs = (struct textureconfig *)(g_TexBase + (uintptr_t)g_TcLightGlareConfigs);
	g_TexSparkConfigs = (struct textureconfig *)(g_TexBase + (uintptr_t)g_TcSparkConfigs);
	g_TexGeneralConfigs = (struct textureconfig *)(g_TexBase + (uintptr_t)g_TcGeneralConfigs);
	g_TexRadarConfigs = (struct textureconfig *)(g_TexBase + (uintptr_t)g_TcRadarConfigs);

	g_TexNumConfigs = (len - (uintptr_t)&g_TcWallhitConfigs + ROM_SIZE * 1024 * 1024) / sizeof(struct textureconfig);
#else
	// HACK: define a big table of pointers we need to fill and sizes of the textureconfig data that goes there
	#define DEFINE_TCPTR(ptr, data) { (void **)&ptr, (const void *)data, sizeof(data), ARRAYCOUNT(data) }
	static const struct {
		void **dst;
		const void *src;
		u32 size;
		s32 count;
	} tcptrs[] = {
		DEFINE_TCPTR(g_TexGdl1, g_TcGdl1),
		DEFINE_TCPTR(g_TexGdl2, g_TcGdl2),
		DEFINE_TCPTR(g_TexGdl3, g_TcGdl3),
		DEFINE_TCPTR(g_ExplosionTexturePairs, g_TcExplosionTexturePairs),
		DEFINE_TCPTR(g_TexWallhitConfigs, g_TcWallhitConfigs),
		DEFINE_TCPTR(g_TexBeamConfigs, g_TcBeamConfigs),
		DEFINE_TCPTR(g_TexLaserConfigs, g_TcLaserConfigs),
		DEFINE_TCPTR(g_TexGroup03Configs, g_TcGroup03Configs),
		DEFINE_TCPTR(g_TexGeCrosshairConfigs, g_TcGeCrosshairConfigs),
		DEFINE_TCPTR(g_TexRedLinesConfigs, g_TcRedLineConfigs),
		DEFINE_TCPTR(g_TexShadowConfigs, g_TcShadowConfigs),
		DEFINE_TCPTR(g_TexShieldConfigs, g_TcShieldConfigs),
		DEFINE_TCPTR(g_TexShardConfigs, g_TcShardConfigs),
		DEFINE_TCPTR(g_TexScreenConfigs, g_TcScreenConfigs),
		DEFINE_TCPTR(g_TexSkyWaterConfigs, g_TcSkyWaterConfigs),
		DEFINE_TCPTR(g_TexGroup11Configs, g_TcGroup11Configs),
		DEFINE_TCPTR(g_TexLightGlareConfigs, g_TcLightGlareConfigs),
		DEFINE_TCPTR(g_TexSparkConfigs, g_TcSparkConfigs),
		DEFINE_TCPTR(g_TexGeneralConfigs, g_TcGeneralConfigs),
		DEFINE_TCPTR(g_TexRadarConfigs, g_TcRadarConfigs),
	};
	#undef DEFINE_TCPTR

	// calculate total length, should be 0xb50 on a 32-bit platform with an unmodified textureconfig.c
	u32 len = 0;
	for (s32 i = 0; i < ARRAYCOUNT(tcptrs); ++i) {
		len += tcptrs[i].size;
	}

	g_TextureConfigSegment = mempAlloc(len, MEMPOOL_STAGE);
	g_TexBase = 0; // unused

	// set up pointers and fill them in
	u32 tcofs = 0;
	for (s32 i = 0; i < ARRAYCOUNT(tcptrs); ++i) {
		*tcptrs[i].dst = g_TextureConfigSegment + tcofs;
		bcopy(tcptrs[i].src, *tcptrs[i].dst, tcptrs[i].size);
		tcofs += tcptrs[i].size;
	}

	// calculate tc count, skipping the gdls and explosion pairs
	g_TexNumConfigs = 0;
	for (s32 i = 4; i < ARRAYCOUNT(tcptrs); ++i) {
		g_TexNumConfigs += tcptrs[i].count;
	}

	// reset backend texture cache
	videoResetTextureCache();

	// get backend texture settings
	g_TexFilter2D = videoGetTextureFilter2D() ? G_TF_BILERP : G_TF_POINT;
#endif

	g_TexWords = mempAlloc(ALIGN16(g_TexNumConfigs * 4), MEMPOOL_STAGE);

	for (i = 0; i < g_TexNumConfigs; i++) {
		g_TexWords[i] = NULL;
	}

	for (i = 0; i < ARRAYCOUNT(g_TcExplosionTexturePairs); i++) {
		texLoad(&g_ExplosionTexturePairs[i].texturenum1, NULL, false);
		texLoad(&g_ExplosionTexturePairs[i].texturenum2, NULL, false);
	}

	texLoadFromDisplayList(g_TexGdl1, 0, 0);
	texLoadFromDisplayList(g_TexGdl3, 0, 0);

	stage = mainGetStageNum();

	if (IS4MB() && stage != STAGE_TITLE && stage != STAGE_CITRAINING && stage != STAGE_4MBMENU) {
		for (i = 0; i < ARRAYCOUNT(g_TcWallhitConfigs); i++) {
			texLoadFromConfig(&g_TexWallhitConfigs[i]);
		}

		for (i = 0; i < ARRAYCOUNT(g_TcBeamConfigs); i++) {
			texLoadFromConfig(&g_TexBeamConfigs[i]);
		}

		for (i = 0; i < ARRAYCOUNT(g_TcSkyWaterConfigs); i++) {
			texLoadFromConfig(&g_TexSkyWaterConfigs[i]);
		}

		for (i = 0; i < ARRAYCOUNT(g_TcShadowConfigs); i++) {
			texLoadFromConfig(&g_TexShadowConfigs[i]);
		}

		for (i = 0; i < ARRAYCOUNT(g_TcShieldConfigs); i++) {
			texLoadFromConfig(&g_TexShieldConfigs[i]);
		}

		for (i = 0; i < ARRAYCOUNT(g_TcRadarConfigs); i++) {
			texLoadFromConfig(&g_TexRadarConfigs[i]);
		}

		for (i = 0; i < ARRAYCOUNT(g_TcSparkConfigs); i++) {
			texLoadFromConfig(&g_TexSparkConfigs[i]);
		}
	}
}