vulkan_test2: d34f84bede17 src/vku.c

vulkan_test2

view src/vku.c @ 13:d34f84bede17

pipeline madness

author	John Tsiombikas <nuclear@member.fsf.org>
date	Mon, 25 Jun 2018 08:00:57 +0300
parents	e17abe477616
children	9fb6c24691ea

line source

1 #include <stdio.h>

2 #include <stdlib.h>

3 #include <string.h>

4 #include <stdint.h>

5 #include "vku.h"

7 static const char *get_device_name(VkPhysicalDeviceType type);

8 static const char *get_mem_prop_flag_string(VkMemoryPropertyFlags flags);

9 static const char *get_queue_flag_string(VkQueueFlagBits flags);

10 static int ver_major(uint32_t ver);

11 static int ver_minor(uint32_t ver);

12 static int ver_patch(uint32_t ver);

13 static const char *mem_size_str(long sz);

15 VkInstance vk;

16 VkDevice vkdev;

17 VkQueue vkq;

19 static VkPhysicalDevice *phys_devices;

20 static int sel_dev, sel_qfamily;

22 static VkExtensionProperties *vkext, *vkdevext;

23 static uint32_t vkext_count, vkdevext_count;

26 int vku_have_extension(const char *name)

27 {

28 int i;

30 if(!vkext) {

31 vkext_count = 0;

32 vkEnumerateInstanceExtensionProperties(0, &vkext_count, 0);

33 if(vkext_count) {

34 if(!(vkext = malloc(vkext_count * sizeof *vkext))) {

35 perror("failed to allocate instance extension list");

36 return 0;

37 }

38 vkEnumerateInstanceExtensionProperties(0, &vkext_count, vkext);

40 printf("instance extensions:\n");

41 for(i=0; i<(int)vkext_count; i++) {

42 printf(" %s (ver: %u)\n", vkext[i].extensionName, (unsigned int)vkext[i].specVersion);

43 }

44 }

45 }

47 for(i=0; i<(int)vkext_count; i++) {

48 if(strcmp(vkext[i].extensionName, name) == 0) {

49 return 1;

50 }

51 }

52 return 0;

53 }

55 int vku_have_device_extension(const char *name)

56 {

57 int i;

59 if(sel_dev < 0) return 0;

61 if(!vkdevext) {

62 vkdevext_count = 0;

63 vkEnumerateDeviceExtensionProperties(phys_devices[sel_dev], 0, &vkdevext_count, 0);

64 if(vkdevext_count) {

65 if(!(vkdevext = malloc(vkdevext_count * sizeof *vkdevext))) {

66 perror("failed to allocate device extension list");

67 return 0;

68 }

69 vkEnumerateDeviceExtensionProperties(phys_devices[sel_dev], 0, &vkdevext_count, vkdevext);

71 printf("selected device extensions:\n");

72 for(i=0; i<(int)vkdevext_count; i++) {

73 printf(" %s (ver: %u)\n", vkdevext[i].extensionName, (unsigned int)vkdevext[i].specVersion);

74 }

75 }

76 }

78 for(i=0; i<(int)vkdevext_count; i++) {

79 if(strcmp(vkdevext[i].extensionName, name) == 0) {

80 return 1;

81 }

82 }

83 return 0;

84 }

86 int vku_create_dev(void)

87 {

88 int i, j;

89 VkInstanceCreateInfo inst_info;

90 VkDeviceCreateInfo dev_info;

91 VkDeviceQueueCreateInfo queue_info;

92 VkCommandPoolCreateInfo cmdpool_info;

93 uint32_t num_devices;

94 float qprio = 0.0f;

96 static const char *ext_names[] = {

97 #ifdef VK_USE_PLATFORM_XLIB_KHR

98 "VK_KHR_xlib_surface",

99 #endif

100 "VK_KHR_surface"

101 };

102 static const char *devext_names[] = {

103 "VK_KHR_swapchain"

104 };

105

106 sel_dev = -1;

107 sel_qfamily = -1;

108

109 for(i=0; i<sizeof ext_names / sizeof *ext_names; i++) {

110 if(!vku_have_extension(ext_names[i])) {

111 fprintf(stderr, "required extension (%s) not found\n", ext_names[i]);

112 return -1;

113 }

114 }

115

116 memset(&inst_info, 0, sizeof inst_info);

117 inst_info.sType = VK_STRUCTURE_TYPE_INSTANCE_CREATE_INFO;

118 inst_info.ppEnabledExtensionNames = ext_names;

119 inst_info.enabledExtensionCount = sizeof ext_names / sizeof *ext_names;

120

121 if(vkCreateInstance(&inst_info, 0, &vk) != 0) {

122 fprintf(stderr, "failed to create vulkan instance\n");

123 return -1;

124 }

125 printf("created vulkan instance\n");

126

127 if(vkEnumeratePhysicalDevices(vk, &num_devices, 0) != 0) {

128 fprintf(stderr, "failed to enumerate vulkan physical devices\n");

129 return -1;

130 }

131 phys_devices = malloc(num_devices * sizeof *phys_devices);

132 if(vkEnumeratePhysicalDevices(vk, &num_devices, phys_devices) != 0) {

133 fprintf(stderr, "failed to enumerate vulkan physical devices\n");

134 return -1;

135 }

136 printf("found %u physical device(s)\n", (unsigned int)num_devices);

137

138 for(i=0; i<(int)num_devices; i++) {

139 VkPhysicalDeviceProperties dev_prop;

140 VkPhysicalDeviceMemoryProperties mem_prop;

141 VkQueueFamilyProperties *qprop;

142 uint32_t qprop_count;

143

144 vkGetPhysicalDeviceProperties(phys_devices[i], &dev_prop);

145

146 printf("Device %d: %s\n", i, dev_prop.deviceName);

147 printf(" type: %s\n", get_device_name(dev_prop.deviceType));

148 printf(" API version: %d.%d.%d\n", ver_major(dev_prop.apiVersion), ver_minor(dev_prop.apiVersion),

149 ver_patch(dev_prop.apiVersion));

150 printf(" driver version: %d.%d.%d\n", ver_major(dev_prop.driverVersion), ver_minor(dev_prop.driverVersion),

151 ver_patch(dev_prop.driverVersion));

152 printf(" vendor id: %x device id: %x\n", dev_prop.vendorID, dev_prop.deviceID);

153

154

155 vkGetPhysicalDeviceMemoryProperties(phys_devices[i], &mem_prop);

156 printf(" %d memory heaps:\n", mem_prop.memoryHeapCount);

157 for(j=0; j<mem_prop.memoryHeapCount; j++) {

158 VkMemoryHeap heap = mem_prop.memoryHeaps[j];

159 printf(" Heap %d - size: %s, flags: %s\n", j, mem_size_str(heap.size),

160 heap.flags & VK_MEMORY_HEAP_DEVICE_LOCAL_BIT ? "device-local" : "-");

161 }

162 printf(" %d memory types:\n", mem_prop.memoryTypeCount);

163 for(j=0; j<mem_prop.memoryTypeCount; j++) {

164 VkMemoryType type = mem_prop.memoryTypes[j];

165 printf(" Type %d - heap: %d, flags: %s\n", j, type.heapIndex,

166 get_mem_prop_flag_string(type.propertyFlags));

167 }

168

169 vkGetPhysicalDeviceQueueFamilyProperties(phys_devices[i], &qprop_count, 0);

170 if(qprop_count <= 0) {

171 continue;

172 }

173 qprop = malloc(qprop_count * sizeof *qprop);

174 vkGetPhysicalDeviceQueueFamilyProperties(phys_devices[i], &qprop_count, qprop);

175

176 for(j=0; j<qprop_count; j++) {

177 printf(" Queue family %d:\n", j);

178 printf(" flags: %s\n", get_queue_flag_string(qprop[j].queueFlags));

179 printf(" num queues: %u\n", qprop[j].queueCount);

180

181 if(qprop[j].queueFlags & VK_QUEUE_GRAPHICS_BIT) {

182 sel_dev = i;

183 sel_qfamily = j;

184 }

185 }

186 free(qprop);

187 }

188

189 if(sel_dev < 0 || sel_qfamily < 0) {

190 fprintf(stderr, "failed to find any device with a graphics-capable command queue\n");

191 vkDestroyDevice(vkdev, 0);

192 return -1;

193 }

194

195 for(i=0; i<sizeof devext_names / sizeof *devext_names; i++) {

196 if(!vku_have_device_extension(devext_names[i])) {

197 fprintf(stderr, "required extension (%s) not found on the selected device (%d)\n",

198 ext_names[i], sel_dev);

199 return -1;

200 }

201 }

202

203 /* create device & command queue */

204 memset(&queue_info, 0, sizeof queue_info);

205 queue_info.sType = VK_STRUCTURE_TYPE_DEVICE_QUEUE_CREATE_INFO;

206 queue_info.queueFamilyIndex = sel_qfamily;

207 queue_info.queueCount = 1;

208 queue_info.pQueuePriorities = &qprio;

209

210 memset(&dev_info, 0, sizeof dev_info);

211 dev_info.sType = VK_STRUCTURE_TYPE_DEVICE_CREATE_INFO;

212 dev_info.queueCreateInfoCount = 1;

213 dev_info.pQueueCreateInfos = &queue_info;

214 dev_info.enabledExtensionCount = sizeof devext_names / sizeof *devext_names;

215 dev_info.ppEnabledExtensionNames = devext_names;

216

217 if(vkCreateDevice(phys_devices[sel_dev], &dev_info, 0, &vkdev) != 0) {

218 fprintf(stderr, "failed to create device %d\n", sel_dev);

219 return -1;

220 }

221 printf("created device %d\n", sel_dev);

222

223 vkGetDeviceQueue(vkdev, sel_qfamily, 0, &vkq);

224

225 /* create command buffer pool */

226 memset(&cmdpool_info, 0, sizeof cmdpool_info);

227 cmdpool_info.sType = VK_STRUCTURE_TYPE_COMMAND_POOL_CREATE_INFO;

228 cmdpool_info.flags = VK_COMMAND_POOL_CREATE_RESET_COMMAND_BUFFER_BIT;

229 cmdpool_info.queueFamilyIndex = sel_qfamily;

230

231 if(vkCreateCommandPool(vkdev, &cmdpool_info, 0, &vkcmdpool) != 0) {

232 fprintf(stderr, "failed to get command quque!\n");

233 return -1;

234 }

235

236 if(!(vkcmdbuf = vku_alloc_cmdbuf(vkcmdpool, VK_COMMAND_BUFFER_LEVEL_PRIMARY))) {

237 fprintf(stderr, "failed to create primary command buffer\n");

238 return -1;

239 }

240

241 return 0;

242 }

243

244 void vku_cleanup(void)

245 {

246 if(vk) {

247 vkDeviceWaitIdle(vkdev);

248 vkDestroyCommandPool(vkdev, vkcmdpool, 0);

249 vkDestroyDevice(vkdev, 0);

250 vkDestroyInstance(vk, 0);

251 vk = 0;

252 }

253

254 free(phys_devices);

255 phys_devices = 0;

256 }

257

258 VkCommandBuffer vku_alloc_cmdbuf(VkCommandPool pool, VkCommandBufferLevel level)

259 {

260 VkCommandBuffer cmdbuf;

261 VkCommandBufferAllocateInfo inf;

262

263 memset(&inf, 0, sizeof inf);

264 inf.sType = VK_STRUCTURE_TYPE_COMMAND_BUFFER_ALLOCATE_INFO;

265 inf.commandPool = pool;

266 inf.level = level;

267 inf.commandBufferCount = 1;

268

269 if(vkAllocateCommandBuffers(vkdev, &inf, &cmdbuf) != 0) {

270 return 0;

271 }

272 return cmdbuf;

273 }

274

275 void vku_free_cmdbuf(VkCommandPool pool, VkCommandBuffer buf)

276 {

277 vkFreeCommandBuffers(vkdev, pool, 1, &buf);

278 }

279

280 void vku_begin_cmdbuf(VkCommandBuffer buf, unsigned int flags)

281 {

282 VkCommandBufferBeginInfo inf;

283

284 memset(&inf, 0, sizeof inf);

285 inf.sType = VK_STRUCTURE_TYPE_COMMAND_BUFFER_BEGIN_INFO;

286 inf.flags = flags;

287

288 vkBeginCommandBuffer(buf, &inf);

289 }

290

291

292 void vku_end_cmdbuf(VkCommandBuffer buf)

293 {

294 vkEndCommandBuffer(buf);

295 }

296

297 void vku_reset_cmdbuf(VkCommandBuffer buf)

298 {

299 vkResetCommandBuffer(buf, 0);

300 }

301

302 void vku_submit_cmdbuf(VkQueue q, VkCommandBuffer buf, VkFence done_fence)

303 {

304 VkSubmitInfo info;

305

306 memset(&info, 0, sizeof info);

307 info.sType = VK_STRUCTURE_TYPE_SUBMIT_INFO;

308 info.commandBufferCount = 1;

309 info.pCommandBuffers = &buf;

310

311 vkQueueSubmit(q, 1, &info, done_fence);

312 }

313

314 VkSwapchainKHR vku_create_swapchain(VkSurfaceKHR surf, int xsz, int ysz, int n,

315 VkPresentModeKHR pmode, VkSwapchainKHR prev)

316 {

317 VkSwapchainKHR sc;

318 VkSwapchainCreateInfoKHR inf;

319

320 memset(&inf, 0, sizeof inf);

321 inf.sType = VK_STRUCTURE_TYPE_SWAPCHAIN_CREATE_INFO_KHR;

322 inf.surface = surf;

323 inf.minImageCount = n;

324 inf.imageFormat = VK_FORMAT_B8G8R8A8_UNORM; /* TODO enumerate and choose */

325 inf.imageColorSpace = VK_COLORSPACE_SRGB_NONLINEAR_KHR;

326 inf.imageExtent.width = xsz;

327 inf.imageExtent.height = ysz;

328 inf.imageArrayLayers = 1;

329 inf.imageUsage = VK_IMAGE_USAGE_COLOR_ATTACHMENT_BIT;

330 inf.imageSharingMode = VK_SHARING_MODE_EXCLUSIVE; /* XXX make this an option? */

331 inf.preTransform = VK_SURFACE_TRANSFORM_INHERIT_BIT_KHR;

332 inf.compositeAlpha = VK_COMPOSITE_ALPHA_OPAQUE_BIT_KHR;

333 inf.presentMode = pmode;

334 inf.oldSwapchain = prev;

335

336 if(vkCreateSwapchainKHR(vkdev, &inf, 0, &sc) != 0) {

337 return 0;

338 }

339 return sc;

340 }

341

342 VkImage *vku_get_swapchain_images(VkSwapchainKHR sc, int *count)

343 {

344 uint32_t nimg;

345 VkImage *images;

346

347 if(vkGetSwapchainImagesKHR(vkdev, sc, &nimg, 0) != 0) {

348 return 0;

349 }

350 if(!(images = malloc(nimg * sizeof *images))) {

351 return 0;

352 }

353 vkGetSwapchainImagesKHR(vkdev, sc, &nimg, images);

354

355 if(count) *count = (int)nimg;

356 return images;

357 }

358

359 int vku_get_next_image(VkSwapchainKHR sc)

360 {

361 uint32_t next;

362

363 if(vkAcquireNextImageKHR(vkdev, sc, UINT64_MAX, 0, 0, &next) != 0) {

364 return -1;

365 }

366 return (int)next;

367 }

368

369 void vku_present(VkSwapchainKHR sc, int img_idx)

370 {

371 VkPresentInfoKHR inf;

372 VkResult res;

373 uint32_t index = img_idx;

374

375 memset(&inf, 0, sizeof inf);

376 inf.sType = VK_STRUCTURE_TYPE_PRESENT_INFO_KHR;

377 inf.swapchainCount = 1;

378 inf.pSwapchains = &sc;

379 inf.pImageIndices = &index;

380 inf.pResults = &res;

381

382 vkQueuePresentKHR(vkq, &inf);

383 }

384

385 struct vku_buffer *vku_create_buffer(int sz, unsigned int usage)

386 {

387 struct vku_buffer *buf;

388 VkBufferCreateInfo binfo;

389

390 if(!(buf = malloc(sizeof *buf))) {

391 perror("failed to allocate vk_buffer structure");

392 return 0;

393 }

394

395 memset(&binfo, 0, sizeof binfo);

396 binfo.sType = VK_STRUCTURE_TYPE_BUFFER_CREATE_INFO;

397 binfo.size = sz;

398 binfo.usage = usage;

399 binfo.sharingMode = VK_SHARING_MODE_EXCLUSIVE;

400

401 if(vkCreateBuffer(vkdev, &binfo, 0, &buf->buf) != 0) {

402 fprintf(stderr, "failed to create %d byte buffer (usage: %x)\n", sz, usage);

403 return 0;

404 }

405 // TODO back with memory

406 return buf;

407 }

408

409 void vku_destroy_buffer(struct vku_buffer *buf)

410 {

411 if(buf) {

412 vkDestroyBuffer(vkdev, buf->buf, 0);

413 free(buf);

414 }

415 }

416

417 void vku_cmd_copybuf(VkCommandBuffer cmdbuf, VkBuffer dest, int doffs,

418 VkBuffer src, int soffs, int size)

419 {

420 VkBufferCopy copy;

421 copy.size = size;

422 copy.srcOffset = soffs;

423 copy.dstOffset = doffs;

424

425 vkCmdCopyBuffer(cmdbuf, src, dest, 1, &copy);

426 }

427

428

429 VkRenderPass vku_create_renderpass(VkFormat cfmt, VkFormat dsfmt)

430 {

431 int count = 1; /* always assume we have a color attachment for now */

432 VkAttachmentDescription at[2];

433 VkAttachmentReference colref, dsref;

434 VkSubpassDescription subpass;

435 VkRenderPass pass;

436 VkRenderPassCreateInfo rpinf;

437

438 colref.attachment = 0;

439 colref.layout = VK_IMAGE_LAYOUT_COLOR_ATTACHMENT_OPTIMAL;

440 dsref.attachment = 1;

441 dsref.layout = VK_IMAGE_LAYOUT_DEPTH_STENCIL_ATTACHMENT_OPTIMAL;

442

443 memset(&subpass, 0, sizeof subpass);

444 subpass.pipelineBindPoint = VK_PIPELINE_BIND_POINT_GRAPHICS;

445 subpass.colorAttachmentCount = 1;

446 subpass.pColorAttachments = &colref;

447

448 at[0].format = cfmt;

449 at[0].samples = VK_SAMPLE_COUNT_1_BIT; /* TODO multisampling */

450 at[0].loadOp = VK_ATTACHMENT_LOAD_OP_DONT_CARE;

451 at[0].storeOp = VK_ATTACHMENT_STORE_OP_STORE;

452 at[0].stencilLoadOp = VK_ATTACHMENT_LOAD_OP_DONT_CARE;

453 at[0].stencilStoreOp = VK_ATTACHMENT_STORE_OP_DONT_CARE;

454 at[0].initialLayout = VK_IMAGE_LAYOUT_UNDEFINED;

455 at[0].finalLayout = VK_IMAGE_LAYOUT_PRESENT_SRC_KHR;

456

457 if(dsfmt != VK_FORMAT_UNDEFINED) {

458 at[1].format = dsfmt;

459 at[1].samples = VK_SAMPLE_COUNT_1_BIT;

460 at[1].loadOp = VK_ATTACHMENT_LOAD_OP_DONT_CARE;

461 at[1].storeOp = VK_ATTACHMENT_STORE_OP_STORE;

462 at[1].stencilLoadOp = VK_ATTACHMENT_LOAD_OP_DONT_CARE;

463 at[1].stencilStoreOp = VK_ATTACHMENT_STORE_OP_STORE;

464 at[1].initialLayout = VK_IMAGE_LAYOUT_UNDEFINED;

465 at[1].finalLayout = VK_IMAGE_LAYOUT_UNDEFINED;

466

467 subpass.pDepthStencilAttachment = &dsref;

468 count++;

469 }

470

471 memset(&rpinf, 0, sizeof rpinf);

472 rpinf.sType = VK_STRUCTURE_TYPE_RENDER_PASS_CREATE_INFO;

473 rpinf.attachmentCount = count;

474 rpinf.pAttachments = at;

475 rpinf.subpassCount = 1;

476 rpinf.pSubpasses = &subpass;

477

478 if(vkCreateRenderPass(vkdev, &rpinf, 0, &pass) != 0) {

479 fprintf(stderr, "vku_create_renderpass: failed to create renderpass\n");

480 return 0;

481 }

482

483 return pass;

484 }

485

486 void vku_destroy_renderpass(VkRenderPass rpass)

487 {

488 vkDestroyRenderPass(vkdev, rpass, 0);

489 }

490

491

492 #ifdef VK_USE_PLATFORM_XLIB_KHR

493 int vku_xlib_usable_visual(Display *dpy, VisualID vid)

494 {

495 return vkGetPhysicalDeviceXlibPresentationSupportKHR(phys_devices[sel_dev],

496 sel_qfamily, dpy, vid);

497 }

498

499 VkSurfaceKHR vku_xlib_create_surface(Display *dpy, Window win)

500 {

501 VkSurfaceKHR surf;

502 VkXlibSurfaceCreateInfoKHR inf;

503

504 memset(&inf, 0, sizeof inf);

505 inf.sType = VK_STRUCTURE_TYPE_XLIB_SURFACE_CREATE_INFO_KHR;

506 inf.dpy = dpy;

507 inf.window = win;

508

509 if(vkCreateXlibSurfaceKHR(vk, &inf, 0, &surf) != 0) {

510 return 0;

511 }

512 return surf;

513 }

514

515 #endif /* VK_USE_PLATFORM_XLIB_KHR */

516

517 static const char *get_device_name(VkPhysicalDeviceType type)

518 {

519 switch(type) {

520 case VK_PHYSICAL_DEVICE_TYPE_INTEGRATED_GPU:

521 return "integrated GPU";

522 case VK_PHYSICAL_DEVICE_TYPE_DISCRETE_GPU:

523 return "discrete GPU";

524 case VK_PHYSICAL_DEVICE_TYPE_VIRTUAL_GPU:

525 return "virtual GPU";

526 case VK_PHYSICAL_DEVICE_TYPE_CPU:

527 return "CPU";

528 default:

529 break;

530 }

531 return "unknown";

532 }

533

534 static const char *get_mem_prop_flag_string(VkMemoryPropertyFlags flags)

535 {

536 static char str[128];

537

538 str[0] = 0;

539 if(flags & VK_MEMORY_PROPERTY_DEVICE_LOCAL_BIT) {

540 strcat(str, "device-local ");

541 }

542 if(flags & VK_MEMORY_PROPERTY_HOST_VISIBLE_BIT) {

543 strcat(str, "host-visible ");

544 }

545 if(flags & VK_MEMORY_PROPERTY_HOST_COHERENT_BIT) {

546 strcat(str, "host-coherent ");

547 }

548 if(flags & VK_MEMORY_PROPERTY_HOST_CACHED_BIT) {

549 strcat(str, "host-cached ");

550 }

551 if(flags & VK_MEMORY_PROPERTY_LAZILY_ALLOCATED_BIT) {

552 strcat(str, "lazily-allocated ");

553 }

554

555 if(!*str) {

556 strcat(str, "-");

557 }

558 return str;

559 }

560

561 static const char *get_queue_flag_string(VkQueueFlagBits flags)

562 {

563 static char str[128];

564

565 str[0] = 0;

566 if(flags & VK_QUEUE_GRAPHICS_BIT) {

567 strcat(str, "graphics ");

568 }

569 if(flags & VK_QUEUE_COMPUTE_BIT) {

570 strcat(str, "compute ");

571 }

572 if(flags & VK_QUEUE_TRANSFER_BIT) {

573 strcat(str, "transfer ");

574 }

575 if(flags & VK_QUEUE_SPARSE_BINDING_BIT) {

576 strcat(str, "sparse-binding ");

577 }

578 if(!*str) {

579 strcat(str, "-");

580 }

581 return str;

582 }

583

584 static int ver_major(uint32_t ver)

585 {

586 return (ver >> 22) & 0x3ff;

587 }

588

589 static int ver_minor(uint32_t ver)

590 {

591 return (ver >> 12) & 0x3ff;

592 }

593

594 static int ver_patch(uint32_t ver)

595 {

596 return ver & 0xfff;

597 }

598

599 static const char *mem_size_str(long sz)

600 {

601 static char str[64];

602 static const char *unitstr[] = { "bytes", "KB", "MB", "GB", "TB", "PB", 0 };

603 int uidx = 0;

604 sz *= 10;

605

606 while(sz >= 10240 && unitstr[uidx + 1]) {

607 sz /= 1024;

608 ++uidx;

609 }

610 sprintf(str, "%ld.%ld %s", sz / 10, sz % 10, unitstr[uidx]);

611 return str;

612 }