/*
This Source Code Form is subject to the
terms of the Mozilla Public License, v.
2.0. If a copy of the MPL was not
distributed with this file, You can
obtain one at
http://mozilla.org/MPL/2.0/.
*/

#include "stdafx.h"
#include "driveruilayer.h"

#include "Globals.h"
#include "application.h"
#include "translation.h"
#include "simulation.h"
#include "Train.h"
#include "AnimModel.h"
#include "renderer.h"

driver_ui::driver_ui()
{

	clear_panels();
	// bind the panels with ui object. maybe not the best place for this but, eh
	add_external_panel(&m_aidpanel);
	add_external_panel(&m_scenariopanel);
	add_external_panel(&m_timetablepanel);
	add_external_panel(&m_debugpanel);
	if (Global.gui_showtranscripts)
		add_external_panel(&m_transcriptspanel);

	add_external_panel(&m_trainingcardpanel);
	add_external_panel(&m_vehiclelist);
	add_external_panel(&m_timepanel);
	add_external_panel(&m_mappanel);
	add_external_panel(&m_logpanel);
	add_external_panel(&m_perfgraphpanel);
	add_external_panel(&m_cameraviewpanel);
	m_logpanel.is_open = false;

	m_aidpanel.title = STR("Driving Aid");

	m_scenariopanel.title = STR("Scenario");
	m_scenariopanel.size_min = {435, 85};
	m_scenariopanel.size_max = {Global.fb_size.x * 0.95f, Global.fb_size.y * 0.95};

	m_timetablepanel.title = STR("%-*.*s    Time: %d:%02d:%02d");
	m_timetablepanel.size_min = {435, 70};
	m_timetablepanel.size_max = {435, Global.fb_size.y * 0.95};

	m_transcriptspanel.title = STR("Transcripts");
	m_transcriptspanel.size_min = {435, 85};
	m_transcriptspanel.size_max = {Global.fb_size.x * 0.95, Global.fb_size.y * 0.95};

	if (Global.gui_defaultwindows)
	{
		m_aidpanel.is_open = true;
		m_scenariopanel.is_open = true;
	}

	if (Global.gui_trainingdefault)
	{
		m_mappanel.is_open = true;
		m_trainingcardpanel.is_open = true;
		m_vehiclelist.is_open = true;
	}
}

void driver_ui::render_menu_contents()
{
	ui_layer::render_menu_contents();

	if (ImGui::BeginMenu(STR_C("Mode windows")))
	{
		ImGui::MenuItem(m_aidpanel.title.c_str(), "F1", &m_aidpanel.is_open);
		ImGui::MenuItem(m_scenariopanel.title.c_str(), "F1", &m_aidpanel.is_open);
		ImGui::MenuItem(STR_C("Timetable"), "F2", &m_timetablepanel.is_open);
		ImGui::MenuItem(m_debugpanel.name().c_str(), "F12", &m_debugpanel.is_open);
		ImGui::MenuItem(m_mappanel.name().c_str(), "Tab", &m_mappanel.is_open);
		ImGui::MenuItem(m_vehiclelist.name().c_str(), nullptr, &m_vehiclelist.is_open);
		ImGui::MenuItem(m_trainingcardpanel.name().c_str(), nullptr, &m_trainingcardpanel.is_open);
		ImGui::MenuItem(m_cameraviewpanel.name().c_str(), nullptr, &m_cameraviewpanel.is_open);
		if (DebugModeFlag)
			ImGui::MenuItem(m_perfgraphpanel.name().c_str(), nullptr, &m_perfgraphpanel.is_open);

		if (ImGui::MenuItem(m_timepanel.name().c_str()))
			m_timepanel.open();

		ImGui::EndMenu();
	}
}

void driver_ui::showDebugUI()
{
	m_debugpanel.is_open = !m_debugpanel.is_open;
}

// potentially processes provided input key. returns: true if key was processed, false otherwise
bool driver_ui::on_key(int const Key, int const Action)
{
	if (ui_layer::on_key(Key, Action))
		return true;

	switch (Key)
	{
	case GLFW_KEY_TAB:
	case GLFW_KEY_F1:
	case GLFW_KEY_F2:
	case GLFW_KEY_F3:
	case GLFW_KEY_F10:
	case GLFW_KEY_F12:
	{ // ui mode selectors

		if ((true == Global.ctrlState) || (true == Global.shiftState))
		{
			// only react to keys without modifiers
			return false;
		}

		if (Action != GLFW_PRESS)
		{
			return true;
		} // recognized, but ignored
	}

	default:
	{ // everything else
		break;
	}
	}

	switch (Key)
	{

	case GLFW_KEY_TAB:
	{
		m_mappanel.is_open = !m_mappanel.is_open;

		return true;
	}

	case GLFW_KEY_F1:
	{
		// basic consist info
		auto state = ((m_aidpanel.is_open == false) ? 0 : (m_aidpanel.is_expanded == false) ? 1 : 2);
		state = clamp_circular(++state, 3);

		m_aidpanel.is_open = (state > 0);
		m_aidpanel.is_expanded = (state > 1);

		return true;
	}

	case GLFW_KEY_F2:
	{
		// timetable
		auto state = ((m_timetablepanel.is_open == false) ? 0 : (m_timetablepanel.is_expanded == false) ? 1 : 2);
		state = clamp_circular(++state, 3);

		m_timetablepanel.is_open = (state > 0);
		m_timetablepanel.is_expanded = (state > 1);

		return true;
	}

	case GLFW_KEY_F3:
	{
		// debug panel
		m_scenariopanel.is_open = !m_scenariopanel.is_open;
		return true;
	}

	case GLFW_KEY_F12:
	{
		// debug panel
		if (Global.shiftState)
		{
			m_debugpanel.is_open = !m_debugpanel.is_open;
			return true;
		}
	}

	default:
	{
		break;
	}
	}

	return false;
}

// potentially processes provided mouse movement. returns: true if the input was processed, false otherwise
bool driver_ui::on_cursor_pos(double const Horizontal, double const Vertical)
{
	// intercept mouse movement when the pause window is on
	return m_paused;
}

// potentially processes provided mouse button. returns: true if the input was processed, false otherwise
bool driver_ui::on_mouse_button(int const Button, int const Action)
{
	// intercept mouse movement when the pause window is on
	return m_paused;
}

// updates state of UI elements
void driver_ui::update()
{

	auto const pausemask{1 | 2};
	auto ispaused{(false == DebugModeFlag) && ((Global.iPause & pausemask) != 0)};
	if ((ispaused != m_paused) && (false == Global.ControlPicking))
	{
		set_cursor(ispaused);
	}
	m_paused = ispaused;

	ui_layer::update();
}

void driver_ui::set_cursor(bool const Visible)
{

	if (Visible)
	{
		Application.set_cursor(GLFW_CURSOR_NORMAL);
		Application.set_cursor_pos(Global.window_size.x / 2, Global.window_size.y / 2);
	}
	else
	{
		Application.set_cursor(GLFW_CURSOR_DISABLED);
		Application.set_cursor_pos(0, 0);
	}
}

// render() subclass details
void driver_ui::render_()
{
	const std::string *rec_name = m_trainingcardpanel.is_recording();
	if (rec_name && m_cameraviewpanel.set_state(true))
	{
		m_cameraviewpanel.rec_name = *rec_name;
		m_cameraviewpanel.is_open = true;
	}
	else if (!rec_name)
		m_cameraviewpanel.set_state(false);

	// pause/quit modal
	auto const popupheader{STR_C("Simulation Paused")};

	ImGui::SetNextWindowSize(ImVec2(-1, -1));
	if (ImGui::BeginPopupModal(popupheader, nullptr, 0))
	{
		if ((ImGui::Button(STR_C("Resume"), ImVec2(150, 0))) || (ImGui::IsKeyReleased(ImGui::GetKeyIndex(ImGuiKey_Escape))))
		{
			m_relay.post(user_command::pausetoggle, 0.0, 0.0, GLFW_RELEASE, 0);
		}
		if (ImGui::Button(STR_C("Quit"), ImVec2(150, 0)))
		{
			Application.queue_quit(false);
		}
		if (!m_paused)
			ImGui::CloseCurrentPopup();
		ImGui::EndPopup();
	}
	else if (m_paused)
	{
		ImGui::OpenPopup(popupheader);
	}

	if (Global.desync != 0.0f)
	{
		ImGui::SetNextWindowSize(ImVec2(-1, -1));
		if (ImGui::Begin("network", nullptr, ImGuiWindowFlags_NoCollapse))
			ImGui::Text("desync: %0.2f", Global.desync);
		ImGui::End();
	}
}