cube.c revision 8a399e92cdfb72bd3f025cf63b7fc7681decd32b
1#define _GNU_SOURCE 2#include <stdio.h> 3#include <stdlib.h> 4#include <string.h> 5#include <stdbool.h> 6#include <assert.h> 7 8#ifdef _WIN32 9#pragma comment(linker, "/subsystem:windows") 10#include <windows.h> 11#define APP_NAME_STR_LEN 80 12#else // _WIN32 13#include <xcb/xcb.h> 14#endif // _WIN32 15 16#include <vulkan.h> 17#include <vkDbg.h> 18#include <vk_wsi_lunarg.h> 19 20#include "icd-spv.h" 21 22#include "linmath.h" 23#include <png.h> 24 25#define DEMO_BUFFER_COUNT 2 26#define DEMO_TEXTURE_COUNT 1 27 28#if defined(NDEBUG) && defined(__GNUC__) 29#define U_ASSERT_ONLY __attribute__((unused)) 30#else 31#define U_ASSERT_ONLY 32#endif 33 34/* 35 * structure to track all objects related to a texture. 36 */ 37struct texture_object { 38 VkSampler sampler; 39 40 VkImage image; 41 VkImageLayout imageLayout; 42 43 uint32_t num_mem; 44 VkDeviceMemory *mem; 45 VkImageView view; 46 int32_t tex_width, tex_height; 47}; 48 49static char *tex_files[] = { 50 "lunarg-logo-256x256-solid.png" 51}; 52 53struct vkcube_vs_uniform { 54 // Must start with MVP 55 float mvp[4][4]; 56 float position[12*3][4]; 57 float color[12*3][4]; 58}; 59 60struct vktexcube_vs_uniform { 61 // Must start with MVP 62 float mvp[4][4]; 63 float position[12*3][4]; 64 float attr[12*3][4]; 65}; 66 67//-------------------------------------------------------------------------------------- 68// Mesh and VertexFormat Data 69//-------------------------------------------------------------------------------------- 70struct Vertex 71{ 72 float posX, posY, posZ, posW; // Position data 73 float r, g, b, a; // Color 74}; 75 76struct VertexPosTex 77{ 78 float posX, posY, posZ, posW; // Position data 79 float u, v, s, t; // Texcoord 80}; 81 82#define XYZ1(_x_, _y_, _z_) (_x_), (_y_), (_z_), 1.f 83#define UV(_u_, _v_) (_u_), (_v_), 0.f, 1.f 84 85static const float g_vertex_buffer_data[] = { 86 -1.0f,-1.0f,-1.0f, // Vertex 0 87 -1.0f,-1.0f, 1.0f, 88 -1.0f, 1.0f, 1.0f, 89 90 -1.0f, 1.0f, 1.0f, // Vertex 1 91 -1.0f, 1.0f,-1.0f, 92 -1.0f,-1.0f,-1.0f, 93 94 -1.0f,-1.0f,-1.0f, // Vertex 2 95 1.0f, 1.0f,-1.0f, 96 1.0f,-1.0f,-1.0f, 97 98 -1.0f,-1.0f,-1.0f, // Vertex 3 99 -1.0f, 1.0f,-1.0f, 100 1.0f, 1.0f,-1.0f, 101 102 -1.0f,-1.0f,-1.0f, // Vertex 4 103 1.0f,-1.0f,-1.0f, 104 1.0f,-1.0f, 1.0f, 105 106 -1.0f,-1.0f,-1.0f, // Vertex 5 107 1.0f,-1.0f, 1.0f, 108 -1.0f,-1.0f, 1.0f, 109 110 -1.0f, 1.0f,-1.0f, // Vertex 6 111 -1.0f, 1.0f, 1.0f, 112 1.0f, 1.0f, 1.0f, 113 114 -1.0f, 1.0f,-1.0f, // Vertex 7 115 1.0f, 1.0f, 1.0f, 116 1.0f, 1.0f,-1.0f, 117 118 1.0f, 1.0f,-1.0f, // Vertex 8 119 1.0f, 1.0f, 1.0f, 120 1.0f,-1.0f, 1.0f, 121 122 1.0f,-1.0f, 1.0f, // Vertex 9 123 1.0f,-1.0f,-1.0f, 124 1.0f, 1.0f,-1.0f, 125 126 -1.0f, 1.0f, 1.0f, // Vertex 10 127 -1.0f,-1.0f, 1.0f, 128 1.0f, 1.0f, 1.0f, 129 130 -1.0f,-1.0f, 1.0f, // Vertex 11 131 1.0f,-1.0f, 1.0f, 132 1.0f, 1.0f, 1.0f, 133}; 134 135static const float g_uv_buffer_data[] = { 136 1.0f, 0.0f, // Vertex 0 137 0.0f, 0.0f, 138 0.0f, 1.0f, 139 140 0.0f, 1.0f, // Vertex 1 141 1.0f, 1.0f, 142 1.0f, 0.0f, 143 144// 0.0f, 1.0f, // Vertex 2 145// 1.0f, 0.0f, 146// 0.0f, 0.0f, 147 148// 0.0f, 1.0f, // Vertex 3 149// 1.0f, 0.0f, 150// 1.0f, 1.0f, 151 152 0.0f, 0.0f, // Vertex 2 153 1.0f, 1.0f, 154 1.0f, 0.0f, 155 156 0.0f, 0.0f, // Vertex 3 157 0.0f, 1.0f, 158 1.0f, 1.0f, 159 160 0.0f, 1.0f, // Vertex 4 161 0.0f, 0.0f, 162 1.0f, 0.0f, 163 164 0.0f, 1.0f, // Vertex 5 165 1.0f, 0.0f, 166 1.0f, 1.0f, 167 168 0.0f, 1.0f, // Vertex 6 169 1.0f, 1.0f, 170 1.0f, 0.0f, 171 172 0.0f, 1.0f, // Vertex 7 173 1.0f, 0.0f, 174 0.0f, 0.0f, 175 176 0.0f, 1.0f, // Vertex 8 177 1.0f, 1.0f, 178 1.0f, 0.0f, 179 180 1.0f, 0.0f, // Vertex 9 181 0.0f, 0.0f, 182 0.0f, 1.0f, 183 184 1.0f, 1.0f, // Vertex 10 185 1.0f, 0.0f, 186 0.0f, 1.0f, 187 188 1.0f, 0.0f, // Vertex 11 189 0.0f, 0.0f, 190 0.0f, 1.0f, 191}; 192 193void dumpMatrix(const char *note, mat4x4 MVP) 194{ 195 int i; 196 197 printf("%s: \n", note); 198 for (i=0; i<4; i++) { 199 printf("%f, %f, %f, %f\n", MVP[i][0], MVP[i][1], MVP[i][2], MVP[i][3]); 200 } 201 printf("\n"); 202 fflush(stdout); 203} 204 205void dumpVec4(const char *note, vec4 vector) 206{ 207 printf("%s: \n", note); 208 printf("%f, %f, %f, %f\n", vector[0], vector[1], vector[2], vector[3]); 209 printf("\n"); 210 fflush(stdout); 211} 212 213struct demo { 214#ifdef _WIN32 215#define APP_NAME_STR_LEN 80 216 HINSTANCE connection; // hInstance - Windows Instance 217 char name[APP_NAME_STR_LEN]; // Name to put on the window/icon 218 HWND window; // hWnd - window handle 219#else // _WIN32 220 xcb_connection_t *connection; 221 xcb_screen_t *screen; 222 xcb_window_t window; 223 xcb_intern_atom_reply_t *atom_wm_delete_window; 224#endif // _WIN32 225 bool prepared; 226 bool use_staging_buffer; 227 228 VkInstance inst; 229 VkPhysicalDevice gpu; 230 VkDevice device; 231 VkQueue queue; 232 uint32_t graphics_queue_node_index; 233 VkPhysicalDeviceProperties *gpu_props; 234 VkPhysicalDeviceQueueProperties *queue_props; 235 236 VkFramebuffer framebuffer; 237 int width, height; 238 VkFormat format; 239 240 VkSwapChainWSI swap_chain; 241 struct { 242 VkImage image; 243 VkDeviceMemory mem; 244 VkCmdBuffer cmd; 245 246 VkColorAttachmentView view; 247 } buffers[DEMO_BUFFER_COUNT]; 248 249 struct { 250 VkFormat format; 251 252 VkImage image; 253 uint32_t num_mem; 254 VkDeviceMemory *mem; 255 VkDepthStencilView view; 256 } depth; 257 258 struct texture_object textures[DEMO_TEXTURE_COUNT]; 259 260 struct { 261 VkBuffer buf; 262 uint32_t num_mem; 263 VkDeviceMemory *mem; 264 VkBufferView view; 265 VkBufferViewAttachInfo attach; 266 } uniform_data; 267 268 VkCmdBuffer cmd; // Buffer for initialization commands 269 VkPipelineLayout pipeline_layout; 270 VkDescriptorSetLayout desc_layout; 271 VkPipeline pipeline; 272 273 VkDynamicVpState viewport; 274 VkDynamicRsState raster; 275 VkDynamicCbState color_blend; 276 VkDynamicDsState depth_stencil; 277 278 mat4x4 projection_matrix; 279 mat4x4 view_matrix; 280 mat4x4 model_matrix; 281 282 float spin_angle; 283 float spin_increment; 284 bool pause; 285 286 VkDescriptorPool desc_pool; 287 VkDescriptorSet desc_set; 288 289 bool quit; 290 uint32_t current_buffer; 291}; 292 293static void demo_flush_init_cmd(struct demo *demo) 294{ 295 VkResult U_ASSERT_ONLY err; 296 297 if (demo->cmd == VK_NULL_HANDLE) 298 return; 299 300 err = vkEndCommandBuffer(demo->cmd); 301 assert(!err); 302 303 const VkCmdBuffer cmd_bufs[] = { demo->cmd }; 304 305 err = vkQueueSubmit(demo->queue, 1, cmd_bufs, VK_NULL_HANDLE); 306 assert(!err); 307 308 err = vkQueueWaitIdle(demo->queue); 309 assert(!err); 310 311 vkDestroyObject(demo->device, VK_OBJECT_TYPE_COMMAND_BUFFER, demo->cmd); 312 demo->cmd = VK_NULL_HANDLE; 313} 314 315static void demo_add_mem_refs( 316 struct demo *demo, 317 int num_refs, VkDeviceMemory *mem) 318{ 319 vkQueueAddMemReferences(demo->queue, num_refs, mem); 320} 321 322static void demo_remove_mem_refs( 323 struct demo *demo, 324 int num_refs, VkDeviceMemory *mem) 325{ 326 vkQueueRemoveMemReferences(demo->queue, num_refs, mem); 327} 328 329static void demo_set_image_layout( 330 struct demo *demo, 331 VkImage image, 332 VkImageLayout old_image_layout, 333 VkImageLayout new_image_layout) 334{ 335 VkResult U_ASSERT_ONLY err; 336 337 if (demo->cmd == VK_NULL_HANDLE) { 338 const VkCmdBufferCreateInfo cmd = { 339 .sType = VK_STRUCTURE_TYPE_CMD_BUFFER_CREATE_INFO, 340 .pNext = NULL, 341 .queueNodeIndex = demo->graphics_queue_node_index, 342 .flags = 0, 343 }; 344 345 err = vkCreateCommandBuffer(demo->device, &cmd, &demo->cmd); 346 assert(!err); 347 348 VkCmdBufferBeginInfo cmd_buf_info = { 349 .sType = VK_STRUCTURE_TYPE_CMD_BUFFER_BEGIN_INFO, 350 .pNext = NULL, 351 .flags = VK_CMD_BUFFER_OPTIMIZE_SMALL_BATCH_BIT | 352 VK_CMD_BUFFER_OPTIMIZE_ONE_TIME_SUBMIT_BIT, 353 }; 354 err = vkBeginCommandBuffer(demo->cmd, &cmd_buf_info); 355 } 356 357 VkImageMemoryBarrier image_memory_barrier = { 358 .sType = VK_STRUCTURE_TYPE_IMAGE_MEMORY_BARRIER, 359 .pNext = NULL, 360 .outputMask = 0, 361 .inputMask = 0, 362 .oldLayout = old_image_layout, 363 .newLayout = new_image_layout, 364 .image = image, 365 .subresourceRange = { VK_IMAGE_ASPECT_COLOR, 0, 1, 0, 0 } 366 }; 367 368 if (new_image_layout == VK_IMAGE_LAYOUT_TRANSFER_DESTINATION_OPTIMAL) { 369 /* Make sure anything that was copying from this image has completed */ 370 image_memory_barrier.inputMask = VK_MEMORY_INPUT_TRANSFER_BIT; 371 } 372 373 if (new_image_layout == VK_IMAGE_LAYOUT_SHADER_READ_ONLY_OPTIMAL) { 374 /* Make sure any Copy or CPU writes to image are flushed */ 375 image_memory_barrier.outputMask = VK_MEMORY_OUTPUT_CPU_WRITE_BIT | VK_MEMORY_OUTPUT_TRANSFER_BIT; 376 } 377 378 VkImageMemoryBarrier *pmemory_barrier = &image_memory_barrier; 379 380 VkPipeEvent set_events[] = { VK_PIPE_EVENT_TOP_OF_PIPE }; 381 382 vkCmdPipelineBarrier(demo->cmd, VK_WAIT_EVENT_TOP_OF_PIPE, 1, set_events, 1, (const void **)&pmemory_barrier); 383} 384 385static void demo_draw_build_cmd(struct demo *demo, VkCmdBuffer cmd_buf) 386{ 387 const VkColorAttachmentBindInfo color_attachment = { 388 .view = demo->buffers[demo->current_buffer].view, 389 .layout = VK_IMAGE_LAYOUT_COLOR_ATTACHMENT_OPTIMAL, 390 }; 391 const VkDepthStencilBindInfo depth_stencil = { 392 .view = demo->depth.view, 393 .layout = VK_IMAGE_LAYOUT_DEPTH_STENCIL_ATTACHMENT_OPTIMAL, 394 }; 395 const VkClearColor clear_color = { 396 .color.floatColor = { 0.2f, 0.2f, 0.2f, 0.2f }, 397 .useRawValue = false, 398 }; 399 const float clear_depth = 1.0f; 400 VkImageSubresourceRange clear_range; 401 VkCmdBufferBeginInfo cmd_buf_info = { 402 .sType = VK_STRUCTURE_TYPE_CMD_BUFFER_BEGIN_INFO, 403 .pNext = NULL, 404 .flags = VK_CMD_BUFFER_OPTIMIZE_SMALL_BATCH_BIT | 405 VK_CMD_BUFFER_OPTIMIZE_ONE_TIME_SUBMIT_BIT, 406 }; 407 VkResult U_ASSERT_ONLY err; 408 VkAttachmentLoadOp load_op = VK_ATTACHMENT_LOAD_OP_DONT_CARE; 409 VkAttachmentStoreOp store_op = VK_ATTACHMENT_STORE_OP_DONT_CARE; 410 const VkFramebufferCreateInfo fb_info = { 411 .sType = VK_STRUCTURE_TYPE_FRAMEBUFFER_CREATE_INFO, 412 .pNext = NULL, 413 .colorAttachmentCount = 1, 414 .pColorAttachments = (VkColorAttachmentBindInfo*) &color_attachment, 415 .pDepthStencilAttachment = (VkDepthStencilBindInfo*) &depth_stencil, 416 .sampleCount = 1, 417 .width = demo->width, 418 .height = demo->height, 419 .layers = 1, 420 }; 421 VkRenderPassCreateInfo rp_info; 422 VkRenderPassBegin rp_begin; 423 424 memset(&rp_info, 0 , sizeof(rp_info)); 425 err = vkCreateFramebuffer(demo->device, &fb_info, &rp_begin.framebuffer); 426 assert(!err); 427 rp_info.sType = VK_STRUCTURE_TYPE_RENDER_PASS_CREATE_INFO; 428 rp_info.renderArea.extent.width = demo->width; 429 rp_info.renderArea.extent.height = demo->height; 430 rp_info.colorAttachmentCount = fb_info.colorAttachmentCount; 431 rp_info.pColorFormats = &demo->format; 432 rp_info.pColorLayouts = &color_attachment.layout; 433 rp_info.pColorLoadOps = &load_op; 434 rp_info.pColorStoreOps = &store_op; 435 rp_info.pColorLoadClearValues = &clear_color; 436 rp_info.depthStencilFormat = VK_FORMAT_D16_UNORM; 437 rp_info.depthStencilLayout = depth_stencil.layout; 438 rp_info.depthLoadOp = VK_ATTACHMENT_LOAD_OP_DONT_CARE; 439 rp_info.depthLoadClearValue = clear_depth; 440 rp_info.depthStoreOp = VK_ATTACHMENT_STORE_OP_DONT_CARE; 441 rp_info.stencilLoadOp = VK_ATTACHMENT_LOAD_OP_DONT_CARE; 442 rp_info.stencilLoadClearValue = 0; 443 rp_info.stencilStoreOp = VK_ATTACHMENT_STORE_OP_DONT_CARE; 444 err = vkCreateRenderPass(demo->device, &rp_info, &rp_begin.renderPass); 445 assert(!err); 446 447 err = vkBeginCommandBuffer(cmd_buf, &cmd_buf_info); 448 assert(!err); 449 450 vkCmdBindPipeline(cmd_buf, VK_PIPELINE_BIND_POINT_GRAPHICS, 451 demo->pipeline); 452 vkCmdBindDescriptorSets(cmd_buf, VK_PIPELINE_BIND_POINT_GRAPHICS, 453 0, 1, &demo->desc_set, 0, NULL); 454 455 vkCmdBindDynamicStateObject(cmd_buf, VK_STATE_BIND_POINT_VIEWPORT, demo->viewport); 456 vkCmdBindDynamicStateObject(cmd_buf, VK_STATE_BIND_POINT_RASTER, demo->raster); 457 vkCmdBindDynamicStateObject(cmd_buf, VK_STATE_BIND_POINT_COLOR_BLEND, 458 demo->color_blend); 459 vkCmdBindDynamicStateObject(cmd_buf, VK_STATE_BIND_POINT_DEPTH_STENCIL, 460 demo->depth_stencil); 461 462 vkCmdBeginRenderPass(cmd_buf, &rp_begin); 463 clear_range.aspect = VK_IMAGE_ASPECT_COLOR; 464 clear_range.baseMipLevel = 0; 465 clear_range.mipLevels = 1; 466 clear_range.baseArraySlice = 0; 467 clear_range.arraySize = 1; 468 vkCmdClearColorImage(cmd_buf, 469 demo->buffers[demo->current_buffer].image, 470 VK_IMAGE_LAYOUT_CLEAR_OPTIMAL, 471 clear_color, 1, &clear_range); 472 473 clear_range.aspect = VK_IMAGE_ASPECT_DEPTH; 474 vkCmdClearDepthStencil(cmd_buf, demo->depth.image, 475 VK_IMAGE_LAYOUT_CLEAR_OPTIMAL, 476 clear_depth, 0, 1, &clear_range); 477 478 vkCmdDraw(cmd_buf, 0, 12 * 3, 0, 1); 479 vkCmdEndRenderPass(cmd_buf, rp_begin.renderPass); 480 481 err = vkEndCommandBuffer(cmd_buf); 482 assert(!err); 483 484 vkDestroyObject(demo->device, VK_OBJECT_TYPE_RENDER_PASS, rp_begin.renderPass); 485 vkDestroyObject(demo->device, VK_OBJECT_TYPE_FRAMEBUFFER, rp_begin.framebuffer); 486} 487 488 489void demo_update_data_buffer(struct demo *demo) 490{ 491 mat4x4 MVP, Model, VP; 492 int matrixSize = sizeof(MVP); 493 uint8_t *pData; 494 VkResult U_ASSERT_ONLY err; 495 496 mat4x4_mul(VP, demo->projection_matrix, demo->view_matrix); 497 498 // Rotate 22.5 degrees around the Y axis 499 mat4x4_dup(Model, demo->model_matrix); 500 mat4x4_rotate(demo->model_matrix, Model, 0.0f, 1.0f, 0.0f, (float)degreesToRadians(demo->spin_angle)); 501 mat4x4_mul(MVP, VP, demo->model_matrix); 502 503 assert(demo->uniform_data.num_mem == 1); 504 err = vkMapMemory(demo->device, demo->uniform_data.mem[0], 0, 0, 0, (void **) &pData); 505 assert(!err); 506 507 memcpy(pData, (const void*) &MVP[0][0], matrixSize); 508 509 err = vkUnmapMemory(demo->device, demo->uniform_data.mem[0]); 510 assert(!err); 511} 512 513static void demo_draw(struct demo *demo) 514{ 515 const VkPresentInfoWSI present = { 516 .sType = VK_STRUCTURE_TYPE_PRESENT_INFO_WSI, 517 .pNext = NULL, 518 .image = demo->buffers[demo->current_buffer].image, 519 .flipInterval = 0, 520 }; 521 VkResult U_ASSERT_ONLY err; 522 523 err = vkQueueSubmit(demo->queue, 1, &demo->buffers[demo->current_buffer].cmd, 524 VK_NULL_HANDLE); 525 assert(!err); 526 527 err = vkQueuePresentWSI(demo->queue, &present); 528 assert(!err); 529 530 demo->current_buffer = (demo->current_buffer + 1) % DEMO_BUFFER_COUNT; 531 532 err = vkQueueWaitIdle(demo->queue); 533 assert(err == VK_SUCCESS); 534} 535 536static void demo_prepare_buffers(struct demo *demo) 537{ 538 const VkSwapChainCreateInfoWSI swap_chain = { 539 .sType = VK_STRUCTURE_TYPE_SWAP_CHAIN_CREATE_INFO_WSI, 540 .pNext = NULL, 541 .pNativeWindowSystemHandle = demo->connection, 542 .pNativeWindowHandle = (void *) (intptr_t) demo->window, 543 .imageCount = DEMO_BUFFER_COUNT, 544 .imageFormat = demo->format, 545 .imageExtent = { 546 .width = demo->width, 547 .height = demo->height, 548 }, 549 .imageArraySize = 1, 550 .imageUsageFlags = VK_IMAGE_USAGE_COLOR_ATTACHMENT_BIT, 551 }; 552 VkSwapChainImageInfoWSI images[DEMO_BUFFER_COUNT]; 553 size_t images_size = sizeof(images); 554 VkResult U_ASSERT_ONLY err; 555 uint32_t i; 556 557 err = vkCreateSwapChainWSI(demo->device, &swap_chain, &demo->swap_chain); 558 assert(!err); 559 560 err = vkGetSwapChainInfoWSI(demo->swap_chain, 561 VK_SWAP_CHAIN_INFO_TYPE_PERSISTENT_IMAGES_WSI, 562 &images_size, images); 563 assert(!err && images_size == sizeof(images)); 564 565 for (i = 0; i < DEMO_BUFFER_COUNT; i++) { 566 VkColorAttachmentViewCreateInfo color_attachment_view = { 567 .sType = VK_STRUCTURE_TYPE_COLOR_ATTACHMENT_VIEW_CREATE_INFO, 568 .pNext = NULL, 569 .format = demo->format, 570 .mipLevel = 0, 571 .baseArraySlice = 0, 572 .arraySize = 1, 573 }; 574 575 demo->buffers[i].image = images[i].image; 576 demo->buffers[i].mem = images[i].memory; 577 578 demo_add_mem_refs(demo, 1, &demo->buffers[i].mem); 579 580 demo_set_image_layout(demo, demo->buffers[i].image, 581 VK_IMAGE_LAYOUT_UNDEFINED, 582 VK_IMAGE_LAYOUT_COLOR_ATTACHMENT_OPTIMAL); 583 584 color_attachment_view.image = demo->buffers[i].image; 585 586 err = vkCreateColorAttachmentView(demo->device, 587 &color_attachment_view, &demo->buffers[i].view); 588 assert(!err); 589 } 590} 591 592static void demo_prepare_depth(struct demo *demo) 593{ 594 const VkFormat depth_format = VK_FORMAT_D16_UNORM; 595 const VkImageCreateInfo image = { 596 .sType = VK_STRUCTURE_TYPE_IMAGE_CREATE_INFO, 597 .pNext = NULL, 598 .imageType = VK_IMAGE_TYPE_2D, 599 .format = depth_format, 600 .extent = { demo->width, demo->height, 1 }, 601 .mipLevels = 1, 602 .arraySize = 1, 603 .samples = 1, 604 .tiling = VK_IMAGE_TILING_OPTIMAL, 605 .usage = VK_IMAGE_USAGE_DEPTH_STENCIL_BIT, 606 .flags = 0, 607 }; 608 VkMemoryAllocInfo mem_alloc = { 609 .sType = VK_STRUCTURE_TYPE_MEMORY_ALLOC_INFO, 610 .pNext = NULL, 611 .allocationSize = 0, 612 .memProps = VK_MEMORY_PROPERTY_DEVICE_ONLY, 613 .memPriority = VK_MEMORY_PRIORITY_NORMAL, 614 }; 615 VkDepthStencilViewCreateInfo view = { 616 .sType = VK_STRUCTURE_TYPE_DEPTH_STENCIL_VIEW_CREATE_INFO, 617 .pNext = NULL, 618 .image = VK_NULL_HANDLE, 619 .mipLevel = 0, 620 .baseArraySlice = 0, 621 .arraySize = 1, 622 .flags = 0, 623 }; 624 625 VkMemoryRequirements *mem_reqs; 626 size_t mem_reqs_size = sizeof(VkMemoryRequirements); 627 VkResult U_ASSERT_ONLY err; 628 uint32_t num_allocations = 0; 629 size_t num_alloc_size = sizeof(num_allocations); 630 631 demo->depth.format = depth_format; 632 633 /* create image */ 634 err = vkCreateImage(demo->device, &image, 635 &demo->depth.image); 636 assert(!err); 637 638 err = vkGetObjectInfo(demo->device, 639 VK_OBJECT_TYPE_IMAGE, demo->depth.image, 640 VK_OBJECT_INFO_TYPE_MEMORY_ALLOCATION_COUNT, 641 &num_alloc_size, &num_allocations); 642 assert(!err && num_alloc_size == sizeof(num_allocations)); 643 mem_reqs = malloc(num_allocations * sizeof(VkMemoryRequirements)); 644 demo->depth.mem = malloc(num_allocations * sizeof(VkDeviceMemory)); 645 demo->depth.num_mem = num_allocations; 646 err = vkGetObjectInfo(demo->device, 647 VK_OBJECT_TYPE_IMAGE, demo->depth.image, 648 VK_OBJECT_INFO_TYPE_MEMORY_REQUIREMENTS, 649 &mem_reqs_size, mem_reqs); 650 assert(!err && mem_reqs_size == num_allocations * sizeof(VkMemoryRequirements)); 651 for (uint32_t i = 0; i < num_allocations; i ++) { 652 mem_alloc.allocationSize = mem_reqs[i].size; 653 654 /* allocate memory */ 655 err = vkAllocMemory(demo->device, &mem_alloc, 656 &(demo->depth.mem[i])); 657 assert(!err); 658 659 /* bind memory */ 660 err = vkQueueBindObjectMemory(demo->queue, 661 VK_OBJECT_TYPE_IMAGE, demo->depth.image, 662 i, demo->depth.mem[i], 0); 663 assert(!err); 664 } 665 666 demo_set_image_layout(demo, demo->depth.image, 667 VK_IMAGE_LAYOUT_UNDEFINED, 668 VK_IMAGE_LAYOUT_DEPTH_STENCIL_ATTACHMENT_OPTIMAL); 669 670 demo_add_mem_refs(demo, demo->depth.num_mem, demo->depth.mem); 671 672 /* create image view */ 673 view.image = demo->depth.image; 674 err = vkCreateDepthStencilView(demo->device, &view, 675 &demo->depth.view); 676 assert(!err); 677} 678 679/** loadTexture 680 * loads a png file into an memory object, using cstdio , libpng. 681 * 682 * \param demo : Needed to access VK calls 683 * \param filename : the png file to be loaded 684 * \param width : width of png, to be updated as a side effect of this function 685 * \param height : height of png, to be updated as a side effect of this function 686 * 687 * \return bool : an opengl texture id. true if successful?, 688 * should be validated by the client of this function. 689 * 690 * Source: http://en.wikibooks.org/wiki/OpenGL_Programming/Intermediate/Textures 691 * Modified to copy image to memory 692 * 693 */ 694bool loadTexture(const char *filename, uint8_t *rgba_data, 695 VkSubresourceLayout *layout, 696 int32_t *width, int32_t *height) 697{ 698 //header for testing if it is a png 699 png_byte header[8]; 700 int is_png, bit_depth, color_type, rowbytes; 701 size_t retval; 702 png_uint_32 i, twidth, theight; 703 png_structp png_ptr; 704 png_infop info_ptr, end_info; 705 png_byte *image_data; 706 png_bytep *row_pointers; 707 708 //open file as binary 709 FILE *fp = fopen(filename, "rb"); 710 if (!fp) { 711 return false; 712 } 713 714 //read the header 715 retval = fread(header, 1, 8, fp); 716 if (retval != 8) { 717 fclose(fp); 718 return false; 719 } 720 721 //test if png 722 is_png = !png_sig_cmp(header, 0, 8); 723 if (!is_png) { 724 fclose(fp); 725 return false; 726 } 727 728 //create png struct 729 png_ptr = png_create_read_struct(PNG_LIBPNG_VER_STRING, NULL, 730 NULL, NULL); 731 if (!png_ptr) { 732 fclose(fp); 733 return (false); 734 } 735 736 //create png info struct 737 info_ptr = png_create_info_struct(png_ptr); 738 if (!info_ptr) { 739 png_destroy_read_struct(&png_ptr, (png_infopp) NULL, (png_infopp) NULL); 740 fclose(fp); 741 return (false); 742 } 743 744 //create png info struct 745 end_info = png_create_info_struct(png_ptr); 746 if (!end_info) { 747 png_destroy_read_struct(&png_ptr, &info_ptr, (png_infopp) NULL); 748 fclose(fp); 749 return (false); 750 } 751 752 //png error stuff, not sure libpng man suggests this. 753 if (setjmp(png_jmpbuf(png_ptr))) { 754 png_destroy_read_struct(&png_ptr, &info_ptr, &end_info); 755 fclose(fp); 756 return (false); 757 } 758 759 //init png reading 760 png_init_io(png_ptr, fp); 761 762 //let libpng know you already read the first 8 bytes 763 png_set_sig_bytes(png_ptr, 8); 764 765 // read all the info up to the image data 766 png_read_info(png_ptr, info_ptr); 767 768 // get info about png 769 png_get_IHDR(png_ptr, info_ptr, &twidth, &theight, &bit_depth, &color_type, 770 NULL, NULL, NULL); 771 772 //update width and height based on png info 773 *width = twidth; 774 *height = theight; 775 776 // Require that incoming texture be 8bits per color component 777 // and 4 components (RGBA). 778 if (png_get_bit_depth(png_ptr, info_ptr) != 8 || 779 png_get_channels(png_ptr, info_ptr) != 4) { 780 return false; 781 } 782 783 if (rgba_data == NULL) { 784 // If data pointer is null, we just want the width & height 785 // clean up memory and close stuff 786 png_destroy_read_struct(&png_ptr, &info_ptr, &end_info); 787 fclose(fp); 788 789 return true; 790 } 791 792 // Update the png info struct. 793 png_read_update_info(png_ptr, info_ptr); 794 795 // Row size in bytes. 796 rowbytes = png_get_rowbytes(png_ptr, info_ptr); 797 798 // Allocate the image_data as a big block, to be given to opengl 799 image_data = (png_byte *)malloc(rowbytes * theight * sizeof(png_byte)); 800 if (!image_data) { 801 //clean up memory and close stuff 802 png_destroy_read_struct(&png_ptr, &info_ptr, &end_info); 803 fclose(fp); 804 return false; 805 } 806 807 // row_pointers is for pointing to image_data for reading the png with libpng 808 row_pointers = (png_bytep *)malloc(theight * sizeof(png_bytep)); 809 if (!row_pointers) { 810 //clean up memory and close stuff 811 png_destroy_read_struct(&png_ptr, &info_ptr, &end_info); 812 // delete[] image_data; 813 fclose(fp); 814 return false; 815 } 816 // set the individual row_pointers to point at the correct offsets of image_data 817 for (i = 0; i < theight; ++i) 818 row_pointers[theight - 1 - i] = rgba_data + i * layout->rowPitch; 819 820 // read the png into image_data through row_pointers 821 png_read_image(png_ptr, row_pointers); 822 823 // clean up memory and close stuff 824 png_destroy_read_struct(&png_ptr, &info_ptr, &end_info); 825 free(row_pointers); 826 free(image_data); 827 fclose(fp); 828 829 return true; 830} 831 832static void demo_prepare_texture_image(struct demo *demo, 833 const char *filename, 834 struct texture_object *tex_obj, 835 VkImageTiling tiling, 836 VkImageUsageFlags usage, 837 VkFlags mem_props) 838{ 839 const VkFormat tex_format = VK_FORMAT_B8G8R8A8_UNORM; 840 int32_t tex_width; 841 int32_t tex_height; 842 VkResult U_ASSERT_ONLY err; 843 844 err = loadTexture(filename, NULL, NULL, &tex_width, &tex_height); 845 assert(err); 846 847 tex_obj->tex_width = tex_width; 848 tex_obj->tex_height = tex_height; 849 850 const VkImageCreateInfo image_create_info = { 851 .sType = VK_STRUCTURE_TYPE_IMAGE_CREATE_INFO, 852 .pNext = NULL, 853 .imageType = VK_IMAGE_TYPE_2D, 854 .format = tex_format, 855 .extent = { tex_width, tex_height, 1 }, 856 .mipLevels = 1, 857 .arraySize = 1, 858 .samples = 1, 859 .tiling = tiling, 860 .usage = usage, 861 .flags = 0, 862 }; 863 VkMemoryAllocInfo mem_alloc = { 864 .sType = VK_STRUCTURE_TYPE_MEMORY_ALLOC_INFO, 865 .pNext = NULL, 866 .allocationSize = 0, 867 .memProps = mem_props, 868 .memPriority = VK_MEMORY_PRIORITY_NORMAL, 869 }; 870 871 VkMemoryRequirements *mem_reqs; 872 size_t mem_reqs_size = sizeof(VkMemoryRequirements); 873 uint32_t num_allocations = 0; 874 size_t num_alloc_size = sizeof(num_allocations); 875 876 err = vkCreateImage(demo->device, &image_create_info, 877 &tex_obj->image); 878 assert(!err); 879 880 err = vkGetObjectInfo(demo->device, 881 VK_OBJECT_TYPE_IMAGE, tex_obj->image, 882 VK_OBJECT_INFO_TYPE_MEMORY_ALLOCATION_COUNT, 883 &num_alloc_size, &num_allocations); 884 assert(!err && num_alloc_size == sizeof(num_allocations)); 885 mem_reqs = malloc(num_allocations * sizeof(VkMemoryRequirements)); 886 tex_obj->mem = malloc(num_allocations * sizeof(VkDeviceMemory)); 887 err = vkGetObjectInfo(demo->device, 888 VK_OBJECT_TYPE_IMAGE, tex_obj->image, 889 VK_OBJECT_INFO_TYPE_MEMORY_REQUIREMENTS, 890 &mem_reqs_size, mem_reqs); 891 assert(!err && mem_reqs_size == num_allocations * sizeof(VkMemoryRequirements)); 892 for (uint32_t j = 0; j < num_allocations; j ++) { 893 mem_alloc.allocationSize = mem_reqs[j].size; 894 895 /* allocate memory */ 896 err = vkAllocMemory(demo->device, &mem_alloc, 897 &(tex_obj->mem[j])); 898 assert(!err); 899 900 /* bind memory */ 901 err = vkQueueBindObjectMemory(demo->queue, 902 VK_OBJECT_TYPE_IMAGE, tex_obj->image, 903 j, tex_obj->mem[j], 0); 904 assert(!err); 905 } 906 free(mem_reqs); 907 mem_reqs = NULL; 908 909 tex_obj->num_mem = num_allocations; 910 911 if (mem_props & VK_MEMORY_PROPERTY_HOST_VISIBLE_BIT) { 912 const VkImageSubresource subres = { 913 .aspect = VK_IMAGE_ASPECT_COLOR, 914 .mipLevel = 0, 915 .arraySlice = 0, 916 }; 917 VkSubresourceLayout layout; 918 size_t layout_size = sizeof(VkSubresourceLayout); 919 void *data; 920 921 err = vkGetImageSubresourceInfo(demo->device, tex_obj->image, &subres, 922 VK_SUBRESOURCE_INFO_TYPE_LAYOUT, 923 &layout_size, &layout); 924 assert(!err && layout_size == sizeof(layout)); 925 /* Linear texture must be within a single memory object */ 926 assert(num_allocations == 1); 927 928 err = vkMapMemory(demo->device, tex_obj->mem[0], 0, 0, 0, &data); 929 assert(!err); 930 931 if (!loadTexture(filename, data, &layout, &tex_width, &tex_height)) { 932 fprintf(stderr, "Error loading texture: %s\n", filename); 933 } 934 935 err = vkUnmapMemory(demo->device, tex_obj->mem[0]); 936 assert(!err); 937 } 938 939 tex_obj->imageLayout = VK_IMAGE_LAYOUT_SHADER_READ_ONLY_OPTIMAL; 940 demo_set_image_layout(demo, tex_obj->image, 941 VK_IMAGE_LAYOUT_UNDEFINED, 942 tex_obj->imageLayout); 943 /* setting the image layout does not reference the actual memory so no need to add a mem ref */ 944} 945 946static void demo_destroy_texture_image(struct demo *demo, struct texture_object *tex_objs) 947{ 948 /* clean up staging resources */ 949 for (uint32_t j = 0; j < tex_objs->num_mem; j ++) { 950 vkQueueBindObjectMemory(demo->queue, 951 VK_OBJECT_TYPE_IMAGE, tex_objs->image, j, VK_NULL_HANDLE, 0); 952 vkFreeMemory(demo->device, tex_objs->mem[j]); 953 } 954 955 free(tex_objs->mem); 956 vkDestroyObject(demo->device, VK_OBJECT_TYPE_IMAGE, tex_objs->image); 957} 958 959static void demo_prepare_textures(struct demo *demo) 960{ 961 const VkFormat tex_format = VK_FORMAT_R8G8B8A8_UNORM; 962 VkFormatProperties props; 963 size_t size = sizeof(props); 964 VkResult U_ASSERT_ONLY err; 965 uint32_t i; 966 967 err = vkGetFormatInfo(demo->device, tex_format, 968 VK_FORMAT_INFO_TYPE_PROPERTIES, 969 &size, &props); 970 assert(!err); 971 972 for (i = 0; i < DEMO_TEXTURE_COUNT; i++) { 973 974 if (props.linearTilingFeatures & VK_FORMAT_FEATURE_SAMPLED_IMAGE_BIT && !demo->use_staging_buffer) { 975 /* Device can texture using linear textures */ 976 demo_prepare_texture_image(demo, tex_files[i], &demo->textures[i], 977 VK_IMAGE_TILING_LINEAR, VK_IMAGE_USAGE_SAMPLED_BIT, VK_MEMORY_PROPERTY_HOST_VISIBLE_BIT); 978 } else if (props.optimalTilingFeatures & VK_FORMAT_FEATURE_SAMPLED_IMAGE_BIT) { 979 /* Must use staging buffer to copy linear texture to optimized */ 980 struct texture_object staging_texture; 981 982 memset(&staging_texture, 0, sizeof(staging_texture)); 983 demo_prepare_texture_image(demo, tex_files[i], &staging_texture, 984 VK_IMAGE_TILING_LINEAR, VK_IMAGE_USAGE_TRANSFER_SOURCE_BIT, VK_MEMORY_PROPERTY_HOST_VISIBLE_BIT); 985 986 demo_prepare_texture_image(demo, tex_files[i], &demo->textures[i], 987 VK_IMAGE_TILING_OPTIMAL, 988 (VK_IMAGE_USAGE_TRANSFER_DESTINATION_BIT | VK_IMAGE_USAGE_SAMPLED_BIT), 989 VK_MEMORY_PROPERTY_DEVICE_ONLY); 990 991 demo_set_image_layout(demo, staging_texture.image, 992 staging_texture.imageLayout, 993 VK_IMAGE_LAYOUT_TRANSFER_SOURCE_OPTIMAL); 994 995 demo_set_image_layout(demo, demo->textures[i].image, 996 demo->textures[i].imageLayout, 997 VK_IMAGE_LAYOUT_TRANSFER_DESTINATION_OPTIMAL); 998 999 VkImageCopy copy_region = { 1000 .srcSubresource = { VK_IMAGE_ASPECT_COLOR, 0, 0 }, 1001 .srcOffset = { 0, 0, 0 }, 1002 .destSubresource = { VK_IMAGE_ASPECT_COLOR, 0, 0 }, 1003 .destOffset = { 0, 0, 0 }, 1004 .extent = { staging_texture.tex_width, staging_texture.tex_height, 1 }, 1005 }; 1006 vkCmdCopyImage(demo->cmd, 1007 staging_texture.image, VK_IMAGE_LAYOUT_TRANSFER_SOURCE_OPTIMAL, 1008 demo->textures[i].image, VK_IMAGE_LAYOUT_TRANSFER_DESTINATION_OPTIMAL, 1009 1, ©_region); 1010 1011 demo_add_mem_refs(demo, staging_texture.num_mem, staging_texture.mem); 1012 demo_add_mem_refs(demo, demo->textures[i].num_mem, demo->textures[i].mem); 1013 1014 demo_set_image_layout(demo, demo->textures[i].image, 1015 VK_IMAGE_LAYOUT_TRANSFER_DESTINATION_OPTIMAL, 1016 demo->textures[i].imageLayout); 1017 1018 demo_flush_init_cmd(demo); 1019 1020 demo_remove_mem_refs(demo, staging_texture.num_mem, staging_texture.mem); 1021 demo_destroy_texture_image(demo, &staging_texture); 1022 } else { 1023 /* Can't support VK_FORMAT_B8G8R8A8_UNORM !? */ 1024 assert(!"No support for tB8G8R8A8_UNORM as texture image format"); 1025 } 1026 1027 const VkSamplerCreateInfo sampler = { 1028 .sType = VK_STRUCTURE_TYPE_SAMPLER_CREATE_INFO, 1029 .pNext = NULL, 1030 .magFilter = VK_TEX_FILTER_NEAREST, 1031 .minFilter = VK_TEX_FILTER_NEAREST, 1032 .mipMode = VK_TEX_MIPMAP_MODE_BASE, 1033 .addressU = VK_TEX_ADDRESS_CLAMP, 1034 .addressV = VK_TEX_ADDRESS_CLAMP, 1035 .addressW = VK_TEX_ADDRESS_CLAMP, 1036 .mipLodBias = 0.0f, 1037 .maxAnisotropy = 1, 1038 .compareOp = VK_COMPARE_OP_NEVER, 1039 .minLod = 0.0f, 1040 .maxLod = 0.0f, 1041 .borderColor = VK_BORDER_COLOR_OPAQUE_WHITE, 1042 }; 1043 1044 VkImageViewCreateInfo view = { 1045 .sType = VK_STRUCTURE_TYPE_IMAGE_VIEW_CREATE_INFO, 1046 .pNext = NULL, 1047 .image = VK_NULL_HANDLE, 1048 .viewType = VK_IMAGE_VIEW_TYPE_2D, 1049 .format = tex_format, 1050 .channels = { VK_CHANNEL_SWIZZLE_R, 1051 VK_CHANNEL_SWIZZLE_G, 1052 VK_CHANNEL_SWIZZLE_B, 1053 VK_CHANNEL_SWIZZLE_A, }, 1054 .subresourceRange = { VK_IMAGE_ASPECT_COLOR, 0, 1, 0, 1 }, 1055 .minLod = 0.0f, 1056 }; 1057 1058 /* create sampler */ 1059 err = vkCreateSampler(demo->device, &sampler, 1060 &demo->textures[i].sampler); 1061 assert(!err); 1062 1063 /* create image view */ 1064 view.image = demo->textures[i].image; 1065 err = vkCreateImageView(demo->device, &view, 1066 &demo->textures[i].view); 1067 assert(!err); 1068 } 1069} 1070 1071void demo_prepare_cube_data_buffer(struct demo *demo) 1072{ 1073 VkBufferCreateInfo buf_info; 1074 VkBufferViewCreateInfo view_info; 1075 VkMemoryAllocInfo alloc_info = { 1076 .sType = VK_STRUCTURE_TYPE_MEMORY_ALLOC_INFO, 1077 .pNext = NULL, 1078 .allocationSize = 0, 1079 .memProps = VK_MEMORY_PROPERTY_HOST_VISIBLE_BIT, 1080 .memPriority = VK_MEMORY_PRIORITY_NORMAL, 1081 }; 1082 VkMemoryRequirements *mem_reqs; 1083 size_t mem_reqs_size = sizeof(VkMemoryRequirements); 1084 uint32_t num_allocations = 0; 1085 size_t num_alloc_size = sizeof(num_allocations); 1086 uint8_t *pData; 1087 int i; 1088 mat4x4 MVP, VP; 1089 VkResult U_ASSERT_ONLY err; 1090 struct vktexcube_vs_uniform data; 1091 1092 mat4x4_mul(VP, demo->projection_matrix, demo->view_matrix); 1093 mat4x4_mul(MVP, VP, demo->model_matrix); 1094 memcpy(data.mvp, MVP, sizeof(MVP)); 1095// dumpMatrix("MVP", MVP); 1096 1097 for (i=0; i<12*3; i++) { 1098 data.position[i][0] = g_vertex_buffer_data[i*3]; 1099 data.position[i][1] = g_vertex_buffer_data[i*3+1]; 1100 data.position[i][2] = g_vertex_buffer_data[i*3+2]; 1101 data.position[i][3] = 1.0f; 1102 data.attr[i][0] = g_uv_buffer_data[2*i]; 1103 data.attr[i][1] = g_uv_buffer_data[2*i + 1]; 1104 data.attr[i][2] = 0; 1105 data.attr[i][3] = 0; 1106 } 1107 1108 memset(&buf_info, 0, sizeof(buf_info)); 1109 buf_info.sType = VK_STRUCTURE_TYPE_BUFFER_CREATE_INFO; 1110 buf_info.size = sizeof(data); 1111 buf_info.usage = VK_BUFFER_USAGE_UNIFORM_BUFFER_BIT; 1112 err = vkCreateBuffer(demo->device, &buf_info, &demo->uniform_data.buf); 1113 assert(!err); 1114 1115 err = vkGetObjectInfo(demo->device, 1116 VK_OBJECT_TYPE_BUFFER, demo->uniform_data.buf, 1117 VK_OBJECT_INFO_TYPE_MEMORY_ALLOCATION_COUNT, 1118 &num_alloc_size, &num_allocations); 1119 assert(!err && num_alloc_size == sizeof(num_allocations)); 1120 mem_reqs = malloc(num_allocations * sizeof(VkMemoryRequirements)); 1121 demo->uniform_data.mem = malloc(num_allocations * sizeof(VkDeviceMemory)); 1122 demo->uniform_data.num_mem = num_allocations; 1123 err = vkGetObjectInfo(demo->device, 1124 VK_OBJECT_TYPE_BUFFER, demo->uniform_data.buf, 1125 VK_OBJECT_INFO_TYPE_MEMORY_REQUIREMENTS, 1126 &mem_reqs_size, mem_reqs); 1127 assert(!err && mem_reqs_size == num_allocations * sizeof(*mem_reqs)); 1128 for (uint32_t i = 0; i < num_allocations; i ++) { 1129 alloc_info.allocationSize = mem_reqs[i].size; 1130 1131 err = vkAllocMemory(demo->device, &alloc_info, &(demo->uniform_data.mem[i])); 1132 assert(!err); 1133 1134 err = vkMapMemory(demo->device, demo->uniform_data.mem[i], 0, 0, 0, (void **) &pData); 1135 assert(!err); 1136 1137 memcpy(pData, &data, sizeof data); 1138 1139 err = vkUnmapMemory(demo->device, demo->uniform_data.mem[i]); 1140 assert(!err); 1141 1142 err = vkQueueBindObjectMemory(demo->queue, 1143 VK_OBJECT_TYPE_BUFFER, demo->uniform_data.buf, 1144 i, demo->uniform_data.mem[i], 0); 1145 assert(!err); 1146 } 1147 demo_add_mem_refs(demo, demo->uniform_data.num_mem, demo->uniform_data.mem); 1148 1149 memset(&view_info, 0, sizeof(view_info)); 1150 view_info.sType = VK_STRUCTURE_TYPE_BUFFER_VIEW_CREATE_INFO; 1151 view_info.buffer = demo->uniform_data.buf; 1152 view_info.viewType = VK_BUFFER_VIEW_TYPE_RAW; 1153 view_info.offset = 0; 1154 view_info.range = sizeof(data); 1155 1156 err = vkCreateBufferView(demo->device, &view_info, &demo->uniform_data.view); 1157 assert(!err); 1158 1159 demo->uniform_data.attach.sType = VK_STRUCTURE_TYPE_BUFFER_VIEW_ATTACH_INFO; 1160 demo->uniform_data.attach.view = demo->uniform_data.view; 1161} 1162 1163static void demo_prepare_descriptor_layout(struct demo *demo) 1164{ 1165 const VkDescriptorSetLayoutBinding layout_bindings[2] = { 1166 [0] = { 1167 .descriptorType = VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER, 1168 .count = 1, 1169 .stageFlags = VK_SHADER_STAGE_VERTEX_BIT, 1170 .pImmutableSamplers = NULL, 1171 }, 1172 [1] = { 1173 .descriptorType = VK_DESCRIPTOR_TYPE_COMBINED_IMAGE_SAMPLER, 1174 .count = DEMO_TEXTURE_COUNT, 1175 .stageFlags = VK_SHADER_STAGE_FRAGMENT_BIT, 1176 .pImmutableSamplers = NULL, 1177 }, 1178 }; 1179 const VkDescriptorSetLayoutCreateInfo descriptor_layout = { 1180 .sType = VK_STRUCTURE_TYPE_DESCRIPTOR_SET_LAYOUT_CREATE_INFO, 1181 .pNext = NULL, 1182 .count = 2, 1183 .pBinding = layout_bindings, 1184 }; 1185 VkResult U_ASSERT_ONLY err; 1186 1187 err = vkCreateDescriptorSetLayout(demo->device, 1188 &descriptor_layout, &demo->desc_layout); 1189 assert(!err); 1190 1191 const VkPipelineLayoutCreateInfo pPipelineLayoutCreateInfo = { 1192 .sType = VK_STRUCTURE_TYPE_PIPELINE_LAYOUT_CREATE_INFO, 1193 .pNext = NULL, 1194 .descriptorSetCount = 1, 1195 .pSetLayouts = &demo->desc_layout, 1196 }; 1197 1198 err = vkCreatePipelineLayout(demo->device, 1199 &pPipelineLayoutCreateInfo, 1200 &demo->pipeline_layout); 1201 assert(!err); 1202} 1203 1204static VkShader demo_prepare_shader(struct demo *demo, 1205 VkShaderStage stage, 1206 const void *code, 1207 size_t size) 1208{ 1209 VkShaderCreateInfo createInfo; 1210 VkShader shader; 1211 VkResult err; 1212 1213 1214 createInfo.sType = VK_STRUCTURE_TYPE_SHADER_CREATE_INFO; 1215 createInfo.pNext = NULL; 1216 1217#ifdef EXTERNAL_SPV 1218 createInfo.codeSize = size; 1219 createInfo.pCode = code; 1220 createInfo.flags = 0; 1221 1222 err = vkCreateShader(demo->device, &createInfo, &shader); 1223 if (err) { 1224 free((void *) createInfo.pCode); 1225 } 1226#else 1227 // Create fake SPV structure to feed GLSL 1228 // to the driver "under the covers" 1229 createInfo.codeSize = 3 * sizeof(uint32_t) + size + 1; 1230 createInfo.pCode = malloc(createInfo.codeSize); 1231 createInfo.flags = 0; 1232 1233 /* try version 0 first: VkShaderStage followed by GLSL */ 1234 ((uint32_t *) createInfo.pCode)[0] = ICD_SPV_MAGIC; 1235 ((uint32_t *) createInfo.pCode)[1] = 0; 1236 ((uint32_t *) createInfo.pCode)[2] = stage; 1237 memcpy(((uint32_t *) createInfo.pCode + 3), code, size + 1); 1238 1239 err = vkCreateShader(demo->device, &createInfo, &shader); 1240 if (err) { 1241 free((void *) createInfo.pCode); 1242 return (VkShader) VK_NULL_HANDLE; 1243 } 1244#endif 1245 1246 return shader; 1247} 1248 1249char *demo_read_spv(const char *filename, size_t *psize) 1250{ 1251 long int size; 1252 size_t U_ASSERT_ONLY retval; 1253 void *shader_code; 1254 1255 FILE *fp = fopen(filename, "rb"); 1256 if (!fp) return NULL; 1257 1258 fseek(fp, 0L, SEEK_END); 1259 size = ftell(fp); 1260 1261 fseek(fp, 0L, SEEK_SET); 1262 1263 shader_code = malloc(size); 1264 retval = fread(shader_code, size, 1, fp); 1265 assert(retval == 1); 1266 1267 *psize = size; 1268 1269 return shader_code; 1270} 1271 1272static VkShader demo_prepare_vs(struct demo *demo) 1273{ 1274#ifdef EXTERNAL_SPV 1275 void *vertShaderCode; 1276 size_t size; 1277 1278 vertShaderCode = demo_read_spv("cube-vert.spv", &size); 1279 1280 return demo_prepare_shader(demo, VK_SHADER_STAGE_VERTEX, 1281 vertShaderCode, size); 1282#else 1283 static const char *vertShaderText = 1284 "#version 140\n" 1285 "#extension GL_ARB_separate_shader_objects : enable\n" 1286 "#extension GL_ARB_shading_language_420pack : enable\n" 1287 "\n" 1288 "layout(binding = 0) uniform buf {\n" 1289 " mat4 MVP;\n" 1290 " vec4 position[12*3];\n" 1291 " vec4 attr[12*3];\n" 1292 "} ubuf;\n" 1293 "\n" 1294 "layout (location = 0) out vec4 texcoord;\n" 1295 "\n" 1296 "void main() \n" 1297 "{\n" 1298 " texcoord = ubuf.attr[gl_VertexID];\n" 1299 " gl_Position = ubuf.MVP * ubuf.position[gl_VertexID];\n" 1300 "}\n"; 1301 1302 return demo_prepare_shader(demo, VK_SHADER_STAGE_VERTEX, 1303 (const void *) vertShaderText, 1304 strlen(vertShaderText)); 1305#endif 1306} 1307 1308static VkShader demo_prepare_fs(struct demo *demo) 1309{ 1310#ifdef EXTERNAL_SPV 1311 void *fragShaderCode; 1312 size_t size; 1313 1314 fragShaderCode = demo_read_spv("cube-frag.spv", &size); 1315 1316 return demo_prepare_shader(demo, VK_SHADER_STAGE_FRAGMENT, 1317 fragShaderCode, size); 1318#else 1319 static const char *fragShaderText = 1320 "#version 140\n" 1321 "#extension GL_ARB_separate_shader_objects : enable\n" 1322 "#extension GL_ARB_shading_language_420pack : enable\n" 1323 "layout (binding = 1) uniform sampler2D tex;\n" 1324 "\n" 1325 "layout (location = 0) in vec4 texcoord;\n" 1326 "void main() {\n" 1327 " gl_FragColor = texture(tex, texcoord.xy);\n" 1328 "}\n"; 1329 1330 return demo_prepare_shader(demo, VK_SHADER_STAGE_FRAGMENT, 1331 (const void *) fragShaderText, 1332 strlen(fragShaderText)); 1333#endif 1334} 1335 1336static void demo_prepare_pipeline(struct demo *demo) 1337{ 1338 VkGraphicsPipelineCreateInfo pipeline; 1339 VkPipelineIaStateCreateInfo ia; 1340 VkPipelineRsStateCreateInfo rs; 1341 VkPipelineCbStateCreateInfo cb; 1342 VkPipelineDsStateCreateInfo ds; 1343 VkPipelineShaderStageCreateInfo vs; 1344 VkPipelineShaderStageCreateInfo fs; 1345 VkPipelineVpStateCreateInfo vp; 1346 VkPipelineMsStateCreateInfo ms; 1347 VkResult U_ASSERT_ONLY err; 1348 1349 memset(&pipeline, 0, sizeof(pipeline)); 1350 pipeline.sType = VK_STRUCTURE_TYPE_GRAPHICS_PIPELINE_CREATE_INFO; 1351 pipeline.layout = demo->pipeline_layout; 1352 1353 memset(&ia, 0, sizeof(ia)); 1354 ia.sType = VK_STRUCTURE_TYPE_PIPELINE_IA_STATE_CREATE_INFO; 1355 ia.topology = VK_PRIMITIVE_TOPOLOGY_TRIANGLE_LIST; 1356 1357 memset(&rs, 0, sizeof(rs)); 1358 rs.sType = VK_STRUCTURE_TYPE_PIPELINE_RS_STATE_CREATE_INFO; 1359 rs.fillMode = VK_FILL_MODE_SOLID; 1360 rs.cullMode = VK_CULL_MODE_BACK; 1361 rs.frontFace = VK_FRONT_FACE_CCW; 1362 1363 memset(&cb, 0, sizeof(cb)); 1364 cb.sType = VK_STRUCTURE_TYPE_PIPELINE_CB_STATE_CREATE_INFO; 1365 VkPipelineCbAttachmentState att_state[1]; 1366 memset(att_state, 0, sizeof(att_state)); 1367 att_state[0].format = demo->format; 1368 att_state[0].channelWriteMask = 0xf; 1369 att_state[0].blendEnable = VK_FALSE; 1370 cb.attachmentCount = 1; 1371 cb.pAttachments = att_state; 1372 1373 memset(&vp, 0, sizeof(vp)); 1374 vp.sType = VK_STRUCTURE_TYPE_PIPELINE_VP_STATE_CREATE_INFO; 1375 vp.viewportCount = 1; 1376 vp.clipOrigin = VK_COORDINATE_ORIGIN_LOWER_LEFT; 1377 1378 memset(&ds, 0, sizeof(ds)); 1379 ds.sType = VK_STRUCTURE_TYPE_PIPELINE_DS_STATE_CREATE_INFO; 1380 ds.format = demo->depth.format; 1381 ds.depthTestEnable = VK_TRUE; 1382 ds.depthWriteEnable = VK_TRUE; 1383 ds.depthCompareOp = VK_COMPARE_OP_LESS_EQUAL; 1384 ds.depthBoundsEnable = VK_FALSE; 1385 ds.back.stencilFailOp = VK_STENCIL_OP_KEEP; 1386 ds.back.stencilPassOp = VK_STENCIL_OP_KEEP; 1387 ds.back.stencilCompareOp = VK_COMPARE_OP_ALWAYS; 1388 ds.stencilTestEnable = VK_FALSE; 1389 ds.front = ds.back; 1390 1391 memset(&vs, 0, sizeof(vs)); 1392 vs.sType = VK_STRUCTURE_TYPE_PIPELINE_SHADER_STAGE_CREATE_INFO; 1393 vs.shader.stage = VK_SHADER_STAGE_VERTEX; 1394 vs.shader.shader = demo_prepare_vs(demo); 1395 assert(vs.shader.shader != VK_NULL_HANDLE); 1396 1397 memset(&fs, 0, sizeof(fs)); 1398 fs.sType = VK_STRUCTURE_TYPE_PIPELINE_SHADER_STAGE_CREATE_INFO; 1399 fs.shader.stage = VK_SHADER_STAGE_FRAGMENT; 1400 fs.shader.shader = demo_prepare_fs(demo); 1401 assert(fs.shader.shader != VK_NULL_HANDLE); 1402 1403 memset(&ms, 0, sizeof(ms)); 1404 ms.sType = VK_STRUCTURE_TYPE_PIPELINE_MS_STATE_CREATE_INFO; 1405 ms.sampleMask = 1; 1406 ms.multisampleEnable = VK_FALSE; 1407 ms.samples = 1; 1408 1409 pipeline.pNext = (const void *) &ia; 1410 ia.pNext = (const void *) &rs; 1411 rs.pNext = (const void *) &cb; 1412 cb.pNext = (const void *) &ms; 1413 ms.pNext = (const void *) &vp; 1414 vp.pNext = (const void *) &ds; 1415 ds.pNext = (const void *) &vs; 1416 vs.pNext = (const void *) &fs; 1417 1418 err = vkCreateGraphicsPipeline(demo->device, &pipeline, &demo->pipeline); 1419 assert(!err); 1420 1421 vkDestroyObject(demo->device, VK_OBJECT_TYPE_SHADER, vs.shader.shader); 1422 vkDestroyObject(demo->device, VK_OBJECT_TYPE_SHADER, fs.shader.shader); 1423} 1424 1425static void demo_prepare_dynamic_states(struct demo *demo) 1426{ 1427 VkDynamicVpStateCreateInfo viewport_create; 1428 VkDynamicRsStateCreateInfo raster; 1429 VkDynamicCbStateCreateInfo color_blend; 1430 VkDynamicDsStateCreateInfo depth_stencil; 1431 VkResult U_ASSERT_ONLY err; 1432 1433 memset(&viewport_create, 0, sizeof(viewport_create)); 1434 viewport_create.sType = VK_STRUCTURE_TYPE_DYNAMIC_VP_STATE_CREATE_INFO; 1435 viewport_create.viewportAndScissorCount = 1; 1436 VkViewport viewport; 1437 memset(&viewport, 0, sizeof(viewport)); 1438 viewport.height = (float) demo->height; 1439 viewport.width = (float) demo->width; 1440 viewport.minDepth = (float) 0.0f; 1441 viewport.maxDepth = (float) 1.0f; 1442 viewport_create.pViewports = &viewport; 1443 VkRect scissor; 1444 memset(&scissor, 0, sizeof(scissor)); 1445 scissor.extent.width = demo->width; 1446 scissor.extent.height = demo->height; 1447 scissor.offset.x = 0; 1448 scissor.offset.y = 0; 1449 viewport_create.pScissors = &scissor; 1450 1451 memset(&raster, 0, sizeof(raster)); 1452 raster.sType = VK_STRUCTURE_TYPE_DYNAMIC_RS_STATE_CREATE_INFO; 1453 raster.pointSize = 1.0; 1454 raster.lineWidth = 1.0; 1455 1456 memset(&color_blend, 0, sizeof(color_blend)); 1457 color_blend.sType = VK_STRUCTURE_TYPE_DYNAMIC_CB_STATE_CREATE_INFO; 1458 color_blend.blendConst[0] = 1.0f; 1459 color_blend.blendConst[1] = 1.0f; 1460 color_blend.blendConst[2] = 1.0f; 1461 color_blend.blendConst[3] = 1.0f; 1462 1463 memset(&depth_stencil, 0, sizeof(depth_stencil)); 1464 depth_stencil.sType = VK_STRUCTURE_TYPE_DYNAMIC_DS_STATE_CREATE_INFO; 1465 depth_stencil.minDepth = 0.0f; 1466 depth_stencil.maxDepth = 1.0f; 1467 depth_stencil.stencilBackRef = 0; 1468 depth_stencil.stencilFrontRef = 0; 1469 depth_stencil.stencilReadMask = 0xff; 1470 depth_stencil.stencilWriteMask = 0xff; 1471 1472 err = vkCreateDynamicViewportState(demo->device, &viewport_create, &demo->viewport); 1473 assert(!err); 1474 1475 err = vkCreateDynamicRasterState(demo->device, &raster, &demo->raster); 1476 assert(!err); 1477 1478 err = vkCreateDynamicColorBlendState(demo->device, 1479 &color_blend, &demo->color_blend); 1480 assert(!err); 1481 1482 err = vkCreateDynamicDepthStencilState(demo->device, 1483 &depth_stencil, &demo->depth_stencil); 1484 assert(!err); 1485} 1486 1487static void demo_prepare_descriptor_pool(struct demo *demo) 1488{ 1489 const VkDescriptorTypeCount type_counts[2] = { 1490 [0] = { 1491 .type = VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER, 1492 .count = 1, 1493 }, 1494 [1] = { 1495 .type = VK_DESCRIPTOR_TYPE_COMBINED_IMAGE_SAMPLER, 1496 .count = DEMO_TEXTURE_COUNT, 1497 }, 1498 }; 1499 const VkDescriptorPoolCreateInfo descriptor_pool = { 1500 .sType = VK_STRUCTURE_TYPE_DESCRIPTOR_POOL_CREATE_INFO, 1501 .pNext = NULL, 1502 .count = 2, 1503 .pTypeCount = type_counts, 1504 }; 1505 VkResult U_ASSERT_ONLY err; 1506 1507 err = vkCreateDescriptorPool(demo->device, 1508 VK_DESCRIPTOR_POOL_USAGE_ONE_SHOT, 1, 1509 &descriptor_pool, &demo->desc_pool); 1510 assert(!err); 1511} 1512 1513static void demo_prepare_descriptor_set(struct demo *demo) 1514{ 1515 VkImageViewAttachInfo view_info[DEMO_TEXTURE_COUNT]; 1516 VkSamplerImageViewInfo combined_info[DEMO_TEXTURE_COUNT]; 1517 VkUpdateSamplerTextures update_fs; 1518 VkUpdateBuffers update_vs; 1519 const void *update_array[2] = { &update_vs, &update_fs }; 1520 VkResult U_ASSERT_ONLY err; 1521 uint32_t count; 1522 uint32_t i; 1523 1524 for (i = 0; i < DEMO_TEXTURE_COUNT; i++) { 1525 view_info[i].sType = VK_STRUCTURE_TYPE_IMAGE_VIEW_ATTACH_INFO; 1526 view_info[i].pNext = NULL; 1527 view_info[i].view = demo->textures[i].view, 1528 view_info[i].layout = VK_IMAGE_LAYOUT_GENERAL; 1529 1530 combined_info[i].sampler = demo->textures[i].sampler; 1531 combined_info[i].pImageView = &view_info[i]; 1532 } 1533 1534 memset(&update_vs, 0, sizeof(update_vs)); 1535 update_vs.sType = VK_STRUCTURE_TYPE_UPDATE_BUFFERS; 1536 update_vs.pNext = &update_fs; 1537 update_vs.descriptorType = VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER; 1538 update_vs.count = 1; 1539 update_vs.pBufferViews = &demo->uniform_data.attach; 1540 1541 memset(&update_fs, 0, sizeof(update_fs)); 1542 update_fs.sType = VK_STRUCTURE_TYPE_UPDATE_SAMPLER_TEXTURES; 1543 update_fs.binding = 1; 1544 update_fs.count = DEMO_TEXTURE_COUNT; 1545 update_fs.pSamplerImageViews = combined_info; 1546 1547 err = vkAllocDescriptorSets(demo->device, demo->desc_pool, 1548 VK_DESCRIPTOR_SET_USAGE_STATIC, 1549 1, &demo->desc_layout, 1550 &demo->desc_set, &count); 1551 assert(!err && count == 1); 1552 1553 vkBeginDescriptorPoolUpdate(demo->device, 1554 VK_DESCRIPTOR_UPDATE_MODE_FASTEST); 1555 1556 vkClearDescriptorSets(demo->device, demo->desc_pool, 1, &demo->desc_set); 1557 vkUpdateDescriptors(demo->device, demo->desc_set, 2, update_array); 1558 1559 vkEndDescriptorPoolUpdate(demo->device, demo->buffers[demo->current_buffer].cmd); 1560} 1561 1562static void demo_prepare(struct demo *demo) 1563{ 1564 const VkCmdBufferCreateInfo cmd = { 1565 .sType = VK_STRUCTURE_TYPE_CMD_BUFFER_CREATE_INFO, 1566 .pNext = NULL, 1567 .queueNodeIndex = demo->graphics_queue_node_index, 1568 .flags = 0, 1569 }; 1570 VkResult U_ASSERT_ONLY err; 1571 1572 demo_prepare_buffers(demo); 1573 demo_prepare_depth(demo); 1574 demo_prepare_textures(demo); 1575 demo_prepare_cube_data_buffer(demo); 1576 1577 demo_prepare_descriptor_layout(demo); 1578 demo_prepare_pipeline(demo); 1579 demo_prepare_dynamic_states(demo); 1580 1581 for (int i = 0; i < DEMO_BUFFER_COUNT; i++) { 1582 err = vkCreateCommandBuffer(demo->device, &cmd, &demo->buffers[i].cmd); 1583 assert(!err); 1584 } 1585 1586 demo_prepare_descriptor_pool(demo); 1587 demo_prepare_descriptor_set(demo); 1588 1589 1590 for (int i = 0; i < DEMO_BUFFER_COUNT; i++) { 1591 demo->current_buffer = i; 1592 demo_draw_build_cmd(demo, demo->buffers[i].cmd); 1593 } 1594 1595 /* 1596 * Prepare functions above may generate pipeline commands 1597 * that need to be flushed before beginning the render loop. 1598 */ 1599 demo_flush_init_cmd(demo); 1600 1601 demo->current_buffer = 0; 1602 demo->prepared = true; 1603} 1604 1605#ifdef _WIN32 1606static void demo_run(struct demo *demo) 1607{ 1608 if (!demo->prepared) 1609 return; 1610 // Wait for work to finish before updating MVP. 1611 vkDeviceWaitIdle(demo->device); 1612 demo_update_data_buffer(demo); 1613 1614 demo_draw(demo); 1615 1616 // Wait for work to finish before updating MVP. 1617 vkDeviceWaitIdle(demo->device); 1618} 1619 1620// On MS-Windows, make this a global, so it's available to WndProc() 1621struct demo demo; 1622 1623// MS-Windows event handling function: 1624LRESULT CALLBACK WndProc(HWND hWnd, 1625 UINT uMsg, 1626 WPARAM wParam, 1627 LPARAM lParam) 1628{ 1629 char tmp_str[] = "Test Vulkan Cube Program"; 1630 1631 switch(uMsg) 1632 { 1633 case WM_CREATE: 1634 return 0; 1635 case WM_CLOSE: 1636 PostQuitMessage(0); 1637 return 0; 1638 case WM_PAINT: 1639 demo_run(&demo); 1640 return 0; 1641 default: 1642 break; 1643 } 1644 return (DefWindowProc(hWnd, uMsg, wParam, lParam)); 1645} 1646 1647static void demo_create_window(struct demo *demo) 1648{ 1649 WNDCLASSEX win_class; 1650 1651 // Initialize the window class structure: 1652 win_class.cbSize = sizeof(WNDCLASSEX); 1653 win_class.style = CS_HREDRAW | CS_VREDRAW; 1654 win_class.lpfnWndProc = WndProc; 1655 win_class.cbClsExtra = 0; 1656 win_class.cbWndExtra = 0; 1657 win_class.hInstance = demo->connection; // hInstance 1658 win_class.hIcon = LoadIcon(NULL, IDI_APPLICATION); 1659 win_class.hCursor = LoadCursor(NULL, IDC_ARROW); 1660 win_class.hbrBackground = (HBRUSH)GetStockObject(WHITE_BRUSH); 1661 win_class.lpszMenuName = NULL; 1662 win_class.lpszClassName = demo->name; 1663 win_class.hIconSm = LoadIcon(NULL, IDI_WINLOGO); 1664 // Register window class: 1665 if (!RegisterClassEx(&win_class)) { 1666 // It didn't work, so try to give a useful error: 1667 printf("Unexpected error trying to start the application!\n"); 1668 fflush(stdout); 1669 exit(1); 1670 } 1671 // Create window with the registered class: 1672 demo->window = CreateWindowEx(0, 1673 demo->name, // class name 1674 demo->name, // app name 1675 WS_OVERLAPPEDWINDOW | // window style 1676 WS_VISIBLE | 1677 WS_SYSMENU, 1678 100,100, // x/y coords 1679 demo->width, // width 1680 demo->height, // height 1681 NULL, // handle to parent 1682 NULL, // handle to menu 1683 demo->connection, // hInstance 1684 NULL); // no extra parameters 1685 if (!demo->window) { 1686 // It didn't work, so try to give a useful error: 1687 printf("Cannot create a window in which to draw!\n"); 1688 fflush(stdout); 1689 exit(1); 1690 } 1691} 1692#else // _WIN32 1693static void demo_handle_event(struct demo *demo, 1694 const xcb_generic_event_t *event) 1695{ 1696 uint8_t event_code = event->response_type & 0x7f; 1697 switch (event_code) { 1698 case XCB_EXPOSE: 1699 // TODO: Resize window 1700 break; 1701 case XCB_CLIENT_MESSAGE: 1702 if((*(xcb_client_message_event_t*)event).data.data32[0] == 1703 (*demo->atom_wm_delete_window).atom) { 1704 demo->quit = true; 1705 } 1706 break; 1707 case XCB_KEY_RELEASE: 1708 { 1709 const xcb_key_release_event_t *key = 1710 (const xcb_key_release_event_t *) event; 1711 1712 switch (key->detail) { 1713 case 0x9: // Escape 1714 demo->quit = true; 1715 break; 1716 case 0x71: // left arrow key 1717 demo->spin_angle += demo->spin_increment; 1718 break; 1719 case 0x72: // right arrow key 1720 demo->spin_angle -= demo->spin_increment; 1721 break; 1722 case 0x41: 1723 demo->pause = !demo->pause; 1724 break; 1725 } 1726 } 1727 break; 1728 default: 1729 break; 1730 } 1731} 1732 1733static void demo_run(struct demo *demo) 1734{ 1735 xcb_flush(demo->connection); 1736 1737 while (!demo->quit) { 1738 xcb_generic_event_t *event; 1739 1740 if (demo->pause) { 1741 event = xcb_wait_for_event(demo->connection); 1742 } else { 1743 event = xcb_poll_for_event(demo->connection); 1744 } 1745 if (event) { 1746 demo_handle_event(demo, event); 1747 free(event); 1748 } 1749 1750 // Wait for work to finish before updating MVP. 1751 vkDeviceWaitIdle(demo->device); 1752 demo_update_data_buffer(demo); 1753 1754 demo_draw(demo); 1755 1756 // Wait for work to finish before updating MVP. 1757 vkDeviceWaitIdle(demo->device); 1758 } 1759} 1760 1761static void demo_create_window(struct demo *demo) 1762{ 1763 uint32_t value_mask, value_list[32]; 1764 1765 demo->window = xcb_generate_id(demo->connection); 1766 1767 value_mask = XCB_CW_BACK_PIXEL | XCB_CW_EVENT_MASK; 1768 value_list[0] = demo->screen->black_pixel; 1769 value_list[1] = XCB_EVENT_MASK_KEY_RELEASE | 1770 XCB_EVENT_MASK_EXPOSURE; 1771 1772 xcb_create_window(demo->connection, 1773 XCB_COPY_FROM_PARENT, 1774 demo->window, demo->screen->root, 1775 0, 0, demo->width, demo->height, 0, 1776 XCB_WINDOW_CLASS_INPUT_OUTPUT, 1777 demo->screen->root_visual, 1778 value_mask, value_list); 1779 1780 /* Magic code that will send notification when window is destroyed */ 1781 xcb_intern_atom_cookie_t cookie = xcb_intern_atom(demo->connection, 1, 12, 1782 "WM_PROTOCOLS"); 1783 xcb_intern_atom_reply_t* reply = xcb_intern_atom_reply(demo->connection, cookie, 0); 1784 1785 xcb_intern_atom_cookie_t cookie2 = xcb_intern_atom(demo->connection, 0, 16, "WM_DELETE_WINDOW"); 1786 demo->atom_wm_delete_window = xcb_intern_atom_reply(demo->connection, cookie2, 0); 1787 1788 xcb_change_property(demo->connection, XCB_PROP_MODE_REPLACE, 1789 demo->window, (*reply).atom, 4, 32, 1, 1790 &(*demo->atom_wm_delete_window).atom); 1791 free(reply); 1792 1793 xcb_map_window(demo->connection, demo->window); 1794} 1795#endif // _WIN32 1796 1797static void demo_init_vk(struct demo *demo) 1798{ 1799 VkResult err; 1800 // Extensions to enable 1801 const char *ext_names[] = { 1802 "VK_WSI_LunarG", 1803 }; 1804 size_t extSize = sizeof(uint32_t); 1805 uint32_t extCount = 0; 1806 err = vkGetGlobalExtensionInfo(VK_EXTENSION_INFO_TYPE_COUNT, 0, &extSize, &extCount); 1807 assert(!err); 1808 1809 VkExtensionProperties extProp; 1810 extSize = sizeof(VkExtensionProperties); 1811 bool32_t U_ASSERT_ONLY extFound = 0; 1812 for (uint32_t i = 0; i < extCount; i++) { 1813 err = vkGetGlobalExtensionInfo(VK_EXTENSION_INFO_TYPE_PROPERTIES, i, &extSize, &extProp); 1814 if (!strcmp(ext_names[0], extProp.extName)) 1815 extFound = 1; 1816 } 1817 assert(extFound); 1818 const VkApplicationInfo app = { 1819 .sType = VK_STRUCTURE_TYPE_APPLICATION_INFO, 1820 .pNext = NULL, 1821 .pAppName = "cube", 1822 .appVersion = 0, 1823 .pEngineName = "cube", 1824 .engineVersion = 0, 1825 .apiVersion = VK_API_VERSION, 1826 }; 1827 const VkInstanceCreateInfo inst_info = { 1828 .sType = VK_STRUCTURE_TYPE_INSTANCE_CREATE_INFO, 1829 .pNext = NULL, 1830 .pAppInfo = &app, 1831 .pAllocCb = NULL, 1832 .extensionCount = 1, 1833 .ppEnabledExtensionNames = ext_names, 1834 }; 1835 const VkDeviceQueueCreateInfo queue = { 1836 .queueNodeIndex = 0, 1837 .queueCount = 1, 1838 }; 1839 1840 const VkDeviceCreateInfo device = { 1841 .sType = VK_STRUCTURE_TYPE_DEVICE_CREATE_INFO, 1842 .pNext = NULL, 1843 .queueRecordCount = 1, 1844 .pRequestedQueues = &queue, 1845 .extensionCount = 1, 1846 .ppEnabledExtensionNames = ext_names, 1847 .flags = VK_DEVICE_CREATE_VALIDATION_BIT, 1848 }; 1849 uint32_t gpu_count; 1850 uint32_t i; 1851 size_t data_size; 1852 uint32_t queue_count; 1853 1854 err = vkCreateInstance(&inst_info, &demo->inst); 1855 if (err) { 1856#ifdef _WIN32 1857 MessageBox(NULL, "vkCreateInstance failed - do you have a Vulkan graphics driver installed?", 1858 "vkCreateInstance Failure", MB_OK); 1859#else 1860 printf("vkCreateInstance failed - Do you have a Vulkan graphics driver installed?" 1861 "(\nExiting ...\n"); 1862 fflush(stdout); 1863#endif 1864 exit(1); 1865 } 1866 1867 gpu_count = 1; 1868 err = vkEnumeratePhysicalDevices(demo->inst, &gpu_count, &demo->gpu); 1869 assert(!err && gpu_count == 1); 1870 1871 err = vkCreateDevice(demo->gpu, &device, &demo->device); 1872 assert(!err); 1873 1874 err = vkGetPhysicalDeviceInfo(demo->gpu, VK_PHYSICAL_DEVICE_INFO_TYPE_PROPERTIES, 1875 &data_size, NULL); 1876 assert(!err); 1877 1878 demo->gpu_props = (VkPhysicalDeviceProperties *) malloc(data_size); 1879 err = vkGetPhysicalDeviceInfo(demo->gpu, VK_PHYSICAL_DEVICE_INFO_TYPE_PROPERTIES, 1880 &data_size, demo->gpu_props); 1881 assert(!err); 1882 1883 err = vkGetPhysicalDeviceInfo(demo->gpu, VK_PHYSICAL_DEVICE_INFO_TYPE_QUEUE_PROPERTIES, 1884 &data_size, NULL); 1885 assert(!err); 1886 1887 demo->queue_props = (VkPhysicalDeviceQueueProperties *) malloc(data_size); 1888 err = vkGetPhysicalDeviceInfo(demo->gpu, VK_PHYSICAL_DEVICE_INFO_TYPE_QUEUE_PROPERTIES, 1889 &data_size, demo->queue_props); 1890 assert(!err); 1891 queue_count = (uint32_t)(data_size / sizeof(VkPhysicalDeviceQueueProperties)); 1892 assert(queue_count >= 1); 1893 1894 // Graphics queue and MemMgr queue can be separate. 1895 // TODO: Add support for separate queues, including synchronization, 1896 // and appropriate tracking for QueueSubmit and QueueBindObjectMemory 1897 for (i = 0; i < queue_count; i++) { 1898 if ((demo->queue_props[i].queueFlags & VK_QUEUE_GRAPHICS_BIT) && 1899 (demo->queue_props[i].queueFlags & VK_QUEUE_MEMMGR_BIT) ) 1900 break; 1901 } 1902 assert(i < queue_count); 1903 demo->graphics_queue_node_index = i; 1904 1905 err = vkGetDeviceQueue(demo->device, demo->graphics_queue_node_index, 1906 0, &demo->queue); 1907 assert(!err); 1908} 1909 1910static void demo_init_connection(struct demo *demo) 1911{ 1912#ifndef _WIN32 1913 const xcb_setup_t *setup; 1914 xcb_screen_iterator_t iter; 1915 int scr; 1916 1917 demo->connection = xcb_connect(NULL, &scr); 1918 if (demo->connection == NULL) { 1919 printf("Cannot find a compatible Vulkan installable client driver " 1920 "(ICD).\nExiting ...\n"); 1921 fflush(stdout); 1922 exit(1); 1923 } 1924 1925 setup = xcb_get_setup(demo->connection); 1926 iter = xcb_setup_roots_iterator(setup); 1927 while (scr-- > 0) 1928 xcb_screen_next(&iter); 1929 1930 demo->screen = iter.data; 1931#endif // _WIN32 1932} 1933 1934#ifdef _WIN32 1935static void demo_init(struct demo *demo, HINSTANCE hInstance, LPSTR pCmdLine) 1936#else // _WIN32 1937static void demo_init(struct demo *demo, int argc, char **argv) 1938#endif // _WIN32 1939{ 1940 vec3 eye = {0.0f, 3.0f, 5.0f}; 1941 vec3 origin = {0, 0, 0}; 1942 vec3 up = {0.0f, -1.0f, 0.0}; 1943 bool argv_error = false; 1944 1945 memset(demo, 0, sizeof(*demo)); 1946 1947#ifdef _WIN32 1948 demo->connection = hInstance; 1949 strncpy(demo->name, "cube", APP_NAME_STR_LEN); 1950 1951 if (strncmp(pCmdLine, "--use_staging", strlen("--use_staging")) == 0) 1952 demo->use_staging_buffer = true; 1953 else if (strlen(pCmdLine) != 0) { 1954 fprintf(stderr, "Do not recognize argument \"%s\".\n", pCmdLine); 1955 argv_error = true; 1956 } 1957#else // _WIN32 1958 for (int i = 1; i < argc; i++) { 1959 if (strncmp(argv[i], "--use_staging", strlen("--use_staging")) == 0) 1960 demo->use_staging_buffer = true; 1961 else { 1962 fprintf(stderr, "Do not recognize argument \"%s\".\n", argv[i]); 1963 argv_error = true; 1964 } 1965 } 1966#endif // _WIN32 1967 if (argv_error) { 1968 fprintf(stderr, "Usage:\n cube [--use_staging]\n"); 1969 fflush(stderr); 1970 exit(1); 1971 } 1972 1973 demo_init_connection(demo); 1974 demo_init_vk(demo); 1975 1976 demo->width = 500; 1977 demo->height = 500; 1978 demo->format = VK_FORMAT_B8G8R8A8_UNORM; 1979 1980 demo->spin_angle = 0.01f; 1981 demo->spin_increment = 0.01f; 1982 demo->pause = false; 1983 1984 mat4x4_perspective(demo->projection_matrix, (float)degreesToRadians(45.0f), 1.0f, 0.1f, 100.0f); 1985 mat4x4_look_at(demo->view_matrix, eye, origin, up); 1986 mat4x4_identity(demo->model_matrix); 1987} 1988 1989static void demo_cleanup(struct demo *demo) 1990{ 1991 uint32_t i, j; 1992 1993 vkDestroyObject(demo->device, VK_OBJECT_TYPE_DESCRIPTOR_SET, demo->desc_set); 1994 vkDestroyObject(demo->device, VK_OBJECT_TYPE_DESCRIPTOR_POOL, demo->desc_pool); 1995 1996 vkDestroyObject(demo->device, VK_OBJECT_TYPE_DYNAMIC_VP_STATE, demo->viewport); 1997 vkDestroyObject(demo->device, VK_OBJECT_TYPE_DYNAMIC_RS_STATE, demo->raster); 1998 vkDestroyObject(demo->device, VK_OBJECT_TYPE_DYNAMIC_CB_STATE, demo->color_blend); 1999 vkDestroyObject(demo->device, VK_OBJECT_TYPE_DYNAMIC_DS_STATE, demo->depth_stencil); 2000 2001 vkDestroyObject(demo->device, VK_OBJECT_TYPE_PIPELINE, demo->pipeline); 2002 vkDestroyObject(demo->device, VK_OBJECT_TYPE_PIPELINE_LAYOUT, demo->pipeline_layout); 2003 vkDestroyObject(demo->device, VK_OBJECT_TYPE_DESCRIPTOR_SET_LAYOUT, demo->desc_layout); 2004 2005 for (i = 0; i < DEMO_TEXTURE_COUNT; i++) { 2006 vkDestroyObject(demo->device, VK_OBJECT_TYPE_IMAGE_VIEW, demo->textures[i].view); 2007 vkQueueBindObjectMemory(demo->queue, VK_OBJECT_TYPE_IMAGE, demo->textures[i].image, 0, VK_NULL_HANDLE, 0); 2008 vkDestroyObject(demo->device, VK_OBJECT_TYPE_IMAGE, demo->textures[i].image); 2009 demo_remove_mem_refs(demo, demo->textures[i].num_mem, demo->textures[i].mem); 2010 for (j = 0; j < demo->textures[i].num_mem; j++) 2011 vkFreeMemory(demo->device, demo->textures[i].mem[j]); 2012 free(demo->textures[i].mem); 2013 vkDestroyObject(demo->device, VK_OBJECT_TYPE_SAMPLER, demo->textures[i].sampler); 2014 } 2015 vkDestroySwapChainWSI(demo->swap_chain); 2016 2017 vkDestroyObject(demo->device, VK_OBJECT_TYPE_DEPTH_STENCIL_VIEW, demo->depth.view); 2018 vkQueueBindObjectMemory(demo->queue, VK_OBJECT_TYPE_IMAGE, demo->depth.image, 0, VK_NULL_HANDLE, 0); 2019 demo_remove_mem_refs(demo, demo->depth.num_mem, demo->depth.mem); 2020 vkDestroyObject(demo->device, VK_OBJECT_TYPE_IMAGE, demo->depth.image); 2021 for (j = 0; j < demo->depth.num_mem; j++) { 2022 vkFreeMemory(demo->device, demo->depth.mem[j]); 2023 } 2024 2025 vkDestroyObject(demo->device, VK_OBJECT_TYPE_BUFFER_VIEW, demo->uniform_data.view); 2026 vkQueueBindObjectMemory(demo->queue, VK_OBJECT_TYPE_BUFFER, demo->uniform_data.buf, 0, VK_NULL_HANDLE, 0); 2027 vkDestroyObject(demo->device, VK_OBJECT_TYPE_BUFFER, demo->uniform_data.buf); 2028 demo_remove_mem_refs(demo, demo->uniform_data.num_mem, demo->uniform_data.mem); 2029 for (j = 0; j < demo->uniform_data.num_mem; j++) 2030 vkFreeMemory(demo->device, demo->uniform_data.mem[j]); 2031 2032 for (i = 0; i < DEMO_BUFFER_COUNT; i++) { 2033 vkDestroyObject(demo->device, VK_OBJECT_TYPE_COLOR_ATTACHMENT_VIEW, demo->buffers[i].view); 2034 vkDestroyObject(demo->device, VK_OBJECT_TYPE_COMMAND_BUFFER, demo->buffers[i].cmd); 2035 demo_remove_mem_refs(demo, 1, &demo->buffers[i].mem); 2036 } 2037 2038 vkDestroyDevice(demo->device); 2039 vkDestroyInstance(demo->inst); 2040 2041#ifndef _WIN32 2042 xcb_destroy_window(demo->connection, demo->window); 2043 xcb_disconnect(demo->connection); 2044#endif // _WIN32 2045} 2046 2047#ifdef _WIN32 2048int APIENTRY WinMain(HINSTANCE hInstance, 2049 HINSTANCE hPrevInstance, 2050 LPSTR pCmdLine, 2051 int nCmdShow) 2052{ 2053 MSG msg; // message 2054 bool done; // flag saying when app is complete 2055 2056 demo_init(&demo, hInstance, pCmdLine); 2057 demo_create_window(&demo); 2058 2059 demo_prepare(&demo); 2060 2061 done = false; //initialize loop condition variable 2062 /* main message loop*/ 2063 while(!done) 2064 { 2065 PeekMessage(&msg,0,0,0,PM_REMOVE); 2066 if (msg.message == WM_QUIT) //check for a quit message 2067 { 2068 done = true; //if found, quit app 2069 } 2070 else 2071 { 2072 /* Translate and dispatch to event queue*/ 2073 TranslateMessage(&msg); 2074 DispatchMessage(&msg); 2075 } 2076 } 2077 2078 demo_cleanup(&demo); 2079 2080 return (int) msg.wParam; 2081} 2082#else // _WIN32 2083int main(int argc, char **argv) 2084{ 2085 struct demo demo; 2086 2087 demo_init(&demo, argc, argv); 2088 demo_create_window(&demo); 2089 2090 demo_prepare(&demo); 2091 demo_run(&demo); 2092 2093 demo_cleanup(&demo); 2094 2095 return 0; 2096} 2097#endif // _WIN32 2098