imgui_impl_allegro5.cpp 12 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316
  1. // ImGui Allegro 5 bindings
  2. // Implemented features:
  3. // [X] User texture binding. Use 'ALLEGRO_BITMAP*' as ImTextureID. Read the FAQ about ImTextureID in imgui.cpp.
  4. // Missing features:
  5. // [ ] Clipboard support via al_set_clipboard_text/al_clipboard_has_text.
  6. // You can copy and use unmodified imgui_impl_* files in your project. See main.cpp for an example of using this.
  7. // If you use this binding you'll need to call 4 functions: ImGui_ImplXXXX_Init(), ImGui_ImplXXXX_NewFrame(), ImGui::Render() and ImGui_ImplXXXX_Shutdown().
  8. // If you are new to ImGui, see examples/README.txt and documentation at the top of imgui.cpp.
  9. // https://github.com/ocornut/imgui, Original code by @birthggd
  10. // CHANGELOG
  11. // (minor and older changes stripped away, please see git history for details)
  12. // 2018-04-18: Misc: Renamed file from imgui_impl_a5.cpp to imgui_impl_allegro5.cpp.
  13. // 2018-04-18: Misc: Added support for 32-bits vertex indices to avoid conversion at runtime. Added imconfig_allegro5.h to enforce 32-bit indices when included from imgui.h.
  14. // 2018-02-16: Misc: Obsoleted the io.RenderDrawListsFn callback and exposed ImGui_ImplA5_RenderDrawData() in the .h file so you can call it yourself.
  15. // 2018-02-06: Misc: Removed call to ImGui::Shutdown() which is not available from 1.60 WIP, user needs to call CreateContext/DestroyContext themselves.
  16. // 2018-02-06: Inputs: Added mapping for ImGuiKey_Space.
  17. #include <stdint.h> // uint64_t
  18. #include <cstring> // memcpy
  19. #include "imgui.h"
  20. #include "imgui_impl_allegro5.h"
  21. #include <allegro5/allegro.h>
  22. #include <allegro5/allegro_primitives.h>
  23. #ifdef _WIN32
  24. #include <allegro5/allegro_windows.h>
  25. #endif
  26. // Data
  27. static ALLEGRO_DISPLAY* g_Display = NULL;
  28. static ALLEGRO_BITMAP* g_Texture = NULL;
  29. static double g_Time = 0.0;
  30. static ALLEGRO_MOUSE_CURSOR* g_MouseCursorInvisible = NULL;
  31. static ALLEGRO_VERTEX_DECL* g_VertexDecl = NULL;
  32. struct ImDrawVertAllegro
  33. {
  34. ImVec2 pos;
  35. ImVec2 uv;
  36. ALLEGRO_COLOR col;
  37. };
  38. // Render function.
  39. // (this used to be set in io.RenderDrawListsFn and called by ImGui::Render(), but you can now call this directly from your main loop)
  40. void ImGui_ImplA5_RenderDrawData(ImDrawData* draw_data)
  41. {
  42. int op, src, dst;
  43. al_get_blender(&op, &src, &dst);
  44. al_set_blender(ALLEGRO_ADD, ALLEGRO_ALPHA, ALLEGRO_INVERSE_ALPHA);
  45. for (int n = 0; n < draw_data->CmdListsCount; n++)
  46. {
  47. const ImDrawList* cmd_list = draw_data->CmdLists[n];
  48. // FIXME-OPT: Unfortunately Allegro doesn't support 32-bits packed colors so we have to convert them to 4 floats
  49. static ImVector<ImDrawVertAllegro> vertices;
  50. vertices.resize(cmd_list->VtxBuffer.Size);
  51. for (int i = 0; i < cmd_list->VtxBuffer.Size; ++i)
  52. {
  53. const ImDrawVert &dv = cmd_list->VtxBuffer[i];
  54. ImDrawVertAllegro v;
  55. v.pos = dv.pos;
  56. v.uv = dv.uv;
  57. unsigned char *c = (unsigned char*)&dv.col;
  58. v.col = al_map_rgba(c[0], c[1], c[2], c[3]);
  59. vertices[i] = v;
  60. }
  61. const int* indices = NULL;
  62. if (sizeof(ImDrawIdx) == 2)
  63. {
  64. // FIXME-OPT: Unfortunately Allegro doesn't support 16-bit indices.. You can '#define ImDrawIdx int' in imconfig.h to request ImGui to output 32-bit indices.
  65. // Otherwise, we convert them from 16-bit to 32-bit at runtime here, which works perfectly but is a little wasteful.
  66. static ImVector<int> indices_converted;
  67. indices_converted.resize(cmd_list->IdxBuffer.Size);
  68. for (int i = 0; i < cmd_list->IdxBuffer.Size; ++i)
  69. indices_converted[i] = (int)cmd_list->IdxBuffer.Data[i];
  70. indices = indices_converted.Data;
  71. }
  72. else if (sizeof(ImDrawIdx) == 4)
  73. {
  74. indices = (const int*)cmd_list->IdxBuffer.Data;
  75. }
  76. int idx_offset = 0;
  77. for (int cmd_i = 0; cmd_i < cmd_list->CmdBuffer.Size; cmd_i++)
  78. {
  79. const ImDrawCmd* pcmd = &cmd_list->CmdBuffer[cmd_i];
  80. if (pcmd->UserCallback)
  81. {
  82. pcmd->UserCallback(cmd_list, pcmd);
  83. }
  84. else
  85. {
  86. ALLEGRO_BITMAP* texture = (ALLEGRO_BITMAP*)pcmd->TextureId;
  87. al_set_clipping_rectangle(pcmd->ClipRect.x, pcmd->ClipRect.y, pcmd->ClipRect.z-pcmd->ClipRect.x, pcmd->ClipRect.w-pcmd->ClipRect.y);
  88. al_draw_indexed_prim(&vertices[0], g_VertexDecl, texture, &indices[idx_offset], pcmd->ElemCount, ALLEGRO_PRIM_TRIANGLE_LIST);
  89. }
  90. idx_offset += pcmd->ElemCount;
  91. }
  92. }
  93. // Restore modified state
  94. al_set_blender(op, src, dst);
  95. al_set_clipping_rectangle(0, 0, al_get_display_width(g_Display), al_get_display_height(g_Display));
  96. }
  97. bool Imgui_ImplA5_CreateDeviceObjects()
  98. {
  99. // Build texture atlas
  100. ImGuiIO &io = ImGui::GetIO();
  101. unsigned char *pixels;
  102. int width, height;
  103. io.Fonts->GetTexDataAsRGBA32(&pixels, &width, &height);
  104. // Create texture
  105. int flags = al_get_new_bitmap_flags();
  106. int fmt = al_get_new_bitmap_format();
  107. al_set_new_bitmap_flags(ALLEGRO_MEMORY_BITMAP|ALLEGRO_MIN_LINEAR|ALLEGRO_MAG_LINEAR);
  108. al_set_new_bitmap_format(ALLEGRO_PIXEL_FORMAT_ABGR_8888_LE);
  109. ALLEGRO_BITMAP* img = al_create_bitmap(width, height);
  110. al_set_new_bitmap_flags(flags);
  111. al_set_new_bitmap_format(fmt);
  112. if (!img)
  113. return false;
  114. ALLEGRO_LOCKED_REGION *locked_img = al_lock_bitmap(img, al_get_bitmap_format(img), ALLEGRO_LOCK_WRITEONLY);
  115. if (!locked_img)
  116. {
  117. al_destroy_bitmap(img);
  118. return false;
  119. }
  120. memcpy(locked_img->data, pixels, sizeof(int)*width*height);
  121. al_unlock_bitmap(img);
  122. // Convert software texture to hardware texture.
  123. ALLEGRO_BITMAP* cloned_img = al_clone_bitmap(img);
  124. al_destroy_bitmap(img);
  125. if (!cloned_img)
  126. return false;
  127. // Store our identifier
  128. io.Fonts->TexID = (void*)cloned_img;
  129. g_Texture = cloned_img;
  130. // Create an invisible mouse cursor
  131. // Because al_hide_mouse_cursor() seems to mess up with the actual inputs..
  132. ALLEGRO_BITMAP* mouse_cursor = al_create_bitmap(8,8);
  133. g_MouseCursorInvisible = al_create_mouse_cursor(mouse_cursor, 0, 0);
  134. al_destroy_bitmap(mouse_cursor);
  135. return true;
  136. }
  137. void ImGui_ImplA5_InvalidateDeviceObjects()
  138. {
  139. if (g_Texture)
  140. {
  141. al_destroy_bitmap(g_Texture);
  142. ImGui::GetIO().Fonts->TexID = NULL;
  143. g_Texture = NULL;
  144. }
  145. if (g_MouseCursorInvisible)
  146. {
  147. al_destroy_mouse_cursor(g_MouseCursorInvisible);
  148. g_MouseCursorInvisible = NULL;
  149. }
  150. }
  151. bool ImGui_ImplA5_Init(ALLEGRO_DISPLAY* display)
  152. {
  153. g_Display = display;
  154. // Create custom vertex declaration.
  155. // Unfortunately Allegro doesn't support 32-bits packed colors so we have to convert them to 4 floats.
  156. // We still use a custom declaration to use 'ALLEGRO_PRIM_TEX_COORD' instead of 'ALLEGRO_PRIM_TEX_COORD_PIXEL' else we can't do a reliable conversion.
  157. ALLEGRO_VERTEX_ELEMENT elems[] =
  158. {
  159. { ALLEGRO_PRIM_POSITION, ALLEGRO_PRIM_FLOAT_2, IM_OFFSETOF(ImDrawVertAllegro, pos) },
  160. { ALLEGRO_PRIM_TEX_COORD, ALLEGRO_PRIM_FLOAT_2, IM_OFFSETOF(ImDrawVertAllegro, uv) },
  161. { ALLEGRO_PRIM_COLOR_ATTR, 0, IM_OFFSETOF(ImDrawVertAllegro, col) },
  162. { 0, 0, 0 }
  163. };
  164. g_VertexDecl = al_create_vertex_decl(elems, sizeof(ImDrawVertAllegro));
  165. ImGuiIO& io = ImGui::GetIO();
  166. io.KeyMap[ImGuiKey_Tab] = ALLEGRO_KEY_TAB;
  167. io.KeyMap[ImGuiKey_LeftArrow] = ALLEGRO_KEY_LEFT;
  168. io.KeyMap[ImGuiKey_RightArrow] = ALLEGRO_KEY_RIGHT;
  169. io.KeyMap[ImGuiKey_UpArrow] = ALLEGRO_KEY_UP;
  170. io.KeyMap[ImGuiKey_DownArrow] = ALLEGRO_KEY_DOWN;
  171. io.KeyMap[ImGuiKey_PageUp] = ALLEGRO_KEY_PGUP;
  172. io.KeyMap[ImGuiKey_PageDown] = ALLEGRO_KEY_PGDN;
  173. io.KeyMap[ImGuiKey_Home] = ALLEGRO_KEY_HOME;
  174. io.KeyMap[ImGuiKey_End] = ALLEGRO_KEY_END;
  175. io.KeyMap[ImGuiKey_Insert] = ALLEGRO_KEY_INSERT;
  176. io.KeyMap[ImGuiKey_Delete] = ALLEGRO_KEY_DELETE;
  177. io.KeyMap[ImGuiKey_Backspace] = ALLEGRO_KEY_BACKSPACE;
  178. io.KeyMap[ImGuiKey_Space] = ALLEGRO_KEY_SPACE;
  179. io.KeyMap[ImGuiKey_Enter] = ALLEGRO_KEY_ENTER;
  180. io.KeyMap[ImGuiKey_Escape] = ALLEGRO_KEY_ESCAPE;
  181. io.KeyMap[ImGuiKey_A] = ALLEGRO_KEY_A;
  182. io.KeyMap[ImGuiKey_C] = ALLEGRO_KEY_C;
  183. io.KeyMap[ImGuiKey_V] = ALLEGRO_KEY_V;
  184. io.KeyMap[ImGuiKey_X] = ALLEGRO_KEY_X;
  185. io.KeyMap[ImGuiKey_Y] = ALLEGRO_KEY_Y;
  186. io.KeyMap[ImGuiKey_Z] = ALLEGRO_KEY_Z;
  187. #ifdef _WIN32
  188. io.ImeWindowHandle = al_get_win_window_handle(g_Display);
  189. #endif
  190. return true;
  191. }
  192. void ImGui_ImplA5_Shutdown()
  193. {
  194. ImGui_ImplA5_InvalidateDeviceObjects();
  195. }
  196. // You can read the io.WantCaptureMouse, io.WantCaptureKeyboard flags to tell if dear imgui wants to use your inputs.
  197. // - When io.WantCaptureMouse is true, do not dispatch mouse input data to your main application.
  198. // - When io.WantCaptureKeyboard is true, do not dispatch keyboard input data to your main application.
  199. // Generally you may always pass all inputs to dear imgui, and hide them from your application based on those two flags.
  200. bool ImGui_ImplA5_ProcessEvent(ALLEGRO_EVENT *ev)
  201. {
  202. ImGuiIO &io = ImGui::GetIO();
  203. switch (ev->type)
  204. {
  205. case ALLEGRO_EVENT_MOUSE_AXES:
  206. io.MouseWheel += ev->mouse.dz;
  207. io.MouseWheelH += ev->mouse.dw;
  208. return true;
  209. case ALLEGRO_EVENT_KEY_CHAR:
  210. if (ev->keyboard.display == g_Display)
  211. if (ev->keyboard.unichar > 0 && ev->keyboard.unichar < 0x10000)
  212. io.AddInputCharacter((unsigned short)ev->keyboard.unichar);
  213. return true;
  214. case ALLEGRO_EVENT_KEY_DOWN:
  215. case ALLEGRO_EVENT_KEY_UP:
  216. if (ev->keyboard.display == g_Display)
  217. io.KeysDown[ev->keyboard.keycode] = (ev->type == ALLEGRO_EVENT_KEY_DOWN);
  218. return true;
  219. }
  220. return false;
  221. }
  222. void ImGui_ImplA5_NewFrame()
  223. {
  224. if (!g_Texture)
  225. Imgui_ImplA5_CreateDeviceObjects();
  226. ImGuiIO &io = ImGui::GetIO();
  227. // Setup display size (every frame to accommodate for window resizing)
  228. int w, h;
  229. w = al_get_display_width(g_Display);
  230. h = al_get_display_height(g_Display);
  231. io.DisplaySize = ImVec2((float)w, (float)h);
  232. // Setup time step
  233. double current_time = al_get_time();
  234. io.DeltaTime = g_Time > 0.0 ? (float)(current_time - g_Time) : (float)(1.0f/60.0f);
  235. g_Time = current_time;
  236. // Setup inputs
  237. ALLEGRO_KEYBOARD_STATE keys;
  238. al_get_keyboard_state(&keys);
  239. io.KeyCtrl = al_key_down(&keys, ALLEGRO_KEY_LCTRL) || al_key_down(&keys, ALLEGRO_KEY_RCTRL);
  240. io.KeyShift = al_key_down(&keys, ALLEGRO_KEY_LSHIFT) || al_key_down(&keys, ALLEGRO_KEY_RSHIFT);
  241. io.KeyAlt = al_key_down(&keys, ALLEGRO_KEY_ALT) || al_key_down(&keys, ALLEGRO_KEY_ALTGR);
  242. io.KeySuper = al_key_down(&keys, ALLEGRO_KEY_LWIN) || al_key_down(&keys, ALLEGRO_KEY_RWIN);
  243. ALLEGRO_MOUSE_STATE mouse;
  244. if (keys.display == g_Display)
  245. {
  246. al_get_mouse_state(&mouse);
  247. io.MousePos = ImVec2((float)mouse.x, (float)mouse.y);
  248. }
  249. else
  250. {
  251. io.MousePos = ImVec2(-FLT_MAX, -FLT_MAX);
  252. }
  253. al_get_mouse_state(&mouse);
  254. io.MouseDown[0] = mouse.buttons & (1 << 0);
  255. io.MouseDown[1] = mouse.buttons & (1 << 1);
  256. io.MouseDown[2] = mouse.buttons & (1 << 2);
  257. // Hide OS mouse cursor if ImGui is drawing it
  258. if (io.MouseDrawCursor)
  259. {
  260. al_set_mouse_cursor(g_Display, g_MouseCursorInvisible);
  261. }
  262. else
  263. {
  264. ALLEGRO_SYSTEM_MOUSE_CURSOR cursor_id = ALLEGRO_SYSTEM_MOUSE_CURSOR_DEFAULT;
  265. switch (ImGui::GetMouseCursor())
  266. {
  267. case ImGuiMouseCursor_TextInput: cursor_id = ALLEGRO_SYSTEM_MOUSE_CURSOR_EDIT; break;
  268. case ImGuiMouseCursor_ResizeAll: cursor_id = ALLEGRO_SYSTEM_MOUSE_CURSOR_MOVE; break;
  269. case ImGuiMouseCursor_ResizeNS: cursor_id = ALLEGRO_SYSTEM_MOUSE_CURSOR_RESIZE_N; break;
  270. case ImGuiMouseCursor_ResizeEW: cursor_id = ALLEGRO_SYSTEM_MOUSE_CURSOR_RESIZE_E; break;
  271. case ImGuiMouseCursor_ResizeNESW: cursor_id = ALLEGRO_SYSTEM_MOUSE_CURSOR_RESIZE_NE; break;
  272. case ImGuiMouseCursor_ResizeNWSE: cursor_id = ALLEGRO_SYSTEM_MOUSE_CURSOR_RESIZE_NW; break;
  273. }
  274. al_set_system_mouse_cursor(g_Display, cursor_id);
  275. }
  276. // Start the frame. This call will update the io.WantCaptureMouse, io.WantCaptureKeyboard flag that you can use to dispatch inputs (or not) to your application.
  277. ImGui::NewFrame();
  278. }