#include "koopatlas/hud.h"

void CharToWChar(const char *input, wchar_t *output, int length) { for (int i = 0; i < length; i++) output[i] = input[i]; }

dWMHud_c *dWMHud_c::instance = 0;

dWMHud_c *dWMHud_c::build() {

	void *buffer = AllocFromGameHeap1(sizeof(dWMHud_c));
	dWMHud_c *c = new(buffer) dWMHud_c;


	instance = c;
	return c;
}



dWMHud_c::dWMHud_c() {
	layoutLoaded = false;
}


// TODO: Need to define these in a better way, somehow
#define ANIM_BUTTON_1 0
#define ANIM_BUTTON_2 1
#define ANIM_BOTTOM_SHOW 2
#define ANIM_BOTTOM_HIDE 3
#define ANIM_TOP_SHOW 4
#define ANIM_TOP_HIDE 5

int dWMHud_c::onCreate() {
	if (!layoutLoaded) {
		bool gotFile = layout.loadArc("maphud.arc", false);
		if (!gotFile)
			return false;

		//static const char *brlanNames[3] = {"maphud_hitbutton.brlan", "maphud_in.brlan", "maphud_out.brlan"};
		static const char *brlanNames[5] = {"maphud_hitbutton.brlan", "bottom_in.brlan", "bottom_out.brlan", "top_in.brlan", "top_out.brlan"};
		static const char *groupNames[6] = {"B01_Button", "B02_Button", "A00_Window", "A00_Window", "A01_Window", "A01_Window"};

		bool output = layout.build("maphud.brlyt");

		if (!IsWideScreen()) {
			layout.clippingEnabled = true;
			layout.clipX = 0;
			layout.clipY = 52;
			layout.clipWidth = 640;
			layout.clipHeight = 352;
			layout.layout.rootPane->scale.x = 0.7711f;
			layout.layout.rootPane->scale.y = 0.7711f;
		}

		layout.loadAnimations(brlanNames, 5);
		layout.loadGroups(groupNames, (int[6]){0, 0, 1, 2, 3, 4}, 6);
		layout.disableAllAnimations();
		layout.enableNonLoopAnim(ANIM_BOTTOM_SHOW);

		hidePointBar();
		setWorldText(" ");
		setWorldName();

		if (dScKoopatlas_c::instance->pathManager.mustComplainToMapCreator)
			dWMHud_c::instance->setLevelText("Please Fix Your Missing Entrance. Thanks");

		layoutLoaded = true;
	}

	return true;
}


int dWMHud_c::onDelete() {
	return layout.free();
}


int dWMHud_c::onExecute() {
	updateLives();
	setWorldName();
	checkPointStatus();
	setPointName();

	layout.execAnimations();
	layout.update();

	return true;
}


int dWMHud_c::onDraw() {
	layout.scheduleForDrawing();

	return true;
}


void dWMHud_c::updateLives() {

	static const char *textID[4] = {"M", "L", "B", "Y"};
	static const char *picID[4] = {"P_mariopic", "P_luigipic", "P_toadBlue", "P_toadyellow"};

	for (int i = 0; i < 4; i++) {

		char boxName [13];
		sprintf(boxName, "T_%slifes_01", textID[i]);
		nw4r::lyt::TextBox *box = layout.findTextBoxByName(boxName);
		nw4r::lyt::Pane *pic = layout.findPictureByName(picID[i]);

		if (Player_Active[i] != 0) {
			box->alpha = 0xFF;
			pic->alpha = 0xFF;

			char lives [3];
			sprintf(lives, "%02d", Player_Lives[Player_ID[i]]);
			const char *loaves = lives;
			wchar_t life;

			CharToWChar(loaves, &life, 3);

			box->SetString(&life);
		}
		else {
			box->alpha = 0;
			pic->alpha = 0;
		}
	}
}

void dWMHud_c::setLevelText(const char *str, int length) {
	if (str == 0) {
		setLevelText("--NULL STRING--");
		return;
	}

	if (length == -1) {
		length = strlen(str);
	}

	wchar_t newString[128];

	int i;
	for (i = 0; i < length && i < 128; i++) {
		newString[i] = str[i];
	}
	newString[i] = 0;

	setLevelText(newString, i);
}

void dWMHud_c::setLevelText(const wchar_t *str, int length) {
	if (str == 0) {
		setLevelText("--NULL STRING--");
		return;
	}

	if (length == -1) {
		length = wcslen(str);
	}

	nw4r::lyt::TextBox *box = layout.findTextBoxByName("T_levelname_01");

	nw4r::ut::TextWriter tw;
	tw.font = box->font;
	tw.SetFontSize(box->fontSizeX, box->fontSizeY);
	tw.somethingRelatedToLineHeight = box->lineSpace;
	tw.charSpace = box->charSpace;
	if (box->tagProc != 0)
		tw.tagProcessorMaybe = box->tagProc;

	float width = tw.CalcStringWidth(str, length);
	SpammyReport("Text width: %f\n", width);

	layout.findWindowByName("W_levelname")->size.x = width + 22;
	layout.findPictureByName("P_topleftboxbg")->size.x = width;
	layout.findPictureByName("P_topthinboxbg")->size.x = 597 - width;

	box->SetString(str);
}


void dWMHud_c::setPointName() {
	// figure this out...
	dKPNode_s *node = dScKoopatlas_c::instance->pathManager.currentNode;

	if (node->type == dKPNode_s::LEVEL) {
		dLevelInfo_c *li = &dScKoopatlas_c::instance->levelInfo;
		dLevelInfo_c::entry_s *entry = li->search(node->levelNumber[0] - 1, node->levelNumber[1] - 1);

		setLevelText(li->getNameForLevel(entry));
	} else {
		setLevelText("   ");
		hidePointBar();
	}
}

void dWMHud_c::checkPointStatus() {
	dKPNode_s *node = dScKoopatlas_c::instance->pathManager.currentNode;

	if (node->type == dKPNode_s::LEVEL) {
		SaveBlock *save = GetSaveFile()->GetBlock(-1);
		
		int world = node->levelNumber[0];
		int level = node->levelNumber[1];

		u32 conds = save->GetLevelCondition(world-1, level-1);
		nw4r::lyt::Pane *pic;

		if (conds & COND_COIN1) {
			pic = layout.findPictureByName("P_coin_on_01");
			pic->alpha = 0xFF;

			pic = layout.findPictureByName("P_coin_off_01");
			pic->alpha = 0;
		}
		else {
			pic = layout.findPictureByName("P_coin_on_01");
			pic->alpha = 0;

			pic = layout.findPictureByName("P_coin_off_01");
			pic->alpha = 0xFF;
		}

		if (conds & COND_COIN2) {
			pic = layout.findPictureByName("P_coin_on_02");
			pic->alpha = 0xFF;

			pic = layout.findPictureByName("P_coin_off_02");
			pic->alpha = 0;
		}
		else {
			pic = layout.findPictureByName("P_coin_on_02");
			pic->alpha = 0;

			pic = layout.findPictureByName("P_coin_off_02");
			pic->alpha = 0xFF;
		}

		if (conds & COND_COIN3) {
			pic = layout.findPictureByName("P_coin_on_03");
			pic->alpha = 0xFF;

			pic = layout.findPictureByName("P_coin_off_03");
			pic->alpha = 0;
		}
		else {
			pic = layout.findPictureByName("P_coin_on_03");
			pic->alpha = 0;

			pic = layout.findPictureByName("P_coin_off_03");
			pic->alpha = 0xFF;
		}

		if (conds & COND_NORMAL) {
			pic = layout.findPictureByName("P_normalexitflag");
			pic->alpha = 0xFF;
		}
		else {
			pic = layout.findPictureByName("P_normalexitflag");
			pic->alpha = 0;
		}

		if (conds & COND_SECRET) {
			pic = layout.findPictureByName("P_secretexitflag");
			pic->alpha = 0xFF;
		}
		else {
			pic = layout.findPictureByName("P_secretexitflag");
			pic->alpha = 0;
		}

	} else {
		nw4r::lyt::Pane *pic;

		static const char *picNames[8] = {"P_coin_on_01", "P_coin_off_01", "P_coin_on_02", "P_coin_off_02", 
										  "P_coin_on_03", "P_coin_off_03", "P_normalexitflag", "P_secretexitflag"};

		for (int i = 0; i < 8; i++) {
			pic = layout.findPictureByName(picNames[i]);
			pic->alpha = 0;
		}
	}
}

void dWMHud_c::setWorldText(const char *str, int length) {
	if (str == 0) {
		setWorldText("--NULL STRING--");
		return;
	}

	if (length == -1) {
		length = strlen(str);
	}

	wchar_t newString[128];

	int i;
	for (i = 0; i < length && i < 128; i++) {
		newString[i] = str[i];
	}
	newString[i] = 0;

	setWorldText(newString, i);
}

void dWMHud_c::setWorldText(const wchar_t *str, int length) {
	if (str == 0) {
		setWorldText("--NULL STRING--");
		return;
	}

	if (length == -1) {
		length = wcslen(str);
	}

	nw4r::lyt::TextBox *box = layout.findTextBoxByName("T_area_01");

	nw4r::ut::TextWriter tw;
	tw.font = box->font;
	tw.SetFontSize(box->fontSizeX, box->fontSizeY);
	tw.somethingRelatedToLineHeight = box->lineSpace;
	tw.charSpace = box->charSpace;
	if (box->tagProc != 0)
		tw.tagProcessorMaybe = box->tagProc;

	float width = tw.CalcStringWidth(str, length);
	SpammyReport("Text width: %f\n", width);

	box->SetString(str);
}

void dWMHud_c::setWorldName() {
	// figure this out...
	dKPNode_s *node = dScKoopatlas_c::instance->pathManager.currentNode;

	if (node->type == dKPNode_s::LEVEL) {
		int world = node->levelNumber[0];
		int level = node->levelNumber[1];

		switch (world) {
			case 1:		
				setWorldText("Yoshi's\nIsland");
				break;	

			case 2:
				if (((level > 1) && (level < 5)) || (level == 15) || (level == 33) || (level == 34))
					setWorldText("Soggy\nSewers");
				else
					setWorldText("Rubble\nRuins");
				break;	

			case 3:		
				setWorldText("Mushroom\nPeaks");
				break;	

			case 4:		
				setWorldText("Sakura\nVillage");
				break;	

			case 5:		
				if ((level < 6) || (level == 15) || (level == 33) || (level == 34))
					setWorldText("Freeze Flame\nValley");
				else
					setWorldText("Freeze Flame\nVolcano");
				break;	

			case 6:		
				setWorldText("Ghost\nRealm");
				break;	

			case 7:		
				if (level < 4)
					setWorldText("Sky\nMountain");
				else
					setWorldText("Starry\nSkies");
				break;	

			case 8:		
				if ((level < 6) || (level == 15) || (level == 25) || (level == 33) || (level == 34))
					setWorldText("Koopa\nPlanet");
				else
					setWorldText("Koopa\nCore");
				break;	

			case 9:		
				setWorldText("Bonus\nLand");
				break;	

			case 10:		
				if      ((level < 6)  || (level == 33) || (level == 34))
					setWorldText("Goldwood\nForest");
				else if ((level < 11) || (level == 35) || (level == 36))
					setWorldText("Mini-Mega\nIsland");
				else if ((level < 16) || (level == 37) || (level == 38) || (level == 31) )
					setWorldText("Crystal\nCaves");
				else if ((level < 19) || (level == 39) || (level == 40))
					setWorldText("Bombard\nCliffs");
				else
					setWorldText("Sky\nCity");
				break;	

			default:
				setWorldText("Unknown\nWorld");
				break;
		}
	}
}

void dWMHud_c::showPointBar() {
	dKPNode_s *node = dScKoopatlas_c::instance->pathManager.currentNode;

	if (node->type == dKPNode_s::LEVEL) {
		isPointBarShown = true;

		layout.enableNonLoopAnim(ANIM_TOP_SHOW);
	}
}


void dWMHud_c::hidePointBar() {
	if (isPointBarShown) {
		isPointBarShown = false;

		layout.enableNonLoopAnim(ANIM_TOP_HIDE);
	}
}