main.cpp 32 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764
  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. #define IMGUI_MAX_POSSIBLE_BACK_BUFFERS 16
  11. #define IMGUI_UNLIMITED_FRAME_RATE
  12. #ifdef _DEBUG
  13. #define IMGUI_VULKAN_DEBUG_REPORT
  14. #endif
  15. static VkAllocationCallbacks* g_Allocator = NULL;
  16. static VkInstance g_Instance = VK_NULL_HANDLE;
  17. static VkSurfaceKHR g_Surface = VK_NULL_HANDLE;
  18. static VkPhysicalDevice g_Gpu = VK_NULL_HANDLE;
  19. static VkDevice g_Device = VK_NULL_HANDLE;
  20. static VkSwapchainKHR g_Swapchain = VK_NULL_HANDLE;
  21. static VkRenderPass g_RenderPass = VK_NULL_HANDLE;
  22. static uint32_t g_QueueFamily = 0;
  23. static VkQueue g_Queue = VK_NULL_HANDLE;
  24. static VkDebugReportCallbackEXT g_DebugReport = VK_NULL_HANDLE;
  25. static VkSurfaceFormatKHR g_SurfaceFormat;
  26. static VkImageSubresourceRange g_ImageRange = { VK_IMAGE_ASPECT_COLOR_BIT, 0, 1, 0, 1 };
  27. static VkPresentModeKHR g_PresentMode;
  28. static VkPipelineCache g_PipelineCache = VK_NULL_HANDLE;
  29. static VkDescriptorPool g_DescriptorPool = VK_NULL_HANDLE;
  30. static int fb_width, fb_height;
  31. static uint32_t g_BackbufferIndices[IMGUI_VK_QUEUED_FRAMES]; // keep track of recently rendered swapchain frame indices
  32. static uint32_t g_BackBufferCount = 0;
  33. static VkImage g_BackBuffer[IMGUI_MAX_POSSIBLE_BACK_BUFFERS] = {};
  34. static VkImageView g_BackBufferView[IMGUI_MAX_POSSIBLE_BACK_BUFFERS] = {};
  35. static VkFramebuffer g_Framebuffer[IMGUI_MAX_POSSIBLE_BACK_BUFFERS] = {};
  36. static uint32_t g_FrameIndex = 0;
  37. static VkCommandPool g_CommandPool[IMGUI_VK_QUEUED_FRAMES];
  38. static VkCommandBuffer g_CommandBuffer[IMGUI_VK_QUEUED_FRAMES];
  39. static VkFence g_Fence[IMGUI_VK_QUEUED_FRAMES];
  40. static VkSemaphore g_PresentCompleteSemaphore[IMGUI_VK_QUEUED_FRAMES];
  41. static VkSemaphore g_RenderCompleteSemaphore[IMGUI_VK_QUEUED_FRAMES];
  42. static VkClearValue g_ClearValue = {};
  43. static void check_vk_result(VkResult err)
  44. {
  45. if (err == 0) return;
  46. printf("VkResult %d\n", err);
  47. if (err < 0)
  48. abort();
  49. }
  50. static void resize_vulkan(SDL_Window*, int w, int h)
  51. {
  52. VkResult err;
  53. VkSwapchainKHR old_swapchain = g_Swapchain;
  54. err = vkDeviceWaitIdle(g_Device);
  55. check_vk_result(err);
  56. // Destroy old Framebuffer:
  57. for (uint32_t i = 0; i < g_BackBufferCount; i++)
  58. if (g_BackBufferView[i])
  59. vkDestroyImageView(g_Device, g_BackBufferView[i], g_Allocator);
  60. for (uint32_t i = 0; i < g_BackBufferCount; i++)
  61. if (g_Framebuffer[i])
  62. vkDestroyFramebuffer(g_Device, g_Framebuffer[i], g_Allocator);
  63. if (g_RenderPass)
  64. vkDestroyRenderPass(g_Device, g_RenderPass, g_Allocator);
  65. // Create Swapchain:
  66. {
  67. VkSwapchainCreateInfoKHR info = {};
  68. info.sType = VK_STRUCTURE_TYPE_SWAPCHAIN_CREATE_INFO_KHR;
  69. info.surface = g_Surface;
  70. info.imageFormat = g_SurfaceFormat.format;
  71. info.imageColorSpace = g_SurfaceFormat.colorSpace;
  72. info.imageArrayLayers = 1;
  73. info.imageUsage |= VK_IMAGE_USAGE_COLOR_ATTACHMENT_BIT;
  74. info.imageSharingMode = VK_SHARING_MODE_EXCLUSIVE;
  75. info.preTransform = VK_SURFACE_TRANSFORM_IDENTITY_BIT_KHR;
  76. info.compositeAlpha = VK_COMPOSITE_ALPHA_OPAQUE_BIT_KHR;
  77. info.presentMode = g_PresentMode;
  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_SurfaceFormat.format;
  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_PRESENT_SRC_KHR;
  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_SurfaceFormat.format;
  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. #ifdef IMGUI_VULKAN_DEBUG_REPORT
  175. 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)
  176. {
  177. (void)flags; (void)object; (void)location; (void)messageCode; (void)pUserData; (void)pLayerPrefix; // Unused arguemnts
  178. printf("[vulkan] ObjectType: %i\nMessage: %s\n\n", objectType, pMessage);
  179. return VK_FALSE;
  180. }
  181. #endif // IMGUI_VULKAN_DEBUG_REPORT
  182. static void setup_vulkan(SDL_Window* window, const char** extensions, uint32_t extensions_count)
  183. {
  184. VkResult err;
  185. // Create Vulkan Instance
  186. {
  187. VkInstanceCreateInfo create_info = {};
  188. create_info.sType = VK_STRUCTURE_TYPE_INSTANCE_CREATE_INFO;
  189. create_info.enabledExtensionCount = extensions_count;
  190. create_info.ppEnabledExtensionNames = extensions;
  191. #ifdef IMGUI_VULKAN_DEBUG_REPORT
  192. // Enabling multiple validation layers grouped as LunarG standard validation
  193. const char* layers[] = { "VK_LAYER_LUNARG_standard_validation" };
  194. create_info.enabledLayerCount = 1;
  195. create_info.ppEnabledLayerNames = layers;
  196. // Enable debug report extension (we need additional storage, so we duplicate the user array to add our new extension to it)
  197. const char** extensions_ext = (const char**)malloc(sizeof(const char*) * (extensions_count + 1));
  198. memcpy(extensions_ext, extensions, extensions_count * sizeof(const char*));
  199. extensions_ext[extensions_count] = "VK_EXT_debug_report";
  200. create_info.enabledExtensionCount = extensions_count + 1;
  201. create_info.ppEnabledExtensionNames = extensions_ext;
  202. #endif // IMGUI_VULKAN_DEBUG_REPORT
  203. err = vkCreateInstance(&create_info, g_Allocator, &g_Instance);
  204. check_vk_result(err);
  205. #ifdef IMGUI_VULKAN_DEBUG_REPORT
  206. free(extensions_ext);
  207. // Get the function pointer (required for any extensions)
  208. auto vkCreateDebugReportCallbackEXT = (PFN_vkCreateDebugReportCallbackEXT)vkGetInstanceProcAddr(g_Instance, "vkCreateDebugReportCallbackEXT");
  209. IM_ASSERT(vkCreateDebugReportCallbackEXT != NULL);
  210. // Setup the debug report callback
  211. VkDebugReportCallbackCreateInfoEXT debug_report_ci = {};
  212. debug_report_ci.sType = VK_STRUCTURE_TYPE_DEBUG_REPORT_CALLBACK_CREATE_INFO_EXT;
  213. debug_report_ci.flags = VK_DEBUG_REPORT_ERROR_BIT_EXT | VK_DEBUG_REPORT_WARNING_BIT_EXT | VK_DEBUG_REPORT_PERFORMANCE_WARNING_BIT_EXT;
  214. debug_report_ci.pfnCallback = debug_report;
  215. debug_report_ci.pUserData = NULL;
  216. err = vkCreateDebugReportCallbackEXT(g_Instance, &debug_report_ci, g_Allocator, &g_DebugReport);
  217. check_vk_result(err);
  218. #endif // IMGUI_VULKAN_DEBUG_REPORT
  219. }
  220. // Create Window Surface (with SDL)
  221. {
  222. SDL_bool result = SDL_Vulkan_CreateSurface(window, g_Instance, &g_Surface);
  223. if (result == 0)
  224. {
  225. printf("failed to create Vulkan surface\n");
  226. abort();
  227. }
  228. }
  229. // Get GPU
  230. {
  231. uint32_t gpu_count;
  232. err = vkEnumeratePhysicalDevices(g_Instance, &gpu_count, NULL);
  233. check_vk_result(err);
  234. VkPhysicalDevice* gpus = (VkPhysicalDevice*)malloc(sizeof(VkPhysicalDevice) * gpu_count);
  235. err = vkEnumeratePhysicalDevices(g_Instance, &gpu_count, gpus);
  236. check_vk_result(err);
  237. // If a number >1 of GPUs got reported, you should find the best fit GPU for your purpose
  238. // e.g. VK_PHYSICAL_DEVICE_TYPE_DISCRETE_GPU if available, or with the greatest memory available, etc.
  239. // for sake of simplicity we'll just take the first one, assuming it has a graphics queue family.
  240. g_Gpu = gpus[0];
  241. free(gpus);
  242. }
  243. // Get queue
  244. {
  245. uint32_t count;
  246. vkGetPhysicalDeviceQueueFamilyProperties(g_Gpu, &count, NULL);
  247. VkQueueFamilyProperties* queues = (VkQueueFamilyProperties*)malloc(sizeof(VkQueueFamilyProperties) * count);
  248. vkGetPhysicalDeviceQueueFamilyProperties(g_Gpu, &count, queues);
  249. for (uint32_t i = 0; i < count; i++)
  250. {
  251. if (queues[i].queueFlags & VK_QUEUE_GRAPHICS_BIT)
  252. {
  253. g_QueueFamily = i;
  254. break;
  255. }
  256. }
  257. free(queues);
  258. }
  259. // Check for WSI support
  260. {
  261. VkBool32 res;
  262. vkGetPhysicalDeviceSurfaceSupportKHR(g_Gpu, g_QueueFamily, g_Surface, &res);
  263. if (res != VK_TRUE)
  264. {
  265. fprintf(stderr, "Error no WSI support on physical device 0\n");
  266. exit(-1);
  267. }
  268. }
  269. // Get Surface Format
  270. {
  271. // Per Spec Format and View Format are expected to be the same unless VK_IMAGE_CREATE_MUTABLE_BIT was set at image creation
  272. // Assuming that the default behavior is without setting this bit, there is no need for separate Spawchain image and image view format
  273. // additionally several new color spaces were introduced with Vulkan Spec v1.0.40
  274. // hence we must make sure that a format with the mostly available color space, VK_COLOR_SPACE_SRGB_NONLINEAR_KHR, is found and used
  275. uint32_t count;
  276. vkGetPhysicalDeviceSurfaceFormatsKHR(g_Gpu, g_Surface, &count, NULL);
  277. VkSurfaceFormatKHR *formats = (VkSurfaceFormatKHR*)malloc(sizeof(VkSurfaceFormatKHR) * count);
  278. vkGetPhysicalDeviceSurfaceFormatsKHR(g_Gpu, g_Surface, &count, formats);
  279. // first check if only one format, VK_FORMAT_UNDEFINED, is available, which would imply that any format is available
  280. if (count == 1)
  281. {
  282. if (formats[0].format == VK_FORMAT_UNDEFINED)
  283. {
  284. g_SurfaceFormat.format = VK_FORMAT_B8G8R8A8_UNORM;
  285. g_SurfaceFormat.colorSpace = VK_COLORSPACE_SRGB_NONLINEAR_KHR;
  286. }
  287. else
  288. { // no point in searching another format
  289. g_SurfaceFormat = formats[0];
  290. }
  291. }
  292. else
  293. {
  294. // request several formats, the first found will be used
  295. VkFormat requestSurfaceImageFormat[] = { VK_FORMAT_B8G8R8A8_UNORM, VK_FORMAT_R8G8B8A8_UNORM, VK_FORMAT_B8G8R8_UNORM, VK_FORMAT_R8G8B8_UNORM };
  296. VkColorSpaceKHR requestSurfaceColorSpace = VK_COLORSPACE_SRGB_NONLINEAR_KHR;
  297. bool requestedFound = false;
  298. for (size_t i = 0; i < sizeof(requestSurfaceImageFormat) / sizeof(requestSurfaceImageFormat[0]); i++)
  299. {
  300. if (requestedFound)
  301. break;
  302. for (uint32_t j = 0; j < count; j++)
  303. {
  304. if (formats[j].format == requestSurfaceImageFormat[i] && formats[j].colorSpace == requestSurfaceColorSpace)
  305. {
  306. g_SurfaceFormat = formats[j];
  307. requestedFound = true;
  308. }
  309. }
  310. }
  311. // if none of the requested image formats could be found, use the first available
  312. if (!requestedFound)
  313. g_SurfaceFormat = formats[0];
  314. }
  315. free(formats);
  316. }
  317. // Get Present Mode
  318. {
  319. // Requst a certain mode and confirm that it is available. If not use VK_PRESENT_MODE_FIFO_KHR which is mandatory
  320. #ifdef IMGUI_UNLIMITED_FRAME_RATE
  321. g_PresentMode = VK_PRESENT_MODE_IMMEDIATE_KHR;
  322. #else
  323. g_PresentMode = VK_PRESENT_MODE_FIFO_KHR;
  324. #endif
  325. uint32_t count = 0;
  326. vkGetPhysicalDeviceSurfacePresentModesKHR(g_Gpu, g_Surface, &count, nullptr);
  327. VkPresentModeKHR* presentModes = (VkPresentModeKHR*)malloc(sizeof(VkQueueFamilyProperties) * count);
  328. vkGetPhysicalDeviceSurfacePresentModesKHR(g_Gpu, g_Surface, &count, presentModes);
  329. bool presentModeAvailable = false;
  330. for (size_t i = 0; i < count; i++)
  331. {
  332. if (presentModes[i] == g_PresentMode)
  333. {
  334. presentModeAvailable = true;
  335. break;
  336. }
  337. }
  338. if (!presentModeAvailable)
  339. g_PresentMode = VK_PRESENT_MODE_FIFO_KHR; // always available
  340. }
  341. // Create Logical Device
  342. {
  343. int device_extension_count = 1;
  344. const char* device_extensions[] = { "VK_KHR_swapchain" };
  345. const uint32_t queue_index = 0;
  346. const uint32_t queue_count = 1;
  347. const float queue_priority[] = { 1.0f };
  348. VkDeviceQueueCreateInfo queue_info[1] = {};
  349. queue_info[0].sType = VK_STRUCTURE_TYPE_DEVICE_QUEUE_CREATE_INFO;
  350. queue_info[0].queueFamilyIndex = g_QueueFamily;
  351. queue_info[0].queueCount = queue_count;
  352. queue_info[0].pQueuePriorities = queue_priority;
  353. VkDeviceCreateInfo create_info = {};
  354. create_info.sType = VK_STRUCTURE_TYPE_DEVICE_CREATE_INFO;
  355. create_info.queueCreateInfoCount = sizeof(queue_info) / sizeof(queue_info[0]);
  356. create_info.pQueueCreateInfos = queue_info;
  357. create_info.enabledExtensionCount = device_extension_count;
  358. create_info.ppEnabledExtensionNames = device_extensions;
  359. err = vkCreateDevice(g_Gpu, &create_info, g_Allocator, &g_Device);
  360. check_vk_result(err);
  361. vkGetDeviceQueue(g_Device, g_QueueFamily, queue_index, &g_Queue);
  362. }
  363. // Create Framebuffers
  364. {
  365. int w, h;
  366. SDL_GetWindowSize(window, &w, &h);
  367. resize_vulkan(window, w, h);
  368. }
  369. // Create Command Buffers
  370. for (int i = 0; i < IMGUI_VK_QUEUED_FRAMES; i++)
  371. {
  372. {
  373. VkCommandPoolCreateInfo info = {};
  374. info.sType = VK_STRUCTURE_TYPE_COMMAND_POOL_CREATE_INFO;
  375. info.flags = VK_COMMAND_POOL_CREATE_RESET_COMMAND_BUFFER_BIT;
  376. info.queueFamilyIndex = g_QueueFamily;
  377. err = vkCreateCommandPool(g_Device, &info, g_Allocator, &g_CommandPool[i]);
  378. check_vk_result(err);
  379. }
  380. {
  381. VkCommandBufferAllocateInfo info = {};
  382. info.sType = VK_STRUCTURE_TYPE_COMMAND_BUFFER_ALLOCATE_INFO;
  383. info.commandPool = g_CommandPool[i];
  384. info.level = VK_COMMAND_BUFFER_LEVEL_PRIMARY;
  385. info.commandBufferCount = 1;
  386. err = vkAllocateCommandBuffers(g_Device, &info, &g_CommandBuffer[i]);
  387. check_vk_result(err);
  388. }
  389. {
  390. VkFenceCreateInfo info = {};
  391. info.sType = VK_STRUCTURE_TYPE_FENCE_CREATE_INFO;
  392. info.flags = VK_FENCE_CREATE_SIGNALED_BIT;
  393. err = vkCreateFence(g_Device, &info, g_Allocator, &g_Fence[i]);
  394. check_vk_result(err);
  395. }
  396. {
  397. VkSemaphoreCreateInfo info = {};
  398. info.sType = VK_STRUCTURE_TYPE_SEMAPHORE_CREATE_INFO;
  399. err = vkCreateSemaphore(g_Device, &info, g_Allocator, &g_PresentCompleteSemaphore[i]);
  400. check_vk_result(err);
  401. err = vkCreateSemaphore(g_Device, &info, g_Allocator, &g_RenderCompleteSemaphore[i]);
  402. check_vk_result(err);
  403. }
  404. }
  405. // Create Descriptor Pool
  406. {
  407. VkDescriptorPoolSize pool_size[11] =
  408. {
  409. { VK_DESCRIPTOR_TYPE_SAMPLER, 1000 },
  410. { VK_DESCRIPTOR_TYPE_COMBINED_IMAGE_SAMPLER, 1000 },
  411. { VK_DESCRIPTOR_TYPE_SAMPLED_IMAGE, 1000 },
  412. { VK_DESCRIPTOR_TYPE_STORAGE_IMAGE, 1000 },
  413. { VK_DESCRIPTOR_TYPE_UNIFORM_TEXEL_BUFFER, 1000 },
  414. { VK_DESCRIPTOR_TYPE_STORAGE_TEXEL_BUFFER, 1000 },
  415. { VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER, 1000 },
  416. { VK_DESCRIPTOR_TYPE_STORAGE_BUFFER, 1000 },
  417. { VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER_DYNAMIC, 1000 },
  418. { VK_DESCRIPTOR_TYPE_STORAGE_BUFFER_DYNAMIC, 1000 },
  419. { VK_DESCRIPTOR_TYPE_INPUT_ATTACHMENT, 1000 }
  420. };
  421. VkDescriptorPoolCreateInfo pool_info = {};
  422. pool_info.sType = VK_STRUCTURE_TYPE_DESCRIPTOR_POOL_CREATE_INFO;
  423. pool_info.flags = VK_DESCRIPTOR_POOL_CREATE_FREE_DESCRIPTOR_SET_BIT;
  424. pool_info.maxSets = 1000 * 11;
  425. pool_info.poolSizeCount = 11;
  426. pool_info.pPoolSizes = pool_size;
  427. err = vkCreateDescriptorPool(g_Device, &pool_info, g_Allocator, &g_DescriptorPool);
  428. check_vk_result(err);
  429. }
  430. }
  431. static void cleanup_vulkan()
  432. {
  433. vkDestroyDescriptorPool(g_Device, g_DescriptorPool, g_Allocator);
  434. for (int i = 0; i < IMGUI_VK_QUEUED_FRAMES; i++)
  435. {
  436. vkDestroyFence(g_Device, g_Fence[i], g_Allocator);
  437. vkFreeCommandBuffers(g_Device, g_CommandPool[i], 1, &g_CommandBuffer[i]);
  438. vkDestroyCommandPool(g_Device, g_CommandPool[i], g_Allocator);
  439. vkDestroySemaphore(g_Device, g_PresentCompleteSemaphore[i], g_Allocator);
  440. vkDestroySemaphore(g_Device, g_RenderCompleteSemaphore[i], g_Allocator);
  441. }
  442. for (uint32_t i = 0; i < g_BackBufferCount; i++)
  443. {
  444. vkDestroyImageView(g_Device, g_BackBufferView[i], g_Allocator);
  445. vkDestroyFramebuffer(g_Device, g_Framebuffer[i], g_Allocator);
  446. }
  447. vkDestroyRenderPass(g_Device, g_RenderPass, g_Allocator);
  448. vkDestroySwapchainKHR(g_Device, g_Swapchain, g_Allocator);
  449. vkDestroySurfaceKHR(g_Instance, g_Surface, g_Allocator);
  450. #ifdef IMGUI_VULKAN_DEBUG_REPORT
  451. // Remove the debug report callback
  452. auto vkDestroyDebugReportCallbackEXT = (PFN_vkDestroyDebugReportCallbackEXT)vkGetInstanceProcAddr(g_Instance, "vkDestroyDebugReportCallbackEXT");
  453. vkDestroyDebugReportCallbackEXT(g_Instance, g_DebugReport, g_Allocator);
  454. #endif // IMGUI_VULKAN_DEBUG_REPORT
  455. vkDestroyDevice(g_Device, g_Allocator);
  456. vkDestroyInstance(g_Instance, g_Allocator);
  457. }
  458. static void frame_begin()
  459. {
  460. VkResult err;
  461. for (;;)
  462. {
  463. err = vkWaitForFences(g_Device, 1, &g_Fence[g_FrameIndex], VK_TRUE, 100);
  464. if (err == VK_SUCCESS) break;
  465. if (err == VK_TIMEOUT) continue;
  466. check_vk_result(err);
  467. }
  468. {
  469. err = vkAcquireNextImageKHR(g_Device, g_Swapchain, UINT64_MAX, g_PresentCompleteSemaphore[g_FrameIndex], VK_NULL_HANDLE, &g_BackbufferIndices[g_FrameIndex]);
  470. check_vk_result(err);
  471. }
  472. {
  473. err = vkResetCommandPool(g_Device, g_CommandPool[g_FrameIndex], 0);
  474. check_vk_result(err);
  475. VkCommandBufferBeginInfo info = {};
  476. info.sType = VK_STRUCTURE_TYPE_COMMAND_BUFFER_BEGIN_INFO;
  477. info.flags |= VK_COMMAND_BUFFER_USAGE_ONE_TIME_SUBMIT_BIT;
  478. err = vkBeginCommandBuffer(g_CommandBuffer[g_FrameIndex], &info);
  479. check_vk_result(err);
  480. }
  481. {
  482. VkRenderPassBeginInfo info = {};
  483. info.sType = VK_STRUCTURE_TYPE_RENDER_PASS_BEGIN_INFO;
  484. info.renderPass = g_RenderPass;
  485. info.framebuffer = g_Framebuffer[g_BackbufferIndices[g_FrameIndex]];
  486. info.renderArea.extent.width = fb_width;
  487. info.renderArea.extent.height = fb_height;
  488. info.clearValueCount = 1;
  489. info.pClearValues = &g_ClearValue;
  490. vkCmdBeginRenderPass(g_CommandBuffer[g_FrameIndex], &info, VK_SUBPASS_CONTENTS_INLINE);
  491. }
  492. }
  493. static void frame_end()
  494. {
  495. VkResult err;
  496. vkCmdEndRenderPass(g_CommandBuffer[g_FrameIndex]);
  497. {
  498. VkPipelineStageFlags wait_stage = VK_PIPELINE_STAGE_COLOR_ATTACHMENT_OUTPUT_BIT;
  499. VkSubmitInfo info = {};
  500. info.sType = VK_STRUCTURE_TYPE_SUBMIT_INFO;
  501. info.waitSemaphoreCount = 1;
  502. info.pWaitSemaphores = &g_PresentCompleteSemaphore[g_FrameIndex];
  503. info.pWaitDstStageMask = &wait_stage;
  504. info.commandBufferCount = 1;
  505. info.pCommandBuffers = &g_CommandBuffer[g_FrameIndex];
  506. info.signalSemaphoreCount = 1;
  507. info.pSignalSemaphores = &g_RenderCompleteSemaphore[g_FrameIndex];
  508. err = vkEndCommandBuffer(g_CommandBuffer[g_FrameIndex]);
  509. check_vk_result(err);
  510. err = vkResetFences(g_Device, 1, &g_Fence[g_FrameIndex]);
  511. check_vk_result(err);
  512. err = vkQueueSubmit(g_Queue, 1, &info, g_Fence[g_FrameIndex]);
  513. check_vk_result(err);
  514. }
  515. }
  516. static void frame_present()
  517. {
  518. VkResult err;
  519. // If IMGUI_UNLIMITED_FRAME_RATE is defined we present the latest but one frame. Otherwise we present the latest rendered frame
  520. #ifdef IMGUI_UNLIMITED_FRAME_RATE
  521. uint32_t PresentIndex = (g_FrameIndex + IMGUI_VK_QUEUED_FRAMES - 1) % IMGUI_VK_QUEUED_FRAMES;
  522. #else
  523. uint32_t PresentIndex = g_FrameIndex;
  524. #endif // IMGUI_UNLIMITED_FRAME_RATE
  525. VkSwapchainKHR swapchains[1] = { g_Swapchain };
  526. uint32_t indices[1] = { g_BackbufferIndices[PresentIndex] };
  527. VkPresentInfoKHR info = {};
  528. info.sType = VK_STRUCTURE_TYPE_PRESENT_INFO_KHR;
  529. info.waitSemaphoreCount = 1;
  530. info.pWaitSemaphores = &g_RenderCompleteSemaphore[PresentIndex];
  531. info.swapchainCount = 1;
  532. info.pSwapchains = swapchains;
  533. info.pImageIndices = indices;
  534. err = vkQueuePresentKHR(g_Queue, &info);
  535. check_vk_result(err);
  536. g_FrameIndex = (g_FrameIndex + 1) % IMGUI_VK_QUEUED_FRAMES;
  537. }
  538. int main(int, char**)
  539. {
  540. // Setup SDL
  541. if (SDL_Init(SDL_INIT_VIDEO|SDL_INIT_TIMER) != 0)
  542. {
  543. printf("Error: %s\n", SDL_GetError());
  544. return 1;
  545. }
  546. // Setup window
  547. SDL_DisplayMode current;
  548. SDL_GetCurrentDisplayMode(0, &current);
  549. SDL_Window* window = SDL_CreateWindow("ImGui SDL2+Vulkan example", SDL_WINDOWPOS_CENTERED, SDL_WINDOWPOS_CENTERED, 1280, 720, SDL_WINDOW_VULKAN|SDL_WINDOW_RESIZABLE);
  550. // Setup Vulkan
  551. uint32_t extensions_count;
  552. SDL_Vulkan_GetInstanceExtensions(window, &extensions_count, nullptr);
  553. const char** sdl_extensions = new const char*[extensions_count];
  554. SDL_Vulkan_GetInstanceExtensions(window, &extensions_count, sdl_extensions);
  555. setup_vulkan(window, sdl_extensions, extensions_count);
  556. delete[] sdl_extensions;
  557. // Setup ImGui binding
  558. ImGui::CreateContext();
  559. ImGui_ImplVulkan_InitData init_data = {};
  560. init_data.allocator = g_Allocator;
  561. init_data.gpu = g_Gpu;
  562. init_data.device = g_Device;
  563. init_data.render_pass = g_RenderPass;
  564. init_data.pipeline_cache = g_PipelineCache;
  565. init_data.descriptor_pool = g_DescriptorPool;
  566. init_data.check_vk_result = check_vk_result;
  567. ImGuiIO& io = ImGui::GetIO(); (void)io;
  568. //io.ConfigFlags |= ImGuiConfigFlags_NavEnableKeyboard; // Enable Keyboard Controls
  569. ImGui_ImplVulkan_Init(&init_data);
  570. ImGui_ImplSDL2_Init(window);
  571. // Setup style
  572. ImGui::StyleColorsDark();
  573. //ImGui::StyleColorsClassic();
  574. // Load Fonts
  575. // - 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.
  576. // - AddFontFromFileTTF() will return the ImFont* so you can store it if you need to select the font among multiple.
  577. // - 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).
  578. // - 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.
  579. // - Read 'misc/fonts/README.txt' for more instructions and details.
  580. // - Remember that in C/C++ if you want to include a backslash \ in a string literal you need to write a double backslash \\ !
  581. //io.Fonts->AddFontDefault();
  582. //io.Fonts->AddFontFromFileTTF("../../misc/fonts/Roboto-Medium.ttf", 16.0f);
  583. //io.Fonts->AddFontFromFileTTF("../../misc/fonts/Cousine-Regular.ttf", 15.0f);
  584. //io.Fonts->AddFontFromFileTTF("../../misc/fonts/DroidSans.ttf", 16.0f);
  585. //io.Fonts->AddFontFromFileTTF("../../misc/fonts/ProggyTiny.ttf", 10.0f);
  586. //ImFont* font = io.Fonts->AddFontFromFileTTF("c:\\Windows\\Fonts\\ArialUni.ttf", 18.0f, NULL, io.Fonts->GetGlyphRangesJapanese());
  587. //IM_ASSERT(font != NULL);
  588. // Upload Fonts
  589. {
  590. VkResult err;
  591. err = vkResetCommandPool(g_Device, g_CommandPool[g_FrameIndex], 0);
  592. check_vk_result(err);
  593. VkCommandBufferBeginInfo begin_info = {};
  594. begin_info.sType = VK_STRUCTURE_TYPE_COMMAND_BUFFER_BEGIN_INFO;
  595. begin_info.flags |= VK_COMMAND_BUFFER_USAGE_ONE_TIME_SUBMIT_BIT;
  596. err = vkBeginCommandBuffer(g_CommandBuffer[g_FrameIndex], &begin_info);
  597. check_vk_result(err);
  598. ImGui_ImplVulkan_CreateFontsTexture(g_CommandBuffer[g_FrameIndex]);
  599. VkSubmitInfo end_info = {};
  600. end_info.sType = VK_STRUCTURE_TYPE_SUBMIT_INFO;
  601. end_info.commandBufferCount = 1;
  602. end_info.pCommandBuffers = &g_CommandBuffer[g_FrameIndex];
  603. err = vkEndCommandBuffer(g_CommandBuffer[g_FrameIndex]);
  604. check_vk_result(err);
  605. err = vkQueueSubmit(g_Queue, 1, &end_info, VK_NULL_HANDLE);
  606. check_vk_result(err);
  607. err = vkDeviceWaitIdle(g_Device);
  608. check_vk_result(err);
  609. ImGui_ImplVulkan_InvalidateFontUploadObjects();
  610. }
  611. bool show_demo_window = true;
  612. bool show_another_window = false;
  613. ImVec4 clear_color = ImVec4(0.45f, 0.55f, 0.60f, 1.00f);
  614. // 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.
  615. // Hence we must render once and increase the g_FrameIndex without presenting, which we do before entering the render loop.
  616. // This is also the reason why frame_end() is split into frame_end() and frame_present(), the later one not being called here.
  617. #ifdef IMGUI_UNLIMITED_FRAME_RATE
  618. ImGui_ImplVulkan_NewFrame();
  619. ImGui_ImplSDL2_NewFrame(window);
  620. frame_begin();
  621. ImGui_ImplVulkan_Render(g_CommandBuffer[g_FrameIndex]);
  622. frame_end();
  623. g_FrameIndex = (g_FrameIndex + 1) % IMGUI_VK_QUEUED_FRAMES;
  624. #endif // IMGUI_UNLIMITED_FRAME_RATE
  625. // Main loop
  626. bool done = false;
  627. while (!done)
  628. {
  629. // You can read the io.WantCaptureMouse, io.WantCaptureKeyboard flags to tell if dear imgui wants to use your inputs.
  630. // - When io.WantCaptureMouse is true, do not dispatch mouse input data to your main application.
  631. // - When io.WantCaptureKeyboard is true, do not dispatch keyboard input data to your main application.
  632. // Generally you may always pass all inputs to dear imgui, and hide them from your application based on those two flags.
  633. SDL_Event event;
  634. while (SDL_PollEvent(&event))
  635. {
  636. ImGui_ImplSDL2_ProcessEvent(&event);
  637. if (event.type == SDL_QUIT)
  638. done = true;
  639. }
  640. ImGui_ImplVulkan_NewFrame();
  641. ImGui_ImplSDL2_NewFrame(window);
  642. // 1. Show a simple window.
  643. // Tip: if we don't call ImGui::Begin()/ImGui::End() the widgets automatically appears in a window called "Debug".
  644. {
  645. static float f = 0.0f;
  646. static int counter = 0;
  647. ImGui::Text("Hello, world!"); // Display some text (you can use a format string too)
  648. ImGui::SliderFloat("float", &f, 0.0f, 1.0f); // Edit 1 float using a slider from 0.0f to 1.0f
  649. ImGui::ColorEdit3("clear color", (float*)&clear_color); // Edit 3 floats representing a color
  650. ImGui::Checkbox("Demo Window", &show_demo_window); // Edit bools storing our windows open/close state
  651. ImGui::Checkbox("Another Window", &show_another_window);
  652. if (ImGui::Button("Button")) // Buttons return true when clicked (NB: most widgets return true when edited/activated)
  653. counter++;
  654. ImGui::SameLine();
  655. ImGui::Text("counter = %d", counter);
  656. ImGui::Text("Application average %.3f ms/frame (%.1f FPS)", 1000.0f / ImGui::GetIO().Framerate, ImGui::GetIO().Framerate);
  657. }
  658. // 2. Show another simple window. In most cases you will use an explicit Begin/End pair to name your windows.
  659. if (show_another_window)
  660. {
  661. ImGui::Begin("Another Window", &show_another_window);
  662. ImGui::Text("Hello from another window!");
  663. if (ImGui::Button("Close Me"))
  664. show_another_window = false;
  665. ImGui::End();
  666. }
  667. // 3. Show the ImGui demo window. Most of the sample code is in ImGui::ShowDemoWindow(). Read its code to learn more about Dear ImGui!
  668. if (show_demo_window)
  669. {
  670. 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!
  671. ImGui::ShowDemoWindow(&show_demo_window);
  672. }
  673. // Rendering
  674. memcpy(&g_ClearValue.color.float32[0], &clear_color, 4 * sizeof(float));
  675. frame_begin();
  676. ImGui_ImplVulkan_Render(g_CommandBuffer[g_FrameIndex]);
  677. frame_end();
  678. frame_present();
  679. }
  680. // Cleanup
  681. VkResult err = vkDeviceWaitIdle(g_Device);
  682. check_vk_result(err);
  683. ImGui_ImplVulkan_Shutdown();
  684. ImGui_ImplSDL2_Shutdown();
  685. ImGui::DestroyContext();
  686. SDL_DestroyWindow(window);
  687. cleanup_vulkan();
  688. SDL_Quit();
  689. return 0;
  690. }