imgui_impl_a5.cpp 10 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294
  1. // ImGui Allegro 5 bindings
  2. // You can copy and use unmodified imgui_impl_* files in your project.
  3. // If you use this binding you'll need to call 4 functions: ImGui_ImplXXXX_Init(), ImGui_ImplXXXX_NewFrame(), ImGui::Render() and ImGui_ImplXXXX_Shutdown().
  4. // See main.cpp for an example of using this.
  5. // https://github.com/ocornut/imgui
  6. // by @birthggd
  7. #include <stdint.h> // uint64_t
  8. #include <cstring> // memcpy
  9. #include <imgui.h>
  10. #include "imgui_impl_a5.h"
  11. #include <allegro5/allegro.h>
  12. #include <allegro5/allegro_primitives.h>
  13. #ifdef _WIN32
  14. #include <allegro5/allegro_windows.h>
  15. #endif
  16. // Data
  17. static ALLEGRO_DISPLAY* g_Display = NULL;
  18. static ALLEGRO_BITMAP* g_Texture = NULL;
  19. static double g_Time = 0.0;
  20. static ALLEGRO_MOUSE_CURSOR* g_MouseCursorInvisible = NULL;
  21. static ALLEGRO_VERTEX_DECL* g_VertexDecl = NULL;
  22. #define OFFSETOF(TYPE, ELEMENT) ((size_t)&(((TYPE *)0)->ELEMENT))
  23. struct ImDrawVertAllegro
  24. {
  25. ImVec2 pos;
  26. ImVec2 uv;
  27. ALLEGRO_COLOR col;
  28. };
  29. void ImGui_ImplA5_RenderDrawLists(ImDrawData* draw_data)
  30. {
  31. int op, src, dst;
  32. al_get_blender(&op, &src, &dst);
  33. al_set_blender(ALLEGRO_ADD, ALLEGRO_ALPHA, ALLEGRO_INVERSE_ALPHA);
  34. for (int n = 0; n < draw_data->CmdListsCount; n++)
  35. {
  36. const ImDrawList* cmd_list = draw_data->CmdLists[n];
  37. // FIXME-OPT: Unfortunately Allegro doesn't support 32-bits packed colors so we have to convert them to 4 floats
  38. static ImVector<ImDrawVertAllegro> vertices;
  39. vertices.resize(cmd_list->VtxBuffer.size());
  40. for (int i = 0; i < cmd_list->VtxBuffer.size(); ++i)
  41. {
  42. const ImDrawVert &dv = cmd_list->VtxBuffer[i];
  43. ImDrawVertAllegro v;
  44. v.pos = dv.pos;
  45. v.uv = dv.uv;
  46. unsigned char *c = (unsigned char*)&dv.col;
  47. v.col = al_map_rgba(c[0], c[1], c[2], c[3]);
  48. vertices[i] = v;
  49. }
  50. // FIXME-OPT: Unfortunately Allegro doesn't support 16-bit indices
  51. // You can also use '#define ImDrawIdx unsigned int' in imconfig.h and request ImGui to output 32-bit indices
  52. static ImVector<int> indices;
  53. indices.resize(cmd_list->IdxBuffer.size());
  54. for (int i = 0; i < cmd_list->IdxBuffer.size(); ++i)
  55. indices[i] = (int)cmd_list->IdxBuffer.Data[i];
  56. int idx_offset = 0;
  57. for (int cmd_i = 0; cmd_i < cmd_list->CmdBuffer.size(); cmd_i++)
  58. {
  59. const ImDrawCmd* pcmd = &cmd_list->CmdBuffer[cmd_i];
  60. if (pcmd->UserCallback)
  61. {
  62. pcmd->UserCallback(cmd_list, pcmd);
  63. }
  64. else
  65. {
  66. ALLEGRO_BITMAP* texture = (ALLEGRO_BITMAP*)pcmd->TextureId;
  67. al_set_clipping_rectangle(pcmd->ClipRect.x, pcmd->ClipRect.y, pcmd->ClipRect.z-pcmd->ClipRect.x, pcmd->ClipRect.w-pcmd->ClipRect.y);
  68. al_draw_indexed_prim(&vertices[0], g_VertexDecl, texture, &indices[idx_offset], pcmd->ElemCount, ALLEGRO_PRIM_TRIANGLE_LIST);
  69. }
  70. idx_offset += pcmd->ElemCount;
  71. }
  72. }
  73. // Restore modified state
  74. al_set_blender(op, src, dst);
  75. al_set_clipping_rectangle(0, 0, al_get_display_width(g_Display), al_get_display_height(g_Display));
  76. }
  77. bool Imgui_ImplA5_CreateDeviceObjects()
  78. {
  79. ImGuiIO &io = ImGui::GetIO();
  80. // Build texture
  81. unsigned char *pixels;
  82. int width, height;
  83. io.Fonts->GetTexDataAsRGBA32(&pixels, &width, &height);
  84. // Create texture
  85. int flags = al_get_new_bitmap_flags();
  86. int fmt = al_get_new_bitmap_format();
  87. al_set_new_bitmap_flags(ALLEGRO_MEMORY_BITMAP|ALLEGRO_MIN_LINEAR|ALLEGRO_MAG_LINEAR);
  88. al_set_new_bitmap_format(ALLEGRO_PIXEL_FORMAT_ABGR_8888_LE);
  89. ALLEGRO_BITMAP* img = al_create_bitmap(width, height);
  90. al_set_new_bitmap_flags(flags);
  91. al_set_new_bitmap_format(fmt);
  92. if (!img)
  93. return false;
  94. ALLEGRO_LOCKED_REGION *locked_img = al_lock_bitmap(img, al_get_bitmap_format(img), ALLEGRO_LOCK_WRITEONLY);
  95. if (!locked_img)
  96. {
  97. al_destroy_bitmap(img);
  98. return false;
  99. }
  100. memcpy(locked_img->data, pixels, sizeof(int)*width*height);
  101. al_unlock_bitmap(img);
  102. // Convert software texture to hardware texture.
  103. ALLEGRO_BITMAP* cloned_img = al_clone_bitmap(img);
  104. al_destroy_bitmap(img);
  105. if (!cloned_img)
  106. return false;
  107. // Store our identifier
  108. io.Fonts->TexID = (void*)cloned_img;
  109. g_Texture = cloned_img;
  110. // Cleanup (don't clear the input data if you want to append new fonts later)
  111. io.Fonts->ClearInputData();
  112. io.Fonts->ClearTexData();
  113. // Create an invisible mouse cursor
  114. // Because al_hide_mouse_cursor() seems to mess up with the actual inputs..
  115. ALLEGRO_BITMAP* mouse_cursor = al_create_bitmap(8,8);
  116. g_MouseCursorInvisible = al_create_mouse_cursor(mouse_cursor, 0, 0);
  117. al_destroy_bitmap(mouse_cursor);
  118. return true;
  119. }
  120. void ImGui_ImplA5_InvalidateDeviceObjects()
  121. {
  122. if (g_Texture)
  123. {
  124. al_destroy_bitmap(g_Texture);
  125. ImGui::GetIO().Fonts->TexID = NULL;
  126. g_Texture = NULL;
  127. }
  128. if (g_MouseCursorInvisible)
  129. {
  130. al_destroy_mouse_cursor(g_MouseCursorInvisible);
  131. g_MouseCursorInvisible = NULL;
  132. }
  133. }
  134. bool ImGui_ImplA5_Init(ALLEGRO_DISPLAY* display)
  135. {
  136. g_Display = display;
  137. // Create custom vertex declaration.
  138. // Unfortunately Allegro doesn't support 32-bits packed colors so we have to convert them to 4 floats.
  139. // 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.
  140. ALLEGRO_VERTEX_ELEMENT elems[] =
  141. {
  142. { ALLEGRO_PRIM_POSITION, ALLEGRO_PRIM_FLOAT_2, OFFSETOF(ImDrawVertAllegro, pos) },
  143. { ALLEGRO_PRIM_TEX_COORD, ALLEGRO_PRIM_FLOAT_2, OFFSETOF(ImDrawVertAllegro, uv) },
  144. { ALLEGRO_PRIM_COLOR_ATTR, 0, OFFSETOF(ImDrawVertAllegro, col) },
  145. { 0, 0, 0 }
  146. };
  147. g_VertexDecl = al_create_vertex_decl(elems, sizeof(ImDrawVertAllegro));
  148. ImGuiIO& io = ImGui::GetIO();
  149. io.KeyMap[ImGuiKey_Tab] = ALLEGRO_KEY_TAB;
  150. io.KeyMap[ImGuiKey_LeftArrow] = ALLEGRO_KEY_LEFT;
  151. io.KeyMap[ImGuiKey_RightArrow] = ALLEGRO_KEY_RIGHT;
  152. io.KeyMap[ImGuiKey_UpArrow] = ALLEGRO_KEY_UP;
  153. io.KeyMap[ImGuiKey_DownArrow] = ALLEGRO_KEY_DOWN;
  154. io.KeyMap[ImGuiKey_PageUp] = ALLEGRO_KEY_PGUP;
  155. io.KeyMap[ImGuiKey_PageDown] = ALLEGRO_KEY_PGDN;
  156. io.KeyMap[ImGuiKey_Home] = ALLEGRO_KEY_HOME;
  157. io.KeyMap[ImGuiKey_End] = ALLEGRO_KEY_END;
  158. io.KeyMap[ImGuiKey_Delete] = ALLEGRO_KEY_DELETE;
  159. io.KeyMap[ImGuiKey_Backspace] = ALLEGRO_KEY_BACKSPACE;
  160. io.KeyMap[ImGuiKey_Enter] = ALLEGRO_KEY_ENTER;
  161. io.KeyMap[ImGuiKey_Escape] = ALLEGRO_KEY_ESCAPE;
  162. io.KeyMap[ImGuiKey_A] = ALLEGRO_KEY_A;
  163. io.KeyMap[ImGuiKey_C] = ALLEGRO_KEY_C;
  164. io.KeyMap[ImGuiKey_V] = ALLEGRO_KEY_V;
  165. io.KeyMap[ImGuiKey_X] = ALLEGRO_KEY_X;
  166. io.KeyMap[ImGuiKey_Y] = ALLEGRO_KEY_Y;
  167. io.KeyMap[ImGuiKey_Z] = ALLEGRO_KEY_Z;
  168. io.RenderDrawListsFn = ImGui_ImplA5_RenderDrawLists; // Alternatively you can set this to NULL and call ImGui::GetDrawData() after ImGui::Render() to get the same ImDrawData pointer.
  169. #ifdef _WIN32
  170. io.ImeWindowHandle = al_get_win_window_handle(g_Display);
  171. #endif
  172. return true;
  173. }
  174. void ImGui_ImplA5_Shutdown()
  175. {
  176. ImGui_ImplA5_InvalidateDeviceObjects();
  177. ImGui::Shutdown();
  178. }
  179. bool ImGui_ImplA5_ProcessEvent(ALLEGRO_EVENT *ev)
  180. {
  181. ImGuiIO &io = ImGui::GetIO();
  182. switch (ev->type)
  183. {
  184. case ALLEGRO_EVENT_MOUSE_AXES:
  185. io.MouseWheel += ev->mouse.dz;
  186. return true;
  187. case ALLEGRO_EVENT_KEY_CHAR:
  188. if (ev->keyboard.display == g_Display)
  189. if (ev->keyboard.unichar > 0 && ev->keyboard.unichar < 0x10000)
  190. io.AddInputCharacter((unsigned short)ev->keyboard.unichar);
  191. return true;
  192. case ALLEGRO_EVENT_KEY_DOWN:
  193. case ALLEGRO_EVENT_KEY_UP:
  194. if (ev->keyboard.display == g_Display)
  195. io.KeysDown[ev->keyboard.keycode] = (ev->type == ALLEGRO_EVENT_KEY_DOWN);
  196. return true;
  197. }
  198. return false;
  199. }
  200. void ImGui_ImplA5_NewFrame()
  201. {
  202. if (!g_Texture)
  203. Imgui_ImplA5_CreateDeviceObjects();
  204. ImGuiIO &io = ImGui::GetIO();
  205. // Setup display size (every frame to accommodate for window resizing)
  206. int w, h;
  207. w = al_get_display_width(g_Display);
  208. h = al_get_display_height(g_Display);
  209. io.DisplaySize = ImVec2((float)w, (float)h);
  210. // Setup time step
  211. double current_time = al_get_time();
  212. io.DeltaTime = g_Time > 0.0 ? (float)(current_time - g_Time) : (float)(1.0f/60.0f);
  213. g_Time = current_time;
  214. // Setup inputs
  215. ALLEGRO_KEYBOARD_STATE keys;
  216. al_get_keyboard_state(&keys);
  217. io.KeyCtrl = al_key_down(&keys, ALLEGRO_KEY_LCTRL) || al_key_down(&keys, ALLEGRO_KEY_RCTRL);
  218. io.KeyShift = al_key_down(&keys, ALLEGRO_KEY_LSHIFT) || al_key_down(&keys, ALLEGRO_KEY_RSHIFT);
  219. io.KeyAlt = al_key_down(&keys, ALLEGRO_KEY_ALT) || al_key_down(&keys, ALLEGRO_KEY_ALTGR);
  220. ALLEGRO_MOUSE_STATE mouse;
  221. if (keys.display == g_Display)
  222. {
  223. al_get_mouse_state(&mouse);
  224. io.MousePos = ImVec2((float)mouse.x, (float)mouse.y);
  225. }
  226. else
  227. {
  228. io.MousePos = ImVec2(-1, -1);
  229. }
  230. al_get_mouse_state(&mouse);
  231. io.MouseDown[0] = mouse.buttons & (1 << 0);
  232. io.MouseDown[1] = mouse.buttons & (1 << 1);
  233. io.MouseDown[2] = mouse.buttons & (1 << 2);
  234. // Hide OS mouse cursor if ImGui is drawing it
  235. if (io.MouseDrawCursor)
  236. {
  237. al_set_mouse_cursor(g_Display, g_MouseCursorInvisible);
  238. }
  239. else
  240. {
  241. ALLEGRO_SYSTEM_MOUSE_CURSOR cursor_id = ALLEGRO_SYSTEM_MOUSE_CURSOR_DEFAULT;
  242. switch (ImGui::GetMouseCursor())
  243. {
  244. case ImGuiMouseCursor_TextInput: cursor_id = ALLEGRO_SYSTEM_MOUSE_CURSOR_EDIT; break;
  245. case ImGuiMouseCursor_Move: cursor_id = ALLEGRO_SYSTEM_MOUSE_CURSOR_MOVE; break;
  246. case ImGuiMouseCursor_ResizeNS: cursor_id = ALLEGRO_SYSTEM_MOUSE_CURSOR_RESIZE_N; break;
  247. case ImGuiMouseCursor_ResizeEW: cursor_id = ALLEGRO_SYSTEM_MOUSE_CURSOR_RESIZE_E; break;
  248. case ImGuiMouseCursor_ResizeNESW: cursor_id = ALLEGRO_SYSTEM_MOUSE_CURSOR_RESIZE_NE; break;
  249. case ImGuiMouseCursor_ResizeNWSE: cursor_id = ALLEGRO_SYSTEM_MOUSE_CURSOR_RESIZE_NW; break;
  250. }
  251. al_set_system_mouse_cursor(g_Display, cursor_id);
  252. }
  253. // Start the frame
  254. ImGui::NewFrame();
  255. }