//
// Copyright (c) 2009-2010 Mikko Mononen memon@inside.org
//
// This software is provided 'as-is', without any express or implied
// warranty.  In no event will the authors be held liable for any damages
// arising from the use of this software.
// Permission is granted to anyone to use this software for any purpose,
// including commercial applications, and to alter it and redistribute it
// freely, subject to the following restrictions:
// 1. The origin of this software must not be misrepresented; you must not
//    claim that you wrote the original software. If you use this software
//    in a product, an acknowledgment in the product documentation would be
//    appreciated but is not required.
// 2. Altered source versions must be plainly marked as such, and must not be
//    misrepresented as being the original software.
// 3. This notice may not be removed or altered from any source distribution.
//

#include "SDL.h"
#include "SDL_keycode.h"
#include "SDL_opengl.h"

#include <algorithm>
#include <cstdio>
#include <functional>
#include <string>
#include <vector>
#ifdef __APPLE__
#	include <OpenGL/glu.h>
#else
#	include <GL/glu.h>
#endif

#include "InputGeom.h"
#include "Recast.h"
#include "Sample_SoloMesh.h"
#include "Sample_TempObstacles.h"
#include "Sample_TileMesh.h"
#include "TestCase.h"
#include "imguiHelpers.h"

#include <imgui.h>
#include <imgui_impl_opengl2.h>
#include <imgui_impl_sdl2.h>

#ifdef WIN32
#	define snprintf _snprintf
#endif

struct SampleItem
{
	std::string name;
	std::function<Sample*()> create;
};

// Constants
namespace
{
constexpr float UPDATE_TIME = 1.0f / 60.0f;  // update at 60Hz
constexpr float fogColor[4] = {0.32f, 0.31f, 0.30f, 1.0f};

SampleItem g_samples[] = {
	{"Solo Mesh",      []() { return new Sample_SoloMesh(); }     },
	{"Tile Mesh",      []() { return new Sample_TileMesh(); }     },
	{"Temp Obstacles", []() { return new Sample_TempObstacles(); }},
};
constexpr int g_nsamples = sizeof(g_samples) / sizeof(SampleItem);

constexpr ImGuiWindowFlags staticWindowFlags = ImGuiWindowFlags_NoMove
	| ImGuiWindowFlags_NoResize
	| ImGuiWindowFlags_NoSavedSettings
	| ImGuiWindowFlags_NoCollapse;
}

struct AppData
{
	SDL_Window* window;
	SDL_GLContext glContext;

	GLdouble projectionMatrix[16];
	GLint viewport[4];

	// Drawable width vs logical width (important for high-dpi screens)
	int width;
	int height;
	int drawableWidth;
	int drawableHeight;

	// Recast data, samples, and test cases
	BuildContext buildContext;
	InputGeom* inputGeometry = nullptr;
	Sample* sample = nullptr;
	TestCase* testCase = nullptr;

	// Time
	float timeAcc = 0.0f;
	Uint32 prevFrameTime = 0;

	// Input
	int mousePos[2]{0, 0};
	int origMousePos[2]{0, 0};  // Used to compute mouse movement totals across frames.

	// Camera
	float cameraEulers[2]{45, -45};
	float cameraPos[3] = {0, 0, 0};  // world space
	float camr = 1000;
	float origCameraEulers[2] = {0, 0};  // Used to compute rotational changes across frames.
	float scrollZoom = 0;

	// Movement
	float moveFront = 0.0f;
	float moveBack = 0.0f;
	float moveLeft = 0.0f;
	float moveRight = 0.0f;
	float moveUp = 0.0f;
	float moveDown = 0.0f;

	// Input state
	bool isRotatingCamera = false;
	bool movedDuringRotate = false;
	bool mouseOverMenu = false;

	// Raycasts
	float rayStart[3];  // world space
	float rayEnd[3];    // world space

	// UI state
	int sampleIndex = -1;
	std::string meshName = "Choose Mesh...";
	bool showMenu = true;
	bool showLog = false;
	bool showTools = true;
	bool showTestCases = false;
	int logScroll = 0;

	// Files
	std::vector<std::string> files;
	std::string meshesFolder = "Meshes";
	std::string testCasesFolder = "TestCases";

	// Markers
	float markerPosition[3] = {0, 0, 0};
	bool markerPositionSet = false;

	~AppData()
	{
		delete sample;
		sample = nullptr;
		delete inputGeometry;
		inputGeometry = nullptr;
	}

	void resetCamera()
	{
		// Reset camera and fog to match the mesh bounds.
		if (inputGeometry)
		{
			const float* boundsMin = inputGeometry->getNavMeshBoundsMin();
			const float* boundsMax = inputGeometry->getNavMeshBoundsMax();

			camr = sqrtf(rcSqr(boundsMax[0] - boundsMin[0]) + rcSqr(boundsMax[1] - boundsMin[1]) + rcSqr(boundsMax[2] - boundsMin[2])) / 2;
			cameraPos[0] = (boundsMax[0] + boundsMin[0]) / 2 + camr;
			cameraPos[1] = (boundsMax[1] + boundsMin[1]) / 2 + camr;
			cameraPos[2] = (boundsMax[2] + boundsMin[2]) / 2 + camr;
			camr *= 3;
		}
		cameraEulers[0] = 45;
		cameraEulers[1] = -45;
		glFogf(GL_FOG_START, camr * 0.1f);
		glFogf(GL_FOG_END, camr * 1.25f);
	}

	void UpdateWindowSize()
	{
		SDL_GetWindowSize(window, &width, &height);
		SDL_GL_GetDrawableSize(window, &drawableWidth, &drawableHeight);

		glViewport(0, 0, drawableWidth, drawableHeight);
		glGetIntegerv(GL_VIEWPORT, viewport);

		glMatrixMode(GL_PROJECTION);
		glLoadIdentity();
		gluPerspective(50.0f, static_cast<float>(width) / static_cast<float>(height), 1.0f, camr);
		glGetDoublev(GL_PROJECTION_MATRIX, projectionMatrix);
	}

	void UpdateUIScale()
	{
		ImGuiIO& io = ImGui::GetIO();
		io.DisplaySize = ImVec2(width, height);
		io.DisplayFramebufferScale = ImVec2(
			static_cast<float>(drawableWidth) / width,
			static_cast<float>(drawableHeight) / height);
	}
};

int main(int /*argc*/, char** /*argv*/)
{
	AppData app;

	// Init SDL
	if (SDL_Init(SDL_INIT_EVERYTHING) != 0)
	{
		printf("Could not initialize SDL.\nError: %s\n", SDL_GetError());
		return -1;
	}

	// Use OpenGL render driver.
	SDL_SetHint(SDL_HINT_RENDER_DRIVER, "opengl");

	// Enable depth buffer.
	SDL_GL_SetAttribute(SDL_GL_DOUBLEBUFFER, 1);
	SDL_GL_SetAttribute(SDL_GL_DEPTH_SIZE, 24);

	// Set color channel depth.
	SDL_GL_SetAttribute(SDL_GL_RED_SIZE, 8);
	SDL_GL_SetAttribute(SDL_GL_GREEN_SIZE, 8);
	SDL_GL_SetAttribute(SDL_GL_BLUE_SIZE, 8);
	SDL_GL_SetAttribute(SDL_GL_ALPHA_SIZE, 8);

	// 4x MSAA.
	SDL_GL_SetAttribute(SDL_GL_MULTISAMPLEBUFFERS, 1);
	SDL_GL_SetAttribute(SDL_GL_MULTISAMPLESAMPLES, 4);

	// Create the SDL window with OpenGL support
	SDL_DisplayMode displayMode;
	SDL_GetCurrentDisplayMode(0, &displayMode);
	app.window = SDL_CreateWindow(
		"Recast Demo",
		SDL_WINDOWPOS_CENTERED,
		SDL_WINDOWPOS_CENTERED,
		displayMode.w - 80,
		displayMode.h - 80,
		SDL_WINDOW_OPENGL | SDL_WINDOW_RESIZABLE | SDL_WINDOW_ALLOW_HIGHDPI);

	// Create the OpenGL context
	app.glContext = SDL_GL_CreateContext(app.window);
	SDL_GL_MakeCurrent(app.window, app.glContext);

	if (!app.window || !app.glContext)
	{
		printf("Could not initialize SDL opengl\nError: %s\n", SDL_GetError());
		return -1;
	}

	SDL_GL_SetSwapInterval(1);  // Enable vsync
	app.UpdateWindowSize();

	// Setup Dear ImGui context
	IMGUI_CHECKVERSION();
	ImGui::CreateContext();
	ImGui_ImplSDL2_InitForOpenGL(app.window, app.glContext);
	ImGui_ImplOpenGL2_Init();

	ImGuiIO& io = ImGui::GetIO();
	io.Fonts->AddFontFromFileTTF("DroidSans.ttf", 16.0f);  // Size in pixels
	ImGui::PushFont(io.Fonts->Fonts[0]);

	app.UpdateUIScale();

	// Set style
	ImGui::StyleColorsDark();

	app.prevFrameTime = SDL_GetTicks();

	// Set up fog.
	glEnable(GL_FOG);
	glFogi(GL_FOG_MODE, GL_LINEAR);
	glFogf(GL_FOG_START, app.camr * 0.1f);
	glFogf(GL_FOG_END, app.camr * 1.25f);
	glFogfv(GL_FOG_COLOR, fogColor);

	// OpenGL settings
	glEnable(GL_CULL_FACE);
	glDepthFunc(GL_LEQUAL);

	bool done = false;
	while (!done)
	{
		// Handle input events.
		bool processHitTest = false;
		bool processHitTestShift = false;

		// Per frame input
		app.mouseOverMenu = ImGui::IsWindowHovered(ImGuiHoveredFlags_AnyWindow);
		SDL_Event event;
		while (SDL_PollEvent(&event))
		{
			ImGui_ImplSDL2_ProcessEvent(&event);
			switch (event.type)
			{
			case SDL_KEYDOWN:
				// Handle any key presses here.
				switch (event.key.keysym.sym)
				{
				case SDLK_ESCAPE:
					done = true;
					break;
				case SDLK_t:
					app.showTestCases = true;
					app.files.clear();
					FileIO::scanDirectory(app.testCasesFolder, ".txt", app.files);
					break;
				case SDLK_TAB:
					app.showMenu = !app.showMenu;
					break;
				case SDLK_SPACE:
					if (app.sample)
					{
						app.sample->onToggle();
					}
					break;
				case SDLK_1:
					if (app.sample)
					{
						app.sample->singleStep();
					}
					break;
				case SDLK_9:
					if (app.sample && app.inputGeometry)
					{
						BuildSettings settings;
						rcVcopy(settings.navMeshBMin, app.inputGeometry->getNavMeshBoundsMin());
						rcVcopy(settings.navMeshBMax, app.inputGeometry->getNavMeshBoundsMax());
						app.sample->collectSettings(settings);
						app.inputGeometry->saveGeomSet(&settings);
					}
					break;
				}
				break;

			case SDL_MOUSEWHEEL:
				if (!app.mouseOverMenu)
				{
					app.scrollZoom += static_cast<float>(event.wheel.y);
				}
				break;
			case SDL_MOUSEBUTTONDOWN:
				if (event.button.button == SDL_BUTTON_RIGHT && !app.mouseOverMenu)
				{
					// Rotate view
					app.isRotatingCamera = true;
					app.movedDuringRotate = false;
					app.origMousePos[0] = app.mousePos[0];
					app.origMousePos[1] = app.mousePos[1];
					app.origCameraEulers[0] = app.cameraEulers[0];
					app.origCameraEulers[1] = app.cameraEulers[1];
				}
				break;

			case SDL_MOUSEBUTTONUP:
				// Handle mouse clicks here.
				if (event.button.button == SDL_BUTTON_RIGHT)
				{
					app.isRotatingCamera = false;
					if (!app.mouseOverMenu && !app.movedDuringRotate)
					{
						processHitTest = true;
						processHitTestShift = true;
					}
				}
				else if (event.button.button == SDL_BUTTON_LEFT)
				{
					if (!app.mouseOverMenu)
					{
						processHitTest = true;
						processHitTestShift = (SDL_GetModState() & KMOD_SHIFT) ? true : false;
					}
				}
				break;

			case SDL_MOUSEMOTION:
				app.mousePos[0] = event.motion.x;
				app.mousePos[1] = app.height - 1 - event.motion.y;

				if (app.isRotatingCamera)
				{
					int dx = app.mousePos[0] - app.origMousePos[0];
					int dy = app.mousePos[1] - app.origMousePos[1];
					app.cameraEulers[0] = app.origCameraEulers[0] - static_cast<float>(dy) * 0.25f;
					app.cameraEulers[1] = app.origCameraEulers[1] + static_cast<float>(dx) * 0.25f;
					if (dx * dx + dy * dy > 3 * 3)
					{
						app.movedDuringRotate = true;
					}
				}
				break;

			case SDL_WINDOWEVENT:
			{
				if (event.window.event == SDL_WINDOWEVENT_RESIZED)
				{
					app.UpdateWindowSize();
					app.UpdateUIScale();
				}
			}
			break;

			case SDL_QUIT:
				done = true;
				break;

			default:
				break;
			}
		}

		Uint32 time = SDL_GetTicks();
		float dt = static_cast<float>(time - app.prevFrameTime) / 1000.0f;
		app.prevFrameTime = time;

		// Hit test mesh.
		if (processHitTest && app.inputGeometry && app.sample)
		{
			float hitTime;
			bool hit = app.inputGeometry->raycastMesh(app.rayStart, app.rayEnd, hitTime);

			if (hit)
			{
				if (SDL_GetModState() & KMOD_CTRL)
				{
					// Marker
					app.markerPositionSet = true;
					app.markerPosition[0] = app.rayStart[0] + (app.rayEnd[0] - app.rayStart[0]) * hitTime;
					app.markerPosition[1] = app.rayStart[1] + (app.rayEnd[1] - app.rayStart[1]) * hitTime;
					app.markerPosition[2] = app.rayStart[2] + (app.rayEnd[2] - app.rayStart[2]) * hitTime;
				}
				else
				{
					float hitPos[3];
					hitPos[0] = app.rayStart[0] + (app.rayEnd[0] - app.rayStart[0]) * hitTime;
					hitPos[1] = app.rayStart[1] + (app.rayEnd[1] - app.rayStart[1]) * hitTime;
					hitPos[2] = app.rayStart[2] + (app.rayEnd[2] - app.rayStart[2]) * hitTime;
					app.sample->onClick(app.rayStart, hitPos, processHitTestShift);
				}
			}
			else
			{
				if (SDL_GetModState() & KMOD_CTRL)
				{
					// Marker
					app.markerPositionSet = false;
				}
			}
		}

		if (app.sample)
		{
			// Update sample simulation.
			app.timeAcc = rcClamp(app.timeAcc + dt, -1.0f, 1.0f);
			while (app.timeAcc > UPDATE_TIME)
			{
				app.timeAcc -= UPDATE_TIME;
				app.sample->update(UPDATE_TIME);
			}
		}
		else
		{
			app.timeAcc = 0;
		}

		// Clear the screen
		glClearColor(0.3f, 0.3f, 0.32f, 1.0f);
		glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);
		glEnable(GL_BLEND);
		glBlendFunc(GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA);
		glDisable(GL_TEXTURE_2D);
		glEnable(GL_DEPTH_TEST);

		// Compute the modelview matrix.
		glMatrixMode(GL_MODELVIEW);
		glLoadIdentity();
		glRotatef(app.cameraEulers[0], 1, 0, 0);
		glRotatef(app.cameraEulers[1], 0, 1, 0);
		glTranslatef(-app.cameraPos[0], -app.cameraPos[1], -app.cameraPos[2]);
		GLdouble modelviewMatrix[16];
		glGetDoublev(GL_MODELVIEW_MATRIX, modelviewMatrix);

		// Get hit ray position and direction.
		int mouseLogicalX;
		int mouseLogicalY;
		SDL_GetMouseState(&mouseLogicalX, &mouseLogicalY);

		// Scale mouse coordinates in accordance with high-dpi scale
		const float scaleX = static_cast<float>(app.drawableWidth) / app.width;
		const float scaleY = static_cast<float>(app.drawableHeight) / app.height;
		const float mouseX = mouseLogicalX * scaleX;
		const float mouseY = app.drawableHeight - mouseLogicalY * scaleY;  // Flip Y (OpenGL origin is bottom-left)

		GLdouble x, y, z;
		gluUnProject(mouseX, mouseY, 0.0, modelviewMatrix, app.projectionMatrix, app.viewport, &x, &y, &z);
		app.rayStart[0] = static_cast<float>(x);
		app.rayStart[1] = static_cast<float>(y);
		app.rayStart[2] = static_cast<float>(z);

		gluUnProject(mouseX, mouseY, 1.0, modelviewMatrix, app.projectionMatrix, app.viewport, &x, &y, &z);
		app.rayEnd[0] = static_cast<float>(x);
		app.rayEnd[1] = static_cast<float>(y);
		app.rayEnd[2] = static_cast<float>(z);

		// Keyboard movement.
		const Uint8* keystate = SDL_GetKeyboardState(NULL);
		app.moveFront = rcClamp(app.moveFront + dt * 4 * ((keystate[SDL_SCANCODE_W] || keystate[SDL_SCANCODE_UP]) ? 1.0f : -1.0f), 0.0f, 1.0f);
		app.moveLeft = rcClamp(app.moveLeft + dt * 4 * ((keystate[SDL_SCANCODE_A] || keystate[SDL_SCANCODE_LEFT]) ? 1.0f : -1.0f), 0.0f, 1.0f);
		app.moveBack = rcClamp(app.moveBack + dt * 4 * ((keystate[SDL_SCANCODE_S] || keystate[SDL_SCANCODE_DOWN]) ? 1.0f : -1.0f), 0.0f, 1.0f);
		app.moveRight = rcClamp(app.moveRight + dt * 4 * ((keystate[SDL_SCANCODE_D] || keystate[SDL_SCANCODE_RIGHT]) ? 1.0f : -1.0f), 0.0f, 1.0f);
		app.moveUp = rcClamp(app.moveUp + dt * 4 * ((keystate[SDL_SCANCODE_Q] || keystate[SDL_SCANCODE_PAGEUP]) ? 1.0f : -1.0f), 0.0f, 1.0f);
		app.moveDown = rcClamp(app.moveDown + dt * 4 * ((keystate[SDL_SCANCODE_E] || keystate[SDL_SCANCODE_PAGEDOWN]) ? 1.0f : -1.0f), 0.0f, 1.0f);

		const float keybSpeed = (SDL_GetModState() & KMOD_SHIFT) ? 4.0f : 22.0f;
		float moveX = (app.moveRight - app.moveLeft) * keybSpeed * dt;
		float moveY = (app.moveBack - app.moveFront) * keybSpeed * dt + app.scrollZoom * 2.0f;
		app.scrollZoom = 0;

		app.cameraPos[0] += moveX * static_cast<float>(modelviewMatrix[0]);
		app.cameraPos[1] += moveX * static_cast<float>(modelviewMatrix[4]);
		app.cameraPos[2] += moveX * static_cast<float>(modelviewMatrix[8]);

		app.cameraPos[0] += moveY * static_cast<float>(modelviewMatrix[2]);
		app.cameraPos[1] += moveY * static_cast<float>(modelviewMatrix[6]);
		app.cameraPos[2] += moveY * static_cast<float>(modelviewMatrix[10]);

		app.cameraPos[1] += (app.moveUp - app.moveDown) * keybSpeed * dt;

		// Draw the mesh
		glEnable(GL_FOG);
		if (app.sample)
		{
			app.sample->render();
		}
		if (app.testCase)
		{
			app.testCase->handleRender();
		}
		glDisable(GL_FOG);

		ImGui_ImplOpenGL2_NewFrame();
		ImGui_ImplSDL2_NewFrame();
		ImGui::NewFrame();
		// ImGui::ShowDemoWindow();

		if (app.sample)
		{
			app.sample->renderOverlay(app.projectionMatrix, modelviewMatrix, app.viewport);
		}
		if (app.testCase)
		{
			app.testCase->handleRenderOverlay(app.projectionMatrix, modelviewMatrix, app.viewport);
		}

		bool newMeshSelected = false;
		bool newSampleSelected = false;
		if (app.showMenu)
		{
			// Help text.
			DrawScreenspaceText(280.0f, 20.0f, IM_COL32(255, 255, 255, 128), "W/A/S/D: Move  RMB: Rotate");

			constexpr int uiColumnWidth = 250;
			constexpr int uiWindowPadding = 10;
			// Properties window
			{
				ImGui::SetNextWindowPos(ImVec2(static_cast<float>(app.width - uiColumnWidth - uiWindowPadding), uiWindowPadding), ImGuiCond_Always);
				ImGui::SetNextWindowSize(ImVec2(uiColumnWidth, static_cast<float>(app.height - uiWindowPadding * 2)), ImGuiCond_Always);
				ImGui::Begin("Properties", nullptr, staticWindowFlags);

				ImGui::Text("Show");
				ImGui::Checkbox("Build Log", &app.showLog);
				ImGui::Checkbox("Tools Panel", &app.showTools);

				ImGui::SeparatorText("Sample");

				if (ImGui::BeginCombo("##sampleCombo", app.sampleIndex >= 0 ? g_samples[app.sampleIndex].name.c_str() : "Choose Sample...", 0))
				{
					for (int n = 0; n < IM_ARRAYSIZE(g_samples); n++)
					{
						const bool is_selected = (app.sampleIndex == n);
						if (ImGui::Selectable(g_samples[n].name.c_str(), is_selected))
						{
							newSampleSelected = !is_selected;
							app.sampleIndex = n;
						}

						// Set the initial focus when opening the combo (scrolling + keyboard navigation focus)
						if (is_selected)
						{
							ImGui::SetItemDefaultFocus();
						}
					}
					ImGui::EndCombo();
				}

				ImGui::SeparatorText("Input Mesh");

				if (ImGui::BeginCombo("##inputMesh", app.meshName.c_str(), 0))
				{
					app.files.clear();
					FileIO::scanDirectory(app.meshesFolder, ".obj", app.files);
					FileIO::scanDirectory(app.meshesFolder, ".gset", app.files);

					for (const auto& file : app.files)
					{
						const bool is_selected = (app.meshName == file);
						if (ImGui::Selectable(file.c_str(), is_selected) && !is_selected)
						{
							app.meshName = file;
							newMeshSelected = true;
						}

						// Set the initial focus when opening the combo (scrolling + keyboard navigation focus)
						if (is_selected)
						{
							ImGui::SetItemDefaultFocus();
						}
					}
					ImGui::EndCombo();
				}

				if (app.inputGeometry)
				{
					DrawRightAlignedText(
						"Verts: %.1fk  Tris: %.1fk",
						static_cast<float>(app.inputGeometry->mesh.getVertCount()) / 1000.0f,
						static_cast<float>(app.inputGeometry->mesh.getTriCount()) / 1000.0f);
				}

				if (app.sample)
				{
					if (app.inputGeometry)
					{
						app.sample->drawSettingsUI();

						if (ImGui::Button("Build"))
						{
							app.buildContext.resetLog();
							if (!app.sample->build())
							{
								app.showLog = true;
								app.logScroll = 0;
							}
							app.buildContext.dumpLog("Build log %s:", app.meshName.c_str());

							// Clear test.
							delete app.testCase;
							app.testCase = 0;
						}
					}

					ImGui::SeparatorText("Debug Settings");
					app.sample->drawDebugUI();
				}

				ImGui::End();
			}

			// Log
			if (app.showLog && app.showMenu)
			{
				constexpr int logWindowHeight = 200;
				ImGui::SetNextWindowPos(ImVec2(uiColumnWidth + 2 * uiWindowPadding, static_cast<float>(app.height - logWindowHeight - uiWindowPadding)), ImGuiCond_Always);  // Position in screen space
				ImGui::SetNextWindowSize(ImVec2(static_cast<float>(app.width - 2 * uiColumnWidth - 4 * uiWindowPadding), logWindowHeight), ImGuiCond_Always);     // Size of the window
				ImGui::Begin("Log", nullptr, staticWindowFlags);

				for (int i = 0; i < app.buildContext.getLogCount(); ++i)
				{
					ImGui::TextUnformatted(app.buildContext.getLogText(i));
				}

				ImGui::End();
			}

			// Left column tools menu
			if (!app.showTestCases && app.showTools && app.showMenu)
			{
				ImGui::SetNextWindowPos(ImVec2(10, 10), ImGuiCond_Always);  // Position in screen space
				ImGui::SetNextWindowSize(ImVec2(250, static_cast<float>(app.height - 20)), ImGuiCond_Always);     // Size of the window
				ImGui::Begin("Tools", nullptr, staticWindowFlags);

				if (app.sample)
				{
					app.sample->drawToolsUI();
				}

				ImGui::End();
			}
		}

		if (newSampleSelected)
		{
			delete app.sample;
			app.sample = g_samples[app.sampleIndex].create();
			app.sample->buildContext = &app.buildContext;
			if (app.inputGeometry)
			{
				app.sample->onMeshChanged(app.inputGeometry);
				app.resetCamera();
			}
		}

		if (newMeshSelected)
		{
			std::string path = app.meshesFolder + "/" + app.meshName;

			delete app.inputGeometry;
			app.inputGeometry = new InputGeom;
			if (!app.inputGeometry->load(&app.buildContext, path))
			{
				delete app.inputGeometry;
				app.inputGeometry = nullptr;

				// Destroy the sample if it already had geometry loaded, as we've just deleted it!
				if (app.sample && app.sample->inputGeometry)
				{
					delete app.sample;
					app.sample = nullptr;
				}

				app.showLog = true;
				app.logScroll = 0;
				app.buildContext.dumpLog("geom load log %s:", app.meshName.c_str());
			}
			app.resetCamera();
			if (app.sample)
			{
				app.sample->onMeshChanged(app.inputGeometry);
			}
		}

		// Test cases
		if (app.showTestCases)
		{
			ImGui::SetNextWindowPos(ImVec2(static_cast<float>(app.width - 10 - 250 - 10 - 200), static_cast<float>(app.height - 10 - 450)), ImGuiCond_Always);  // Position in screen space
			ImGui::SetNextWindowSize(ImVec2(200, 450), ImGuiCond_Always);
			ImGui::Begin("Test Cases", nullptr, staticWindowFlags);

			static int currentTest = 0;
			int newTest = currentTest;
			if (ImGui::BeginCombo("Choose Test", app.files[0].c_str()))
			{
				for (int i = 0; i < static_cast<int>(app.files.size()); ++i)
				{
					if (ImGui::Selectable(app.files[i].c_str(), currentTest == i))
					{
						newTest = i;
					}

					if (currentTest == i)
					{
						ImGui::SetItemDefaultFocus();  // Sets keyboard focus
					}
				}
				ImGui::EndCombo();
			}

			if (newTest != currentTest)
			{
				currentTest = newTest;

				std::string path = app.testCasesFolder + "/" + app.files[currentTest];

				// Load the test.
				app.testCase = new TestCase{};
				if (!app.testCase->load(path))
				{
					delete app.testCase;
					app.testCase = 0;
				}

				// Create sample
				Sample* newSample = nullptr;
				for (int i = 0; i < g_nsamples; ++i)
				{
					if (g_samples[i].name == app.testCase->sampleName)
					{
						newSample = g_samples[i].create();
						if (newSample)
						{
							app.sampleIndex = i;
						}
					}
				}

				delete app.sample;
				app.sample = newSample;

				if (app.sample)
				{
					app.sample->buildContext = &app.buildContext;
				}

				// Load geom.
				app.meshName = app.testCase->geomFileName;

				path = app.meshesFolder + "/" + app.meshName;

				delete app.inputGeometry;
				app.inputGeometry = new InputGeom{};
				if (!app.inputGeometry->load(&app.buildContext, path))
				{
					delete app.inputGeometry;
					app.inputGeometry = nullptr;

					delete app.sample;
					app.sample = nullptr;

					app.showLog = true;
					app.logScroll = 0;
					app.buildContext.dumpLog("geom load log %s:", app.meshName.c_str());
				}

				if (app.sample)
				{
					if (app.inputGeometry)
					{
						app.sample->onMeshChanged(app.inputGeometry);
					}

					// This will ensure that tile & poly bits are updated in tiled sample.
					app.sample->drawSettingsUI();

					app.buildContext.resetLog();
					if (!app.sample->build())
					{
						app.buildContext.dumpLog("Build log %s:", app.meshName.c_str());
					}
				}

				if (app.inputGeometry || app.sample)
				{
					app.resetCamera();
				}

				// Do the tests.
				if (app.sample)
				{
					app.testCase->doTests(app.sample->navMesh, app.sample->navQuery);
				}
			}

			ImGui::End();
		}

		// Draw Marker
		if (app.markerPositionSet && gluProject(app.markerPosition[0], app.markerPosition[1], app.markerPosition[2], modelviewMatrix, app.projectionMatrix, app.viewport, &x, &y, &z))
		{
			// Draw marker circle
			glLineWidth(5.0f);
			glColor4ub(240, 220, 0, 196);
			glBegin(GL_LINE_LOOP);
			for (int i = 0; i < 20; ++i)
			{
				constexpr float radius = 25.0f;
				const float arc = static_cast<float>(i) / 20.0f * RC_PI * 2;
				const float vertexX = static_cast<float>(x) + cosf(arc) * radius;
				const float vertexY = static_cast<float>(y) + sinf(arc) * radius;
				glVertex2f(vertexX, vertexY);
			}
			glEnd();
			glLineWidth(1.0f);
		}

		ImGui::Render();
		ImGui_ImplOpenGL2_RenderDrawData(ImGui::GetDrawData());
		SDL_GL_SwapWindow(app.window);
	}

	ImGui_ImplOpenGL2_Shutdown();
	ImGui_ImplSDL2_Shutdown();
	ImGui::DestroyContext();

	SDL_GL_DeleteContext(app.glContext);
	SDL_DestroyWindow(app.window);
	SDL_Quit();

	return 0;
}