345 lines
11 KiB
C++
345 lines
11 KiB
C++
#include"VKDevice.h"
|
||
#include"VKInstance.h"
|
||
#include"VKPhysicalDevice.h"
|
||
#include"VKFramebuffer.h"
|
||
|
||
VK_NAMESPACE_BEGIN
|
||
VkSurfaceKHR CreateRenderDevice(VkInstance,Window *);
|
||
|
||
namespace
|
||
{
|
||
template<typename T> T Clamp(const T &cur,const T &min_value,const T &max_value)
|
||
{
|
||
if(cur<min_value)return min_value;
|
||
if(cur>max_value)return max_value;
|
||
|
||
return cur;
|
||
}
|
||
|
||
VkExtent2D GetSwapchainExtent(VkSurfaceCapabilitiesKHR &surface_caps,uint32_t width,uint32_t height)
|
||
{
|
||
if(surface_caps.currentExtent.width==UINT32_MAX)
|
||
{
|
||
VkExtent2D swapchain_extent;
|
||
|
||
swapchain_extent.width=Clamp(width,surface_caps.minImageExtent.width,surface_caps.maxImageExtent.width);
|
||
swapchain_extent.height=Clamp(height,surface_caps.minImageExtent.height,surface_caps.maxImageExtent.height);
|
||
|
||
return swapchain_extent;
|
||
}
|
||
else
|
||
{
|
||
return surface_caps.currentExtent;
|
||
}
|
||
}
|
||
|
||
VkDevice CreateDevice(VkInstance instance,VkPhysicalDevice physical_device,uint32_t graphics_family)
|
||
{
|
||
float queue_priorities[1]={0.0};
|
||
|
||
VkDeviceQueueCreateInfo queue_info;
|
||
queue_info.sType=VK_STRUCTURE_TYPE_DEVICE_QUEUE_CREATE_INFO;
|
||
queue_info.pNext=nullptr;
|
||
queue_info.queueFamilyIndex=graphics_family;
|
||
queue_info.queueCount=1;
|
||
queue_info.pQueuePriorities=queue_priorities;
|
||
queue_info.flags=0; //如果这里写VK_DEVICE_QUEUE_CREATE_PROTECTED_BIT，会导致vkGetDeviceQueue调用崩溃
|
||
|
||
VkDeviceCreateInfo create_info={};
|
||
const char *ext_list[1]={VK_KHR_SWAPCHAIN_EXTENSION_NAME};
|
||
create_info.sType=VK_STRUCTURE_TYPE_DEVICE_CREATE_INFO;
|
||
create_info.pNext=nullptr;
|
||
create_info.queueCreateInfoCount=1;
|
||
create_info.pQueueCreateInfos=&queue_info;
|
||
create_info.enabledExtensionCount=1;
|
||
create_info.ppEnabledExtensionNames=ext_list;
|
||
create_info.enabledLayerCount=0;
|
||
create_info.ppEnabledLayerNames=nullptr;
|
||
create_info.pEnabledFeatures=nullptr;
|
||
|
||
VkDevice device;
|
||
|
||
if(vkCreateDevice(physical_device,&create_info,nullptr,&device)==VK_SUCCESS)
|
||
return device;
|
||
|
||
return nullptr;
|
||
}
|
||
|
||
void GetDeviceQueue(DeviceAttribute *attr)
|
||
{
|
||
vkGetDeviceQueue(attr->device,attr->graphics_family,0,&attr->graphics_queue);
|
||
|
||
if(attr->graphics_family==attr->present_family)
|
||
attr->present_queue=attr->graphics_queue;
|
||
else
|
||
vkGetDeviceQueue(attr->device,attr->present_family,0,&attr->present_queue);
|
||
}
|
||
|
||
VkCommandPool CreateCommandPool(VkDevice device,uint32_t graphics_family)
|
||
{
|
||
VkCommandPoolCreateInfo cmd_pool_info={};
|
||
|
||
cmd_pool_info.sType=VK_STRUCTURE_TYPE_COMMAND_POOL_CREATE_INFO;
|
||
cmd_pool_info.pNext=nullptr;
|
||
cmd_pool_info.queueFamilyIndex=graphics_family;
|
||
cmd_pool_info.flags=0;
|
||
|
||
VkCommandPool cmd_pool;
|
||
|
||
if(vkCreateCommandPool(device,&cmd_pool_info,nullptr,&cmd_pool)==VK_SUCCESS)
|
||
return cmd_pool;
|
||
|
||
return(nullptr);
|
||
}
|
||
|
||
VkSwapchainKHR CreateSwapChain(DeviceAttribute *rsa)
|
||
{
|
||
VkSwapchainCreateInfoKHR swapchain_ci={};
|
||
|
||
swapchain_ci.sType=VK_STRUCTURE_TYPE_SWAPCHAIN_CREATE_INFO_KHR;
|
||
swapchain_ci.pNext=nullptr;
|
||
swapchain_ci.surface=rsa->surface;
|
||
swapchain_ci.minImageCount=rsa->surface_caps.minImageCount;
|
||
swapchain_ci.imageFormat=rsa->format;
|
||
swapchain_ci.imageExtent=rsa->swapchain_extent;
|
||
swapchain_ci.preTransform=rsa->preTransform;
|
||
swapchain_ci.compositeAlpha=rsa->compositeAlpha;
|
||
swapchain_ci.imageArrayLayers=1;
|
||
swapchain_ci.presentMode=VK_PRESENT_MODE_FIFO_KHR;
|
||
swapchain_ci.oldSwapchain=VK_NULL_HANDLE;
|
||
swapchain_ci.clipped=true;
|
||
swapchain_ci.imageColorSpace=VK_COLORSPACE_SRGB_NONLINEAR_KHR;
|
||
swapchain_ci.imageUsage=VK_IMAGE_USAGE_COLOR_ATTACHMENT_BIT;
|
||
|
||
uint32_t queueFamilyIndices[2]={rsa->graphics_family, rsa->present_family};
|
||
if(rsa->graphics_family!=rsa->present_family)
|
||
{
|
||
// If the graphics and present queues are from different queue families,
|
||
// we either have to explicitly transfer ownership of images between
|
||
// the queues, or we have to create the swapchain with imageSharingMode
|
||
// as VK_SHARING_MODE_CONCURRENT
|
||
swapchain_ci.imageSharingMode=VK_SHARING_MODE_CONCURRENT;
|
||
swapchain_ci.queueFamilyIndexCount=2;
|
||
swapchain_ci.pQueueFamilyIndices=queueFamilyIndices;
|
||
}
|
||
else
|
||
{
|
||
swapchain_ci.imageSharingMode=VK_SHARING_MODE_EXCLUSIVE;
|
||
}
|
||
|
||
VkSwapchainKHR swap_chain;
|
||
|
||
if(vkCreateSwapchainKHR(rsa->device,&swapchain_ci,nullptr,&swap_chain)==VK_SUCCESS)
|
||
return(swap_chain);
|
||
|
||
return(nullptr);
|
||
}
|
||
|
||
ImageView *Create2DImageView(VkDevice device,VkFormat format,VkImage img=nullptr)
|
||
{
|
||
return CreateImageView(device,VK_IMAGE_VIEW_TYPE_2D,format,VK_IMAGE_ASPECT_COLOR_BIT,img);
|
||
}
|
||
|
||
ImageView *CreateDepthImageView(VkDevice device,VkFormat format,VkImage img=nullptr)
|
||
{
|
||
return CreateImageView(device,VK_IMAGE_VIEW_TYPE_2D,format,VK_IMAGE_ASPECT_DEPTH_BIT,img);
|
||
}
|
||
|
||
bool CreateSwapchainImageView(DeviceAttribute *rsa)
|
||
{
|
||
uint32_t count;
|
||
|
||
if(vkGetSwapchainImagesKHR(rsa->device,rsa->swap_chain,&count,nullptr)!=VK_SUCCESS)
|
||
return(false);
|
||
|
||
rsa->sc_images.SetCount(count);
|
||
|
||
if(vkGetSwapchainImagesKHR(rsa->device,rsa->swap_chain,&count,rsa->sc_images.GetData())!=VK_SUCCESS)
|
||
{
|
||
rsa->sc_images.Clear();
|
||
return(false);
|
||
}
|
||
|
||
VkImage *ip=rsa->sc_images.GetData();
|
||
ImageView *vp;
|
||
for(uint32_t i=0; i<count; i++)
|
||
{
|
||
vp=Create2DImageView(rsa->device,rsa->format,*ip);
|
||
|
||
if(vp==nullptr)
|
||
return(false);
|
||
|
||
rsa->sc_image_views.Add(vp);
|
||
|
||
++ip;
|
||
}
|
||
|
||
return(true);
|
||
}
|
||
|
||
bool CreateDepthBuffer(DeviceAttribute *rsa)
|
||
{
|
||
VkImageCreateInfo image_info={};
|
||
|
||
const VkFormat depth_format=VK_FORMAT_D16_UNORM;
|
||
VkFormatProperties props;
|
||
|
||
vkGetPhysicalDeviceFormatProperties(rsa->physical_device->physical_device,depth_format,&props);
|
||
|
||
if(props.linearTilingFeatures&VK_FORMAT_FEATURE_DEPTH_STENCIL_ATTACHMENT_BIT)
|
||
image_info.tiling=VK_IMAGE_TILING_LINEAR;
|
||
else
|
||
if(props.optimalTilingFeatures&VK_FORMAT_FEATURE_DEPTH_STENCIL_ATTACHMENT_BIT)
|
||
image_info.tiling=VK_IMAGE_TILING_OPTIMAL;
|
||
else
|
||
return(false);
|
||
|
||
image_info.sType=VK_STRUCTURE_TYPE_IMAGE_CREATE_INFO;
|
||
image_info.pNext=nullptr;
|
||
image_info.imageType=VK_IMAGE_TYPE_2D;
|
||
image_info.format=depth_format;
|
||
image_info.extent.width=rsa->swapchain_extent.width;
|
||
image_info.extent.height=rsa->swapchain_extent.height;
|
||
image_info.extent.depth=1;
|
||
image_info.mipLevels=1;
|
||
image_info.arrayLayers=1;
|
||
image_info.samples=VK_SAMPLE_COUNT_1_BIT;
|
||
image_info.initialLayout=VK_IMAGE_LAYOUT_UNDEFINED;
|
||
image_info.usage=VK_IMAGE_USAGE_DEPTH_STENCIL_ATTACHMENT_BIT;
|
||
image_info.queueFamilyIndexCount=0;
|
||
image_info.pQueueFamilyIndices=nullptr;
|
||
image_info.sharingMode=VK_SHARING_MODE_EXCLUSIVE;
|
||
image_info.flags=0;
|
||
|
||
rsa->depth.format=depth_format;
|
||
|
||
if(vkCreateImage(rsa->device,&image_info,nullptr,&rsa->depth.image)!=VK_SUCCESS)
|
||
return(false);
|
||
|
||
VkMemoryRequirements mem_reqs;
|
||
vkGetImageMemoryRequirements(rsa->device,rsa->depth.image,&mem_reqs);
|
||
|
||
VkMemoryAllocateInfo mem_alloc={};
|
||
mem_alloc.sType=VK_STRUCTURE_TYPE_MEMORY_ALLOCATE_INFO;
|
||
mem_alloc.pNext=nullptr;
|
||
mem_alloc.allocationSize=0;
|
||
mem_alloc.memoryTypeIndex=0;
|
||
mem_alloc.allocationSize=mem_reqs.size;
|
||
|
||
if(!rsa->CheckMemoryType(mem_reqs.memoryTypeBits,VK_MEMORY_PROPERTY_DEVICE_LOCAL_BIT,&mem_alloc.memoryTypeIndex))
|
||
return(false);
|
||
|
||
if(vkAllocateMemory(rsa->device,&mem_alloc,nullptr,&rsa->depth.mem)!=VK_SUCCESS)
|
||
return(false);
|
||
|
||
if(vkBindImageMemory(rsa->device,rsa->depth.image,rsa->depth.mem,0)!=VK_SUCCESS)
|
||
return(false);
|
||
|
||
rsa->depth.view=CreateDepthImageView(rsa->device,depth_format,rsa->depth.image);
|
||
|
||
if(rsa->depth.view==nullptr)
|
||
return(false);
|
||
|
||
return(true);
|
||
}
|
||
|
||
VkDescriptorPool CreateDescriptorPool(VkDevice device,int sets_count)
|
||
{
|
||
constexpr size_t DESC_POOL_COUNT=1;
|
||
|
||
VkDescriptorPoolSize pool_size[DESC_POOL_COUNT];
|
||
|
||
for(size_t i=0;i<DESC_POOL_COUNT;i++)
|
||
{
|
||
pool_size[i].type=VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER;
|
||
pool_size[i].descriptorCount=1;
|
||
}
|
||
|
||
VkDescriptorPoolCreateInfo dp_create_info={};
|
||
dp_create_info.sType=VK_STRUCTURE_TYPE_DESCRIPTOR_POOL_CREATE_INFO;
|
||
dp_create_info.pNext=nullptr;
|
||
dp_create_info.maxSets=sets_count;
|
||
dp_create_info.poolSizeCount=DESC_POOL_COUNT;
|
||
dp_create_info.pPoolSizes=pool_size;
|
||
|
||
VkDescriptorPool desc_pool;
|
||
|
||
if(vkCreateDescriptorPool(device,&dp_create_info,nullptr,&desc_pool)!=VK_SUCCESS)
|
||
return(nullptr);
|
||
|
||
return desc_pool;
|
||
}
|
||
|
||
VkPipelineCache CreatePipelineCache(VkDevice device)
|
||
{
|
||
VkPipelineCacheCreateInfo pipelineCache;
|
||
pipelineCache.sType = VK_STRUCTURE_TYPE_PIPELINE_CACHE_CREATE_INFO;
|
||
pipelineCache.pNext = nullptr;
|
||
pipelineCache.initialDataSize = 0;
|
||
pipelineCache.pInitialData = nullptr;
|
||
pipelineCache.flags = 0;
|
||
|
||
VkPipelineCache cache;
|
||
|
||
if(vkCreatePipelineCache(device, &pipelineCache, nullptr, &cache)!=VK_SUCCESS)
|
||
return(nullptr);
|
||
|
||
return cache;
|
||
}
|
||
}//namespace
|
||
|
||
Device *CreateRenderDevice(VkInstance inst,const PhysicalDevice *physical_device,Window *win)
|
||
{
|
||
VkSurfaceKHR surface=CreateRenderDevice(inst,win);
|
||
|
||
if(!surface)
|
||
return(nullptr);
|
||
|
||
DeviceAttribute *attr=new DeviceAttribute(inst,physical_device,surface);
|
||
|
||
AutoDelete<DeviceAttribute> auto_delete(attr);
|
||
|
||
attr->swapchain_extent=GetSwapchainExtent(attr->surface_caps,win->GetWidth(),win->GetHeight());
|
||
|
||
if(attr->graphics_family==ERROR_FAMILY_INDEX)
|
||
return(nullptr);
|
||
|
||
attr->device=CreateDevice(inst,physical_device->physical_device,attr->graphics_family);
|
||
|
||
if(!attr->device)
|
||
return(nullptr);
|
||
|
||
GetDeviceQueue(attr);
|
||
|
||
attr->cmd_pool=CreateCommandPool(attr->device,attr->graphics_family);
|
||
|
||
if(!attr->cmd_pool)
|
||
return(nullptr);
|
||
|
||
attr->swap_chain=CreateSwapChain(attr);
|
||
|
||
if(!attr->swap_chain)
|
||
return(nullptr);
|
||
|
||
if(!CreateSwapchainImageView(attr))
|
||
return(nullptr);
|
||
|
||
if(!CreateDepthBuffer(attr))
|
||
return(nullptr);
|
||
|
||
attr->desc_pool=CreateDescriptorPool(attr->device,1);
|
||
|
||
if(!attr->desc_pool)
|
||
return(nullptr);
|
||
|
||
attr->pipeline_cache=CreatePipelineCache(attr->device);
|
||
|
||
if(!attr->pipeline_cache)
|
||
return(nullptr);
|
||
|
||
auto_delete.Clear();
|
||
|
||
return(new Device(attr));
|
||
}
|
||
VK_NAMESPACE_END
|