crown-engine/tools-imgui/level_editor.cpp

/*
 * Copyright (c) 2012-2017 Daniele Bartolini and individual contributors.
 * License: https://github.com/dbartolini/crown/blob/master/LICENSE
 */

#if CROWN_TOOLS

#include <imgui.h>
#include <iconfontheaders/icons_material_design.h>
#include <time.h>

#include "core/containers/vector.h"
#include "core/filesystem/filesystem_disk.h"
#include "core/filesystem/path.h"
#include "core/filesystem/file.h"
#include "core/json/json.h"
#include "core/json/sjson.h"
#include "core/json/json_object.h"
#include "core/network/socket.h"
#include "core/strings/dynamic_string.h"
#include "device/device.h"
#include "device/device_event_queue.h"
#include "device/input_device.h"
#include "device/input_manager.h"
#include "device/input_types.h"
#include "device/device_options.h"
#include "device/log.h"
#include "resource/resource_manager.h"
#include "resource/texture_resource.h"
#include "device/pipeline.h"

#include "imgui_context.h"
#include "tool_api.h"

#include <sys/time.h>

namespace { const crown::log_internal::System LEVEL_EDITOR = { "LevelEditor" }; }

namespace crown
{
static u16 _width = 1280;
static u16 _height = 720;

//-----------------------------------------------------------------------------
struct Inspector
{
	// Inspector
	char _name[1024];
	float _position[3];
	float _rotation[3];
	float _scale[3];
	char _sprite[1024];
	char _material[1024];
	bool _visible;
	char _state_machine[1024];
	bool _open;

	Inspector()
		: _visible(true)
		, _open(true)
	{
		memset(_name, 0, sizeof(_name));
		memset(_sprite, 0, sizeof(_sprite));
		memset(_material, 0, sizeof(_material));
		memset(_position, 0, sizeof(_position));
		memset(_rotation, 0, sizeof(_rotation));
		memset(_scale, 0, sizeof(_scale));
		memset(_state_machine, 0, sizeof(_state_machine));
	}

	void draw()
	{
		if (!_open) return;
		if (ImGui::BeginDock("Inspector", &_open))
		{
			ImGui::SetNextTreeNodeOpen(true);
			if (ImGui::TreeNode("Unit"))
			{
				ImGui::InputText("Name", _name, sizeof(_name));
				ImGui::TreePop();
			}

			ImGui::SetNextTreeNodeOpen(true);
			if (ImGui::TreeNode("Transform"))
			{
				ImGui::InputFloat3("Position", _position, ImGuiInputTextFlags_CharsDecimal);
				ImGui::InputFloat3("Rotation", _rotation, ImGuiInputTextFlags_CharsDecimal);
				ImGui::InputFloat3("Scale", _scale, ImGuiInputTextFlags_CharsDecimal);

				ImGui::TreePop();
			}

			ImGui::SetNextTreeNodeOpen(true);
			if (ImGui::TreeNode("Renderer"))
			{
				ImGui::InputText("Sprite", _sprite, sizeof(_sprite));
				ImGui::InputText("Material", _material, sizeof(_material));
				ImGui::Checkbox("Visible", &_visible);
				ImGui::TreePop();
			}

			ImGui::SetNextTreeNodeOpen(true);
			if (ImGui::TreeNode("Animation"))
			{
				ImGui::InputText("State Machine", _state_machine, sizeof(_state_machine));
				ImGui::TreePop();
			}
		}
		ImGui::EndDock();
	}
};

//-----------------------------------------------------------------------------
struct SceneView
{
	ImVec2 _pos;
	ImVec2 _size;
	ImVec2 _mouse_curr;
	ImVec2 _mouse_last;
	bool _open;

	SceneView()
		: _open(true)
	{
	}

	void draw()
	{
		if (!_open) return;

		ImGui::PushStyleVar(ImGuiStyleVar_WindowPadding, ImVec2(0.0f, 0.0f));
		ImGui::SetNextWindowPos(ImVec2(0, 25));
		if (ImGui::BeginDock("Scene View"
			, &_open
			, ImGuiWindowFlags_NoScrollbar
			| ImGuiWindowFlags_NoScrollWithMouse
			| ImGuiWindowFlags_NoTitleBar))
		{
			uint16_t w, h;
			device()->resolution(w, h);
			bgfx::TextureHandle txh = device()->_pipeline->_buffers[0];
			if (bgfx::isValid(txh))
			{
				ImTextureID tex_id = (void*)(uintptr_t)txh.idx;
				ImGui::Image(tex_id
					, ImVec2(w, h)
#if CROWN_PLATFORM_WINDOWS
					, ImVec2(0, 0)
					, ImVec2(1, 1)
#else
					, ImVec2(0, 1)
					, ImVec2(1, 0)
#endif // CROWN_PLATFORM_WINDOWS
					, ImColor(255,255,255,255)
					, ImColor(255,255,255,128)
				);
			}

			if (ImGui::IsWindowHovered())
			{
				// send all input to engine
				ImGui::CaptureMouseFromApp(false);
				ImGui::CaptureKeyboardFromApp(false);
			}
			else
			{
				// send all input to imgui
				ImGui::CaptureMouseFromApp(true);
				ImGui::CaptureKeyboardFromApp(true);
			}
		}

		_pos = ImGui::GetWindowPos();
		_size = ImGui::GetWindowSize();

		ImGui::EndDock();
		ImGui::PopStyleVar();
	}
};

//-----------------------------------------------------------------------------
struct UnitList
{
	bool _open;

	UnitList()
		: _open(true)
	{
	}

	void draw()
	{
		if (!_open) return;

		if (ImGui::BeginDock("Unit List", &_open))
		{
			ImGui::SetNextTreeNodeOpen(true);
			if (ImGui::TreeNode("Units"))
			{
				if (ImGui::TreeNode("Objects"))
				{
					for (int i = 0; i < 5; i++)
						if (ImGui::TreeNode((void*)(intptr_t)i, "Child %d", i))
						{
							ImGui::Text("blah blah");
							ImGui::SameLine();
							if (ImGui::SmallButton("print")) printf("Child %d pressed", i);
							ImGui::TreePop();
						}
					ImGui::TreePop();
				}

				if (ImGui::TreeNode("Lights"))
				{
					// ShowHelpMarker("This is a more standard looking tree with selectable nodes.\nClick to select, Ctrl+Click to toggle, click on arrows or double-click to open.");
					static bool align_label_with_current_x_position = false;
					ImGui::Checkbox("Align label with current X position)", &align_label_with_current_x_position);
					ImGui::Text("Hello!");
					if (align_label_with_current_x_position)
						ImGui::Unindent(ImGui::GetTreeNodeToLabelSpacing());

					static int selection_mask = (1 << 2); // Dumb representation of what may be user-side selection state. You may carry selection state inside or outside your objects in whatever format you see fit.
					int node_clicked = -1;                // Temporary storage of what node we have clicked to process selection at the end of the loop. May be a pointer to your own node type, etc.
					ImGui::PushStyleVar(ImGuiStyleVar_IndentSpacing, ImGui::GetFontSize()*3); // Increase spacing to differentiate leaves from expanded contents.
					for (int i = 0; i < 6; i++)
					{
						// Disable the default open on single-click behavior and pass in Selected flag according to our selection state.
						ImGuiTreeNodeFlags node_flags = ImGuiTreeNodeFlags_OpenOnArrow | ImGuiTreeNodeFlags_OpenOnDoubleClick | ((selection_mask & (1 << i)) ? ImGuiTreeNodeFlags_Selected : 0);
						if (i < 3)
						{
							// Node
							bool node_open = ImGui::TreeNodeEx((void*)(intptr_t)i, node_flags, "Selectable Node %d", i);
							if (ImGui::IsItemClicked())
								node_clicked = i;
							if (node_open)
							{
								ImGui::Text("Blah blah\nBlah Blah");
								ImGui::TreePop();
							}
						}
						else
						{
							// Leaf: The only reason we have a TreeNode at all is to allow selection of the leaf. Otherwise we can use BulletText() or TreeAdvanceToLabelPos()+Text().
							ImGui::TreeNodeEx((void*)(intptr_t)i, node_flags | ImGuiTreeNodeFlags_Leaf | ImGuiTreeNodeFlags_NoTreePushOnOpen, "Selectable Leaf %d", i);
							if (ImGui::IsItemClicked())
								node_clicked = i;
						}
					}
					if (node_clicked != -1)
					{
						// Update selection state. Process outside of tree loop to avoid visual inconsistencies during the clicking-frame.
						if (ImGui::GetIO().KeyCtrl)
							selection_mask ^= (1 << node_clicked);          // Ctrl+click to toggle
						else //if (!(selection_mask & (1 << node_clicked))) // Depending on selection behavior you want, this commented bit preserve selection when clicking on item that is part of the selection
							selection_mask = (1 << node_clicked);           // Click to single-select
					}
					ImGui::PopStyleVar();
					if (align_label_with_current_x_position)
						ImGui::Indent(ImGui::GetTreeNodeToLabelSpacing());
					ImGui::TreePop();
				}
				ImGui::TreePop();
			}
		}
		ImGui::EndDock();	// End Object List
	}
};

//-----------------------------------------------------------------------------
struct SpriteAnimator
{
	bool _open;
	bool _add_animation_popup_open;

	Array<const char*> _entities;
	s32 _cur_entity;
	TextureResource* _texture;
	u32 _texture_width;
	u32 _texture_height;

	struct Frame
	{
		char	name[512];
		ImVec2	pivot;
		ImVec4	region;
	};
	Array<Frame> _frames;

	char _anim_name[512];
	f32 _anim_time;
	Array<const char*> _listbox_items;
	s32 _listbox_item_current;
	Array<Frame> _anim_preview_frames;

	f32 _delta;
	u32 current;

	Vector<DynamicString> file_list_sprites;

	FilesystemDisk* _fs;

	SpriteAnimator(const DynamicString& src_dir)
		: _open(true)
		, _add_animation_popup_open(false)
		, _entities(default_allocator())
		, _cur_entity(0)
		, _texture(nullptr)
		, _texture_width(0)
		, _texture_height(0)
		, _frames(default_allocator())
		, _anim_time(0.1f)
		, _listbox_items(default_allocator())
		, _listbox_item_current(0)
		, _anim_preview_frames(default_allocator())
		, _delta(0.0f)
		, current(0)
		, file_list_sprites(default_allocator())
	{
		memset(_anim_name, 0, sizeof(_anim_name));

		_fs = CE_NEW(default_allocator(), FilesystemDisk)(default_allocator());
		_fs->set_prefix(src_dir.c_str());

		get_sprites_list();
	}

	ImVec2 pixel_to_uv(uint32_t tex_w, uint32_t tex_h, float x, float y)
	{
		ImVec2 uv;
		uv.x = (float)x / (float)tex_w;
		uv.y = (float)y / (float)tex_h;
		return uv;
	}

	void get_directory(DynamicString& dir, const char* path)
	{
		CE_ENSURE(NULL != path);

		const char* ls = strrchr(path, '/');
		u32 file_len = strlen(ls+1);
		u32 dir_len = strlen(path) - file_len;

		char buff[1024];
		memcpy(buff, path, dir_len);
		buff[dir_len] = '\0';

		path::reduce(dir, buff);
	}

	void save_sprite_animation()
	{
		TempAllocator4096 ta;

		StringStream ss(ta);
		ss << "frames = [ ";
		for (u32 i = 0; i < array::size(_listbox_items); i++)
		{
			ss << _listbox_items[i] + 7 << " ";
		}
		ss << "]\n";
		ss << "total_time = ";
		ss << _anim_time;

		DynamicString dir(ta);
		get_directory(dir, _entities[_cur_entity]);
		DynamicString file_name(ta);
		path::join(file_name, dir.c_str(), _anim_name);
		file_name += ".sprite_animation";
		File* f = _fs->open(file_name.c_str(), FileOpenMode::WRITE);
		f->write(string_stream::c_str(ss), strlen(string_stream::c_str(ss)));
		f->close();
	}

	void get_files_list(Vector<DynamicString>& out, const char* path="")
	{
		TempAllocator4096 ta;
		Vector<DynamicString> files(ta);

		_fs->list_files(path, files);

		for (u32 i = 0; i < vector::size(files); i++)
		{
			DynamicString join(ta);
			path::join(join, path, files[i].c_str());
			join = join.c_str()[0] == '/' ? join.c_str()+1 : join.c_str();

			if (_fs->is_directory(join.c_str()))
			{
				get_files_list(out, join.c_str());
			}
			else
			{
				vector::push_back(out, join);
			}
		}
	}

	void get_sprites_list()
	{
		TempAllocator4096 ta;
		Vector<DynamicString> files(ta);
		get_files_list(files);

		for (DynamicString* f = vector::begin(files); f != vector::end(files); f++)
			if (f->has_suffix(".sprite"))
				array::push_back(_entities, (const char*) strdup(f->c_str()));
	}

	void draw()
	{
		if (!_open) return;

		if (ImGui::BeginDock("Animator", &_open))
		{
			if (_texture)
			{
				Frame f = _frames[0];
				ImVec2 start = pixel_to_uv(_texture_width, _texture_height, f.region.x, f.region.y);
				ImVec2 end = pixel_to_uv(_texture_width, _texture_height, f.region.x+f.region.z, f.region.y+f.region.w);
				ImGui::Image((void*)(uintptr_t) _texture->handle.idx
					, ImVec2(f.region.z, f.region.w)
					, start
					, end
					, ImColor(255, 255, 255, 55)
				);
			}

			if (ImGui::Combo("Entities", &_cur_entity, (const char* const*) array::begin(_entities), array::size(_entities)))
			{
				array::clear(_frames);

				const char* sprite = _entities[_cur_entity];
				u32 sprite_len = strlen(sprite);
				char entity[1024];
				strncpy(entity, sprite, strlen(sprite));
				entity[sprite_len-7] = '\0';	// remove ".sprite"

				ResourceManager* resman = device()->_resource_manager;
				_texture = (TextureResource*)resman->get(RESOURCE_TYPE_TEXTURE, StringId64(entity));

				File* file = _fs->open(sprite, FileOpenMode::READ);
				const u32 size = file->size();
				Buffer buf(default_allocator());
				array::resize(buf, size);
				file->read(array::begin(buf), size);
				_fs->close(*file);

				JsonObject obj(default_allocator());
				JsonArray list(default_allocator());
				sjson::parse(buf, obj);
				sjson::parse_array(obj["frames"], list);
				_texture_width = sjson::parse_int(obj["width"]);
				_texture_height = sjson::parse_int(obj["height"]);
				for (uint32_t i = 0; i < array::size(list); i++)
				{
					JsonObject frame(default_allocator());
					DynamicString name(default_allocator());
					JsonArray pivot(default_allocator());
					JsonArray region(default_allocator());
					sjson::parse_object(list[i], frame);
					sjson::parse_array(frame["pivot"], pivot);
					sjson::parse_array(frame["region"], region);

					Frame f;
					sjson::parse_string(frame["name"], name);
					strncpy(f.name, name.c_str(), name.length());
					f.name[name.length()] = '\0';
					f.pivot.x = sjson::parse_float(pivot[0]);
					f.pivot.y = sjson::parse_float(pivot[1]);
					f.region.x = sjson::parse_float(region[0]);
					f.region.y = sjson::parse_float(region[1]);
					f.region.z = sjson::parse_float(region[2]);
					f.region.w = sjson::parse_float(region[3]);

					array::push_back(_frames, f);
				}
			}

			if (ImGui::Button("Add animation", ImVec2(100, 25)))
			{
				_add_animation_popup_open = true;
			}

			if (_add_animation_popup_open)
			{
				ImGui::OpenPopup("Add animation");
			}

			if (ImGui::BeginPopup("Add animation"))
			{
				ImGui::InputText("Name", _anim_name, sizeof(_anim_name));
				ImGui::InputFloat("Time", &_anim_time, 0.1f, 0.1f, 1);
				ImGui::ListBox("Animation Frames", &_listbox_item_current, (const char* const*)array::begin(_listbox_items), array::size(_listbox_items));
				if (ImGui::Button("Clear Frames", ImVec2(100.0f, 25.0f)))
				{
					array::clear(_listbox_items);
					array::clear(_anim_preview_frames);
					_delta = 0.0f;
					current = 0;
				}

				if (array::size(_anim_preview_frames) > 0)
				{
					_delta += 1.0f/60.0f;
					if (_delta >= _anim_time/array::size(_anim_preview_frames))
					{
						_delta = 0;
						current++;
						if (current >= array::size(_anim_preview_frames))
							current = 0;
					}

					Frame f = _anim_preview_frames[current];
					ImVec2 start = pixel_to_uv(_texture_width, _texture_height, f.region.x, f.region.y);
					ImVec2 end = pixel_to_uv(_texture_width, _texture_height, f.region.x+f.region.z, f.region.y+f.region.w);
					ImGui::Image(
						  (void*)(uintptr_t) _texture->handle.idx
						, ImVec2(f.region.z, f.region.w)
						, start
						, end
						, ImColor(255, 255, 255, 55)
						);
				}

				ImGui::Separator();

				for (uint32_t i = 0; i < array::size(_frames); i++)
				{
					Frame f = _frames[i];
					ImVec2 start = pixel_to_uv(_texture_width, _texture_height, f.region.x, f.region.y);
					ImVec2 end = pixel_to_uv(_texture_width, _texture_height, f.region.x+f.region.z, f.region.y+f.region.w);

					ImGui::SameLine();
					if (i % 9 == 0) ImGui::NewLine();
					ImGui::BeginGroup();
					ImGui::Image(
						  (void*)(uintptr_t) _texture->handle.idx
						, ImVec2(f.region.z, f.region.w)
						, start
						, end
						, ImColor(255, 255, 255, 55)
						);
					ImGui::NewLine();
					if (ImGui::Button(_frames[i].name, ImVec2(100.0f, 25.0f)))
					{
						array::push_back(_listbox_items, (const char*) strdup(_frames[i].name));
						array::push_back(_anim_preview_frames, _frames[i]);
					}
					ImGui::EndGroup();
				}

				if (ImGui::Button("Save", ImVec2(100, 25)))
				{
					save_sprite_animation();
					ImGui::CloseCurrentPopup();
					_add_animation_popup_open = false;
				}
				ImGui::SameLine();
				if (ImGui::Button("Cancel", ImVec2(100, 25)))
				{
					ImGui::CloseCurrentPopup();
					_add_animation_popup_open = false;
				}

				ImGui::EndPopup();
			}
		}
		ImGui::EndDock();
	}
};

//-----------------------------------------------------------------------------
struct LevelEditor
{
	DynamicString _source_dir;

	// FX
	TextureResource* tex_move;
	TextureResource* tex_place;
	TextureResource* tex_rotate;
	TextureResource* tex_scale;
	TextureResource* tex_ref_world;
	TextureResource* tex_ref_local;
	TextureResource* tex_axis_local;
	TextureResource* tex_axis_world;
	TextureResource* tex_snap_grid;

	// State
	float _grid_size;
	int32_t _rotation_snap;
	bool _show_grid;
	bool _snap_to_grid;
	bool _debug_render_world;
	bool _debug_physics_world;
	tool::ToolType::Enum _tool_type;
	tool::SnapMode::Enum _snap_mode;
	tool::ReferenceSystem::Enum _reference_system;

	ImVec2 _main_menu_pos;
	ImVec2 _main_menu_size;

	ImVec2 _toolbar_pos;
	ImVec2 _toolbar_size;

	Console _console;
	Inspector _inspector;
	SceneView _scene_view;
	UnitList _unit_list;
	SpriteAnimator _animator;

	LevelEditor(const DynamicString& source_dir)
		: _source_dir(default_allocator())
		, _grid_size(1.0f)
		, _rotation_snap(15)
		, _show_grid(true)
		, _snap_to_grid(true)
		, _debug_render_world(false)
		, _debug_physics_world(false)
		, _tool_type(tool::ToolType::MOVE)
		, _snap_mode(tool::SnapMode::RELATIVE)
		, _reference_system(tool::ReferenceSystem::LOCAL)

		, _main_menu_pos(0, 0)
		, _main_menu_size(0, 0)

		, _toolbar_pos(0, 0)
		, _toolbar_size(0, 0)
		, _animator(source_dir)
	{
		ResourceManager* resman = device()->_resource_manager;
		tex_move = (TextureResource*)resman->get(RESOURCE_TYPE_TEXTURE, StringId64("core/editors/gui/tool-move"));
		tex_place = (TextureResource*)resman->get(RESOURCE_TYPE_TEXTURE, StringId64("core/editors/gui/tool-place"));
		tex_rotate = (TextureResource*)resman->get(RESOURCE_TYPE_TEXTURE, StringId64("core/editors/gui/tool-rotate"));
		tex_scale = (TextureResource*)resman->get(RESOURCE_TYPE_TEXTURE, StringId64("core/editors/gui/tool-scale"));
		tex_ref_world = (TextureResource*)resman->get(RESOURCE_TYPE_TEXTURE, StringId64("core/editors/gui/reference-world"));
		tex_ref_local = (TextureResource*)resman->get(RESOURCE_TYPE_TEXTURE, StringId64("core/editors/gui/reference-local"));
		tex_axis_local = (TextureResource*)resman->get(RESOURCE_TYPE_TEXTURE, StringId64("core/editors/gui/axis-local"));
		tex_axis_world = (TextureResource*)resman->get(RESOURCE_TYPE_TEXTURE, StringId64("core/editors/gui/axis-world"));
		tex_snap_grid = (TextureResource*)resman->get(RESOURCE_TYPE_TEXTURE, StringId64("core/editors/gui/snap-to-grid"));

		imgui_create();

		ImGui::LoadDock();
	}

	~LevelEditor()
	{
		ImGui::SaveDock();

		imgui_destroy();
	}

	void update(float dt)
	{
		CE_UNUSED(dt);

		static f32 last_w = 0.0f;
		static f32 last_h = 0.0f;
		if (last_w != _scene_view._size.x || last_h != _scene_view._size.y)
		{
			last_w = _scene_view._size.x;
			last_h = _scene_view._size.y;
			device()->_width  = _scene_view._size.x != 0.0f ? _scene_view._size.x : 128.0f;
			device()->_height = _scene_view._size.y != 0.0f ? _scene_view._size.y : 128.0f;
		}

		TempAllocator4096 ta;

		u32 message_count = 0;

		// Receive response from engine
		for (;;)
		{
			uint32_t msg_len = 0;
			ReadResult rr = _console._client.read_nonblock(&msg_len, sizeof(msg_len));

			if (rr.error == ReadResult::WOULDBLOCK)
				break;

			if (ReadResult::SUCCESS == rr.error)
			{
				char msg[8192];
				rr = _console._client.read(msg, msg_len);
				msg[msg_len] = '\0';
				message_count++;
				// logi(LEVEL_EDITOR, "count: %d", message_count);
				if (ReadResult::SUCCESS == rr.error)
				{
					JsonObject obj(ta);
					DynamicString type(ta);
					json::parse(msg, obj);
					json::parse_string(obj["type"], type);

					if (type == "message")
					{
						DynamicString severity(ta);
						DynamicString message(ta);

						json::parse_string(obj["severity"], severity);
						json::parse_string(obj["message"], message);

						LogSeverity::Enum ls = LogSeverity::COUNT;
						if (strcmp("info", severity.c_str()) == 0)
							ls = LogSeverity::LOG_INFO;
						else if (strcmp("warning", severity.c_str()) == 0)
							ls = LogSeverity::LOG_WARN;
						else if (strcmp("error", severity.c_str()) == 0)
							ls = LogSeverity::LOG_ERROR;
						else
							CE_FATAL("Unknown severity");

						ConsoleLog log(ls, message.c_str());
						vector::push_back(_console._console_items, log);
					}
					else
					{
						ConsoleLog log(LogSeverity::LOG_ERROR, "Unknown message type");
						vector::push_back(_console._console_items, log);
					}

					console_scroll_to_bottom();
				}
			}
		}

		imgui_begin_frame(VIEW_IMGUI, _width, _height);

		float offset_y = _main_menu_size.y;
		ImGui::RootDock(ImVec2(0, offset_y), ImVec2(_width, _height-offset_y));

		main_menu_bar();
		// toolbar();
		_scene_view.draw();
		console_draw(_console);
		_unit_list.draw();
		_inspector.draw();
		_animator.draw();

		imgui_end_frame();
	}

	void send_command(StringStream& ss)
	{
		TempAllocator4096 ta;
		StringStream out(ta);
		out << "{\"type\":\"script\",\"script\":\"";
		out << string_stream::c_str(ss);
		out << "\"}";

		const char* cmd = string_stream::c_str(out);
		const uint32_t size = strlen32(cmd);
		_console._client.write(&size, sizeof(uint32_t));
		_console._client.write(cmd, size);
	}

	void tool_send_state()
	{
		TempAllocator256 ta;
		StringStream ss(ta);
		tool::set_grid_size(ss, _grid_size);
		tool::set_rotation_snap(ss, _rotation_snap);
		tool::enable_show_grid(ss, _show_grid);
		tool::enable_snap_to_grid(ss, _snap_to_grid);
		tool::enable_debug_render_world(ss, _debug_render_world);
		tool::enable_debug_physics_world(ss, _debug_physics_world);
		tool::set_tool_type(ss, _tool_type);
		tool::set_snap_mode(ss, _snap_mode);
		tool::set_reference_system(ss, _reference_system);
		send_command(ss);
	}

	void format_time(uint64_t time, char* str, size_t size)
	{
#if CROWN_PLATFORM_WINDOWS
		// SYSTEMTIME st;
		// FileTimeToSystemTime(&time, &st);
		// sprintf(str, "%d/%d/%d %d:%d"
		// 	, st.wDay
		// 	, st.wMonth
		// 	, st.wYear
		// 	, st.wHour
		// 	, st.wMinute);
#else
		struct tm lt;
		localtime_r((time_t*)&time, &lt);
		strftime(str, size, "%d/%m/%Y %H:%M", &lt);
#endif
	}

	void main_menu_bar()
	{
		// Main Menu
		if (ImGui::BeginMainMenuBar())
		{
			if (ImGui::BeginMenu("File"))
			{
				if (ImGui::MenuItem("New"))
				{

				}
				if (ImGui::MenuItem("Open", "Ctrl+O"))
				{

				}
				if (ImGui::MenuItem("Save", "Ctrl+S"))
				{
				}
				if (ImGui::MenuItem("Save As..."))
				{
				}
				ImGui::Separator();
				if (ImGui::MenuItem("Quit", "Ctrl+Q", false, true))
				{
					TempAllocator64 ta;
					StringStream ss(ta);
					tool::device_quit(ss);
					send_command(ss);
				}
				ImGui::EndMenu();
			}
			if (ImGui::BeginMenu("Edit"))
			{
				if (ImGui::MenuItem("Undo", "Ctrl+Z")) {}
				if (ImGui::MenuItem("Redo", "Ctrl+Y")) {}
				ImGui::EndMenu();
			}
			if (ImGui::BeginMenu("Create"))
			{
				TempAllocator256 ta;
				StringStream ss(ta);

				if (ImGui::BeginMenu("Primitives"))
				{
					if (ImGui::MenuItem("Cube", NULL, false, true))
					{
						_tool_type = tool::ToolType::PLACE;
						tool_send_state();
						tool::set_placeable(ss, "unit", "core/units/primitives/cube");
						send_command(ss);
					}
					if (ImGui::MenuItem("Sphere"))
					{
						_tool_type = tool::ToolType::PLACE;
						tool_send_state();
						tool::set_placeable(ss, "unit", "core/units/primitives/sphere");
						send_command(ss);
					}
					if (ImGui::MenuItem("Cone"))
					{
						_tool_type = tool::ToolType::PLACE;
						tool_send_state();
						tool::set_placeable(ss, "unit", "core/units/primitives/cone");
						send_command(ss);
					}
					if (ImGui::MenuItem("Cylinder"))
					{
						_tool_type = tool::ToolType::PLACE;
						tool_send_state();
						tool::set_placeable(ss, "unit", "core/units/primitives/cylinder");
						send_command(ss);
					}
					if (ImGui::MenuItem("Plane"))
					{
						_tool_type = tool::ToolType::PLACE;
						tool_send_state();
						tool::set_placeable(ss, "unit", "core/units/primitives/plane");
						send_command(ss);
					}
					ImGui::EndMenu();
				}

				if (ImGui::MenuItem("Camera"))
				{
					_tool_type = tool::ToolType::PLACE;
					tool_send_state();
					tool::set_placeable(ss, "unit", "core/units/camera");
					send_command(ss);
				}
				if (ImGui::MenuItem("Light"))
				{
					_tool_type = tool::ToolType::PLACE;
					tool_send_state();
					tool::set_placeable(ss, "unit", "core/units/light");
					send_command(ss);
				}
				if (ImGui::MenuItem("Sound"))
				{
					_tool_type = tool::ToolType::PLACE;
					tool_send_state();
					tool::set_placeable(ss, "sound", "core/units/camera");
					send_command(ss);
				}
				ImGui::EndMenu();
			}
			if (ImGui::BeginMenu("Scene View"))
			{
				TempAllocator256 ta;
				StringStream ss(ta);

				if (ImGui::BeginMenu("Camera View"))
				{
					if (ImGui::MenuItem("Perspective"))
					{
						tool::camera_view_perspective(ss);
						send_command(ss);
					}
					if (ImGui::MenuItem("Front"))
					{
						tool::camera_view_front(ss);
						send_command(ss);
					}
					if (ImGui::MenuItem("Back"))
					{
						tool::camera_view_back(ss);
						send_command(ss);
					}
					if (ImGui::MenuItem("Right"))
					{
						tool::camera_view_right(ss);
						send_command(ss);
					}
					if (ImGui::MenuItem("Left"))
					{
						tool::camera_view_left(ss);
						send_command(ss);
					}
					if (ImGui::MenuItem("Top"))
					{
						tool::camera_view_top(ss);
						send_command(ss);
					}
					if (ImGui::MenuItem("Bottom"))
					{
						tool::camera_view_bottom(ss);
						send_command(ss);
					}
					ImGui::EndMenu();
				}
				ImGui::Separator();
				if (ImGui::MenuItem("Show Grid", NULL, _show_grid))
				{
					_show_grid = !_show_grid;
					tool_send_state();
				}
				if (ImGui::SliderFloat("Grid Size", &_grid_size, 0.1f, 5.0f))
				{
					tool_send_state();
				}
				if (ImGui::SliderInt("Snap Rot", &_rotation_snap, 1, 180))
				{
					tool_send_state();
				}
				ImGui::EndMenu();
			}
			if (ImGui::BeginMenu("Windows"))
			{
				if (ImGui::MenuItem("Scene"))
				{
					_scene_view._open = true;
				}
				if (ImGui::MenuItem("Unit List"))
				{
					_unit_list._open = true;
				}
				if (ImGui::MenuItem("Inspector"))
				{
					_inspector._open = true;
				}
				if (ImGui::MenuItem("Console"))
				{
					_console._open = true;
				}
				if (ImGui::MenuItem("Animator"))
				{
					_animator._open = true;
				}

				ImGui::EndMenu();
			}
			if (ImGui::BeginMenu("Help"))
			{
				ImGui::MenuItem("About", "");
				ImGui::EndMenu();
			}

			_main_menu_pos = ImGui::GetWindowPos();
			_main_menu_size = ImGui::GetWindowSize();

			ImGui::EndMainMenuBar();
		}
	}

	void toolbar()
	{
		if (ImGui::BeginToolbar("Toolbar", _toolbar_pos, _toolbar_size))
		{
			if (ImGui::ToolbarButton((void*)(uintptr_t) tex_place->handle.idx, ImVec4(0, 0, 0, 0), "Place"))
			{
				_tool_type = tool::ToolType::PLACE;
				tool_send_state();
			}

			if (ImGui::ToolbarButton((void*)(uintptr_t) tex_move->handle.idx, ImVec4(0, 0, 0, 0), "Move"))
			{
				_tool_type = tool::ToolType::MOVE;
				tool_send_state();
			}

			if (ImGui::ToolbarButton((void*)(uintptr_t) tex_rotate->handle.idx, ImVec4(0, 0, 0, 0), "Rotate"))
			{
				_tool_type = tool::ToolType::ROTATE;
				tool_send_state();
			}

			if (ImGui::ToolbarButton((void*)(uintptr_t) tex_scale->handle.idx, ImVec4(0, 0, 0, 0), "Scale"))
			{
				_tool_type = tool::ToolType::SCALE;
				tool_send_state();
			}

			if (ImGui::ToolbarButton((void*)(uintptr_t) tex_axis_local->handle.idx, ImVec4(0, 0, 0, 0), "Reference System: Local"))
			{
				_reference_system = tool::ReferenceSystem::LOCAL;
				tool_send_state();
			}

			if (ImGui::ToolbarButton((void*)(uintptr_t) tex_axis_world->handle.idx, ImVec4(0, 0, 0, 0), "Reference System: World"))
			{
				_reference_system = tool::ReferenceSystem::WORLD;
				tool_send_state();
			}

			if (ImGui::ToolbarButton((void*)(uintptr_t) tex_axis_local->handle.idx, ImVec4(0, 0, 0, 0), "Snap Mode: Relative"))
			{
				_snap_mode = tool::SnapMode::RELATIVE;
				tool_send_state();
			}

			if (ImGui::ToolbarButton((void*)(uintptr_t) tex_ref_world->handle.idx, ImVec4(0, 0, 0, 0), "Snap Mode: Absolute"))
			{
				_snap_mode = tool::SnapMode::ABSOLUTE;
				tool_send_state();
			}

			_toolbar_pos = ImGui::GetWindowPos();
			_toolbar_size = ImGui::GetWindowSize();

			ImGui::EndToolbar();
		}
	}
};

LevelEditor* s_editor;

void tool_init()
{
	const DeviceOptions& opt = device()->_device_options;
	s_editor = CE_NEW(default_allocator(), LevelEditor)(opt._source_dir);
}

void tool_update(float dt)
{
	s_editor->update(dt);
}

void tool_shutdown()
{
	CE_DELETE(default_allocator(), s_editor);
}

extern bool next_event(OsEvent& ev);

bool tool_process_events()
{
	ImGuiIO& io = ImGui::GetIO();

	bool exit = false;
	bool reset = false;

	TempAllocator4096 ta;
	StringStream ss(ta);

	OsEvent event;
	while (next_event(event))
	{
		if (event.type == OsEventType::NONE)
			continue;

		switch (event.type)
		{
		case OsEventType::BUTTON:
			{
				const ButtonEvent& ev = event.button;
				switch (ev.device_id)
				{
				case crown::InputDeviceType::KEYBOARD:
					io.KeyCtrl = ((ev.button_num == crown::KeyboardButton::CTRL_LEFT)
						|| (ev.button_num == crown::KeyboardButton::CTRL_RIGHT)) && ev.pressed;
					io.KeyShift = ((ev.button_num == crown::KeyboardButton::SHIFT_LEFT)
						|| (ev.button_num == crown::KeyboardButton::SHIFT_RIGHT)) && ev.pressed;
					io.KeyAlt = ((ev.button_num == crown::KeyboardButton::ALT_LEFT)
						|| (ev.button_num == crown::KeyboardButton::ALT_RIGHT)) && ev.pressed;
					io.KeySuper = ((ev.button_num == crown::KeyboardButton::SUPER_LEFT)
						|| (ev.button_num == crown::KeyboardButton::SUPER_RIGHT)) && ev.pressed;

					io.KeysDown[crown::KeyboardButton::TAB] = (ev.button_num == crown::KeyboardButton::TAB) && ev.pressed;
					io.KeysDown[crown::KeyboardButton::LEFT] = (ev.button_num == crown::KeyboardButton::LEFT) && ev.pressed;
					io.KeysDown[crown::KeyboardButton::RIGHT] = (ev.button_num == crown::KeyboardButton::RIGHT) && ev.pressed;
					io.KeysDown[crown::KeyboardButton::UP] = (ev.button_num == crown::KeyboardButton::UP) && ev.pressed;
					io.KeysDown[crown::KeyboardButton::DOWN] = (ev.button_num == crown::KeyboardButton::DOWN) && ev.pressed;
					io.KeysDown[crown::KeyboardButton::PAGE_UP] = (ev.button_num == crown::KeyboardButton::PAGE_UP) && ev.pressed;
					io.KeysDown[crown::KeyboardButton::PAGE_DOWN] = (ev.button_num == crown::KeyboardButton::PAGE_DOWN) && ev.pressed;
					io.KeysDown[crown::KeyboardButton::HOME] = (ev.button_num == crown::KeyboardButton::HOME) && ev.pressed;
					io.KeysDown[crown::KeyboardButton::END] = (ev.button_num == crown::KeyboardButton::END) && ev.pressed;
					io.KeysDown[crown::KeyboardButton::DEL] = (ev.button_num == crown::KeyboardButton::DEL) && ev.pressed;
					io.KeysDown[crown::KeyboardButton::BACKSPACE] = (ev.button_num == crown::KeyboardButton::BACKSPACE) && ev.pressed;
					io.KeysDown[crown::KeyboardButton::ENTER] = (ev.button_num == crown::KeyboardButton::ENTER) && ev.pressed;
					io.KeysDown[crown::KeyboardButton::ESCAPE] = (ev.button_num == crown::KeyboardButton::ESCAPE) && ev.pressed;
					io.KeysDown[crown::KeyboardButton::A] = (ev.button_num == crown::KeyboardButton::A) && ev.pressed;
					io.KeysDown[crown::KeyboardButton::C] = (ev.button_num == crown::KeyboardButton::C) && ev.pressed;
					io.KeysDown[crown::KeyboardButton::V] = (ev.button_num == crown::KeyboardButton::V) && ev.pressed;
					io.KeysDown[crown::KeyboardButton::X] = (ev.button_num == crown::KeyboardButton::X) && ev.pressed;
					io.KeysDown[crown::KeyboardButton::Y] = (ev.button_num == crown::KeyboardButton::Y) && ev.pressed;
					io.KeysDown[crown::KeyboardButton::Z] = (ev.button_num == crown::KeyboardButton::Z) && ev.pressed;

					if (!io.WantCaptureKeyboard)
					{
						if (ev.pressed)
						{
							if (ev.button_num == crown::KeyboardButton::W)
								tool::keyboard_pressed(ss, 'w');
							if (ev.button_num == crown::KeyboardButton::A)
								tool::keyboard_pressed(ss, 'a');
							if (ev.button_num == crown::KeyboardButton::S)
								tool::keyboard_pressed(ss, 's');
							if (ev.button_num == crown::KeyboardButton::D)
								tool::keyboard_pressed(ss, 'd');
						}
						else
						{
							if (ev.button_num == crown::KeyboardButton::W)
								tool::keyboard_released(ss, 'w');
							if (ev.button_num == crown::KeyboardButton::A)
								tool::keyboard_released(ss, 'a');
							if (ev.button_num == crown::KeyboardButton::S)
								tool::keyboard_released(ss, 's');
							if (ev.button_num == crown::KeyboardButton::D)
								tool::keyboard_released(ss, 'd');
						}
					}

					break;

				case crown::InputDeviceType::MOUSE:
					io.MouseDown[0] = (ev.button_num == crown::MouseButton::LEFT) && ev.pressed;
					io.MouseDown[1] = (ev.button_num == crown::MouseButton::RIGHT) && ev.pressed;
					io.MouseDown[2] = (ev.button_num == crown::MouseButton::MIDDLE) && ev.pressed;

					if (!io.WantCaptureMouse)
					{
						ImVec2& mouse_curr = s_editor->_scene_view._mouse_curr;
						mouse_curr.x = io.MousePos.x - s_editor->_scene_view._pos.x;
						mouse_curr.y = io.MousePos.y - s_editor->_scene_view._pos.y;

						tool::set_mouse_state(ss
							, mouse_curr.x
							, mouse_curr.y
							, io.MouseDown[0]
							, io.MouseDown[2]
							, io.MouseDown[1]
							);

						if (ev.button_num == crown::MouseButton::LEFT)
						{
							if (ev.pressed)
								tool::mouse_down(ss, mouse_curr.x, mouse_curr.y);
							else
								tool::mouse_up(ss, mouse_curr.x, mouse_curr.y);
						}
					}

					break;
				}
			}
			break;

		case OsEventType::AXIS:
			{
				const AxisEvent& ev = event.axis;
				switch(ev.device_id)
				{
				case InputDeviceType::MOUSE:
					{
						switch(ev.axis_num)
						{
						case crown::MouseAxis::CURSOR:
							io.MousePos = ImVec2(ev.axis_x, ev.axis_y);

							if (!io.WantCaptureMouse)
							{
								ImVec2& mouse_curr = s_editor->_scene_view._mouse_curr;
								ImVec2& mouse_last = s_editor->_scene_view._mouse_last;

								mouse_curr.x = io.MousePos.x - s_editor->_scene_view._pos.x;
								mouse_curr.y = io.MousePos.y - s_editor->_scene_view._pos.y;

								float delta_x = mouse_curr.x - mouse_last.x;
								float delta_y = mouse_curr.y - mouse_last.y;

								tool::mouse_move(ss, mouse_curr.x, mouse_curr.y, delta_x, delta_y);

								mouse_last = mouse_curr;
							}

							break;

						case crown::MouseAxis::WHEEL:
							io.MouseWheel += ev.axis_y;

							if (!io.WantCaptureMouse)
							{
								tool::mouse_wheel(ss, io.MouseWheel);
							}

							break;
						}
					}
				}
			}
			break;

		case OsEventType::TEXT:
			{
				const TextEvent& ev = event.text;
				io.AddInputCharactersUTF8((const char*) ev.utf8);
			}
			break;

		case OsEventType::RESOLUTION:
			{
				const ResolutionEvent& ev = event.resolution;
				_width  = ev.width;
				_height = ev.height;
				reset   = true;
			}
			break;

		case OsEventType::EXIT:
			exit = true;
			break;

		default:
			break;
		}

	}

	s_editor->send_command(ss);

	bool vsync = true;
	if (reset)
		bgfx::reset(_width, _height, (vsync ? BGFX_RESET_VSYNC : BGFX_RESET_NONE));

	return exit;
}

} // namespace crown

#endif // CROWN_TOOLS