main.cpp 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602
  1. // ImGui - standalone example application for Glfw + Vulkan, using programmable pipeline
  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 <stdio.h> // printf, fprintf
  5. #include <stdlib.h> // abort
  6. #define GLFW_INCLUDE_NONE
  7. #define GLFW_INCLUDE_VULKAN
  8. #include <GLFW/glfw3.h>
  9. #include "imgui_impl_glfw_vulkan.h"
  10. #define IMGUI_MAX_POSSIBLE_BACK_BUFFERS 16
  11. #define IMGUI_UNLIMITED_FRAME_RATE
  12. static VkAllocationCallbacks* g_Allocator = NULL;
  13. static VkInstance g_Instance = VK_NULL_HANDLE;
  14. static VkSurfaceKHR g_Surface = VK_NULL_HANDLE;
  15. static VkPhysicalDevice g_Gpu = VK_NULL_HANDLE;
  16. static VkDevice g_Device = VK_NULL_HANDLE;
  17. static VkSwapchainKHR g_Swapchain = VK_NULL_HANDLE;
  18. static VkRenderPass g_RenderPass = VK_NULL_HANDLE;
  19. static uint32_t g_QueueFamily = 0;
  20. static VkQueue g_Queue = VK_NULL_HANDLE;
  21. static VkFormat g_ImageFormat = VK_FORMAT_B8G8R8A8_UNORM;
  22. static VkFormat g_ViewFormat = VK_FORMAT_B8G8R8A8_UNORM;
  23. static VkColorSpaceKHR g_ColorSpace = VK_COLORSPACE_SRGB_NONLINEAR_KHR;
  24. static VkImageSubresourceRange g_ImageRange = {VK_IMAGE_ASPECT_COLOR_BIT, 0, 1, 0, 1};
  25. static VkPipelineCache g_PipelineCache = VK_NULL_HANDLE;
  26. static VkDescriptorPool g_DescriptorPool = VK_NULL_HANDLE;
  27. static int fb_width, fb_height;
  28. static uint32_t g_BackBufferIndex = 0;
  29. static uint32_t g_BackBufferCount = 0;
  30. static VkImage g_BackBuffer[IMGUI_MAX_POSSIBLE_BACK_BUFFERS] = {};
  31. static VkImageView g_BackBufferView[IMGUI_MAX_POSSIBLE_BACK_BUFFERS] = {};
  32. static VkFramebuffer g_Framebuffer[IMGUI_MAX_POSSIBLE_BACK_BUFFERS] = {};
  33. static uint32_t g_FrameIndex = 0;
  34. static VkCommandPool g_CommandPool[IMGUI_VK_QUEUED_FRAMES];
  35. static VkCommandBuffer g_CommandBuffer[IMGUI_VK_QUEUED_FRAMES];
  36. static VkFence g_Fence[IMGUI_VK_QUEUED_FRAMES];
  37. static VkSemaphore g_Semaphore[IMGUI_VK_QUEUED_FRAMES];
  38. static VkClearValue g_ClearValue = {};
  39. static void check_vk_result(VkResult err)
  40. {
  41. if (err == 0) return;
  42. printf("VkResult %d\n", err);
  43. if (err < 0)
  44. abort();
  45. }
  46. static void resize_vulkan(GLFWwindow* /*window*/, int w, int h)
  47. {
  48. VkResult err;
  49. VkSwapchainKHR old_swapchain = g_Swapchain;
  50. err = vkDeviceWaitIdle(g_Device);
  51. check_vk_result(err);
  52. // Destroy old Framebuffer:
  53. for (uint32_t i = 0; i < g_BackBufferCount; i++)
  54. if (g_BackBufferView[i])
  55. vkDestroyImageView(g_Device, g_BackBufferView[i], g_Allocator);
  56. for (uint32_t i = 0; i < g_BackBufferCount; i++)
  57. if (g_Framebuffer[i])
  58. vkDestroyFramebuffer(g_Device, g_Framebuffer[i], g_Allocator);
  59. if (g_RenderPass)
  60. vkDestroyRenderPass(g_Device, g_RenderPass, g_Allocator);
  61. // Create Swapchain:
  62. {
  63. VkSwapchainCreateInfoKHR info = {};
  64. info.sType = VK_STRUCTURE_TYPE_SWAPCHAIN_CREATE_INFO_KHR;
  65. info.surface = g_Surface;
  66. info.imageFormat = g_ImageFormat;
  67. info.imageColorSpace = g_ColorSpace;
  68. info.imageArrayLayers = 1;
  69. info.imageUsage |= VK_IMAGE_USAGE_COLOR_ATTACHMENT_BIT;
  70. info.imageSharingMode = VK_SHARING_MODE_EXCLUSIVE;
  71. info.preTransform = VK_SURFACE_TRANSFORM_IDENTITY_BIT_KHR;
  72. info.compositeAlpha = VK_COMPOSITE_ALPHA_OPAQUE_BIT_KHR;
  73. #ifdef IMGUI_UNLIMITED_FRAME_RATE
  74. info.presentMode = VK_PRESENT_MODE_IMMEDIATE_KHR;
  75. #else
  76. info.presentMode = VK_PRESENT_MODE_FIFO_KHR;
  77. #endif
  78. info.clipped = VK_TRUE;
  79. info.oldSwapchain = old_swapchain;
  80. VkSurfaceCapabilitiesKHR cap;
  81. err = vkGetPhysicalDeviceSurfaceCapabilitiesKHR(g_Gpu, g_Surface, &cap);
  82. check_vk_result(err);
  83. if (cap.maxImageCount > 0)
  84. info.minImageCount = (cap.minImageCount + 2 < cap.maxImageCount) ? (cap.minImageCount + 2) : cap.maxImageCount;
  85. else
  86. info.minImageCount = cap.minImageCount + 2;
  87. if (cap.currentExtent.width == 0xffffffff)
  88. {
  89. fb_width = w;
  90. fb_height = h;
  91. info.imageExtent.width = fb_width;
  92. info.imageExtent.height = fb_height;
  93. }
  94. else
  95. {
  96. fb_width = cap.currentExtent.width;
  97. fb_height = cap.currentExtent.height;
  98. info.imageExtent.width = fb_width;
  99. info.imageExtent.height = fb_height;
  100. }
  101. err = vkCreateSwapchainKHR(g_Device, &info, g_Allocator, &g_Swapchain);
  102. check_vk_result(err);
  103. err = vkGetSwapchainImagesKHR(g_Device, g_Swapchain, &g_BackBufferCount, NULL);
  104. check_vk_result(err);
  105. err = vkGetSwapchainImagesKHR(g_Device, g_Swapchain, &g_BackBufferCount, g_BackBuffer);
  106. check_vk_result(err);
  107. }
  108. if (old_swapchain)
  109. vkDestroySwapchainKHR(g_Device, old_swapchain, g_Allocator);
  110. // Create the Render Pass:
  111. {
  112. VkAttachmentDescription attachment = {};
  113. attachment.format = g_ViewFormat;
  114. attachment.samples = VK_SAMPLE_COUNT_1_BIT;
  115. attachment.loadOp = VK_ATTACHMENT_LOAD_OP_CLEAR;
  116. attachment.storeOp = VK_ATTACHMENT_STORE_OP_STORE;
  117. attachment.stencilLoadOp = VK_ATTACHMENT_LOAD_OP_DONT_CARE;
  118. attachment.stencilStoreOp = VK_ATTACHMENT_STORE_OP_DONT_CARE;
  119. attachment.initialLayout = VK_IMAGE_LAYOUT_UNDEFINED;
  120. attachment.finalLayout = VK_IMAGE_LAYOUT_COLOR_ATTACHMENT_OPTIMAL;
  121. VkAttachmentReference color_attachment = {};
  122. color_attachment.attachment = 0;
  123. color_attachment.layout = VK_IMAGE_LAYOUT_COLOR_ATTACHMENT_OPTIMAL;
  124. VkSubpassDescription subpass = {};
  125. subpass.pipelineBindPoint = VK_PIPELINE_BIND_POINT_GRAPHICS;
  126. subpass.colorAttachmentCount = 1;
  127. subpass.pColorAttachments = &color_attachment;
  128. VkRenderPassCreateInfo info = {};
  129. info.sType = VK_STRUCTURE_TYPE_RENDER_PASS_CREATE_INFO;
  130. info.attachmentCount = 1;
  131. info.pAttachments = &attachment;
  132. info.subpassCount = 1;
  133. info.pSubpasses = &subpass;
  134. err = vkCreateRenderPass(g_Device, &info, g_Allocator, &g_RenderPass);
  135. check_vk_result(err);
  136. }
  137. // Create The Image Views
  138. {
  139. VkImageViewCreateInfo info = {};
  140. info.sType = VK_STRUCTURE_TYPE_IMAGE_VIEW_CREATE_INFO;
  141. info.viewType = VK_IMAGE_VIEW_TYPE_2D;
  142. info.format = g_ViewFormat;
  143. info.components.r = VK_COMPONENT_SWIZZLE_R;
  144. info.components.g = VK_COMPONENT_SWIZZLE_G;
  145. info.components.b = VK_COMPONENT_SWIZZLE_B;
  146. info.components.a = VK_COMPONENT_SWIZZLE_A;
  147. info.subresourceRange = g_ImageRange;
  148. for (uint32_t i = 0; i < g_BackBufferCount; i++)
  149. {
  150. info.image = g_BackBuffer[i];
  151. err = vkCreateImageView(g_Device, &info, g_Allocator, &g_BackBufferView[i]);
  152. check_vk_result(err);
  153. }
  154. }
  155. // Create Framebuffer:
  156. {
  157. VkImageView attachment[1];
  158. VkFramebufferCreateInfo info = {};
  159. info.sType = VK_STRUCTURE_TYPE_FRAMEBUFFER_CREATE_INFO;
  160. info.renderPass = g_RenderPass;
  161. info.attachmentCount = 1;
  162. info.pAttachments = attachment;
  163. info.width = fb_width;
  164. info.height = fb_height;
  165. info.layers = 1;
  166. for (uint32_t i = 0; i < g_BackBufferCount; i++)
  167. {
  168. attachment[0] = g_BackBufferView[i];
  169. err = vkCreateFramebuffer(g_Device, &info, g_Allocator, &g_Framebuffer[i]);
  170. check_vk_result(err);
  171. }
  172. }
  173. }
  174. static void setup_vulkan(GLFWwindow* window)
  175. {
  176. VkResult err;
  177. // Create Vulkan Instance
  178. {
  179. uint32_t glfw_extensions_count;
  180. const char** glfw_extensions = glfwGetRequiredInstanceExtensions(&glfw_extensions_count);
  181. VkInstanceCreateInfo create_info = {};
  182. create_info.sType = VK_STRUCTURE_TYPE_INSTANCE_CREATE_INFO;
  183. create_info.enabledExtensionCount = glfw_extensions_count;
  184. create_info.ppEnabledExtensionNames = glfw_extensions;
  185. err = vkCreateInstance(&create_info, g_Allocator, &g_Instance);
  186. check_vk_result(err);
  187. }
  188. // Create Window Surface
  189. {
  190. err = glfwCreateWindowSurface(g_Instance, window, g_Allocator, &g_Surface);
  191. check_vk_result(err);
  192. }
  193. // Get GPU (WARNING here we assume the first gpu is one we can use)
  194. {
  195. uint32_t count = 1;
  196. err = vkEnumeratePhysicalDevices(g_Instance, &count, &g_Gpu);
  197. check_vk_result(err);
  198. }
  199. // Get queue
  200. {
  201. uint32_t count;
  202. vkGetPhysicalDeviceQueueFamilyProperties(g_Gpu, &count, NULL);
  203. VkQueueFamilyProperties* queues = (VkQueueFamilyProperties*)malloc(sizeof(VkQueueFamilyProperties) * count);
  204. vkGetPhysicalDeviceQueueFamilyProperties(g_Gpu, &count, queues);
  205. for (uint32_t i = 0; i < count; i++)
  206. {
  207. if (queues[i].queueFlags & VK_QUEUE_GRAPHICS_BIT)
  208. {
  209. g_QueueFamily = i;
  210. break;
  211. }
  212. }
  213. free(queues);
  214. }
  215. // Check for WSI support
  216. {
  217. VkBool32 res;
  218. vkGetPhysicalDeviceSurfaceSupportKHR(g_Gpu, g_QueueFamily, g_Surface, &res);
  219. if (res != VK_TRUE)
  220. {
  221. fprintf(stderr, "Error no WSI support on physical device 0\n");
  222. exit(-1);
  223. }
  224. }
  225. // Get Surface Format
  226. {
  227. VkFormat image_view_format[][2] = {{VK_FORMAT_B8G8R8A8_UNORM, VK_FORMAT_B8G8R8A8_UNORM}, {VK_FORMAT_B8G8R8A8_SRGB, VK_FORMAT_B8G8R8A8_UNORM}};
  228. uint32_t count;
  229. vkGetPhysicalDeviceSurfaceFormatsKHR(g_Gpu, g_Surface, &count, NULL);
  230. VkSurfaceFormatKHR *formats = (VkSurfaceFormatKHR*)malloc(sizeof(VkSurfaceFormatKHR) * count);
  231. vkGetPhysicalDeviceSurfaceFormatsKHR(g_Gpu, g_Surface, &count, formats);
  232. for (size_t i = 0; i < sizeof(image_view_format) / sizeof(image_view_format[0]); i++)
  233. {
  234. for (uint32_t j = 0; j < count; j++)
  235. {
  236. if (formats[j].format == image_view_format[i][0])
  237. {
  238. g_ImageFormat = image_view_format[i][0];
  239. g_ViewFormat = image_view_format[i][1];
  240. g_ColorSpace = formats[j].colorSpace;
  241. }
  242. }
  243. }
  244. free(formats);
  245. }
  246. // Create Logical Device
  247. {
  248. int device_extension_count = 1;
  249. const char* device_extensions[] = {"VK_KHR_swapchain"};
  250. const uint32_t queue_index = 0;
  251. const uint32_t queue_count = 1;
  252. const float queue_priority[] = {1.0f};
  253. VkDeviceQueueCreateInfo queue_info[1] = {};
  254. queue_info[0].sType = VK_STRUCTURE_TYPE_DEVICE_QUEUE_CREATE_INFO;
  255. queue_info[0].queueFamilyIndex = g_QueueFamily;
  256. queue_info[0].queueCount = queue_count;
  257. queue_info[0].pQueuePriorities = queue_priority;
  258. VkDeviceCreateInfo create_info = {};
  259. create_info.sType = VK_STRUCTURE_TYPE_DEVICE_CREATE_INFO;
  260. create_info.queueCreateInfoCount = sizeof(queue_info)/sizeof(queue_info[0]);
  261. create_info.pQueueCreateInfos = queue_info;
  262. create_info.enabledExtensionCount = device_extension_count;
  263. create_info.ppEnabledExtensionNames = device_extensions;
  264. err = vkCreateDevice(g_Gpu, &create_info, g_Allocator, &g_Device);
  265. check_vk_result(err);
  266. vkGetDeviceQueue(g_Device, g_QueueFamily, queue_index, &g_Queue);
  267. }
  268. // Create Framebuffers
  269. {
  270. int w, h;
  271. glfwGetFramebufferSize(window, &w, &h);
  272. resize_vulkan(window, w, h);
  273. glfwSetFramebufferSizeCallback(window, resize_vulkan);
  274. }
  275. // Create Command Buffers
  276. for (int i = 0; i < IMGUI_VK_QUEUED_FRAMES; i++)
  277. {
  278. {
  279. VkCommandPoolCreateInfo info = {};
  280. info.sType = VK_STRUCTURE_TYPE_COMMAND_POOL_CREATE_INFO;
  281. info.flags = VK_COMMAND_POOL_CREATE_RESET_COMMAND_BUFFER_BIT;
  282. info.queueFamilyIndex = g_QueueFamily;
  283. err = vkCreateCommandPool(g_Device, &info, g_Allocator, &g_CommandPool[i]);
  284. check_vk_result(err);
  285. }
  286. {
  287. VkCommandBufferAllocateInfo info = {};
  288. info.sType = VK_STRUCTURE_TYPE_COMMAND_BUFFER_ALLOCATE_INFO;
  289. info.commandPool = g_CommandPool[i];
  290. info.level = VK_COMMAND_BUFFER_LEVEL_PRIMARY;
  291. info.commandBufferCount = 1;
  292. err = vkAllocateCommandBuffers(g_Device, &info, &g_CommandBuffer[i]);
  293. check_vk_result(err);
  294. }
  295. {
  296. VkFenceCreateInfo info = {};
  297. info.sType = VK_STRUCTURE_TYPE_FENCE_CREATE_INFO;
  298. info.flags = VK_FENCE_CREATE_SIGNALED_BIT;
  299. err = vkCreateFence(g_Device, &info, g_Allocator, &g_Fence[i]);
  300. check_vk_result(err);
  301. }
  302. {
  303. VkSemaphoreCreateInfo info = {};
  304. info.sType = VK_STRUCTURE_TYPE_SEMAPHORE_CREATE_INFO;
  305. err = vkCreateSemaphore(g_Device, &info, g_Allocator, &g_Semaphore[i]);
  306. check_vk_result(err);
  307. }
  308. }
  309. // Create Descriptor Pool
  310. {
  311. VkDescriptorPoolSize pool_size[11] =
  312. {
  313. { VK_DESCRIPTOR_TYPE_SAMPLER, 1000 },
  314. { VK_DESCRIPTOR_TYPE_COMBINED_IMAGE_SAMPLER, 1000 },
  315. { VK_DESCRIPTOR_TYPE_SAMPLED_IMAGE, 1000 },
  316. { VK_DESCRIPTOR_TYPE_STORAGE_IMAGE, 1000 },
  317. { VK_DESCRIPTOR_TYPE_UNIFORM_TEXEL_BUFFER, 1000 },
  318. { VK_DESCRIPTOR_TYPE_STORAGE_TEXEL_BUFFER, 1000 },
  319. { VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER, 1000 },
  320. { VK_DESCRIPTOR_TYPE_STORAGE_BUFFER, 1000 },
  321. { VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER_DYNAMIC, 1000 },
  322. { VK_DESCRIPTOR_TYPE_STORAGE_BUFFER_DYNAMIC, 1000 },
  323. { VK_DESCRIPTOR_TYPE_INPUT_ATTACHMENT, 1000 }
  324. };
  325. VkDescriptorPoolCreateInfo pool_info = {};
  326. pool_info.sType = VK_STRUCTURE_TYPE_DESCRIPTOR_POOL_CREATE_INFO;
  327. pool_info.flags = VK_DESCRIPTOR_POOL_CREATE_FREE_DESCRIPTOR_SET_BIT;
  328. pool_info.maxSets = 1000 * 11;
  329. pool_info.poolSizeCount = 11;
  330. pool_info.pPoolSizes = pool_size;
  331. err = vkCreateDescriptorPool(g_Device, &pool_info, g_Allocator, &g_DescriptorPool);
  332. check_vk_result(err);
  333. }
  334. }
  335. static void cleanup_vulkan()
  336. {
  337. vkDestroyDescriptorPool(g_Device, g_DescriptorPool, g_Allocator);
  338. for (int i = 0; i < IMGUI_VK_QUEUED_FRAMES; i++)
  339. {
  340. vkDestroyFence(g_Device, g_Fence[i], g_Allocator);
  341. vkFreeCommandBuffers(g_Device, g_CommandPool[i], 1, &g_CommandBuffer[i]);
  342. vkDestroyCommandPool(g_Device, g_CommandPool[i], g_Allocator);
  343. vkDestroySemaphore(g_Device, g_Semaphore[i], g_Allocator);
  344. }
  345. for (uint32_t i = 0; i < g_BackBufferCount; i++)
  346. {
  347. vkDestroyImageView(g_Device, g_BackBufferView[i], g_Allocator);
  348. vkDestroyFramebuffer(g_Device, g_Framebuffer[i], g_Allocator);
  349. }
  350. vkDestroyRenderPass(g_Device, g_RenderPass, g_Allocator);
  351. vkDestroySwapchainKHR(g_Device, g_Swapchain, g_Allocator);
  352. vkDestroySurfaceKHR(g_Instance, g_Surface, g_Allocator);
  353. vkDestroyDevice(g_Device, g_Allocator);
  354. vkDestroyInstance(g_Instance, g_Allocator);
  355. }
  356. static void frame_begin()
  357. {
  358. VkResult err;
  359. while (true)
  360. {
  361. err = vkWaitForFences(g_Device, 1, &g_Fence[g_FrameIndex], VK_TRUE, 100);
  362. if (err == VK_SUCCESS) break;
  363. if (err == VK_TIMEOUT) continue;
  364. check_vk_result(err);
  365. }
  366. {
  367. err = vkAcquireNextImageKHR(g_Device, g_Swapchain, UINT64_MAX, g_Semaphore[g_FrameIndex], VK_NULL_HANDLE, &g_BackBufferIndex);
  368. check_vk_result(err);
  369. }
  370. {
  371. err = vkResetCommandPool(g_Device, g_CommandPool[g_FrameIndex], 0);
  372. check_vk_result(err);
  373. VkCommandBufferBeginInfo info = {};
  374. info.sType = VK_STRUCTURE_TYPE_COMMAND_BUFFER_BEGIN_INFO;
  375. info.flags |= VK_COMMAND_BUFFER_USAGE_ONE_TIME_SUBMIT_BIT;
  376. err = vkBeginCommandBuffer(g_CommandBuffer[g_FrameIndex], &info);
  377. check_vk_result(err);
  378. }
  379. {
  380. VkRenderPassBeginInfo info = {};
  381. info.sType = VK_STRUCTURE_TYPE_RENDER_PASS_BEGIN_INFO;
  382. info.renderPass = g_RenderPass;
  383. info.framebuffer = g_Framebuffer[g_BackBufferIndex];
  384. info.renderArea.extent.width = fb_width;
  385. info.renderArea.extent.height = fb_height;
  386. info.clearValueCount = 1;
  387. info.pClearValues = &g_ClearValue;
  388. vkCmdBeginRenderPass(g_CommandBuffer[g_FrameIndex], &info, VK_SUBPASS_CONTENTS_INLINE);
  389. }
  390. }
  391. static void frame_end()
  392. {
  393. VkResult err;
  394. vkCmdEndRenderPass(g_CommandBuffer[g_FrameIndex]);
  395. {
  396. VkImageMemoryBarrier barrier = {};
  397. barrier.sType = VK_STRUCTURE_TYPE_IMAGE_MEMORY_BARRIER;
  398. barrier.srcAccessMask = VK_ACCESS_COLOR_ATTACHMENT_WRITE_BIT;
  399. barrier.dstAccessMask = VK_ACCESS_MEMORY_READ_BIT;
  400. barrier.oldLayout = VK_IMAGE_LAYOUT_COLOR_ATTACHMENT_OPTIMAL;
  401. barrier.newLayout = VK_IMAGE_LAYOUT_PRESENT_SRC_KHR;
  402. barrier.srcQueueFamilyIndex = VK_QUEUE_FAMILY_IGNORED;
  403. barrier.dstQueueFamilyIndex = VK_QUEUE_FAMILY_IGNORED;
  404. barrier.image = g_BackBuffer[g_BackBufferIndex];
  405. barrier.subresourceRange = g_ImageRange;
  406. vkCmdPipelineBarrier(g_CommandBuffer[g_FrameIndex], VK_PIPELINE_STAGE_COLOR_ATTACHMENT_OUTPUT_BIT, VK_PIPELINE_STAGE_BOTTOM_OF_PIPE_BIT, 0, 0, NULL, 0, NULL, 1, &barrier);
  407. }
  408. {
  409. VkPipelineStageFlags wait_stage = VK_PIPELINE_STAGE_COLOR_ATTACHMENT_OUTPUT_BIT;
  410. VkSubmitInfo info = {};
  411. info.sType = VK_STRUCTURE_TYPE_SUBMIT_INFO;
  412. info.waitSemaphoreCount = 1;
  413. info.pWaitSemaphores = &g_Semaphore[g_FrameIndex];
  414. info.pWaitDstStageMask = &wait_stage;
  415. info.commandBufferCount = 1;
  416. info.pCommandBuffers = &g_CommandBuffer[g_FrameIndex];
  417. err = vkEndCommandBuffer(g_CommandBuffer[g_FrameIndex]);
  418. check_vk_result(err);
  419. err = vkResetFences(g_Device, 1, &g_Fence[g_FrameIndex]);
  420. check_vk_result(err);
  421. err = vkQueueSubmit(g_Queue, 1, &info, g_Fence[g_FrameIndex]);
  422. check_vk_result(err);
  423. }
  424. {
  425. VkResult res;
  426. VkSwapchainKHR swapchains[1] = {g_Swapchain};
  427. uint32_t indices[1] = {g_BackBufferIndex};
  428. VkPresentInfoKHR info = {};
  429. info.sType = VK_STRUCTURE_TYPE_PRESENT_INFO_KHR;
  430. info.swapchainCount = 1;
  431. info.pSwapchains = swapchains;
  432. info.pImageIndices = indices;
  433. info.pResults = &res;
  434. err = vkQueuePresentKHR(g_Queue, &info);
  435. check_vk_result(err);
  436. check_vk_result(res);
  437. }
  438. g_FrameIndex = (g_FrameIndex+1) % IMGUI_VK_QUEUED_FRAMES;
  439. }
  440. static void error_callback(int error, const char* description)
  441. {
  442. fprintf(stderr, "Error %d: %s\n", error, description);
  443. }
  444. int main(int, char**)
  445. {
  446. // Setup window
  447. glfwSetErrorCallback(error_callback);
  448. if (!glfwInit())
  449. return 1;
  450. glfwWindowHint(GLFW_CLIENT_API, GLFW_NO_API);
  451. GLFWwindow* window = glfwCreateWindow(1280, 720, "ImGui Vulkan example", NULL, NULL);
  452. // Setup Vulkan
  453. if (!glfwVulkanSupported())
  454. {
  455. printf("GLFW: Vulkan Not Supported\n");
  456. return 1;
  457. }
  458. setup_vulkan(window);
  459. // Setup ImGui binding
  460. ImGui_ImplGlfwVulkan_Init_Data init_data = {};
  461. init_data.allocator = g_Allocator;
  462. init_data.gpu = g_Gpu;
  463. init_data.device = g_Device;
  464. init_data.render_pass = g_RenderPass;
  465. init_data.pipeline_cache = g_PipelineCache;
  466. init_data.descriptor_pool = g_DescriptorPool;
  467. init_data.check_vk_result = check_vk_result;
  468. ImGui_ImplGlfwVulkan_Init(window, true, &init_data);
  469. // Load Fonts
  470. // (there is a default font, this is only if you want to change it. see extra_fonts/README.txt for more details)
  471. //ImGuiIO& io = ImGui::GetIO();
  472. //io.Fonts->AddFontDefault();
  473. //io.Fonts->AddFontFromFileTTF("../../extra_fonts/Cousine-Regular.ttf", 15.0f);
  474. //io.Fonts->AddFontFromFileTTF("../../extra_fonts/DroidSans.ttf", 16.0f);
  475. //io.Fonts->AddFontFromFileTTF("../../extra_fonts/ProggyClean.ttf", 13.0f);
  476. //io.Fonts->AddFontFromFileTTF("../../extra_fonts/ProggyTiny.ttf", 10.0f);
  477. //io.Fonts->AddFontFromFileTTF("c:\\Windows\\Fonts\\ArialUni.ttf", 18.0f, NULL, io.Fonts->GetGlyphRangesJapanese());
  478. // Upload Fonts
  479. {
  480. VkResult err;
  481. err = vkResetCommandPool(g_Device, g_CommandPool[g_FrameIndex], 0);
  482. check_vk_result(err);
  483. VkCommandBufferBeginInfo begin_info = {};
  484. begin_info.sType = VK_STRUCTURE_TYPE_COMMAND_BUFFER_BEGIN_INFO;
  485. begin_info.flags |= VK_COMMAND_BUFFER_USAGE_ONE_TIME_SUBMIT_BIT;
  486. err = vkBeginCommandBuffer(g_CommandBuffer[g_FrameIndex], &begin_info);
  487. check_vk_result(err);
  488. ImGui_ImplGlfwVulkan_CreateFontsTexture(g_CommandBuffer[g_FrameIndex]);
  489. VkSubmitInfo end_info = {};
  490. end_info.sType = VK_STRUCTURE_TYPE_SUBMIT_INFO;
  491. end_info.commandBufferCount = 1;
  492. end_info.pCommandBuffers = &g_CommandBuffer[g_FrameIndex];
  493. err = vkEndCommandBuffer(g_CommandBuffer[g_FrameIndex]);
  494. check_vk_result(err);
  495. err = vkQueueSubmit(g_Queue, 1, &end_info, VK_NULL_HANDLE);
  496. check_vk_result(err);
  497. err = vkDeviceWaitIdle(g_Device);
  498. check_vk_result(err);
  499. ImGui_ImplGlfwVulkan_InvalidateFontUploadObjects();
  500. }
  501. bool show_test_window = true;
  502. bool show_another_window = false;
  503. ImVec4 clear_color = ImColor(114, 144, 154);
  504. // Main loop
  505. while (!glfwWindowShouldClose(window))
  506. {
  507. glfwPollEvents();
  508. ImGui_ImplGlfwVulkan_NewFrame();
  509. // 1. Show a simple window
  510. // Tip: if we don't call ImGui::Begin()/ImGui::End() the widgets appears in a window automatically called "Debug"
  511. {
  512. static float f = 0.0f;
  513. ImGui::Text("Hello, world!");
  514. ImGui::SliderFloat("float", &f, 0.0f, 1.0f);
  515. ImGui::ColorEdit3("clear color", (float*)&clear_color);
  516. if (ImGui::Button("Test Window")) show_test_window ^= 1;
  517. if (ImGui::Button("Another Window")) show_another_window ^= 1;
  518. ImGui::Text("Application average %.3f ms/frame (%.1f FPS)", 1000.0f / ImGui::GetIO().Framerate, ImGui::GetIO().Framerate);
  519. }
  520. // 2. Show another simple window, this time using an explicit Begin/End pair
  521. if (show_another_window)
  522. {
  523. ImGui::SetNextWindowSize(ImVec2(200,100), ImGuiSetCond_FirstUseEver);
  524. ImGui::Begin("Another Window", &show_another_window);
  525. ImGui::Text("Hello");
  526. ImGui::End();
  527. }
  528. // 3. Show the ImGui test window. Most of the sample code is in ImGui::ShowTestWindow()
  529. if (show_test_window)
  530. {
  531. ImGui::SetNextWindowPos(ImVec2(650, 20), ImGuiSetCond_FirstUseEver);
  532. ImGui::ShowTestWindow(&show_test_window);
  533. }
  534. g_ClearValue.color.float32[0] = clear_color.x;
  535. g_ClearValue.color.float32[1] = clear_color.y;
  536. g_ClearValue.color.float32[2] = clear_color.z;
  537. g_ClearValue.color.float32[3] = clear_color.w;
  538. frame_begin();
  539. ImGui_ImplGlfwVulkan_Render(g_CommandBuffer[g_FrameIndex]);
  540. frame_end();
  541. }
  542. // Cleanup
  543. VkResult err = vkDeviceWaitIdle(g_Device);
  544. check_vk_result(err);
  545. ImGui_ImplGlfwVulkan_Shutdown();
  546. cleanup_vulkan();
  547. glfwTerminate();
  548. return 0;
  549. }