#include "stdafx.h"
#include "Textures.h"
#include "StitchedTexture.h"
#include "AbstractTexturePack.h"
#include "TexturePackRepository.h"
#include "..\Minecraft.World\InputOutputStream.h"
#include "..\Minecraft.World\FileInputStream.h"
#include "..\Minecraft.World\StringHelpers.h"

const unordered_map<std::wstring, std::wstring> AbstractTexturePack::INDEXED_TO_JAVA_MAP = {
	{L"res/misc/pumpkinblur.png", L"misc/pumpkinblur.png"},
	//	L"%blur%/misc/vignette",		// Not currently used
	{L"res/misc/shadow", L""},
	//	L"/achievement/bg",				// Not currently used
	{L"art/kz", L""},
	{L"res/environment/clouds.png", L"environment/clouds.png"},
	{L"res/environment/rain.png", L"environment/rain.png"},
	{L"res/environment/snow.png", L"environment/snow.png"},
	{L"gui/gui", L""},
	{L"gui/icons", L""},
	//{L"item/arrows", L""},
	//{L"item/boat", L""},
	//{L"item/cart", L""},
	//{L"item/sign", L""},
	{L"res/misc/mapbg", L""},
	{L"res/misc/mapicons", L""},
	{L"res/misc/water", L""},
	{L"res/misc/footprint", L""},
	//{L"mob/saddle", L""},
	{L"res/res/mob/sheep_fur.png", L"entity/sheep/sheep_fur.png"},
	{L"res/mob/spider_eyes.png", L""},
	{L"res/particles", L""},
	{L"res/mob/chicken.png", L"entity/chicken.png"},
	{L"res/mob/cow.png", L"entity/cow/cow.png"},
	{L"res/mob/pig.png", L"entity/pig/pig.png"},
	{L"res/mob/sheep.png", L"entity/sheep/sheep.png"},
	{L"res/mob/squid.png", L"entity/squid/squid.png"},
	{L"res/mob/wolf.png", L"entity/wolf/wolf.png"},
	{L"res/mob/wolf_tame.png", L"entity/wolf/wolf_tame.png"},
	{L"res/mob/wolf_angry.png", L"entity/wolf/wolf_angry.png"},
	{L"res/mob/creeper.png", L"entity/creeper/creeper.png"},
	{L"res/mob/ghast.png", L"entity/ghast/ghast.png"},
	{L"res/mob/ghast_fire.png", L"entity/ghast/ghast_shooting.png"},
	//{L"res/mob/zombie", L""}, // zombie uses 64x texture
	//{L"res/mob/pigzombie", L""}, // rip
	{L"res/mob/skeleton.png", L"entity/skeleton/skeleton.png"},
	{L"res/mob/slime.png", L"entity/slime/slime.png"},
	{L"res/mob/spider.png", L"entity/spider/spider.png"},
	//{L"mob/char", L""},
	//{L"mob/char1", L""},
	//{L"mob/char2", L""},
	//{L"mob/char3", L""},
	//{L"mob/char4", L""},
	//{L"mob/char5", L""},
	//{L"mob/char6", L""},
	//{L"mob/char7", L""},
	{L"terrain/moon", L""},
	{L"terrain/sun", L""},
	{L"armor/power", L""},

	// 1.8.2
	{L"res/mob/cavespider.png", L"entity/spider/cave_spider.png"},
	{L"res/mob/enderman.png", L"entity/enderman/enderman.png"},
	{L"res/mob/silverfish.png", L"entity/silverfish.png"},
	{L"res/mob/enderman_eyes.png", L"entity/enderman/enderman_eyes.png"},
	//{L"res/misc/explosion", L""}, // not bothering to reatlas them rn
	{L"res/item/xporb.png", L"entity/experience_orb.png"},
	{L"res/item/chest.png", L"entity/chest/normal.png"},
	//{L"item/largechest", L"entity/chest/normal"}, // was split in half

	// 1.3.2 
	{L"res/item/enderchest.png", L"entity/chest/ender.png"},

	// 1.0.1 
	{L"res/mob/redcow.png", L"entity/cow/mooshroom.png"},
	//{L"mob/snowman", L""},
	//{L"mob/enderdragon/ender", L""},
	{L"res/mob/fire.png", L"entity/blaze.png"},
	{L"res/mob/lava.png", L"entity/slime/magmacube.png"},
	//{L"mob/villager/villager", L""},
	//{L"mob/villager/farmer", L""},
	//{L"mob/villager/librarian", L""},
	//{L"mob/villager/priest", L""},
	//{L"mob/villager/smith", L""},
	//{L"mob/villager/butcher", L""},
	{L"res/mob/enderdragon/crystal.png", L"entity/end_crystal/end_crystal.png"},
	//{L"mob/enderdragon/shuffle", L""},
	{L"res/mob/enderdragon/beam.png", L"entity/end_crystal/end_crystal_beam.png"},
	//{L"mob/enderdragon/ender_eyes", L""},
	{L"res/misc/glint.png", L"misc/enchanted_item_glint.png"},
	//{L"item/book", L""},
	{L"res/misc/tunnel.png", L"environment/end_sky.png"},
	{L"res/misc/particlefield.png", L"entity/end_portal.png"},
	{L"res/terrain/moon_phases.png", L"environment/moon_phases.png"},

	// 1.2.3
	{L"res/mob/ozelot.png", L"entity/cat/ocelot.png"},
	{L"res/mob/cat_black.png", L"entity/cat/black.png"},
	{L"res/mob/cat_red.png", L"entity/cat/red.png"},
	{L"res/mob/cat_siamese.png", L"entity/cat/siamese.png"},
	{L"res/mob/villager_golem.png", L"entity/iron_golem.png"},
	{L"res/mob/skeleton_wither.png", L"entity/wither_skeleton.png"},

	// TU 14
	{L"res/mob/wolf_collar.png", L"entity/wolf/wolf_collar.png"},
	//{L"mob/zombie_villager", L""},

	// 1.6.4
	{L"res/item/lead_knot.png", L"entity/lead_knot.png"},

	{L"res/misc/beacon_beam.png", L"entity/beacon_beam.png"},

	//{L"res/mob/bat.png", L"entity/bat.png"}, // new uvs is broken

	// incompatible horse armor
	//{L"mob/horse/donkey", L""},
	//{L"mob/horse/horse_black", L""},
	//{L"mob/horse/horse_brown", L""},
	//{L"mob/horse/horse_chestnut", L""},
	//{L"mob/horse/horse_creamy", L""},
	//{L"mob/horse/horse_darkbrown", L""},
	//{L"mob/horse/horse_gray", L""},
	//{L"mob/horse/horse_markings_blackdots", L""},
	//{L"mob/horse/horse_markings_white", L""},
	//{L"mob/horse/horse_markings_whitedots", L""},
	//{L"mob/horse/horse_markings_whitefield", L""},
	//{L"mob/horse/horse_skeleton", L""},
	//{L"mob/horse/horse_white", L""},
	//{L"mob/horse/horse_zombie", L""},
	//{L"mob/horse/mule", L""},

	//{L"mob/horse/armor/horse_armor_diamond", L""},
	//{L"mob/horse/armor/horse_armor_gold", L""},
	//{L"mob/horse/armor/horse_armor_iron", L""},

	{ L"res/mob/witch", L"entity/witch.png" },

	{ L"res/mob/wither/wither.png", L"entity/wither/wither.png" },
	{ L"res/mob/wither/wither_armor.png", L"entity/wither/wither_armor.png" },
	{ L"res/mob/wither/wither_invulnerable.png", L"entity/wither/wither_invulnerable.png" },

	{ L"res/item/trapped.png", L"entity/chest/trapped.png" },
	//{ L"item/trapped_double", L"" },
	//L"item/christmas",
	//L"item/christmas_double",

#ifdef _LARGE_WORLDS
	//{L"misc/additionalmapicons", L""},
#endif

	//{L"font/Default", L""},
	//{L"font/alternate", L""},

	// skin packs 
/*	{L"/SP1", L""},
	{L"/SP2", L""},
	{L"/SP3", L""},
	{L"/SPF", L""},
	{ L""},
	{// themes L""},
	{L"/ThSt", L""},
	{L"/ThIr", L""},
	{L"/ThGo", L""},
	{L"/ThDi", L""},
	{ L""},
	{// gamerpics L""},
	{L"/GPAn", L""},
	{L"/GPCo", L""},
	{L"/GPEn", L""},
	{L"/GPFo", L""},
	{L"/GPTo", L""},
	{L"/GPBA", L""},
	{L"/GPFa", L""},
	{L"/GPME", L""},
	{L"/GPMF", L""},
	{L"/GPMM", L""},
	{L"/GPSE", L""},
	{ L""},
	{// avatar items L""},
	{ L""},
	{L"/AH_0006", L""},
	{L"/AH_0003", L""},
	{L"/AH_0007", L""},
	{L"/AH_0005", L""},
	{L"/AH_0004", L""},
	{L"/AH_0001", L""},
	{L"/AH_0002", L""},
	{L"/AT_0001", L""},
	{L"/AT_0002", L""},
	{L"/AT_0003", L""},
	{L"/AT_0004", L""},
	{L"/AT_0005", L""},
	{L"/AT_0006", L""},
	{L"/AT_0007", L""},
	{L"/AT_0008", L""},
	{L"/AT_0009", L""},
	{L"/AT_0010", L""},
	{L"/AT_0011", L""},
	{L"/AT_0012", L""},
	{L"/AP_0001", L""},
	{L"/AP_0002", L""},
	{L"/AP_0003", L""},
	{L"/AP_0004", L""},
	{L"/AP_0005", L""},
	{L"/AP_0006", L""},
	{L"/AP_0007", L""},
	{L"/AP_0009", L""},
	{L"/AP_0010", L""},
	{L"/AP_0011", L""},
	{L"/AP_0012", L""},
	{L"/AP_0013", L""},
	{L"/AP_0014", L""},
	{L"/AP_0015", L""},
	{L"/AP_0016", L""},
	{L"/AP_0017", L""},
	{L"/AP_0018", L""},
	{L"/AA_0001", L""},
	{L"/AT_0013", L""},
	{L"/AT_0014", L""},
	{L"/AT_0015", L""},
	{L"/AT_0016", L""},
	{L"/AT_0017", L""},
	{L"/AT_0018", L""},
	{L"/AP_0019", L""},
	{L"/AP_0020", L""},
	{L"/AP_0021", L""},
	{L"/AP_0022", L""},
	{L"/AP_0023", L""},
	{L"/AH_0008", L""},
	{L"/AH_0009", L""},*/

	//{L"gui/items", L""},
	//{L"terrain", L""}
};

AbstractTexturePack::AbstractTexturePack(DWORD id, File *file, const wstring &name, TexturePack *fallback) : id(id), name(name)
{
	// 4J init
	textureId = -1;
	m_colourTable = NULL;


	this->file = file;
	this->fallback = fallback;

	m_iconData = NULL;
	m_iconSize = 0;

	m_comparisonData = NULL;
	m_comparisonSize = 0;

	// 4J Stu - These calls need to be in the most derived version of the class
	//loadIcon();
	//loadDescription();
}

wstring AbstractTexturePack::trim(wstring line)
{
	if (!line.empty() && line.length() > 34)
	{
		line = line.substr(0, 34);
	}
	return line;
}

void AbstractTexturePack::loadIcon()
{
#ifdef _XBOX
	// 4J Stu - Temporary only	
	const DWORD LOCATOR_SIZE = 256; // Use this to allocate space to hold a ResourceLocator string 
	WCHAR szResourceLocator[ LOCATOR_SIZE ];

	const ULONG_PTR c_ModuleHandle = (ULONG_PTR)GetModuleHandle(NULL);
	swprintf(szResourceLocator, LOCATOR_SIZE ,L"section://%X,%ls#%ls",c_ModuleHandle,L"media", L"media/Graphics/TexturePackIcon.png");

	UINT size = 0;
	HRESULT hr = XuiResourceLoadAllNoLoc(szResourceLocator, &m_iconData, &size);
	m_iconSize = size;
#endif
}

void AbstractTexturePack::loadComparison()
{
#ifdef _XBOX
	// 4J Stu - Temporary only	
	const DWORD LOCATOR_SIZE = 256; // Use this to allocate space to hold a ResourceLocator string 
	WCHAR szResourceLocator[ LOCATOR_SIZE ];

	const ULONG_PTR c_ModuleHandle = (ULONG_PTR)GetModuleHandle(NULL);
	swprintf(szResourceLocator, LOCATOR_SIZE ,L"section://%X,%ls#%ls",c_ModuleHandle,L"media", L"media/Graphics/DefaultPack_Comparison.png");

	UINT size = 0;
	HRESULT hr = XuiResourceLoadAllNoLoc(szResourceLocator, &m_comparisonData, &size);
	m_comparisonSize = size;
#endif
}

void AbstractTexturePack::loadDescription()
{
	// 4J Unused currently
#if 0
	InputStream *inputStream = NULL;
	BufferedReader *br = NULL;
	//try {
	inputStream = getResourceImplementation(L"/pack.txt");
	br = new BufferedReader(new InputStreamReader(inputStream));
	desc1 = trim(br->readLine());
	desc2 = trim(br->readLine());
	//} catch (IOException ignored) {
	//} finally {
	// TODO [EB]: use IOUtils.closeSilently()
	//	try {
	if (br != NULL)
	{
		br->close();
		delete br;
	}
	if (inputStream != NULL)
	{
		inputStream->close();
		delete inputStream;
	}
	//	} catch (IOException ignored) {
	//	}
	//}
#endif
}

void AbstractTexturePack::loadName()
{
}

void AbstractTexturePack::checkTexSize() {
	BufferedImage* img = getImageResource(L"dirt.png", true);
	if (img != NULL) {
		int width = img->getWidth();
		int height = img->getHeight();
		if (width != height) {
			app.DebugPrintf("Warning: Texture pack contains texture with bad size: %d x %d\n", width, height);
		} else
			texSize = width;
		delete img;
	} else {
		texSize = 16;
	}
}

InputStream *AbstractTexturePack::getResource(const wstring &name, bool allowFallback) //throws IOException
{
	app.DebugPrintf("texture - %ls\n",name.c_str());
	InputStream *is = getResourceImplementation(name);
	if (is == NULL && fallback != NULL && allowFallback)
	{
		is = fallback->getResource(name, true);
	}

	return is;
}

// 4J Currently removed due to override in TexturePack class
//InputStream *AbstractTexturePack::getResource(const wstring &name) //throws IOException
//{
//	return getResource(name, true);
//}

void AbstractTexturePack::unload(Textures *textures)
{
	if (iconImage != NULL && textureId != -1)
	{
		textures->releaseTexture(textureId);
	}
}

void AbstractTexturePack::load(Textures *textures)
{
	if (iconImage != NULL)
	{
		if (textureId == -1)
		{
			textureId = textures->getTexture(iconImage);
		}
		glBindTexture(GL_TEXTURE_2D, textureId);
		textures->clearLastBoundId();
	}
	else
	{
		// 4J Stu - Don't do this
		//textures->bindTexture(L"/gui/unknown_pack.png");
	}
}

bool AbstractTexturePack::hasFile(const wstring &name, bool allowFallback)
{
	if (name == L"res/terrain.png" && terrainAtlas != nullptr)
		return true;

	if (name == L"res/items.png" && itemAtlas != nullptr)
		return true;

	bool hasFile = this->hasFile(name);

	auto it = INDEXED_TO_JAVA_MAP.find(name);
	if (it != INDEXED_TO_JAVA_MAP.end()) {
		hasFile = this->hasFile(L"assets/minecraft/textures/" + it->second);
	}

	return !hasFile && (allowFallback && fallback != NULL) ? fallback->hasFile(name, allowFallback) : hasFile;
}

DWORD AbstractTexturePack::getId()
{
	return id;
}

wstring AbstractTexturePack::getName()
{
	return texname;
}

wstring AbstractTexturePack::getWorldName()
{
	return m_wsWorldName;
}

wstring AbstractTexturePack::getDesc1()
{
	return desc1;
}

wstring AbstractTexturePack::getDesc2()
{
	return desc2;
}

wstring AbstractTexturePack::getAnimationString(const wstring &textureName, const wstring &path, bool allowFallback)
{
	return getAnimationString(textureName, path);
}

wstring AbstractTexturePack::getAnimationString(const wstring &textureName, const wstring &path)
{
	wstring animationDefinitionFile = textureName + L".txt";

	bool requiresFallback = !hasFile(L"\\" + textureName + L".png", false);
	
	wstring result = L"";

	InputStream *fileStream = getResource(L"\\" + path + animationDefinitionFile, requiresFallback);

	if(fileStream)
	{
		//Minecraft::getInstance()->getLogger().info("Found animation info for: " + animationDefinitionFile);
#ifndef _CONTENT_PACKAGE
		app.DebugPrintf("Found animation info for: %ls\n", animationDefinitionFile.c_str() );
#endif
		InputStreamReader isr(fileStream);
		BufferedReader br(&isr);


		wstring line = br.readLine();
		while (!line.empty())
		{
			line = trimString(line);
			if (line.length() > 0)
			{
				result.append(L",");
				result.append(line);
			}
			line = br.readLine();
		}
		delete fileStream;
	}

	return result;
}

struct UV {
	float x0, y0, x1, y1;

	float width(float imgSize) {
		return (x1 - x0) * imgSize - 1;
	}

	float height(float imgSize) {
		return (y1 - y0) * imgSize - 1;
	}

	UV operator /=(const float other) {
		x0 /= other;
		y0 /= other;
		x1 /= other;
		y1 /= other;
		return *this;
	}
};

BufferedImage* AbstractTexturePack::getBedTex(std::wstring name) {
	
	if (bedTexCache.get() == nullptr) {
		BufferedImage* bedtex = getImageResource(L"assets/minecraft/textures/entity/bed/red.png", true);
		if (bedtex != nullptr) {
			if (bedtex->getWidth() < 0 || bedtex->getHeight() < 0)
				return nullptr;

			int len = bedtex->getWidth() * bedtex->getHeight();
			bedTexCache = std::make_unique<BufferedImage>(bedtex->getWidth(), bedtex->getHeight(), 0);
			memcpy(bedTexCache->getData(), bedtex->getData(), len * sizeof(int));
			delete bedtex;
		}
	}

	if (bedTexCache.get() == nullptr) { // todo: rip from og atlas
		app.DebugPrintf("Failed to load bed texture, returning null\n");
		return nullptr;
	}

	// hardcoded in java and no way to load it here, I hate hardcoding this but its what I got
	UV	feetEnd, feetEndStubL, feetEndStubR,
		feetSide, feetSideStub, 
		headSide, headSideStub,
		headEnd, headEndStubL, headEndStubR,
		feetFace, headFace;

	// This code is extremely dumb, x0 and y0 must always be less than x1 and y1
	// These are the top left of the pixel, so add 1 extra for x1 and y1
	const float baseBedTexSize = 64; // original bed tex size, dont change unless new base game res changes
	headFace = { 6, 6, 22, 22 };
	feetFace = { 6, 28, 22, 44 };
	feetFace /= baseBedTexSize; headFace /= baseBedTexSize;

	feetEnd = { 22, 22, 38, 28 };
	headEnd = { 6, 0, 22, 6 };
	feetEnd /= baseBedTexSize; headEnd /= baseBedTexSize;

	feetEndStubL = { 53, 3, 56, 6 };
	feetEndStubR = { 50, 15, 53, 18 };
	feetEndStubL /= baseBedTexSize; feetEndStubR /= baseBedTexSize;

	headEndStubL = { 53, 21, 56, 24 };
	headEndStubR = { 50, 9, 53, 12 };
	headEndStubL /= baseBedTexSize; headEndStubR /= baseBedTexSize;

	headSide = { 22, 6, 28, 22 };
	headSideStub = { 50, 21, 53, 24 };
	headSide /= baseBedTexSize; headSideStub /= baseBedTexSize;

	feetSide = { 22, 28, 28, 44 };
	feetSideStub = { 53, 15, 56, 18 };
	feetSide /= baseBedTexSize; feetSideStub /= baseBedTexSize;

	// Output uvs

	UV outBedBody, outStubL, outStubR, outBedFace;
	outBedBody = { 0, 7, 16, 13 };
	outStubL = { 0, 13, 3, 16 };
	outStubR = { 13, 13, 16, 16 };
	outBedFace = { 0, 0, 16, 16 };
	outBedBody /= 16; outStubL /= 16; outStubR /= 16; outBedFace /= 16;

	const int bedTexSize = bedTexCache->getWidth();
	auto workTex = new BufferedImage(texSize, texSize, GL_RGBA);
	auto workPix = reinterpret_cast<Pixel*>(workTex->getData());
	auto bedPix = reinterpret_cast<Pixel*>(bedTexCache->getData());
	memset(workPix, 0, texSize * texSize * sizeof(Pixel));

	if (name == L"bed_feet_end") {

		// side body minus legs
		for (int y = 0; y <= outBedBody.height(texSize); y++)
			for (int x = 0; x <= outBedBody.width(texSize); x++) {
				// first get raw src.x0 of tex to sample from current uv pixel, then current uv value * targetUV dimension
				// hopefully this can sample any size as long as dst is always equal or larger res

				int srcx = roundf((feetEnd.x0 * bedTexSize) + (x / outBedBody.width(texSize)) * feetEnd.width(bedTexSize));
				int srcy = roundf((feetEnd.y0 * bedTexSize) + (1 - (y / outBedBody.height(texSize))) * feetEnd.height(bedTexSize));
				int dstx = roundf((outBedBody.x0 * texSize) + (x / outBedBody.width(texSize)) * outBedBody.width(texSize));
				int dsty = roundf((outBedBody.y0 * texSize) + (y / outBedBody.height(texSize)) * outBedBody.height(texSize));

				int srcIdx = srcy * bedTexCache->getWidth() + srcx;
				int dstIdx = dsty * texSize + dstx;

				(&workPix[dstIdx])->raw = (&bedPix[srcIdx])->raw;
			}

		// left leg
		for (int y = 0; y <= outStubL.height(texSize); y++)
			for (int x = 0; x <= outStubL.width(texSize); x++) {
				int srcx = roundf((feetEndStubL.x0 * bedTexSize) + (x / outStubL.height(texSize)) * feetEndStubL.width(bedTexSize));
				int srcy = roundf((feetEndStubL.y0 * bedTexSize) + (y / outStubL.width(texSize)) * feetEndStubL.height(bedTexSize));
				int dstx = roundf((outStubL.x0 * texSize) + (x / outStubL.width(texSize)) * outStubL.width(texSize));
				int dsty = roundf((outStubL.y0 * texSize) + (y / outStubL.height(texSize)) * outStubL.height(texSize));

				int srcIdx = srcy * bedTexCache->getWidth() + srcx;
				int dstIdx = dsty * texSize + dstx;

				(&workPix[dstIdx])->raw = (&bedPix[srcIdx])->raw;
			}

		// right leg
		for (int y = 0; y <= outStubR.height(texSize); y++)
			for (int x = 0; x <= outStubR.width(texSize); x++) {
				int srcx = roundf((feetEndStubR.x0 * bedTexSize) + (x / outStubR.height(texSize)) * feetEndStubR.width(bedTexSize));
				int srcy = roundf((feetEndStubR.y0 * bedTexSize) + (y / outStubR.width(texSize)) * feetEndStubR.height(bedTexSize));
				int dstx = roundf((outStubR.x0 * texSize) + (x / outStubR.width(texSize)) * outStubR.width(texSize));
				int dsty = roundf((outStubR.y0 * texSize) + (y / outStubR.height(texSize)) * outStubR.height(texSize));

				int srcIdx = srcy * bedTexCache->getWidth() + srcx;
				int dstIdx = dsty * texSize + dstx;

				(&workPix[dstIdx])->raw = (&bedPix[srcIdx])->raw;
			}

		return workTex;
	} else if (name == L"bed_feet_side") {

		// side body minus leg
		for (int y = 0; y <= outBedBody.height(texSize); y++)
			for (int x = 0; x <= outBedBody.width(texSize); x++) {
				// first get raw src.x0 of tex to sample from current uv pixel, then current uv value * targetUV dimension
				// hopefully this can sample any size as long as dst is always equal or larger res

				int srcx = roundf((feetSide.x0 * bedTexSize) + (y / outBedBody.height(texSize)) * feetSide.width(bedTexSize));
				int srcy = roundf((feetSide.y0 * bedTexSize) + (1 - (x / outBedBody.width(texSize))) * feetSide.height(bedTexSize));
				int dstx = roundf((outBedBody.x0 * texSize) + (x / outBedBody.width(texSize)) * outBedBody.width(texSize));
				int dsty = roundf((outBedBody.y0 * texSize) + (y / outBedBody.height(texSize)) * outBedBody.height(texSize));

				int srcIdx = srcy * bedTexCache->getWidth() + srcx;
				int dstIdx = dsty * texSize + dstx;

				(&workPix[dstIdx])->raw = (&bedPix[srcIdx])->raw;
			}

		// side leg
		for (int y = 0; y <= outStubL.height(texSize); y++)
			for (int x = 0; x <= outStubL.width(texSize); x++) {
				int srcx = roundf((feetSideStub.x0 * bedTexSize) + (x / outStubL.height(texSize)) * feetSideStub.width(bedTexSize));
				int srcy = roundf((feetSideStub.y0 * bedTexSize) + (y / outStubL.width(texSize)) * feetSideStub.height(bedTexSize));
				int dstx = roundf((outStubL.x0 * texSize) + (x / outStubL.width(texSize)) * outStubL.width(texSize));
				int dsty = roundf((outStubL.y0 * texSize) + (y / outStubL.height(texSize)) * outStubL.height(texSize));

				int srcIdx = srcy * bedTexCache->getWidth() + srcx;
				int dstIdx = dsty * texSize + dstx;

				(&workPix[dstIdx])->raw = (&bedPix[srcIdx])->raw;
			}

		return workTex;
	} else if (name == L"bed_head_side") {

		// side body minus leg
		for (int y = 0; y <= outBedBody.height(texSize); y++)
			for (int x = 0; x <= outBedBody.width(texSize); x++) {
				// first get raw src.x0 of tex to sample from current uv pixel, then current uv value * targetUV dimension
				// hopefully this can sample any size as long as dst is always equal or larger res

				int srcx = roundf((headSide.x0 * bedTexSize) + (y / outBedBody.height(texSize)) * headSide.width(bedTexSize));
				int srcy = roundf((headSide.y0 * bedTexSize) + (1 - (x / outBedBody.width(texSize))) * headSide.height(bedTexSize));
				int dstx = roundf((outBedBody.x0 * texSize) + (x / outBedBody.width(texSize)) * outBedBody.width(texSize));
				int dsty = roundf((outBedBody.y0 * texSize) + (y / outBedBody.height(texSize)) * outBedBody.height(texSize));

				int srcIdx = srcy * bedTexCache->getWidth() + srcx;
				int dstIdx = dsty * texSize + dstx;

				(&workPix[dstIdx])->raw = (&bedPix[srcIdx])->raw;
			}

		// side leg
		for (int y = 0; y <= outStubR.height(texSize); y++)
			for (int x = 0; x <= outStubR.width(texSize); x++) {
				int srcx = roundf((headSideStub.x0 * bedTexSize) + (x / outStubR.height(texSize)) * headSideStub.width(bedTexSize));
				int srcy = roundf((headSideStub.y0 * bedTexSize) + (y / outStubR.width(texSize)) * headSideStub.height(bedTexSize));
				int dstx = roundf((outStubR.x0 * texSize) + (x / outStubR.width(texSize)) * outStubR.width(texSize));
				int dsty = roundf((outStubR.y0 * texSize) + (y / outStubR.height(texSize)) * outStubR.height(texSize));

				int srcIdx = srcy * bedTexCache->getWidth() + srcx;
				int dstIdx = dsty * texSize + dstx;

				(&workPix[dstIdx])->raw = (&bedPix[srcIdx])->raw;
			}


		return workTex;
	} else if (name == L"bed_head_end") {

		// side body minus legs
		for (int y = 0; y <= outBedBody.height(texSize); y++)
			for (int x = 0; x <= outBedBody.width(texSize); x++) {
				// first get raw src.x0 of tex to sample from current uv pixel, then current uv value * targetUV dimension
				// hopefully this can sample any size as long as dst is always equal or larger res

				int srcx = roundf((headEnd.x0 * bedTexSize) + (x / outBedBody.width(texSize)) * headEnd.width(bedTexSize));
				int srcy = roundf((headEnd.y0 * bedTexSize) + (1 - (y / outBedBody.height(texSize))) * headEnd.height(bedTexSize));
				int dstx = roundf((outBedBody.x0 * texSize) + (x / outBedBody.width(texSize)) * outBedBody.width(texSize));
				int dsty = roundf((outBedBody.y0 * texSize) + (y / outBedBody.height(texSize)) * outBedBody.height(texSize));

				int srcIdx = srcy * bedTexCache->getWidth() + srcx;
				int dstIdx = dsty * texSize + dstx;

				(&workPix[dstIdx])->raw = (&bedPix[srcIdx])->raw;
			}

		// left leg
		for (int y = 0; y <= outStubL.height(texSize); y++)
			for (int x = 0; x <= outStubL.width(texSize); x++) {
				int srcx = roundf((headEndStubL.x0 * bedTexSize) + (x / outStubL.height(texSize)) * headEndStubL.width(bedTexSize));
				int srcy = roundf((headEndStubL.y0 * bedTexSize) + (y / outStubL.width(texSize)) * headEndStubL.height(bedTexSize));
				int dstx = roundf((outStubL.x0 * texSize) + (x / outStubL.width(texSize)) * outStubL.width(texSize));
				int dsty = roundf((outStubL.y0 * texSize) + (y / outStubL.height(texSize)) * outStubL.height(texSize));

				int srcIdx = srcy * bedTexCache->getWidth() + srcx;
				int dstIdx = dsty * texSize + dstx;

				(&workPix[dstIdx])->raw = (&bedPix[srcIdx])->raw;
			}

		// right leg
		for (int y = 0; y <= outStubR.height(texSize); y++)
			for (int x = 0; x <= outStubR.width(texSize); x++) {
				int srcx = roundf((headEndStubR.x0 * bedTexSize) + (x / outStubR.height(texSize)) * headEndStubR.width(bedTexSize));
				int srcy = roundf((headEndStubR.y0 * bedTexSize) + (y / outStubR.width(texSize)) * headEndStubR.height(bedTexSize));
				int dstx = roundf((outStubR.x0 * texSize) + (x / outStubR.width(texSize)) * outStubR.width(texSize));
				int dsty = roundf((outStubR.y0 * texSize) + (y / outStubR.height(texSize)) * outStubR.height(texSize));

				int srcIdx = srcy * bedTexCache->getWidth() + srcx;
				int dstIdx = dsty * texSize + dstx;

				(&workPix[dstIdx])->raw = (&bedPix[srcIdx])->raw;
			}

		return workTex;
	} else if (name == L"bed_head_top") {

		for (int y = 0; y <= outBedFace.height(texSize); y++)
			for (int x = 0; x <= outBedFace.width(texSize); x++) {
				int srcx = roundf((headFace.x0 * bedTexSize) + (x / outBedFace.width(texSize)) * headFace.width(bedTexSize));
				int srcy = roundf((headFace.y0 * bedTexSize) + (1 - (y / outBedFace.height(texSize))) * headFace.height(bedTexSize));
				int dstx = roundf((outBedFace.x0 * texSize) + (y / outBedFace.width(texSize)) * outBedFace.width(texSize));
				int dsty = roundf((outBedFace.y0 * texSize) + (x / outBedFace.height(texSize)) * outBedFace.height(texSize));

				int srcIdx = srcy * bedTexCache->getWidth() + srcx;
				int dstIdx = dsty * texSize + dstx;

				(&workPix[dstIdx])->raw = (&bedPix[srcIdx])->raw;
			}

		return workTex;
	} else if (name == L"bed_feet_top") {

		for (int y = 0; y <= outBedFace.height(texSize); y++)
			for (int x = 0; x <= outBedFace.width(texSize); x++) {
				int srcx = roundf((feetFace.x0 * bedTexSize) + (x / outBedFace.width(texSize)) * feetFace.width(bedTexSize));
				int srcy = roundf((feetFace.y0 * bedTexSize) + (1 - (y / outBedFace.height(texSize))) * feetFace.height(bedTexSize));
				int dstx = roundf((outBedFace.x0 * texSize) + (y / outBedFace.width(texSize)) * outBedFace.width(texSize));
				int dsty = roundf((outBedFace.y0 * texSize) + (x / outBedFace.height(texSize)) * outBedFace.height(texSize));

				int srcIdx = srcy * bedTexCache->getWidth() + srcx;
				int dstIdx = dsty * texSize + dstx;

				(&workPix[dstIdx])->raw = (&bedPix[srcIdx])->raw;
			}

		return workTex;
	}

	return workTex;

}

BufferedImage* AbstractTexturePack::grabFromDefault(pair<wstring, Icon*> item, Pixel* ogAtlas, pair<int, int> ogDimensions) {
	auto preStitched = static_cast<StitchedTexture*>(item.second);

	auto imag = new BufferedImage(texSize, texSize, GL_RGBA);
	auto pixels = reinterpret_cast<Pixel*>(imag->getData());

	int x = preStitched->getU0() * ogDimensions.first;
	int y = preStitched->getV0() * ogDimensions.second;

	for (int j = 0; j < texSize * texSize; j++) {
		int srcx = x + ((j % texSize) / (float)texSize) * 16;
		int srcy = y + ((j / texSize) / (float)texSize) * 16;
		(&pixels[j])->raw = ogAtlas[srcx + (srcy * 256)].raw;
	}

	return imag;
}

BufferedImage* getDefaultAtlas(std::wstring atlasFile) {
	auto defaultPack = Minecraft::GetInstance()->skins->getDefault();
	auto defaultPath = defaultPack->getPath(true);
	auto terrainFile = File(defaultPath + L"res\\" + atlasFile);

	byteArray terrainBuf(terrainFile.length());
	FileInputStream stream(terrainFile);
	stream.read(terrainBuf);

	return new BufferedImage(terrainBuf.data, terrainBuf.length);
}

void AbstractTexturePack::generateStitched(unordered_map<wstring, Icon*> texturesByName) {
	app.DebugPrintf("Generating stitched texture based on map\n");

	BufferedImage *atlas, *srcImg, *defaultAtlas; // I hate that they all need to have the star instead of the type getting it
	Pixel *atlasPixels, *defaultPixels, *texPixels, *src, *dst;

	int colCount, rowCount, resW, resH; // filled with hardcoded texture widths
	if (texturesByName.find(L"sand") != texturesByName.end()) { // terrain.png
		if (hasFile(L"res/terrain.png") || terrainAtlas.get() != nullptr)
			return;

		app.DebugPrintf("Generating terrain.png...\n");

		defaultAtlas = getDefaultAtlas(L"terrain.png");
		defaultPixels = reinterpret_cast<Pixel*>(defaultAtlas->getData());

		colCount = 16;
		rowCount = 32;

		resW = colCount * texSize;
		resH = rowCount * texSize;

		terrainAtlas = std::make_unique<BufferedImage>(resW, resH, 0);
		atlas = terrainAtlas.get();
		atlasPixels = reinterpret_cast<Pixel*>(atlas->getData());
		
		for (auto &i : texturesByName) {
			auto preStitched = static_cast<StitchedTexture*>(i.second);
			
			int x = preStitched->getU0() * resW;
			int y = preStitched->getV0() * resH;
			int width = (preStitched->getU1() * resW) - x;
			int height = (preStitched->getV1() * resH) - y;

			if (i.first.find(L"bed_") == 0)
				srcImg = getBedTex(i.first);
			else
				srcImg = getImageResource(L"assets/minecraft/textures/block/" + i.first + L".png", true, false);
			if (srcImg == nullptr || srcImg->getWidth() < 0 || srcImg->getHeight() < 0){
				app.DebugPrintf("Failed to find %ls in resource pack!\n", i.first.c_str());
				srcImg = grabFromDefault(i, defaultPixels, { defaultAtlas->getWidth(), defaultAtlas->getHeight() });
				//__debugbreak();
				//continue;
			}

			int imgW = srcImg->getWidth();
			int imgH = srcImg->getHeight();
			if (imgW != texSize || imgH != texSize) {
				app.DebugPrintf("Texture %ls is wrong size! skipping\n", i.first.c_str());
				delete srcImg;
				continue;
			}


			texPixels = reinterpret_cast<Pixel*>(srcImg->getData());

			for (int j = 0; j < imgH * imgW; j++) {
				Pixel* src = &texPixels[j];
				Pixel* dst = &atlasPixels[(x + j % imgW) + (resW * (y + j / imgH))];
				// int imgx = j % imgW; // debugging vars
				// int imgy = j / imgH;
				// int dstx = x + imgx;
				// int dsty = y + imgy; 
				
				//unsigned char tmp = src->r; // unblue everything for saving, unsure why they are flipped
				//src->r = src->b;
				//src->b = tmp;

				dst->raw =  src->raw;
			}
			
			delete srcImg;
		}

		/*
		for (int i = 0; i < resH * resW; i++) { // Magenta test
			auto pix = &atlasPixels[i];
			pix->r = 255;
			pix->g = 0;
			pix->b = 255;
			pix->a = 255;
		}*/

		delete defaultAtlas;

		// Uncomment these two lines and the unblue section above if you are debugging autostitching of the atlas! 
		// If you forget to uncomment the unblue section the written atlas will have red and blue swapped!
		//D3DXIMAGE_INFO info = { resW, resH };
		//RenderManager.SaveTextureData("autostitch.png", &info, atlas->getData());
	} else { // items.png?
		if (hasFile(L"res/items.png") || itemAtlas.get() != nullptr)
			return;

		defaultAtlas = getDefaultAtlas(L"items.png");
		defaultPixels = reinterpret_cast<Pixel*>(defaultAtlas->getData());

		colCount = rowCount = 16;

		resW = colCount * texSize;
		resH = rowCount * texSize;

		itemAtlas = std::make_unique<BufferedImage>(resW, resH, 0);
		atlas = itemAtlas.get();
		atlasPixels = reinterpret_cast<Pixel*>(atlas->getData());

		for (auto& i : texturesByName) {
			auto preStitched = static_cast<StitchedTexture*>(i.second);

			int x = preStitched->getU0() * resW;
			int y = preStitched->getV0() * resH;
			int width = (preStitched->getU1() * resW) - x;
			int height = (preStitched->getV1() * resH) - y;

			srcImg = getImageResource(L"assets/minecraft/textures/item/" + i.first + L".png", true, false);
			if (srcImg == nullptr || srcImg->getWidth() < 0 || srcImg->getHeight() < 0) {
				app.DebugPrintf("Failed to find %ls in resource pack!\n", i.first.c_str());
				srcImg = grabFromDefault(i, defaultPixels, { defaultAtlas->getWidth(), defaultAtlas->getHeight() });
				//__debugbreak();
				//continue;
			}

			int imgW = srcImg->getWidth();
			int imgH = srcImg->getHeight();
			if (imgW != texSize || imgH != texSize) {
				app.DebugPrintf("Texture %ls is wrong size! skipping\n", i.first.c_str());
				delete srcImg;
				continue;
			}


			texPixels = reinterpret_cast<Pixel*>(srcImg->getData());

			for (int j = 0; j < imgH * imgW; j++) {
				Pixel* src = &texPixels[j];
				Pixel* dst = &atlasPixels[(x + j % imgW) + (resW * (y + j / imgH))];
				// int imgx = j % imgW; // debugging vars
				// int imgy = j / imgH;
				// int dstx = x + imgx;
				// int dsty = y + imgy; 

				//unsigned char tmp = src->r; // unblue everything for saving, unsure why they are flipped
				//src->r = src->b;
				//src->b = tmp;

				dst->raw = src->raw;
			}

			delete srcImg;
		}

		delete defaultAtlas;

		// Uncomment these two lines and the unblue section above if you are debugging autostitching of the atlas! 
		// If you forget to uncomment the unblue section the written atlas will have red and blue swapped!
		//D3DXIMAGE_INFO info = {resW, resH};
		//RenderManager.SaveTextureData("autostitchitem.png", &info, atlas->getData());
	}
}

BufferedImage *AbstractTexturePack::getImageResource(const wstring& File, bool filenameHasExtension /*= false*/, bool bTitleUpdateTexture /*=false*/, const wstring &drive /*=L""*/)
{
	const char *pchTexture=wstringtofilename(File);
	app.DebugPrintf("AbstractTexturePack::getImageResource - %s, drive is %s\n",pchTexture, wstringtofilename(drive));

	return new BufferedImage(TexturePack::getResource(L"/" + File),filenameHasExtension,bTitleUpdateTexture,drive);
}

void AbstractTexturePack::loadDefaultUI()
{
#ifdef _XBOX
	// load from the .xzp file
	const ULONG_PTR c_ModuleHandle = (ULONG_PTR)GetModuleHandle(NULL);

	// Load new skin
	const DWORD LOCATOR_SIZE = 256; // Use this to allocate space to hold a ResourceLocator string 
	WCHAR szResourceLocator[ LOCATOR_SIZE ];

	swprintf(szResourceLocator, LOCATOR_SIZE,L"section://%X,%ls#%ls",c_ModuleHandle,L"media", L"media/skin_Minecraft.xur");
	
	XuiFreeVisuals(L"");
	app.LoadSkin(szResourceLocator,NULL);//L"TexturePack");
	//CXuiSceneBase::GetInstance()->SetVisualPrefix(L"TexturePack");
	CXuiSceneBase::GetInstance()->SkinChanged(CXuiSceneBase::GetInstance()->m_hObj);
#else
	ui.ReloadSkin();
#endif
}

void AbstractTexturePack::loadColourTable()
{
	loadDefaultColourTable();
	loadDefaultHTMLColourTable();
}

void AbstractTexturePack::loadDefaultColourTable()
{
	// Load the file
#ifdef __PS3__
	// need to check if it's a BD build, so pass in the name
	File coloursFile(AbstractTexturePack::getPath(true,app.GetBootedFromDiscPatch()?"colours.col":NULL).append(L"res/colours.col"));

#else
	File coloursFile(AbstractTexturePack::getPath(true).append(L"res/colours.col"));
#endif


	if(coloursFile.exists())
	{
		DWORD dwLength = coloursFile.length();
		byteArray data(dwLength);

		FileInputStream fis(coloursFile);
		fis.read(data,0,dwLength);
		fis.close();
		if(m_colourTable != NULL) delete m_colourTable;
		m_colourTable = new ColourTable(data.data, dwLength);

		delete [] data.data;
	}
	else
	{
		app.DebugPrintf("Failed to load the default colours table\n");
		app.FatalLoadError();
	}
}

void AbstractTexturePack::loadDefaultHTMLColourTable()
{
#ifdef _XBOX
	// load from the .xzp file
	const ULONG_PTR c_ModuleHandle = (ULONG_PTR)GetModuleHandle(NULL);

	const DWORD LOCATOR_SIZE = 256; // Use this to allocate space to hold a ResourceLocator string 
	WCHAR szResourceLocator[ LOCATOR_SIZE ];

	// Try and load the HTMLColours.col based off the common XML first, before the deprecated xuiscene_colourtable	
	wsprintfW(szResourceLocator,L"section://%X,%s#%s",c_ModuleHandle,L"media", L"media/HTMLColours.col");
	BYTE *data;
	UINT dataLength;
	if(XuiResourceLoadAll(szResourceLocator, &data, &dataLength) == S_OK)
	{
		m_colourTable->loadColoursFromData(data,dataLength);

		XuiFree(data);
	}
	else
	{
		wsprintfW(szResourceLocator,L"section://%X,%s#%s",c_ModuleHandle,L"media", L"media/");
		HXUIOBJ hScene;
		HRESULT hr = XuiSceneCreate(szResourceLocator,L"xuiscene_colourtable.xur", NULL, &hScene);

		if(HRESULT_SUCCEEDED(hr))
		{
			loadHTMLColourTableFromXuiScene(hScene);
		}
	}
#else
	if(app.hasArchiveFile(L"HTMLColours.col"))
	{
		byteArray textColours = app.getArchiveFile(L"HTMLColours.col");
		m_colourTable->loadColoursFromData(textColours.data,textColours.length);

		delete [] textColours.data;
	}
#endif
}

#ifdef _XBOX
void AbstractTexturePack::loadHTMLColourTableFromXuiScene(HXUIOBJ hObj)
{
	HXUIOBJ child;
	HRESULT hr = XuiElementGetFirstChild(hObj, &child);

	while(HRESULT_SUCCEEDED(hr) && child != NULL)
	{
		LPCWSTR childName;
		XuiElementGetId(child,&childName);
		m_colourTable->setColour(childName,XuiTextElementGetText(child));

		//eMinecraftTextColours colourIndex = eTextColor_NONE;
		//for(int i = 0; i < (int)eTextColor_MAX; i++)
		//{
		//	if(wcscmp(HTMLColourTableElements[i],childName)==0)
		//	{
		//		colourIndex = (eMinecraftTextColours)i;
		//		break;
		//	}
		//}

		//LPCWSTR stringValue = XuiTextElementGetText(child);

		//m_htmlColourTable[colourIndex] = XuiTextElementGetText(child);

		hr = XuiElementGetNext(child, &child);
	}
}
#endif

void AbstractTexturePack::loadUI()
{
	loadColourTable();
	
#ifdef _XBOX
	CXuiSceneBase::GetInstance()->SkinChanged(CXuiSceneBase::GetInstance()->m_hObj);
#endif
}

void AbstractTexturePack::unloadUI()
{
	// Do nothing
}

wstring AbstractTexturePack::getXuiRootPath()
{
	const ULONG_PTR c_ModuleHandle = (ULONG_PTR)GetModuleHandle(NULL);

	// Load new skin
	const DWORD LOCATOR_SIZE = 256; // Use this to allocate space to hold a ResourceLocator string 
	WCHAR szResourceLocator[ LOCATOR_SIZE ];

	swprintf(szResourceLocator, LOCATOR_SIZE,L"section://%X,%ls#%ls",c_ModuleHandle,L"media", L"media/");
	return szResourceLocator;
}

PBYTE AbstractTexturePack::getPackIcon(DWORD &dwImageBytes)
{
	if(m_iconSize == 0 || m_iconData == NULL) loadIcon();
	dwImageBytes = m_iconSize;
	return m_iconData;
}

PBYTE AbstractTexturePack::getPackComparison(DWORD &dwImageBytes)
{
	if(m_comparisonSize == 0 || m_comparisonData == NULL) loadComparison();
	
	dwImageBytes = m_comparisonSize;
	return m_comparisonData;
}

unsigned int AbstractTexturePack::getDLCParentPackId()
{
	return 0;
}

unsigned char AbstractTexturePack::getDLCSubPackId()
{
	return 0;
}