nuclear@0: #include <stdio.h>
nuclear@0: #include <stdlib.h>
nuclear@0: #include <string.h>
nuclear@0: #include <alloca.h>
nuclear@0: #include <vulkan/vulkan.h>
nuclear@0: 
nuclear@0: const char *get_device_name(VkPhysicalDeviceType type);
nuclear@2: const char *get_mem_prop_flag_string(VkMemoryPropertyFlags flags);
nuclear@0: const char *get_queue_flag_string(VkQueueFlagBits flags);
nuclear@0: int ver_major(uint32_t ver);
nuclear@0: int ver_minor(uint32_t ver);
nuclear@0: int ver_patch(uint32_t ver);
nuclear@2: const char *mem_size_str(long sz);
nuclear@0: 
nuclear@0: int main(void)
nuclear@0: {
nuclear@0: 	int i, j;
nuclear@0: 	VkInstance vk;
nuclear@0: 	VkInstanceCreateInfo inst_info;
nuclear@0: 	VkPhysicalDevice *devices;
nuclear@0: 	VkDevice vkdev;
nuclear@0: 	VkDeviceCreateInfo dev_info;
nuclear@0: 	VkDeviceQueueCreateInfo queue_info;
nuclear@0: 	uint32_t num_devices;
nuclear@0: 	int sel_dev = -1;
nuclear@0: 	int sel_qfamily = -1;
nuclear@0: 	float qprio = 0.0f;
nuclear@0: 
nuclear@0: 	memset(&inst_info, 0, sizeof inst_info);
nuclear@0: 	inst_info.sType = VK_STRUCTURE_TYPE_INSTANCE_CREATE_INFO;
nuclear@0: 
nuclear@0: 	if(vkCreateInstance(&inst_info, 0, &vk) != 0) {
nuclear@0: 		fprintf(stderr, "failed to create vulkan instance\n");
nuclear@0: 		return 1;
nuclear@0: 	}
nuclear@0: 	printf("created vulkan instance\n");
nuclear@0: 
nuclear@0: 	if(vkEnumeratePhysicalDevices(vk, &num_devices, 0) != 0) {
nuclear@0: 		fprintf(stderr, "failed to enumerate vulkan physical devices\n");
nuclear@0: 		return 1;
nuclear@0: 	}
nuclear@0: 	devices = alloca(num_devices * sizeof *devices);
nuclear@0: 	if(vkEnumeratePhysicalDevices(vk, &num_devices, devices) != 0) {
nuclear@0: 		fprintf(stderr, "failed to enumerate vulkan physical devices\n");
nuclear@0: 		return 1;
nuclear@0: 	}
nuclear@0: 	printf("found %u physical device(s)\n", (unsigned int)num_devices);
nuclear@0: 
nuclear@0: 	for(i=0; i<(int)num_devices; i++) {
nuclear@0: 		VkPhysicalDeviceProperties dev_prop;
nuclear@2: 		VkPhysicalDeviceMemoryProperties mem_prop;
nuclear@0: 		VkQueueFamilyProperties *qprop;
nuclear@0: 		uint32_t qprop_count;
nuclear@0: 
nuclear@0: 		vkGetPhysicalDeviceProperties(devices[i], &dev_prop);
nuclear@0: 
nuclear@0: 		printf("Device %d: %s\n", i, dev_prop.deviceName);
nuclear@0: 		printf("  type: %s\n", get_device_name(dev_prop.deviceType));
nuclear@0: 		printf("  API version: %d.%d.%d\n", ver_major(dev_prop.apiVersion), ver_minor(dev_prop.apiVersion),
nuclear@0: 				ver_patch(dev_prop.apiVersion));
nuclear@0: 		printf("  driver version: %d.%d.%d\n", ver_major(dev_prop.driverVersion), ver_minor(dev_prop.driverVersion),
nuclear@0: 				ver_patch(dev_prop.driverVersion));
nuclear@0: 		printf("  vendor id: %x  device id: %x\n", dev_prop.vendorID, dev_prop.deviceID);
nuclear@0: 
nuclear@2: 
nuclear@2: 		vkGetPhysicalDeviceMemoryProperties(devices[i], &mem_prop);
nuclear@2: 		printf("  %d memory heaps:\n", mem_prop.memoryHeapCount);
nuclear@2: 		for(j=0; j<mem_prop.memoryHeapCount; j++) {
nuclear@2: 			VkMemoryHeap heap = mem_prop.memoryHeaps[j];
nuclear@2: 			printf("    Heap %d - size: %s, flags: %s\n", j, mem_size_str(heap.size),
nuclear@2: 					heap.flags & VK_MEMORY_HEAP_DEVICE_LOCAL_BIT ? "device-local" : "-");
nuclear@2: 		}
nuclear@2: 		printf("  %d memory types:\n", mem_prop.memoryTypeCount);
nuclear@2: 		for(j=0; j<mem_prop.memoryTypeCount; j++) {
nuclear@2: 			VkMemoryType type = mem_prop.memoryTypes[j];
nuclear@2: 			printf("    Type %d - heap: %d, flags: %s\n", j, type.heapIndex,
nuclear@2: 					get_mem_prop_flag_string(type.propertyFlags));
nuclear@2: 		}
nuclear@2: 
nuclear@0: 		vkGetPhysicalDeviceQueueFamilyProperties(devices[i], &qprop_count, 0);
nuclear@0: 		if(qprop_count <= 0) {
nuclear@0: 			continue;
nuclear@0: 		}
nuclear@0: 		qprop = malloc(qprop_count * sizeof *qprop);
nuclear@0: 		vkGetPhysicalDeviceQueueFamilyProperties(devices[i], &qprop_count, qprop);
nuclear@0: 
nuclear@0: 		for(j=0; j<qprop_count; j++) {
nuclear@0: 			printf("  Queue family %d:\n", j);
nuclear@0: 			printf("    flags: %s\n", get_queue_flag_string(qprop[j].queueFlags));
nuclear@0: 			printf("    num queues: %u\n", qprop[j].queueCount);
nuclear@0: 
nuclear@0: 			if(qprop[j].queueFlags & VK_QUEUE_GRAPHICS_BIT) {
nuclear@0: 				sel_dev = i;
nuclear@0: 				sel_qfamily = j;
nuclear@0: 			}
nuclear@0: 		}
nuclear@0: 		free(qprop);
nuclear@0: 	}
nuclear@0: 
nuclear@0: 	if(sel_dev < 0 || sel_qfamily < 0) {
nuclear@0: 		fprintf(stderr, "failed to find any device with a graphics-capable command queue\n");
nuclear@0: 		vkDestroyDevice(vkdev, 0);
nuclear@0: 		return 1;
nuclear@0: 	}
nuclear@0: 
nuclear@1: 	// create device & command queue
nuclear@0: 	memset(&queue_info, 0, sizeof queue_info);
nuclear@0: 	queue_info.sType = VK_STRUCTURE_TYPE_DEVICE_QUEUE_CREATE_INFO;
nuclear@0: 	queue_info.queueFamilyIndex = sel_qfamily;
nuclear@0: 	queue_info.queueCount = 1;
nuclear@0: 	queue_info.pQueuePriorities = &qprio;
nuclear@0: 
nuclear@0: 	memset(&dev_info, 0, sizeof dev_info);
nuclear@0: 	dev_info.sType = VK_STRUCTURE_TYPE_DEVICE_CREATE_INFO;
nuclear@0: 	dev_info.queueCreateInfoCount = 1;
nuclear@0: 	dev_info.pQueueCreateInfos = &queue_info;
nuclear@0: 
nuclear@0: 	if(vkCreateDevice(devices[sel_dev], &dev_info, 0, &vkdev) != 0) {
nuclear@0: 		fprintf(stderr, "failed to create device %d\n", sel_dev);
nuclear@0: 		return 1;
nuclear@0: 	}
nuclear@0: 	printf("created device %d\n", sel_dev);
nuclear@0: 
nuclear@0: 	vkDestroyDevice(vkdev, 0);
nuclear@0: 	vkDestroyInstance(vk, 0);
nuclear@0: 	return 0;
nuclear@0: }
nuclear@0: 
nuclear@0: 
nuclear@0: const char *get_device_name(VkPhysicalDeviceType type)
nuclear@0: {
nuclear@0: 	switch(type) {
nuclear@0: 	case VK_PHYSICAL_DEVICE_TYPE_INTEGRATED_GPU:
nuclear@0: 		return "integrated GPU";
nuclear@0: 	case VK_PHYSICAL_DEVICE_TYPE_DISCRETE_GPU:
nuclear@0: 		return "discrete GPU";
nuclear@0: 	case VK_PHYSICAL_DEVICE_TYPE_VIRTUAL_GPU:
nuclear@0: 		return "virtual GPU";
nuclear@0: 	case VK_PHYSICAL_DEVICE_TYPE_CPU:
nuclear@0: 		return "CPU";
nuclear@0: 	default:
nuclear@0: 		break;
nuclear@0: 	}
nuclear@0: 	return "unknown";
nuclear@0: }
nuclear@0: 
nuclear@2: const char *get_mem_prop_flag_string(VkMemoryPropertyFlags flags)
nuclear@2: {
nuclear@2: 	static char str[128];
nuclear@2: 
nuclear@2: 	str[0] = 0;
nuclear@2: 	if(flags & VK_MEMORY_PROPERTY_DEVICE_LOCAL_BIT) {
nuclear@2: 		strcat(str, "device-local ");
nuclear@2: 	}
nuclear@2: 	if(flags & VK_MEMORY_PROPERTY_HOST_VISIBLE_BIT) {
nuclear@2: 		strcat(str, "host-visible ");
nuclear@2: 	}
nuclear@2: 	if(flags & VK_MEMORY_PROPERTY_HOST_COHERENT_BIT) {
nuclear@2: 		strcat(str, "host-coherent ");
nuclear@2: 	}
nuclear@2: 	if(flags & VK_MEMORY_PROPERTY_HOST_CACHED_BIT) {
nuclear@2: 		strcat(str, "host-cached ");
nuclear@2: 	}
nuclear@2: 	if(flags & VK_MEMORY_PROPERTY_LAZILY_ALLOCATED_BIT) {
nuclear@2: 		strcat(str, "lazily-allocated ");
nuclear@2: 	}
nuclear@2: 
nuclear@2: 	if(!*str) {
nuclear@2: 		strcat(str, "-");
nuclear@2: 	}
nuclear@2: 	return str;
nuclear@2: }
nuclear@2: 
nuclear@0: const char *get_queue_flag_string(VkQueueFlagBits flags)
nuclear@0: {
nuclear@0: 	static char str[128];
nuclear@0: 
nuclear@0: 	str[0] = 0;
nuclear@0: 	if(flags & VK_QUEUE_GRAPHICS_BIT) {
nuclear@0: 		strcat(str, "graphics ");
nuclear@0: 	}
nuclear@0: 	if(flags & VK_QUEUE_COMPUTE_BIT) {
nuclear@0: 		strcat(str, "compute ");
nuclear@0: 	}
nuclear@0: 	if(flags & VK_QUEUE_TRANSFER_BIT) {
nuclear@0: 		strcat(str, "transfer ");
nuclear@0: 	}
nuclear@0: 	if(flags & VK_QUEUE_SPARSE_BINDING_BIT) {
nuclear@0: 		strcat(str, "sparse-binding ");
nuclear@0: 	}
nuclear@2: 	if(!*str) {
nuclear@2: 		strcat(str, "-");
nuclear@2: 	}
nuclear@0: 	return str;
nuclear@0: }
nuclear@0: 
nuclear@0: int ver_major(uint32_t ver)
nuclear@0: {
nuclear@0: 	return (ver >> 22) & 0x3ff;
nuclear@0: }
nuclear@0: 
nuclear@0: int ver_minor(uint32_t ver)
nuclear@0: {
nuclear@0: 	return (ver >> 12) & 0x3ff;
nuclear@0: }
nuclear@0: 
nuclear@0: int ver_patch(uint32_t ver)
nuclear@0: {
nuclear@0: 	return ver & 0xfff;
nuclear@0: }
nuclear@2: 
nuclear@2: const char *mem_size_str(long sz)
nuclear@2: {
nuclear@2: 	static char str[64];
nuclear@2: 	static const char *unitstr[] = { "bytes", "KB", "MB", "GB", "TB", "PB", 0 };
nuclear@2: 	int uidx = 0;
nuclear@2: 	sz *= 10;
nuclear@2: 
nuclear@2: 	while(sz >= 10240 && unitstr[uidx + 1]) {
nuclear@2: 		sz /= 1024;
nuclear@2: 		++uidx;
nuclear@2: 	}
nuclear@2: 	sprintf(str, "%ld.%ld %s", sz / 10, sz % 10, unitstr[uidx]);
nuclear@2: 	return str;
nuclear@2: }