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