main.cpp 30 KB

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