main.cpp 31 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750
  1. // ImGui - standalone example application for SDL2 + Vulkan
  2. // If you are new to ImGui, see examples/README.txt and documentation at the top of imgui.cpp.
  3. #include "imgui.h"
  4. #include "../imgui_impl_sdl2.h"
  5. #include "../imgui_impl_vulkan.h"
  6. #include <stdio.h>
  7. #include <SDL.h>
  8. #include <SDL_vulkan.h>
  9. #include <vulkan/vulkan.h>
  10. // FIXME-VULKAN: Resizing with IMGUI_UNLIMITED_FRAME_RATE triggers errors from the validation layer.
  11. #define IMGUI_MAX_POSSIBLE_BACK_BUFFERS 16
  12. #define IMGUI_UNLIMITED_FRAME_RATE
  13. #ifdef _DEBUG
  14. #define IMGUI_VULKAN_DEBUG_REPORT
  15. #endif
  16. static VkAllocationCallbacks* g_Allocator = NULL;
  17. static VkInstance g_Instance = VK_NULL_HANDLE;
  18. static VkPhysicalDevice g_PhysicalDevice = VK_NULL_HANDLE;
  19. static VkDevice g_Device = VK_NULL_HANDLE;
  20. static uint32_t g_QueueFamily = (uint32_t)-1;
  21. static VkQueue g_Queue = VK_NULL_HANDLE;
  22. static VkDebugReportCallbackEXT g_DebugReport = VK_NULL_HANDLE;
  23. static VkPipelineCache g_PipelineCache = VK_NULL_HANDLE;
  24. static VkDescriptorPool g_DescriptorPool = VK_NULL_HANDLE;
  25. struct FrameData
  26. {
  27. uint32_t BackbufferIndex; // keep track of recently rendered swapchain frame indices
  28. VkCommandPool CommandPool;
  29. VkCommandBuffer CommandBuffer;
  30. VkFence Fence;
  31. VkSemaphore PresentCompleteSemaphore;
  32. VkSemaphore RenderCompleteSemaphore;
  33. FrameData()
  34. {
  35. BackbufferIndex = 0;
  36. CommandPool = VK_NULL_HANDLE;
  37. CommandBuffer = VK_NULL_HANDLE;
  38. Fence = VK_NULL_HANDLE;
  39. PresentCompleteSemaphore = VK_NULL_HANDLE;
  40. RenderCompleteSemaphore = VK_NULL_HANDLE;
  41. }
  42. };
  43. struct WindowData
  44. {
  45. int Width, Height;
  46. VkSwapchainKHR Swapchain;
  47. VkSurfaceKHR Surface;
  48. VkSurfaceFormatKHR SurfaceFormat;
  49. VkPresentModeKHR PresentMode;
  50. VkRenderPass RenderPass;
  51. VkClearValue ClearValue;
  52. uint32_t BackBufferCount;
  53. VkImage BackBuffer[IMGUI_MAX_POSSIBLE_BACK_BUFFERS];
  54. VkImageView BackBufferView[IMGUI_MAX_POSSIBLE_BACK_BUFFERS];
  55. VkFramebuffer Framebuffer[IMGUI_MAX_POSSIBLE_BACK_BUFFERS];
  56. uint32_t FrameIndex;
  57. FrameData Frames[IMGUI_VK_QUEUED_FRAMES];
  58. WindowData()
  59. {
  60. Width = Height = 0;
  61. Swapchain = VK_NULL_HANDLE;
  62. Surface = VK_NULL_HANDLE;
  63. memset(&SurfaceFormat, 0, sizeof(SurfaceFormat));
  64. PresentMode = VK_PRESENT_MODE_MAX_ENUM_KHR;
  65. RenderPass = VK_NULL_HANDLE;
  66. memset(&ClearValue, 0, sizeof(ClearValue));
  67. BackBufferCount = 0;
  68. memset(&BackBuffer, 0, sizeof(BackBuffer));
  69. memset(&BackBufferView, 0, sizeof(BackBufferView));
  70. memset(&Framebuffer, 0, sizeof(Framebuffer));
  71. FrameIndex = 0;
  72. }
  73. };
  74. static WindowData g_WindowData;
  75. static void check_vk_result(VkResult err)
  76. {
  77. if (err == 0) return;
  78. printf("VkResult %d\n", err);
  79. if (err < 0)
  80. abort();
  81. }
  82. static void CreateOrResizeSwapChainAndFrameBuffer(WindowData* wd, int w, int h)
  83. {
  84. VkResult err;
  85. VkSwapchainKHR old_swapchain = wd->Swapchain;
  86. err = vkDeviceWaitIdle(g_Device);
  87. check_vk_result(err);
  88. // Destroy old Framebuffer
  89. for (uint32_t i = 0; i < wd->BackBufferCount; i++)
  90. {
  91. if (wd->BackBufferView[i])
  92. vkDestroyImageView(g_Device, wd->BackBufferView[i], g_Allocator);
  93. if (wd->Framebuffer[i])
  94. vkDestroyFramebuffer(g_Device, wd->Framebuffer[i], g_Allocator);
  95. }
  96. wd->BackBufferCount = 0;
  97. if (wd->RenderPass)
  98. vkDestroyRenderPass(g_Device, wd->RenderPass, g_Allocator);
  99. // Create Swapchain
  100. {
  101. VkSwapchainCreateInfoKHR info = {};
  102. info.sType = VK_STRUCTURE_TYPE_SWAPCHAIN_CREATE_INFO_KHR;
  103. info.surface = wd->Surface;
  104. info.imageFormat = wd->SurfaceFormat.format;
  105. info.imageColorSpace = wd->SurfaceFormat.colorSpace;
  106. info.imageArrayLayers = 1;
  107. info.imageUsage = VK_IMAGE_USAGE_COLOR_ATTACHMENT_BIT;
  108. info.imageSharingMode = VK_SHARING_MODE_EXCLUSIVE; // Assume that graphics family == present family
  109. info.preTransform = VK_SURFACE_TRANSFORM_IDENTITY_BIT_KHR;
  110. info.compositeAlpha = VK_COMPOSITE_ALPHA_OPAQUE_BIT_KHR;
  111. info.presentMode = wd->PresentMode;
  112. info.clipped = VK_TRUE;
  113. info.oldSwapchain = old_swapchain;
  114. VkSurfaceCapabilitiesKHR cap;
  115. err = vkGetPhysicalDeviceSurfaceCapabilitiesKHR(g_PhysicalDevice, wd->Surface, &cap);
  116. check_vk_result(err);
  117. if (cap.maxImageCount > 0)
  118. info.minImageCount = (cap.minImageCount + 2 < cap.maxImageCount) ? (cap.minImageCount + 2) : cap.maxImageCount;
  119. else
  120. info.minImageCount = cap.minImageCount + 2;
  121. if (cap.currentExtent.width == 0xffffffff)
  122. {
  123. info.imageExtent.width = wd->Width = w;
  124. info.imageExtent.height = wd->Height = h;
  125. }
  126. else
  127. {
  128. info.imageExtent.width = wd->Width = cap.currentExtent.width;
  129. info.imageExtent.height = wd->Height = cap.currentExtent.height;
  130. }
  131. err = vkCreateSwapchainKHR(g_Device, &info, g_Allocator, &wd->Swapchain);
  132. check_vk_result(err);
  133. err = vkGetSwapchainImagesKHR(g_Device, wd->Swapchain, &wd->BackBufferCount, NULL);
  134. check_vk_result(err);
  135. err = vkGetSwapchainImagesKHR(g_Device, wd->Swapchain, &wd->BackBufferCount, wd->BackBuffer);
  136. check_vk_result(err);
  137. }
  138. if (old_swapchain)
  139. vkDestroySwapchainKHR(g_Device, old_swapchain, g_Allocator);
  140. // Create the Render Pass
  141. {
  142. VkAttachmentDescription attachment = {};
  143. attachment.format = wd->SurfaceFormat.format;
  144. attachment.samples = VK_SAMPLE_COUNT_1_BIT;
  145. attachment.loadOp = VK_ATTACHMENT_LOAD_OP_CLEAR;
  146. attachment.storeOp = VK_ATTACHMENT_STORE_OP_STORE;
  147. attachment.stencilLoadOp = VK_ATTACHMENT_LOAD_OP_DONT_CARE;
  148. attachment.stencilStoreOp = VK_ATTACHMENT_STORE_OP_DONT_CARE;
  149. attachment.initialLayout = VK_IMAGE_LAYOUT_UNDEFINED;
  150. attachment.finalLayout = VK_IMAGE_LAYOUT_PRESENT_SRC_KHR;
  151. VkAttachmentReference color_attachment = {};
  152. color_attachment.attachment = 0;
  153. color_attachment.layout = VK_IMAGE_LAYOUT_COLOR_ATTACHMENT_OPTIMAL;
  154. VkSubpassDescription subpass = {};
  155. subpass.pipelineBindPoint = VK_PIPELINE_BIND_POINT_GRAPHICS;
  156. subpass.colorAttachmentCount = 1;
  157. subpass.pColorAttachments = &color_attachment;
  158. VkRenderPassCreateInfo info = {};
  159. info.sType = VK_STRUCTURE_TYPE_RENDER_PASS_CREATE_INFO;
  160. info.attachmentCount = 1;
  161. info.pAttachments = &attachment;
  162. info.subpassCount = 1;
  163. info.pSubpasses = &subpass;
  164. err = vkCreateRenderPass(g_Device, &info, g_Allocator, &wd->RenderPass);
  165. check_vk_result(err);
  166. }
  167. // Create The Image Views
  168. {
  169. VkImageViewCreateInfo info = {};
  170. info.sType = VK_STRUCTURE_TYPE_IMAGE_VIEW_CREATE_INFO;
  171. info.viewType = VK_IMAGE_VIEW_TYPE_2D;
  172. info.format = wd->SurfaceFormat.format;
  173. info.components.r = VK_COMPONENT_SWIZZLE_R;
  174. info.components.g = VK_COMPONENT_SWIZZLE_G;
  175. info.components.b = VK_COMPONENT_SWIZZLE_B;
  176. info.components.a = VK_COMPONENT_SWIZZLE_A;
  177. VkImageSubresourceRange image_range = { VK_IMAGE_ASPECT_COLOR_BIT, 0, 1, 0, 1 };
  178. info.subresourceRange = image_range;
  179. for (uint32_t i = 0; i < wd->BackBufferCount; i++)
  180. {
  181. info.image = wd->BackBuffer[i];
  182. err = vkCreateImageView(g_Device, &info, g_Allocator, &wd->BackBufferView[i]);
  183. check_vk_result(err);
  184. }
  185. }
  186. // Create Framebuffer
  187. {
  188. VkImageView attachment[1];
  189. VkFramebufferCreateInfo info = {};
  190. info.sType = VK_STRUCTURE_TYPE_FRAMEBUFFER_CREATE_INFO;
  191. info.renderPass = wd->RenderPass;
  192. info.attachmentCount = 1;
  193. info.pAttachments = attachment;
  194. info.width = wd->Width;
  195. info.height = wd->Height;
  196. info.layers = 1;
  197. for (uint32_t i = 0; i < wd->BackBufferCount; i++)
  198. {
  199. attachment[0] = wd->BackBufferView[i];
  200. err = vkCreateFramebuffer(g_Device, &info, g_Allocator, &wd->Framebuffer[i]);
  201. check_vk_result(err);
  202. }
  203. }
  204. }
  205. #ifdef IMGUI_VULKAN_DEBUG_REPORT
  206. static VKAPI_ATTR VkBool32 VKAPI_CALL debug_report(VkDebugReportFlagsEXT flags, VkDebugReportObjectTypeEXT objectType, uint64_t object, size_t location, int32_t messageCode, const char* pLayerPrefix, const char* pMessage, void* pUserData)
  207. {
  208. (void)flags; (void)object; (void)location; (void)messageCode; (void)pUserData; (void)pLayerPrefix; // Unused arguemnts
  209. printf("[vulkan] ObjectType: %i\nMessage: %s\n\n", objectType, pMessage);
  210. return VK_FALSE;
  211. }
  212. #endif // IMGUI_VULKAN_DEBUG_REPORT
  213. static void CreateVulkanInstance(const char** extensions, uint32_t extensions_count)
  214. {
  215. VkResult err;
  216. // Create Vulkan Instance
  217. {
  218. VkInstanceCreateInfo create_info = {};
  219. create_info.sType = VK_STRUCTURE_TYPE_INSTANCE_CREATE_INFO;
  220. create_info.enabledExtensionCount = extensions_count;
  221. create_info.ppEnabledExtensionNames = extensions;
  222. #ifdef IMGUI_VULKAN_DEBUG_REPORT
  223. // Enabling multiple validation layers grouped as LunarG standard validation
  224. const char* layers[] = { "VK_LAYER_LUNARG_standard_validation" };
  225. create_info.enabledLayerCount = 1;
  226. create_info.ppEnabledLayerNames = layers;
  227. // Enable debug report extension (we need additional storage, so we duplicate the user array to add our new extension to it)
  228. const char** extensions_ext = (const char**)malloc(sizeof(const char*) * (extensions_count + 1));
  229. memcpy(extensions_ext, extensions, extensions_count * sizeof(const char*));
  230. extensions_ext[extensions_count] = "VK_EXT_debug_report";
  231. create_info.enabledExtensionCount = extensions_count + 1;
  232. create_info.ppEnabledExtensionNames = extensions_ext;
  233. #endif // IMGUI_VULKAN_DEBUG_REPORT
  234. err = vkCreateInstance(&create_info, g_Allocator, &g_Instance);
  235. check_vk_result(err);
  236. #ifdef IMGUI_VULKAN_DEBUG_REPORT
  237. free(extensions_ext);
  238. // Get the function pointer (required for any extensions)
  239. auto vkCreateDebugReportCallbackEXT = (PFN_vkCreateDebugReportCallbackEXT)vkGetInstanceProcAddr(g_Instance, "vkCreateDebugReportCallbackEXT");
  240. IM_ASSERT(vkCreateDebugReportCallbackEXT != NULL);
  241. // Setup the debug report callback
  242. VkDebugReportCallbackCreateInfoEXT debug_report_ci = {};
  243. debug_report_ci.sType = VK_STRUCTURE_TYPE_DEBUG_REPORT_CALLBACK_CREATE_INFO_EXT;
  244. debug_report_ci.flags = VK_DEBUG_REPORT_ERROR_BIT_EXT | VK_DEBUG_REPORT_WARNING_BIT_EXT | VK_DEBUG_REPORT_PERFORMANCE_WARNING_BIT_EXT;
  245. debug_report_ci.pfnCallback = debug_report;
  246. debug_report_ci.pUserData = NULL;
  247. err = vkCreateDebugReportCallbackEXT(g_Instance, &debug_report_ci, g_Allocator, &g_DebugReport);
  248. check_vk_result(err);
  249. #endif // IMGUI_VULKAN_DEBUG_REPORT
  250. }
  251. }
  252. static void SetupVulkan(WindowData* wd)
  253. {
  254. VkResult err;
  255. // Select GPU
  256. {
  257. uint32_t gpu_count;
  258. err = vkEnumeratePhysicalDevices(g_Instance, &gpu_count, NULL);
  259. check_vk_result(err);
  260. VkPhysicalDevice* gpus = (VkPhysicalDevice*)malloc(sizeof(VkPhysicalDevice) * gpu_count);
  261. err = vkEnumeratePhysicalDevices(g_Instance, &gpu_count, gpus);
  262. check_vk_result(err);
  263. // If a number >1 of GPUs got reported, you should find the best fit GPU for your purpose
  264. // e.g. VK_PHYSICAL_DEVICE_TYPE_DISCRETE_GPU if available, or with the greatest memory available, etc.
  265. // for sake of simplicity we'll just take the first one, assuming it has a graphics queue family.
  266. g_PhysicalDevice = gpus[0];
  267. free(gpus);
  268. }
  269. // Select graphics queue family
  270. {
  271. uint32_t count;
  272. vkGetPhysicalDeviceQueueFamilyProperties(g_PhysicalDevice, &count, NULL);
  273. VkQueueFamilyProperties* queues = (VkQueueFamilyProperties*)malloc(sizeof(VkQueueFamilyProperties) * count);
  274. vkGetPhysicalDeviceQueueFamilyProperties(g_PhysicalDevice, &count, queues);
  275. for (uint32_t i = 0; i < count; i++)
  276. if (queues[i].queueFlags & VK_QUEUE_GRAPHICS_BIT)
  277. {
  278. g_QueueFamily = i;
  279. break;
  280. }
  281. free(queues);
  282. IM_ASSERT(g_QueueFamily != -1);
  283. }
  284. // Check for WSI support
  285. {
  286. VkBool32 res;
  287. vkGetPhysicalDeviceSurfaceSupportKHR(g_PhysicalDevice, g_QueueFamily, wd->Surface, &res);
  288. if (res != VK_TRUE)
  289. {
  290. fprintf(stderr, "Error no WSI support on physical device 0\n");
  291. exit(-1);
  292. }
  293. }
  294. // Get Surface Format
  295. {
  296. const VkFormat requestSurfaceImageFormat[] = { VK_FORMAT_B8G8R8A8_UNORM, VK_FORMAT_R8G8B8A8_UNORM, VK_FORMAT_B8G8R8_UNORM, VK_FORMAT_R8G8B8_UNORM };
  297. const VkColorSpaceKHR requestSurfaceColorSpace = VK_COLORSPACE_SRGB_NONLINEAR_KHR;
  298. wd->SurfaceFormat = ImGui_ImplVulkan_SelectSurfaceFormat(g_PhysicalDevice, wd->Surface, requestSurfaceImageFormat, (size_t)IM_ARRAYSIZE(requestSurfaceImageFormat), requestSurfaceColorSpace);
  299. }
  300. // Get Present Mode
  301. {
  302. #ifdef IMGUI_UNLIMITED_FRAME_RATE
  303. VkPresentModeKHR present_mode = VK_PRESENT_MODE_IMMEDIATE_KHR;
  304. #else
  305. VkPresentModeKHR present_mode = VK_PRESENT_MODE_FIFO_KHR;
  306. #endif
  307. wd->PresentMode = ImGui_ImplVulkan_SelectPresentMode(g_PhysicalDevice, wd->Surface, &present_mode, 1);
  308. }
  309. // Create Logical Device (with 1 queue)
  310. {
  311. int device_extension_count = 1;
  312. const char* device_extensions[] = { "VK_KHR_swapchain" };
  313. const float queue_priority[] = { 1.0f };
  314. VkDeviceQueueCreateInfo queue_info[1] = {};
  315. queue_info[0].sType = VK_STRUCTURE_TYPE_DEVICE_QUEUE_CREATE_INFO;
  316. queue_info[0].queueFamilyIndex = g_QueueFamily;
  317. queue_info[0].queueCount = 1;
  318. queue_info[0].pQueuePriorities = queue_priority;
  319. VkDeviceCreateInfo create_info = {};
  320. create_info.sType = VK_STRUCTURE_TYPE_DEVICE_CREATE_INFO;
  321. create_info.queueCreateInfoCount = sizeof(queue_info) / sizeof(queue_info[0]);
  322. create_info.pQueueCreateInfos = queue_info;
  323. create_info.enabledExtensionCount = device_extension_count;
  324. create_info.ppEnabledExtensionNames = device_extensions;
  325. err = vkCreateDevice(g_PhysicalDevice, &create_info, g_Allocator, &g_Device);
  326. check_vk_result(err);
  327. vkGetDeviceQueue(g_Device, g_QueueFamily, 0, &g_Queue);
  328. }
  329. // Create Command Buffers
  330. for (int i = 0; i < IMGUI_VK_QUEUED_FRAMES; i++)
  331. {
  332. FrameData* fd = &wd->Frames[i];
  333. {
  334. VkCommandPoolCreateInfo info = {};
  335. info.sType = VK_STRUCTURE_TYPE_COMMAND_POOL_CREATE_INFO;
  336. info.flags = VK_COMMAND_POOL_CREATE_RESET_COMMAND_BUFFER_BIT;
  337. info.queueFamilyIndex = g_QueueFamily;
  338. err = vkCreateCommandPool(g_Device, &info, g_Allocator, &fd->CommandPool);
  339. check_vk_result(err);
  340. }
  341. {
  342. VkCommandBufferAllocateInfo info = {};
  343. info.sType = VK_STRUCTURE_TYPE_COMMAND_BUFFER_ALLOCATE_INFO;
  344. info.commandPool = fd->CommandPool;
  345. info.level = VK_COMMAND_BUFFER_LEVEL_PRIMARY;
  346. info.commandBufferCount = 1;
  347. err = vkAllocateCommandBuffers(g_Device, &info, &fd->CommandBuffer);
  348. check_vk_result(err);
  349. }
  350. {
  351. VkFenceCreateInfo info = {};
  352. info.sType = VK_STRUCTURE_TYPE_FENCE_CREATE_INFO;
  353. info.flags = VK_FENCE_CREATE_SIGNALED_BIT;
  354. err = vkCreateFence(g_Device, &info, g_Allocator, &fd->Fence);
  355. check_vk_result(err);
  356. }
  357. {
  358. VkSemaphoreCreateInfo info = {};
  359. info.sType = VK_STRUCTURE_TYPE_SEMAPHORE_CREATE_INFO;
  360. err = vkCreateSemaphore(g_Device, &info, g_Allocator, &fd->PresentCompleteSemaphore);
  361. check_vk_result(err);
  362. err = vkCreateSemaphore(g_Device, &info, g_Allocator, &fd->RenderCompleteSemaphore);
  363. check_vk_result(err);
  364. }
  365. }
  366. // Create Descriptor Pool
  367. {
  368. VkDescriptorPoolSize pool_sizes[] =
  369. {
  370. { VK_DESCRIPTOR_TYPE_SAMPLER, 1000 },
  371. { VK_DESCRIPTOR_TYPE_COMBINED_IMAGE_SAMPLER, 1000 },
  372. { VK_DESCRIPTOR_TYPE_SAMPLED_IMAGE, 1000 },
  373. { VK_DESCRIPTOR_TYPE_STORAGE_IMAGE, 1000 },
  374. { VK_DESCRIPTOR_TYPE_UNIFORM_TEXEL_BUFFER, 1000 },
  375. { VK_DESCRIPTOR_TYPE_STORAGE_TEXEL_BUFFER, 1000 },
  376. { VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER, 1000 },
  377. { VK_DESCRIPTOR_TYPE_STORAGE_BUFFER, 1000 },
  378. { VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER_DYNAMIC, 1000 },
  379. { VK_DESCRIPTOR_TYPE_STORAGE_BUFFER_DYNAMIC, 1000 },
  380. { VK_DESCRIPTOR_TYPE_INPUT_ATTACHMENT, 1000 }
  381. };
  382. VkDescriptorPoolCreateInfo pool_info = {};
  383. pool_info.sType = VK_STRUCTURE_TYPE_DESCRIPTOR_POOL_CREATE_INFO;
  384. pool_info.flags = VK_DESCRIPTOR_POOL_CREATE_FREE_DESCRIPTOR_SET_BIT;
  385. pool_info.maxSets = 1000 * IM_ARRAYSIZE(pool_sizes);
  386. pool_info.poolSizeCount = (uint32_t)IM_ARRAYSIZE(pool_sizes);
  387. pool_info.pPoolSizes = pool_sizes;
  388. err = vkCreateDescriptorPool(g_Device, &pool_info, g_Allocator, &g_DescriptorPool);
  389. check_vk_result(err);
  390. }
  391. }
  392. static void cleanup_vulkan()
  393. {
  394. WindowData* wd = &g_WindowData;
  395. vkDestroyDescriptorPool(g_Device, g_DescriptorPool, g_Allocator);
  396. for (int i = 0; i < IMGUI_VK_QUEUED_FRAMES; i++)
  397. {
  398. FrameData* fd = &wd->Frames[i];
  399. vkDestroyFence(g_Device, fd->Fence, g_Allocator);
  400. vkFreeCommandBuffers(g_Device, fd->CommandPool, 1, &fd->CommandBuffer);
  401. vkDestroyCommandPool(g_Device, fd->CommandPool, g_Allocator);
  402. vkDestroySemaphore(g_Device, fd->PresentCompleteSemaphore, g_Allocator);
  403. vkDestroySemaphore(g_Device, fd->RenderCompleteSemaphore, g_Allocator);
  404. }
  405. for (uint32_t i = 0; i < wd->BackBufferCount; i++)
  406. {
  407. vkDestroyImageView(g_Device, wd->BackBufferView[i], g_Allocator);
  408. vkDestroyFramebuffer(g_Device, wd->Framebuffer[i], g_Allocator);
  409. }
  410. vkDestroyRenderPass(g_Device, wd->RenderPass, g_Allocator);
  411. vkDestroySwapchainKHR(g_Device, wd->Swapchain, g_Allocator);
  412. vkDestroySurfaceKHR(g_Instance, wd->Surface, g_Allocator);
  413. #ifdef IMGUI_VULKAN_DEBUG_REPORT
  414. // Remove the debug report callback
  415. auto vkDestroyDebugReportCallbackEXT = (PFN_vkDestroyDebugReportCallbackEXT)vkGetInstanceProcAddr(g_Instance, "vkDestroyDebugReportCallbackEXT");
  416. vkDestroyDebugReportCallbackEXT(g_Instance, g_DebugReport, g_Allocator);
  417. #endif // IMGUI_VULKAN_DEBUG_REPORT
  418. vkDestroyDevice(g_Device, g_Allocator);
  419. vkDestroyInstance(g_Instance, g_Allocator);
  420. }
  421. static void frame_begin(WindowData* wd)
  422. {
  423. FrameData* fd = &wd->Frames[wd->FrameIndex];
  424. VkResult err;
  425. for (;;)
  426. {
  427. err = vkWaitForFences(g_Device, 1, &fd->Fence, VK_TRUE, 100);
  428. if (err == VK_SUCCESS) break;
  429. if (err == VK_TIMEOUT) continue;
  430. check_vk_result(err);
  431. }
  432. {
  433. err = vkAcquireNextImageKHR(g_Device, wd->Swapchain, UINT64_MAX, fd->PresentCompleteSemaphore, VK_NULL_HANDLE, &fd->BackbufferIndex);
  434. check_vk_result(err);
  435. }
  436. {
  437. err = vkResetCommandPool(g_Device, fd->CommandPool, 0);
  438. check_vk_result(err);
  439. VkCommandBufferBeginInfo info = {};
  440. info.sType = VK_STRUCTURE_TYPE_COMMAND_BUFFER_BEGIN_INFO;
  441. info.flags |= VK_COMMAND_BUFFER_USAGE_ONE_TIME_SUBMIT_BIT;
  442. err = vkBeginCommandBuffer(fd->CommandBuffer, &info);
  443. check_vk_result(err);
  444. }
  445. {
  446. VkRenderPassBeginInfo info = {};
  447. info.sType = VK_STRUCTURE_TYPE_RENDER_PASS_BEGIN_INFO;
  448. info.renderPass = wd->RenderPass;
  449. info.framebuffer = wd->Framebuffer[fd->BackbufferIndex];
  450. info.renderArea.extent.width = wd->Width;
  451. info.renderArea.extent.height = wd->Height;
  452. info.clearValueCount = 1;
  453. info.pClearValues = &wd->ClearValue;
  454. vkCmdBeginRenderPass(fd->CommandBuffer, &info, VK_SUBPASS_CONTENTS_INLINE);
  455. }
  456. }
  457. static void frame_end(WindowData* wd)
  458. {
  459. FrameData* fd = &wd->Frames[wd->FrameIndex];
  460. VkResult err;
  461. vkCmdEndRenderPass(fd->CommandBuffer);
  462. {
  463. VkPipelineStageFlags wait_stage = VK_PIPELINE_STAGE_COLOR_ATTACHMENT_OUTPUT_BIT;
  464. VkSubmitInfo info = {};
  465. info.sType = VK_STRUCTURE_TYPE_SUBMIT_INFO;
  466. info.waitSemaphoreCount = 1;
  467. info.pWaitSemaphores = &fd->PresentCompleteSemaphore;
  468. info.pWaitDstStageMask = &wait_stage;
  469. info.commandBufferCount = 1;
  470. info.pCommandBuffers = &fd->CommandBuffer;
  471. info.signalSemaphoreCount = 1;
  472. info.pSignalSemaphores = &fd->RenderCompleteSemaphore;
  473. err = vkEndCommandBuffer(fd->CommandBuffer);
  474. check_vk_result(err);
  475. err = vkResetFences(g_Device, 1, &fd->Fence);
  476. check_vk_result(err);
  477. err = vkQueueSubmit(g_Queue, 1, &info, fd->Fence);
  478. check_vk_result(err);
  479. }
  480. }
  481. static void frame_present(WindowData* wd)
  482. {
  483. VkResult err;
  484. // If IMGUI_UNLIMITED_FRAME_RATE is defined we present the latest but one frame. Otherwise we present the latest rendered frame
  485. #ifdef IMGUI_UNLIMITED_FRAME_RATE
  486. uint32_t PresentIndex = (wd->FrameIndex + IMGUI_VK_QUEUED_FRAMES - 1) % IMGUI_VK_QUEUED_FRAMES;
  487. #else
  488. uint32_t PresentIndex = g_FrameIndex;
  489. #endif // IMGUI_UNLIMITED_FRAME_RATE
  490. FrameData* fd = &wd->Frames[PresentIndex];
  491. VkPresentInfoKHR info = {};
  492. info.sType = VK_STRUCTURE_TYPE_PRESENT_INFO_KHR;
  493. info.waitSemaphoreCount = 1;
  494. info.pWaitSemaphores = &fd->RenderCompleteSemaphore;
  495. info.swapchainCount = 1;
  496. info.pSwapchains = &wd->Swapchain;
  497. info.pImageIndices = &fd->BackbufferIndex;
  498. err = vkQueuePresentKHR(g_Queue, &info);
  499. check_vk_result(err);
  500. }
  501. int main(int, char**)
  502. {
  503. // Setup SDL
  504. if (SDL_Init(SDL_INIT_VIDEO|SDL_INIT_TIMER) != 0)
  505. {
  506. printf("Error: %s\n", SDL_GetError());
  507. return 1;
  508. }
  509. // Setup window
  510. SDL_DisplayMode current;
  511. SDL_GetCurrentDisplayMode(0, &current);
  512. SDL_Window* window = SDL_CreateWindow("ImGui SDL2+Vulkan example", SDL_WINDOWPOS_CENTERED, SDL_WINDOWPOS_CENTERED, 1280, 720, SDL_WINDOW_VULKAN|SDL_WINDOW_RESIZABLE);
  513. // Setup Vulkan
  514. uint32_t sdl_extensions_count;
  515. SDL_Vulkan_GetInstanceExtensions(window, &sdl_extensions_count, NULL);
  516. const char** sdl_extensions = new const char*[sdl_extensions_count];
  517. SDL_Vulkan_GetInstanceExtensions(window, &sdl_extensions_count, sdl_extensions);
  518. CreateVulkanInstance(sdl_extensions, sdl_extensions_count);
  519. delete[] sdl_extensions;
  520. // Create Window Surface
  521. WindowData* wd = &g_WindowData;
  522. SDL_bool result = SDL_Vulkan_CreateSurface(window, g_Instance, &wd->Surface);
  523. if (result == 0)
  524. {
  525. printf("Failed to create Vulkan surface.\n");
  526. return 1;
  527. }
  528. SetupVulkan(wd);
  529. // Create Framebuffers
  530. {
  531. int w, h;
  532. SDL_GetWindowSize(window, &w, &h);
  533. CreateOrResizeSwapChainAndFrameBuffer(wd, w, h);
  534. }
  535. // Setup ImGui binding
  536. ImGui::CreateContext();
  537. ImGui_ImplVulkan_InitInfo init_info = {};
  538. init_info.Allocator = g_Allocator;
  539. init_info.PhysicalDevice = g_PhysicalDevice;
  540. init_info.Device = g_Device;
  541. init_info.RenderPass = wd->RenderPass;
  542. init_info.PipelineCache = g_PipelineCache;
  543. init_info.DescriptorPool = g_DescriptorPool;
  544. init_info.CheckVkResultFn = check_vk_result;
  545. ImGuiIO& io = ImGui::GetIO(); (void)io;
  546. //io.ConfigFlags |= ImGuiConfigFlags_MultiViewports;
  547. //io.ConfigFlags |= ImGuiConfigFlags_NavEnableKeyboard; // Enable Keyboard Controls
  548. ImGui_ImplVulkan_Init(&init_info);
  549. ImGui_ImplSDL2_Init(window, NULL);
  550. // Setup style
  551. ImGui::StyleColorsDark();
  552. //ImGui::StyleColorsClassic();
  553. // Load Fonts
  554. // - If no fonts are loaded, dear imgui will use the default font. You can also load multiple fonts and use ImGui::PushFont()/PopFont() to select them.
  555. // - AddFontFromFileTTF() will return the ImFont* so you can store it if you need to select the font among multiple.
  556. // - If the file cannot be loaded, the function will return NULL. Please handle those errors in your application (e.g. use an assertion, or display an error and quit).
  557. // - The fonts will be rasterized at a given size (w/ oversampling) and stored into a texture when calling ImFontAtlas::Build()/GetTexDataAsXXXX(), which ImGui_ImplXXXX_NewFrame below will call.
  558. // - Read 'misc/fonts/README.txt' for more instructions and details.
  559. // - Remember that in C/C++ if you want to include a backslash \ in a string literal you need to write a double backslash \\ !
  560. //io.Fonts->AddFontDefault();
  561. //io.Fonts->AddFontFromFileTTF("../../misc/fonts/Roboto-Medium.ttf", 16.0f);
  562. //io.Fonts->AddFontFromFileTTF("../../misc/fonts/Cousine-Regular.ttf", 15.0f);
  563. //io.Fonts->AddFontFromFileTTF("../../misc/fonts/DroidSans.ttf", 16.0f);
  564. //io.Fonts->AddFontFromFileTTF("../../misc/fonts/ProggyTiny.ttf", 10.0f);
  565. //ImFont* font = io.Fonts->AddFontFromFileTTF("c:\\Windows\\Fonts\\ArialUni.ttf", 18.0f, NULL, io.Fonts->GetGlyphRangesJapanese());
  566. //IM_ASSERT(font != NULL);
  567. // Upload Fonts
  568. {
  569. // Use any command queue
  570. VkCommandPool command_pool = wd->Frames[wd->FrameIndex].CommandPool;
  571. VkCommandBuffer command_buffer = wd->Frames[wd->FrameIndex].CommandBuffer;
  572. VkResult err;
  573. err = vkResetCommandPool(g_Device, command_pool, 0);
  574. check_vk_result(err);
  575. VkCommandBufferBeginInfo begin_info = {};
  576. begin_info.sType = VK_STRUCTURE_TYPE_COMMAND_BUFFER_BEGIN_INFO;
  577. begin_info.flags |= VK_COMMAND_BUFFER_USAGE_ONE_TIME_SUBMIT_BIT;
  578. err = vkBeginCommandBuffer(command_buffer, &begin_info);
  579. check_vk_result(err);
  580. ImGui_ImplVulkan_CreateFontsTexture(command_buffer);
  581. VkSubmitInfo end_info = {};
  582. end_info.sType = VK_STRUCTURE_TYPE_SUBMIT_INFO;
  583. end_info.commandBufferCount = 1;
  584. end_info.pCommandBuffers = &command_buffer;
  585. err = vkEndCommandBuffer(command_buffer);
  586. check_vk_result(err);
  587. err = vkQueueSubmit(g_Queue, 1, &end_info, VK_NULL_HANDLE);
  588. check_vk_result(err);
  589. err = vkDeviceWaitIdle(g_Device);
  590. check_vk_result(err);
  591. ImGui_ImplVulkan_InvalidateFontUploadObjects();
  592. }
  593. bool show_demo_window = true;
  594. bool show_another_window = false;
  595. ImVec4 clear_color = ImVec4(0.45f, 0.55f, 0.60f, 1.00f);
  596. bool swap_chain_has_at_least_one_image = false;
  597. // Main loop
  598. bool done = false;
  599. while (!done)
  600. {
  601. // You can read the io.WantCaptureMouse, io.WantCaptureKeyboard flags to tell if dear imgui wants to use your inputs.
  602. // - When io.WantCaptureMouse is true, do not dispatch mouse input data to your main application.
  603. // - When io.WantCaptureKeyboard is true, do not dispatch keyboard input data to your main application.
  604. // Generally you may always pass all inputs to dear imgui, and hide them from your application based on those two flags.
  605. SDL_Event event;
  606. while (SDL_PollEvent(&event))
  607. {
  608. ImGui_ImplSDL2_ProcessEvent(&event);
  609. if (event.type == SDL_QUIT)
  610. done = true;
  611. if (event.type == SDL_WINDOWEVENT && event.window.event == SDL_WINDOWEVENT_RESIZED && event.window.windowID == SDL_GetWindowID(window))
  612. CreateOrResizeSwapChainAndFrameBuffer(wd, (int)event.window.data1, (int)event.window.data2);
  613. }
  614. ImGui_ImplVulkan_NewFrame();
  615. ImGui_ImplSDL2_NewFrame(window);
  616. // 1. Show a simple window.
  617. // Tip: if we don't call ImGui::Begin()/ImGui::End() the widgets automatically appears in a window called "Debug".
  618. {
  619. static float f = 0.0f;
  620. static int counter = 0;
  621. ImGui::Text("Hello, world!"); // Display some text (you can use a format string too)
  622. ImGui::SliderFloat("float", &f, 0.0f, 1.0f); // Edit 1 float using a slider from 0.0f to 1.0f
  623. ImGui::ColorEdit3("clear color", (float*)&clear_color); // Edit 3 floats representing a color
  624. ImGui::Checkbox("Demo Window", &show_demo_window); // Edit bools storing our windows open/close state
  625. ImGui::Checkbox("Another Window", &show_another_window);
  626. if (ImGui::Button("Button")) // Buttons return true when clicked (NB: most widgets return true when edited/activated)
  627. counter++;
  628. ImGui::SameLine();
  629. ImGui::Text("counter = %d", counter);
  630. ImGui::Text("Application average %.3f ms/frame (%.1f FPS)", 1000.0f / ImGui::GetIO().Framerate, ImGui::GetIO().Framerate);
  631. }
  632. // 2. Show another simple window. In most cases you will use an explicit Begin/End pair to name your windows.
  633. if (show_another_window)
  634. {
  635. ImGui::Begin("Another Window", &show_another_window);
  636. ImGui::Text("Hello from another window!");
  637. if (ImGui::Button("Close Me"))
  638. show_another_window = false;
  639. ImGui::End();
  640. }
  641. // 3. Show the ImGui demo window. Most of the sample code is in ImGui::ShowDemoWindow(). Read its code to learn more about Dear ImGui!
  642. if (show_demo_window)
  643. {
  644. ImGui::SetNextWindowPos(ImVec2(650, 20), ImGuiCond_FirstUseEver); // Normally user code doesn't need/want to call this because positions are saved in .ini file anyway. Here we just want to make the demo initial state a bit more friendly!
  645. ImGui::ShowDemoWindow(&show_demo_window);
  646. }
  647. // Rendering
  648. memcpy(&wd->ClearValue.color.float32[0], &clear_color, 4 * sizeof(float));
  649. frame_begin(wd);
  650. ImGui_ImplVulkan_Render(wd->Frames[wd->FrameIndex].CommandBuffer);
  651. frame_end(wd);
  652. #ifdef IMGUI_UNLIMITED_FRAME_RATE
  653. // When IMGUI_UNLIMITED_FRAME_RATE is defined we render into latest image acquired from the swapchain but we display the image which was rendered before.
  654. // Hence we must render once and increase the FrameIndex without presenting.
  655. if (swap_chain_has_at_least_one_image)
  656. frame_present(wd);
  657. #else
  658. frame_present(wd);
  659. #endif
  660. swap_chain_has_at_least_one_image = true;
  661. wd->FrameIndex = (wd->FrameIndex + 1) % IMGUI_VK_QUEUED_FRAMES;
  662. ImGui::RenderAdditionalViewports();
  663. }
  664. // Cleanup
  665. VkResult err = vkDeviceWaitIdle(g_Device);
  666. check_vk_result(err);
  667. ImGui_ImplVulkan_Shutdown();
  668. ImGui_ImplSDL2_Shutdown();
  669. ImGui::DestroyContext();
  670. SDL_DestroyWindow(window);
  671. cleanup_vulkan();
  672. SDL_Quit();
  673. return 0;
  674. }