/*****************************************************************************
* Copyright (c) 2014 Ted John, Duncan Frost
* OpenRCT2, an open source clone of Roller Coaster Tycoon 2.
*
* This file is part of OpenRCT2.
*
* OpenRCT2 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 3 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, see <http://www.gnu.org/licenses/>.
*****************************************************************************/

#include "addresses.h"
#include "game.h"
#include "peep.h"
#include "string_ids.h"
#include "sprite.h"
#include "sprites.h"
#include "viewport.h"
#include "widget.h"
#include "window.h"
#include "window_dropdown.h"

enum WINDOW_STAFF_PEEP_PAGE {
	WINDOW_STAFF_PEEP_OVERVIEW,
	WINDOW_STAFF_PEEP_OPTIONS,
	WINDOW_STAFF_PEEP_STATISTICS,
};

enum WINDOW_STAFF_PEEP_WIDGET_IDX {
	WIDX_BACKGROUND,
	WIDX_TITLE,
	WIDX_CLOSE,
	WIDX_RESIZE,
	WIDX_TAB_1,
	WIDX_TAB_2,
	WIDX_TAB_3,
	WIDX_TAB_4,
	WIDX_VIEWPORT,
	WIDX_BTM_LABEL,
	WIDX_PICKUP,
	WIDX_PATROL,
	WIDX_RENAME,
	WIDX_LOCATE,
	WIDX_FIRE
};

void window_staff_peep_emptysub(){};

rct_widget window_staff_peep_overview_widgets[] = { 
	{ WWT_FRAME,	0, 0,	189,	0,		179,	0x0FFFFFFFF,	STR_NONE },				// Panel / Background
	{ WWT_CAPTION,	0, 1,	188,	1,		14,		0x361,			STR_WINDOW_TITLE_TIP }, // Title
	{ WWT_CLOSEBOX, 0, 177, 187,	2,		13,		0x338,			STR_CLOSE_WINDOW_TIP }, // Close x button
	{ WWT_RESIZE,	1, 0,	189,	43,		179,	0x0FFFFFFFF,	STR_NONE },				// Resize
	{ WWT_TAB,		1, 3,	33,		17,		43,		0x2000144E,		1939 },					// Tab 1
	{ WWT_TAB,		1, 34,	64,		17,		43,		0x2000144E,		1945},					// Tab 2
	{ WWT_TAB,		1, 65,	95,		17,		43,		0x2000144E,		2348},					// Tab 3
	{ WWT_TAB,		1, 96,	126,	17,		43,		0x2000144E,		STR_NONE},				// Tab 4
	{ WWT_VIEWPORT,	1, 3,	164,	47,		166,	0x0FFFFFFFF,	STR_NONE},				// Viewport
	{ WWT_12,		1, 3,	164,	167,	177,	0x0FFFFFFFF,	STR_NONE },				// Label at bottom of viewport
	{ WWT_FLATBTN,	1, 165,	188,	45,		68,		0x1436,			1706},					// Pickup Button
	{ WWT_FLATBTN,	1, 165, 188,	69,		92,		0x1437,			1708},					// Patrol Button
	{ WWT_FLATBTN,	1, 165, 188,	93,		116,	0x1430,			1056},					// Rename Button
	{ WWT_FLATBTN,	1, 165, 188,	117,	140,	0x142F,			1027},					// Locate Button
	{ WWT_FLATBTN,	1, 165, 188,	141,	164,	0x142D,			1705},					// Fire Button
	{ WIDGETS_END },
};

rct_widget *window_staff_peep_page_widgets[] = {
	window_staff_peep_overview_widgets,
	(rct_widget *)0x9AF910,
	(rct_widget *)0x9AF9F4
};

void window_staff_peep_set_page(rct_window* w, int page);
void window_staff_peep_disable_widgets(rct_window* w);

void window_staff_peep_overview_close();
void window_staff_peep_overview_mouseup();
void window_staff_peep_overview_resize();
void window_staff_peep_overview_mousedown(int widgetIndex, rct_window*w, rct_widget* widget);
void window_staff_peep_overview_dropdown();
void window_staff_peep_overview_update(rct_window *w);

void window_staff_peep_orders_mouseup();

void window_staff_peep_stats_mouseup();
void window_staff_peep_stats_resize();

// 0x992AEC
static void* window_staff_peep_overview_events[] = {
	window_staff_peep_overview_close,
	window_staff_peep_overview_mouseup,
	window_staff_peep_overview_resize,
	window_staff_peep_overview_mousedown,
	window_staff_peep_overview_dropdown,
	window_staff_peep_emptysub,
	window_staff_peep_overview_update,
	window_staff_peep_emptysub,
	window_staff_peep_emptysub,
	(void*)0x6BDFD8,
	(void*)0x6BDFC3,
	window_staff_peep_emptysub,
	window_staff_peep_emptysub,
	(void*)0x6BDFAE,
	window_staff_peep_emptysub,
	window_staff_peep_emptysub,
	window_staff_peep_emptysub,
	window_staff_peep_emptysub,
	window_staff_peep_emptysub,
	(void*)0x6BDFED,
	(void*)0x6BE5FC,
	window_staff_peep_emptysub,
	window_staff_peep_emptysub,
	window_staff_peep_emptysub,
	window_staff_peep_emptysub,
	(void*)0x6BDD91, //Invalidate
	(void*)0x6BDEAF, //Paint
	window_staff_peep_emptysub
};

// 0x992B5C
static void* window_staff_peep_orders_events[] = {
	window_staff_peep_emptysub,
	window_staff_peep_orders_mouseup,
	window_staff_peep_stats_resize,
	(void*)0x6BE802,
	(void*)0x6BE809,
	(void*)0x6BE9DA,
	(void*)0x6BE960, // update
	window_staff_peep_emptysub,
	window_staff_peep_emptysub,
	window_staff_peep_emptysub,
	window_staff_peep_emptysub,
	window_staff_peep_emptysub,
	window_staff_peep_emptysub,
	window_staff_peep_emptysub,
	window_staff_peep_emptysub,
	window_staff_peep_emptysub,
	window_staff_peep_emptysub,
	window_staff_peep_emptysub,
	window_staff_peep_emptysub,
	window_staff_peep_emptysub,
	window_staff_peep_emptysub,
	window_staff_peep_emptysub,
	window_staff_peep_emptysub,
	window_staff_peep_emptysub,
	window_staff_peep_emptysub,
	(void*)0x6BE62B, //Invalidate
	(void*)0x6BE7C6, //Paint
	window_staff_peep_emptysub
};

// 0x992BCC
static void* window_staff_peep_stats_events[] = {
	window_staff_peep_emptysub,
	window_staff_peep_stats_mouseup,
	window_staff_peep_stats_resize,
	window_staff_peep_emptysub,
	window_staff_peep_emptysub,
	(void*)0x6BEC80,
	(void*)0x6BEBEA, // update
	window_staff_peep_emptysub,
	window_staff_peep_emptysub,
	window_staff_peep_emptysub,
	window_staff_peep_emptysub,
	window_staff_peep_emptysub,
	window_staff_peep_emptysub,
	window_staff_peep_emptysub,
	window_staff_peep_emptysub,
	window_staff_peep_emptysub,
	window_staff_peep_emptysub,
	window_staff_peep_emptysub,
	window_staff_peep_emptysub,
	window_staff_peep_emptysub,
	window_staff_peep_emptysub,
	window_staff_peep_emptysub,
	window_staff_peep_emptysub,
	window_staff_peep_emptysub,
	window_staff_peep_emptysub,
	(void*)0x6BE9E9, //Invalidate
	(void*)0x6BEA86, //Paint
	window_staff_peep_emptysub
};

void* window_staff_peep_page_events[] = {
	window_staff_peep_overview_events,
	window_staff_peep_orders_events,
	window_staff_peep_stats_events
};

uint32 window_staff_peep_page_enabled_widgets[] = {
	(1 << WIDX_CLOSE) |
	(1 << WIDX_TAB_1) |
	(1 << WIDX_TAB_2) |
	(1 << WIDX_TAB_3) |
	(1 << WIDX_PICKUP) |
	(1 << WIDX_PATROL) |
	(1 << WIDX_RENAME) |
	(1 << WIDX_LOCATE) |
	(1 << WIDX_FIRE),

	(1 << WIDX_CLOSE) |
	(1 << WIDX_TAB_1) |
	(1 << WIDX_TAB_2) |
	(1 << WIDX_TAB_3),

	(1 << WIDX_CLOSE) |
	(1 << WIDX_TAB_1) |
	(1 << WIDX_TAB_2) |
	(1 << WIDX_TAB_3)
};

/**
*
*  rct2: 0x006BEE98
*/
void window_staff_peep_open(rct_peep* peep)
{
	rct_window* w = window_bring_to_front_by_id(WC_PEEP, peep->sprite_index);
	if (w == NULL) {
		w = window_create_auto_pos(190, 180, (uint32*)window_staff_peep_overview_events, WC_PEEP, (uint16)0x400);

		w->widgets = RCT2_GLOBAL(0x9AF81C, rct_widget*);
		w->enabled_widgets = RCT2_GLOBAL(0x9929B0, uint32);
		w->number = peep->sprite_index;
		w->page = 0;
		w->viewport_focus_coordinates.y = 0;
		w->frame_no = 0;

		RCT2_GLOBAL((int*)w + 0x496, uint16) = 0; // missing, var_494 should perhaps be uint16?

		window_staff_peep_disable_widgets(w);

		w->min_width = 190;
		w->min_height = 180;
		w->max_width = 500;
		w->max_height = 450;

		w->flags = 1 << 8;

		w->colours[0] = 1;
		w->colours[1] = 4;
		w->colours[2] = 4;
	}
	w->page = 0;
	window_invalidate(w);

	w->widgets = window_staff_peep_overview_widgets;
	w->enabled_widgets = window_staff_peep_page_enabled_widgets[0];
	w->var_020 = RCT2_GLOBAL(0x9929BC, uint32);
	w->event_handlers = window_staff_peep_page_events[0];
	w->pressed_widgets = 0;
	window_staff_peep_disable_widgets(w);
	window_init_scroll_widgets(w);
	RCT2_CALLPROC_X(0x006BEDA3, 0, 0, 0, 0, (int)w, 0, 0);
	if (g_sprite_list[w->number].peep.state == PEEP_STATE_PICKED) {
		RCT2_CALLPROC_X(w->event_handlers[WE_MOUSE_UP], 0, 0, 0, 10, (int)w, 0, 0);
	}
}

/**
* rct2: 0x006BED21
* Disable the staff pickup if not in pickup state.
*/
void window_staff_peep_disable_widgets(rct_window* w)
{
	rct_peep* peep = &g_sprite_list[w->number].peep;

	int eax = 0 | 0x80;

	if (peep->staff_type == 2) {
		eax |= 0x20;
	}

	//RCT2_CALLFUNC_X(0x698827, 0, 0, 0, 0, 0, 0, 0);
	// sub_698827
	// This is here due to needing the Carry Flag.

	int CF = 0;
	int res = RCT2_GLOBAL(0x982004 + peep->state, uint8) & 1;

	if (res == 0) {
		CF = 1;
	}
	else {
		eax = eax & eax;
	}

	// end sub_698827

	int a = 0;

	// pop esi
	if (CF == 1 && w->page == 0) {
		eax |= 0x400; //or      eax, 400h

		a = w->disabled_widgets & (1 << 0xA); //bt      dword ptr[esi + 10h], 0Ah

	}

	if (a == 0) {
		CF = w->disabled_widgets & (1 << 0xA); //bt      dword ptr [esi+10h], 0Ah
		if (CF == 1) {
			window_invalidate(w);
		}
	}

	w->disabled_widgets = eax;
}

/**
 * Same as window_peep_close.
 * rct2: 0x006BDFF8
 */
void window_staff_peep_overview_close()
{
	rct_window* w;

	window_get_register(w);

	if (RCT2_GLOBAL(0x9DE518, uint32) & (1 << 3)){
		if (w->classification == RCT2_GLOBAL(RCT2_ADDRESS_TOOL_WINDOWCLASS, rct_windowclass) &&
			w->number == RCT2_GLOBAL(RCT2_ADDRESS_TOOL_WINDOWNUMBER, rct_windownumber))
			tool_cancel();
	}
}

/** rct2: 0x6C0A77 */
void window_staff_peep_fire(rct_window* w)
{
	// Check if the confirm window already exists.
	if (window_bring_to_front_by_id(0x1A, w->number)) {
		return;
	}

	// Find center of the screen.
	int screen_height = RCT2_GLOBAL(RCT2_ADDRESS_SCREEN_HEIGHT, sint16);
	int screen_width = RCT2_GLOBAL(RCT2_ADDRESS_SCREEN_WIDTH, sint16);
	int x = screen_width/2 - 100;
	int y = screen_height/2 - 50;

	rct_window* window_prompt = window_create(x, y, 200, 100, (uint32*)0x992C3C, 0x1A, 0);
	window_prompt->widgets = (rct_widget*)0x9AFB4C;
	window_prompt->enabled_widgets |= 0x4;
	window_prompt->enabled_widgets |= 0x8;
	window_prompt->enabled_widgets |= 0x10;

	window_init_scroll_widgets(window_prompt);

	window_prompt->flags |= 0x10;
	window_prompt->number = w->number;
	window_prompt->colours[0] = 0x9A;
}

/**
 * Mostly similar to window_peep_set_page.
 * rct2: 0x006BE023
 */
void window_staff_peep_set_page(rct_window* w, int page)
{
	if (RCT2_GLOBAL(0x9DE518,uint32) & (1 << 3))
	{
		if(w->number == RCT2_GLOBAL(RCT2_ADDRESS_TOOL_WINDOWNUMBER, rct_windownumber) &&
		   w->classification == RCT2_GLOBAL(RCT2_ADDRESS_TOOL_WINDOWCLASS, rct_windowclass))
			tool_cancel();
	
	}
	
	int listen = 0;
	if (page == WINDOW_STAFF_PEEP_OVERVIEW && w->page == WINDOW_STAFF_PEEP_OVERVIEW && w->viewport){
		if (!(w->viewport->flags & VIEWPORT_FLAG_SOUND_ON))
			listen = 1;
	}

	
	w->page = page;
	w->frame_no = 0;

	rct_viewport* viewport = w->viewport;
	w->viewport = 0;
	if (viewport){
		viewport->width = 0;
	}

	w->enabled_widgets = window_staff_peep_page_enabled_widgets[page];
	w->var_020 = RCT2_ADDRESS(0x9929BC, uint32)[page];
	w->event_handlers = window_staff_peep_page_events[page];
	w->pressed_widgets = 0;
	w->widgets = window_staff_peep_page_widgets[page];

	window_staff_peep_disable_widgets(w);
	window_invalidate(w);

	RCT2_CALLPROC_X(w->event_handlers[WE_RESIZE], 0, 0, 0, 0, (int)w, 0, 0);
	RCT2_CALLPROC_X(w->event_handlers[WE_INVALIDATE], 0, 0, 0, 0, (int)w, 0, 0);

	window_init_scroll_widgets(w);
	window_invalidate(w);

	if (listen && w->viewport) w->viewport->flags |= VIEWPORT_FLAG_SOUND_ON;
}

/** rct2: 0x006BDF55 */
void window_staff_peep_overview_mouseup()
{
	short widgetIndex;
	rct_window* w;
	window_widget_get_registers(w, widgetIndex);
	rct_peep* peep = GET_PEEP(w->number);

	switch (widgetIndex) {
	case WIDX_CLOSE:
		window_close(w);
		break;
	case WIDX_TAB_1:
	case WIDX_TAB_2:
	case WIDX_TAB_3:
		window_staff_peep_set_page(w, widgetIndex - WIDX_TAB_1);
		break;
	case WIDX_LOCATE: // 0xD
		window_scroll_to_viewport(w);
		break;
	case WIDX_PICKUP: // 0xA
		// 0x6BE236
		if (tool_set(w, widgetIndex, 7)) {
			return;
		}

		w->var_48C = peep->sprite_identifier;

		RCT2_CALLPROC_X(0x0069A512, 0, 0, 0, 0, (int)peep, 0, 0);
		RCT2_CALLPROC_X(0x006EC473, 0, 0, 0, 0, (int)peep, 0, 0);

		RCT2_CALLPROC_X(0x0069E9D3, 0x8000, 0, peep->y, peep->z, (int)peep, 0, 0);
		RCT2_CALLPROC_X(0x0069A409, 0, 0, 0, 0, (int)peep, 0, 0);
		peep->state = 9;
		RCT2_CALLPROC_X(0x0069A42F, 0, 0, 0, 0, (int)peep, 0, 0);
		break;
	case WIDX_FIRE: // 0xE
		window_staff_peep_fire(w);
		break;
	case WIDX_RENAME: // 0xC
		// 6BE4BC
		window_show_textinput(w, (int)widgetIndex, 0xBA1, 0xBA2, peep->name_string_idx);
		break;
	}
}

/** rct2: 0x006BE558 */
void window_staff_peep_overview_resize()
{
	rct_window* w;
	window_get_register(w);

	window_staff_peep_disable_widgets(w);

	w->min_width = 190;
	w->max_width = 500;
	w->min_height = 180;
	w->max_height = 450;

	if (w->width < w->min_width) {
		w->width = w->min_width;
		window_invalidate(w);
	}

	if (w->width > w->max_width) {
		window_invalidate(w);
		w->width = w->max_width;
	}

	if (w->height < w->min_height) {
		w->height = w->min_height;
		window_invalidate(w);
	}

	if (w->height > w->max_height) {
		window_invalidate(w);
		w->height = w->max_height;
	}

	rct_viewport* viewport = w->viewport;

	if (viewport) {
		int new_width = w->width - 30;
		int new_height = w->height - 62;

		// Update the viewport size
		if (viewport->width != new_width || viewport->height != new_height) {
			viewport->width = new_width;
			viewport->height = new_height;
			viewport->view_width = new_width << viewport->zoom;
			viewport->view_height = new_height << viewport->zoom;
		}
	}

	RCT2_CALLPROC_X(0x006BEDA3, 0, 0, 0, 0, (int)w, 0, 0);
}

/** 
 * Handle the dropdown of patrol button.
 * rct2: 0x006BDF98
 */
void window_staff_peep_overview_mousedown(int widgetIndex, rct_window* w, rct_widget* widget)
{
	if (widgetIndex != WIDX_PATROL) {
		return;
	}

	// Dropdown names
	gDropdownItemsFormat[0] = 0xD75;
	gDropdownItemsFormat[1] = 0xD76;

	int x = widget->left + w->x;
	int y = widget->top + w->y;;
	int extray = widget->bottom - widget->top + 1;
	window_dropdown_show_text(x, y, extray, w->colours[1], 0, 2);
	RCT2_GLOBAL(0x009DEBA2, sint16) = 0;

	rct_peep* peep = GET_PEEP(w->number);

	// Disable clear patrol area if no area is set.
	if (!(RCT2_ADDRESS(0x013CA672, uint8)[peep->var_C5] & 2)) {
		RCT2_GLOBAL(0x009DED34, sint32) |= 1 << 1;
	}
}

/** rct2: 0x006BDFA3 */
void window_staff_peep_overview_dropdown()
{
	short widgetIndex, dropdownIndex;
	rct_window* w;

	window_dropdown_get_registers(w, widgetIndex, dropdownIndex);

	if (widgetIndex != WIDX_PATROL) {
		return;
	}

	// Clear patrol
	if (dropdownIndex == 1) {
		rct_peep* peep = GET_PEEP(w->number);
		int edi = peep->var_C5;
		int ebx = edi << 9;

		for (int i = 0; i < 128; i++)
		{
			RCT2_GLOBAL(0x13B0E72 + ebx + i * 4, uint32) = 0;
		}
		RCT2_GLOBAL(0x13CA672 + edi, uint16) &= 0xFD; // bug??

		window_invalidate(w);
		RCT2_CALLPROC_EBPSAFE(0x006C0C3F);
	}
	else {
		if (!tool_set(w, widgetIndex, 22)) {
			show_gridlines();
			RCT2_GLOBAL(0x009DEA50, sint16) = w->number;
			window_invalidate(w);
		}
	}
}

/**
 * Update the animation frame of the tab icon.
 * rct2: 0x6BE602 
 */
void window_staff_peep_overview_update(rct_window* w)
{
	int var_496 = RCT2_GLOBAL((int)w + 0x496, uint16);
	var_496++;
	if (var_496 >= 24) {
		var_496 = 0;
	}
	RCT2_GLOBAL((int)w + 0x496, uint16) = var_496;
	window_invalidate_by_id(0x497, w->number);
}

/** rct2: 0x006BE814 */
void window_staff_peep_set_order(rct_window* w, int order_id)
{
	int eax = 1 << order_id;

	rct_peep* peep = GET_PEEP(w->number);

	int ax = peep->var_C6 ^ eax;
	int flags = (ax << 8) | 1;

	game_do_command(peep->x, flags, peep->y, w->number, GAME_COMMAND_SET_STAFF_ORDER, (int)peep, 0);
}

/** rct2: 0x006BE7DB */
void window_staff_peep_orders_mouseup()
{
	short widgetIndex;
	rct_window* w;
	window_widget_get_registers(w, widgetIndex);

	switch (widgetIndex) {
	case WIDX_CLOSE:
		window_close(w);
		break;
	case WIDX_TAB_1:
	case WIDX_TAB_2:
	case WIDX_TAB_3:
		window_staff_peep_set_page(w, widgetIndex - WIDX_TAB_1);
		break;
	case WIDX_VIEWPORT:
	case WIDX_BTM_LABEL:
	case WIDX_PICKUP:
	case WIDX_PATROL:
	case WIDX_RENAME:
		window_staff_peep_set_order(w, widgetIndex - 8);
		break;
	}
}

/** rct2: 0x006BEBCF */
void window_staff_peep_stats_mouseup()
{
	short widgetIndex;
	rct_window* w;
	window_widget_get_registers(w, widgetIndex);

	switch (widgetIndex) {
	case WIDX_CLOSE:
		window_close(w);
		break;
	case WIDX_TAB_1:
	case WIDX_TAB_2:
	case WIDX_TAB_3:
		window_staff_peep_set_page(w, widgetIndex - WIDX_TAB_1);
		break;
	}
}

/** rct2: 0x006BEC1B and rct2: 0x006BE975 */
void window_staff_peep_stats_resize()
{
	rct_window* w;
	window_get_register(w);

	w->min_width = 190;
	w->max_width = 190; 
	w->min_height = 119;
	w->max_height = 119;
	
	if (w->width < w->min_width) {
		w->width = w->min_width;
		window_invalidate(w);
	}

	if (w->width > w->max_width) {
		window_invalidate(w);
		w->width = w->max_width;
	}

	if (w->height < w->min_height) {
		w->height = w->min_height;
		window_invalidate(w);
	}

	if (w->height > w->max_height) {
		window_invalidate(w);
		w->height = w->max_height;
	}
}