// Copyright(c) 2019, NVIDIA CORPORATION. All rights reserved. // // Licensed under the Apache License, Version 2.0 (the "License"); // you may not use this file except in compliance with the License. // You may obtain a copy of the License at // // http://www.apache.org/licenses/LICENSE-2.0 // // Unless required by applicable law or agreed to in writing, software // distributed under the License is distributed on an "AS IS" BASIS, // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. // See the License for the specific language governing permissions and // limitations under the License. // #if defined( _MSC_VER ) // no need to ignore any warnings with MSVC #elif defined( __clang__ ) # pragma clang diagnostic ignored "-Wmissing-braces" #elif defined( __GNUC__ ) // no need to ignore any warnings with GCC #else // unknow compiler... just ignore the warnings for yourselves ;) #endif #include "utils.hpp" #include #include #include #if defined( VULKAN_HPP_NO_TO_STRING ) # include #endif #include #if ( VULKAN_HPP_DISPATCH_LOADER_DYNAMIC == 1 ) VULKAN_HPP_DEFAULT_DISPATCH_LOADER_DYNAMIC_STORAGE #endif namespace vk { namespace su { vk::DeviceMemory allocateDeviceMemory( vk::Device const & device, vk::PhysicalDeviceMemoryProperties const & memoryProperties, vk::MemoryRequirements const & memoryRequirements, vk::MemoryPropertyFlags memoryPropertyFlags ) { uint32_t memoryTypeIndex = findMemoryType( memoryProperties, memoryRequirements.memoryTypeBits, memoryPropertyFlags ); return device.allocateMemory( vk::MemoryAllocateInfo( memoryRequirements.size, memoryTypeIndex ) ); } bool contains( std::vector const & extensionProperties, std::string const & extensionName ) { auto propertyIterator = std::find_if( extensionProperties.begin(), extensionProperties.end(), [&extensionName]( vk::ExtensionProperties const & ep ) { return extensionName == ep.extensionName; } ); return ( propertyIterator != extensionProperties.end() ); } vk::DescriptorPool createDescriptorPool( vk::Device const & device, std::vector const & poolSizes ) { assert( !poolSizes.empty() ); uint32_t maxSets = std::accumulate( poolSizes.begin(), poolSizes.end(), 0, []( uint32_t sum, vk::DescriptorPoolSize const & dps ) { return sum + dps.descriptorCount; } ); assert( 0 < maxSets ); vk::DescriptorPoolCreateInfo descriptorPoolCreateInfo( vk::DescriptorPoolCreateFlagBits::eFreeDescriptorSet, maxSets, poolSizes ); return device.createDescriptorPool( descriptorPoolCreateInfo ); } vk::DescriptorSetLayout createDescriptorSetLayout( vk::Device const & device, std::vector> const & bindingData, vk::DescriptorSetLayoutCreateFlags flags ) { std::vector bindings( bindingData.size() ); for ( size_t i = 0; i < bindingData.size(); i++ ) { bindings[i] = vk::DescriptorSetLayoutBinding( checked_cast( i ), std::get<0>( bindingData[i] ), std::get<1>( bindingData[i] ), std::get<2>( bindingData[i] ) ); } return device.createDescriptorSetLayout( vk::DescriptorSetLayoutCreateInfo( flags, bindings ) ); } vk::Device createDevice( vk::PhysicalDevice const & physicalDevice, uint32_t queueFamilyIndex, std::vector const & extensions, vk::PhysicalDeviceFeatures const * physicalDeviceFeatures, void const * pNext ) { std::vector enabledExtensions; enabledExtensions.reserve( extensions.size() ); for ( auto const & ext : extensions ) { enabledExtensions.push_back( ext.data() ); } float queuePriority = 0.0f; vk::DeviceQueueCreateInfo deviceQueueCreateInfo( {}, queueFamilyIndex, 1, &queuePriority ); vk::DeviceCreateInfo deviceCreateInfo( {}, deviceQueueCreateInfo, {}, enabledExtensions, physicalDeviceFeatures, pNext ); vk::Device device = physicalDevice.createDevice( deviceCreateInfo ); #if ( VULKAN_HPP_DISPATCH_LOADER_DYNAMIC == 1 ) // initialize function pointers for instance VULKAN_HPP_DEFAULT_DISPATCHER.init( device ); #endif return device; } std::vector createFramebuffers( vk::Device const & device, vk::RenderPass & renderPass, std::vector const & imageViews, vk::ImageView const & depthImageView, vk::Extent2D const & extent ) { vk::ImageView attachments[2]; attachments[1] = depthImageView; vk::FramebufferCreateInfo framebufferCreateInfo( vk::FramebufferCreateFlags(), renderPass, depthImageView ? 2 : 1, attachments, extent.width, extent.height, 1 ); std::vector framebuffers; framebuffers.reserve( imageViews.size() ); for ( auto const & view : imageViews ) { attachments[0] = view; framebuffers.push_back( device.createFramebuffer( framebufferCreateInfo ) ); } return framebuffers; } vk::Pipeline createGraphicsPipeline( vk::Device const & device, vk::PipelineCache const & pipelineCache, std::pair const & vertexShaderData, std::pair const & fragmentShaderData, uint32_t vertexStride, std::vector> const & vertexInputAttributeFormatOffset, vk::FrontFace frontFace, bool depthBuffered, vk::PipelineLayout const & pipelineLayout, vk::RenderPass const & renderPass ) { std::array pipelineShaderStageCreateInfos = { vk::PipelineShaderStageCreateInfo( vk::PipelineShaderStageCreateFlags(), vk::ShaderStageFlagBits::eVertex, vertexShaderData.first, "main", vertexShaderData.second ), vk::PipelineShaderStageCreateInfo( vk::PipelineShaderStageCreateFlags(), vk::ShaderStageFlagBits::eFragment, fragmentShaderData.first, "main", fragmentShaderData.second ) }; std::vector vertexInputAttributeDescriptions; vk::PipelineVertexInputStateCreateInfo pipelineVertexInputStateCreateInfo; vk::VertexInputBindingDescription vertexInputBindingDescription( 0, vertexStride ); if ( 0 < vertexStride ) { vertexInputAttributeDescriptions.reserve( vertexInputAttributeFormatOffset.size() ); for ( uint32_t i = 0; i < vertexInputAttributeFormatOffset.size(); i++ ) { vertexInputAttributeDescriptions.emplace_back( i, 0, vertexInputAttributeFormatOffset[i].first, vertexInputAttributeFormatOffset[i].second ); } pipelineVertexInputStateCreateInfo.setVertexBindingDescriptions( vertexInputBindingDescription ); pipelineVertexInputStateCreateInfo.setVertexAttributeDescriptions( vertexInputAttributeDescriptions ); } vk::PipelineInputAssemblyStateCreateInfo pipelineInputAssemblyStateCreateInfo( vk::PipelineInputAssemblyStateCreateFlags(), vk::PrimitiveTopology::eTriangleList ); vk::PipelineViewportStateCreateInfo pipelineViewportStateCreateInfo( vk::PipelineViewportStateCreateFlags(), 1, nullptr, 1, nullptr ); vk::PipelineRasterizationStateCreateInfo pipelineRasterizationStateCreateInfo( vk::PipelineRasterizationStateCreateFlags(), false, false, vk::PolygonMode::eFill, vk::CullModeFlagBits::eBack, frontFace, false, 0.0f, 0.0f, 0.0f, 1.0f ); vk::PipelineMultisampleStateCreateInfo pipelineMultisampleStateCreateInfo( {}, vk::SampleCountFlagBits::e1 ); vk::StencilOpState stencilOpState( vk::StencilOp::eKeep, vk::StencilOp::eKeep, vk::StencilOp::eKeep, vk::CompareOp::eAlways ); vk::PipelineDepthStencilStateCreateInfo pipelineDepthStencilStateCreateInfo( vk::PipelineDepthStencilStateCreateFlags(), depthBuffered, depthBuffered, vk::CompareOp::eLessOrEqual, false, false, stencilOpState, stencilOpState ); vk::ColorComponentFlags colorComponentFlags( vk::ColorComponentFlagBits::eR | vk::ColorComponentFlagBits::eG | vk::ColorComponentFlagBits::eB | vk::ColorComponentFlagBits::eA ); vk::PipelineColorBlendAttachmentState pipelineColorBlendAttachmentState( false, vk::BlendFactor::eZero, vk::BlendFactor::eZero, vk::BlendOp::eAdd, vk::BlendFactor::eZero, vk::BlendFactor::eZero, vk::BlendOp::eAdd, colorComponentFlags ); vk::PipelineColorBlendStateCreateInfo pipelineColorBlendStateCreateInfo( vk::PipelineColorBlendStateCreateFlags(), false, vk::LogicOp::eNoOp, pipelineColorBlendAttachmentState, { { 1.0f, 1.0f, 1.0f, 1.0f } } ); std::array dynamicStates = { vk::DynamicState::eViewport, vk::DynamicState::eScissor }; vk::PipelineDynamicStateCreateInfo pipelineDynamicStateCreateInfo( vk::PipelineDynamicStateCreateFlags(), dynamicStates ); vk::GraphicsPipelineCreateInfo graphicsPipelineCreateInfo( vk::PipelineCreateFlags(), pipelineShaderStageCreateInfos, &pipelineVertexInputStateCreateInfo, &pipelineInputAssemblyStateCreateInfo, nullptr, &pipelineViewportStateCreateInfo, &pipelineRasterizationStateCreateInfo, &pipelineMultisampleStateCreateInfo, &pipelineDepthStencilStateCreateInfo, &pipelineColorBlendStateCreateInfo, &pipelineDynamicStateCreateInfo, pipelineLayout, renderPass ); auto result = device.createGraphicsPipeline( pipelineCache, graphicsPipelineCreateInfo ); assert( result.result == vk::Result::eSuccess ); return result.value; } std::vector gatherExtensions( std::vector const & extensions #if !defined( NDEBUG ) , std::vector const & extensionProperties #endif ) { std::vector enabledExtensions; enabledExtensions.reserve( extensions.size() ); for ( auto const & ext : extensions ) { assert( std::any_of( extensionProperties.begin(), extensionProperties.end(), [ext]( vk::ExtensionProperties const & ep ) { return ext == ep.extensionName; } ) ); enabledExtensions.push_back( ext.data() ); } #if !defined( NDEBUG ) if ( std::none_of( extensions.begin(), extensions.end(), []( std::string const & extension ) { return extension == VK_EXT_DEBUG_UTILS_EXTENSION_NAME; } ) && std::any_of( extensionProperties.begin(), extensionProperties.end(), []( vk::ExtensionProperties const & ep ) { return ( strcmp( VK_EXT_DEBUG_UTILS_EXTENSION_NAME, ep.extensionName ) == 0 ); } ) ) { enabledExtensions.push_back( VK_EXT_DEBUG_UTILS_EXTENSION_NAME ); } #endif return enabledExtensions; } std::vector gatherLayers( std::vector const & layers #if !defined( NDEBUG ) , std::vector const & layerProperties #endif ) { std::vector enabledLayers; enabledLayers.reserve( layers.size() ); for ( auto const & layer : layers ) { assert( std::any_of( layerProperties.begin(), layerProperties.end(), [layer]( vk::LayerProperties const & lp ) { return layer == lp.layerName; } ) ); enabledLayers.push_back( layer.data() ); } #if !defined( NDEBUG ) // Enable standard validation layer to find as much errors as possible! if ( std::none_of( layers.begin(), layers.end(), []( std::string const & layer ) { return layer == "VK_LAYER_KHRONOS_validation"; } ) && std::any_of( layerProperties.begin(), layerProperties.end(), []( vk::LayerProperties const & lp ) { return ( strcmp( "VK_LAYER_KHRONOS_validation", lp.layerName ) == 0 ); } ) ) { enabledLayers.push_back( "VK_LAYER_KHRONOS_validation" ); } #endif return enabledLayers; } vk::Instance createInstance( std::string const & appName, std::string const & engineName, std::vector const & layers, std::vector const & extensions, uint32_t apiVersion ) { #if ( VULKAN_HPP_DISPATCH_LOADER_DYNAMIC == 1 ) static vk::DynamicLoader dl; PFN_vkGetInstanceProcAddr vkGetInstanceProcAddr = dl.getProcAddress( "vkGetInstanceProcAddr" ); VULKAN_HPP_DEFAULT_DISPATCHER.init( vkGetInstanceProcAddr ); #endif vk::ApplicationInfo applicationInfo( appName.c_str(), 1, engineName.c_str(), 1, apiVersion ); std::vector enabledLayers = vk::su::gatherLayers( layers #if !defined( NDEBUG ) , vk::enumerateInstanceLayerProperties() #endif ); std::vector enabledExtensions = vk::su::gatherExtensions( extensions #if !defined( NDEBUG ) , vk::enumerateInstanceExtensionProperties() #endif ); vk::Instance instance = vk::createInstance( makeInstanceCreateInfoChain( applicationInfo, enabledLayers, enabledExtensions ).get() ); #if ( VULKAN_HPP_DISPATCH_LOADER_DYNAMIC == 1 ) // initialize function pointers for instance VULKAN_HPP_DEFAULT_DISPATCHER.init( instance ); #endif return instance; } vk::RenderPass createRenderPass( vk::Device const & device, vk::Format colorFormat, vk::Format depthFormat, vk::AttachmentLoadOp loadOp, vk::ImageLayout colorFinalLayout ) { std::vector attachmentDescriptions; assert( colorFormat != vk::Format::eUndefined ); attachmentDescriptions.emplace_back( vk::AttachmentDescriptionFlags(), colorFormat, vk::SampleCountFlagBits::e1, loadOp, vk::AttachmentStoreOp::eStore, vk::AttachmentLoadOp::eDontCare, vk::AttachmentStoreOp::eDontCare, vk::ImageLayout::eUndefined, colorFinalLayout ); if ( depthFormat != vk::Format::eUndefined ) { attachmentDescriptions.emplace_back( vk::AttachmentDescriptionFlags(), depthFormat, vk::SampleCountFlagBits::e1, loadOp, vk::AttachmentStoreOp::eDontCare, vk::AttachmentLoadOp::eDontCare, vk::AttachmentStoreOp::eDontCare, vk::ImageLayout::eUndefined, vk::ImageLayout::eDepthStencilAttachmentOptimal ); } vk::AttachmentReference colorAttachment( 0, vk::ImageLayout::eColorAttachmentOptimal ); vk::AttachmentReference depthAttachment( 1, vk::ImageLayout::eDepthStencilAttachmentOptimal ); vk::SubpassDescription subpassDescription( vk::SubpassDescriptionFlags(), vk::PipelineBindPoint::eGraphics, {}, colorAttachment, {}, ( depthFormat != vk::Format::eUndefined ) ? &depthAttachment : nullptr ); return device.createRenderPass( vk::RenderPassCreateInfo( vk::RenderPassCreateFlags(), attachmentDescriptions, subpassDescription ) ); } VKAPI_ATTR VkBool32 VKAPI_CALL debugUtilsMessengerCallback( VkDebugUtilsMessageSeverityFlagBitsEXT messageSeverity, VkDebugUtilsMessageTypeFlagsEXT messageTypes, VkDebugUtilsMessengerCallbackDataEXT const * pCallbackData, void * /*pUserData*/ ) { #if !defined( NDEBUG ) if ( static_cast(pCallbackData->messageIdNumber) == 0x822806fa ) { // Validation Warning: vkCreateInstance(): to enable extension VK_EXT_debug_utils, but this extension is intended to support use by applications when // debugging and it is strongly recommended that it be otherwise avoided. return vk::False; } else if ( static_cast(pCallbackData->messageIdNumber) == 0xe8d1a9fe ) { // Validation Performance Warning: Using debug builds of the validation layers *will* adversely affect performance. return vk::False; } #endif std::cerr << vk::to_string( static_cast( messageSeverity ) ) << ": " << vk::to_string( static_cast( messageTypes ) ) << ":\n"; std::cerr << std::string( "\t" ) << "messageIDName = <" << pCallbackData->pMessageIdName << ">\n"; std::cerr << std::string( "\t" ) << "messageIdNumber = " << pCallbackData->messageIdNumber << "\n"; std::cerr << std::string( "\t" ) << "message = <" << pCallbackData->pMessage << ">\n"; if ( 0 < pCallbackData->queueLabelCount ) { std::cerr << std::string( "\t" ) << "Queue Labels:\n"; for ( uint32_t i = 0; i < pCallbackData->queueLabelCount; i++ ) { std::cerr << std::string( "\t\t" ) << "labelName = <" << pCallbackData->pQueueLabels[i].pLabelName << ">\n"; } } if ( 0 < pCallbackData->cmdBufLabelCount ) { std::cerr << std::string( "\t" ) << "CommandBuffer Labels:\n"; for ( uint32_t i = 0; i < pCallbackData->cmdBufLabelCount; i++ ) { std::cerr << std::string( "\t\t" ) << "labelName = <" << pCallbackData->pCmdBufLabels[i].pLabelName << ">\n"; } } if ( 0 < pCallbackData->objectCount ) { std::cerr << std::string( "\t" ) << "Objects:\n"; for ( uint32_t i = 0; i < pCallbackData->objectCount; i++ ) { std::cerr << std::string( "\t\t" ) << "Object " << i << "\n"; std::cerr << std::string( "\t\t\t" ) << "objectType = " << vk::to_string( static_cast( pCallbackData->pObjects[i].objectType ) ) << "\n"; std::cerr << std::string( "\t\t\t" ) << "objectHandle = " << pCallbackData->pObjects[i].objectHandle << "\n"; if ( pCallbackData->pObjects[i].pObjectName ) { std::cerr << std::string( "\t\t\t" ) << "objectName = <" << pCallbackData->pObjects[i].pObjectName << ">\n"; } } } return vk::False; } uint32_t findGraphicsQueueFamilyIndex( std::vector const & queueFamilyProperties ) { // get the first index into queueFamiliyProperties which supports graphics std::vector::const_iterator graphicsQueueFamilyProperty = std::find_if( queueFamilyProperties.begin(), queueFamilyProperties.end(), []( vk::QueueFamilyProperties const & qfp ) { return qfp.queueFlags & vk::QueueFlagBits::eGraphics; } ); assert( graphicsQueueFamilyProperty != queueFamilyProperties.end() ); return static_cast( std::distance( queueFamilyProperties.begin(), graphicsQueueFamilyProperty ) ); } std::pair findGraphicsAndPresentQueueFamilyIndex( vk::PhysicalDevice physicalDevice, vk::SurfaceKHR const & surface ) { std::vector queueFamilyProperties = physicalDevice.getQueueFamilyProperties(); assert( queueFamilyProperties.size() < std::numeric_limits::max() ); uint32_t graphicsQueueFamilyIndex = findGraphicsQueueFamilyIndex( queueFamilyProperties ); if ( physicalDevice.getSurfaceSupportKHR( graphicsQueueFamilyIndex, surface ) ) { return std::make_pair( graphicsQueueFamilyIndex, graphicsQueueFamilyIndex ); // the first graphicsQueueFamilyIndex does also support presents } // the graphicsQueueFamilyIndex doesn't support present -> look for an other family index that supports both // graphics and present for ( size_t i = 0; i < queueFamilyProperties.size(); i++ ) { if ( ( queueFamilyProperties[i].queueFlags & vk::QueueFlagBits::eGraphics ) && physicalDevice.getSurfaceSupportKHR( static_cast( i ), surface ) ) { return std::make_pair( static_cast( i ), static_cast( i ) ); } } // there's nothing like a single family index that supports both graphics and present -> look for an other family // index that supports present for ( size_t i = 0; i < queueFamilyProperties.size(); i++ ) { if ( physicalDevice.getSurfaceSupportKHR( static_cast( i ), surface ) ) { return std::make_pair( graphicsQueueFamilyIndex, static_cast( i ) ); } } throw std::runtime_error( "Could not find queues for both graphics or present -> terminating" ); } uint32_t findMemoryType( vk::PhysicalDeviceMemoryProperties const & memoryProperties, uint32_t typeBits, vk::MemoryPropertyFlags requirementsMask ) { uint32_t typeIndex = uint32_t( ~0 ); for ( uint32_t i = 0; i < memoryProperties.memoryTypeCount; i++ ) { if ( ( typeBits & 1 ) && ( ( memoryProperties.memoryTypes[i].propertyFlags & requirementsMask ) == requirementsMask ) ) { typeIndex = i; break; } typeBits >>= 1; } assert( typeIndex != uint32_t( ~0 ) ); return typeIndex; } std::vector getDeviceExtensions() { return { VK_KHR_SWAPCHAIN_EXTENSION_NAME }; } std::vector getInstanceExtensions() { std::vector extensions; extensions.push_back( VK_KHR_SURFACE_EXTENSION_NAME ); #if defined( VK_USE_PLATFORM_ANDROID_KHR ) extensions.push_back( VK_KHR_ANDROID_SURFACE_EXTENSION_NAME ); #elif defined( VK_USE_PLATFORM_IOS_MVK ) extensions.push_back( VK_MVK_IOS_SURFACE_EXTENSION_NAME ); #elif defined( VK_USE_PLATFORM_MACOS_MVK ) extensions.push_back( VK_MVK_MACOS_SURFACE_EXTENSION_NAME ); #elif defined( VK_USE_PLATFORM_MIR_KHR ) extensions.push_back( VK_KHR_MIR_SURFACE_EXTENSION_NAME ); #elif defined( VK_USE_PLATFORM_VI_NN ) extensions.push_back( VK_NN_VI_SURFACE_EXTENSION_NAME ); #elif defined( VK_USE_PLATFORM_WAYLAND_KHR ) extensions.push_back( VK_KHR_WAYLAND_SURFACE_EXTENSION_NAME ); #elif defined( VK_USE_PLATFORM_WIN32_KHR ) extensions.push_back( VK_KHR_WIN32_SURFACE_EXTENSION_NAME ); #elif defined( VK_USE_PLATFORM_XCB_KHR ) extensions.push_back( VK_KHR_XCB_SURFACE_EXTENSION_NAME ); #elif defined( VK_USE_PLATFORM_XLIB_KHR ) extensions.push_back( VK_KHR_XLIB_SURFACE_EXTENSION_NAME ); #elif defined( VK_USE_PLATFORM_XLIB_XRANDR_EXT ) extensions.push_back( VK_EXT_ACQUIRE_XLIB_DISPLAY_EXTENSION_NAME ); #endif return extensions; } vk::Format pickDepthFormat( vk::PhysicalDevice const & physicalDevice ) { std::vector candidates = { vk::Format::eD32Sfloat, vk::Format::eD32SfloatS8Uint, vk::Format::eD24UnormS8Uint }; for ( vk::Format format : candidates ) { vk::FormatProperties props = physicalDevice.getFormatProperties( format ); if ( props.optimalTilingFeatures & vk::FormatFeatureFlagBits::eDepthStencilAttachment ) { return format; } } throw std::runtime_error( "failed to find supported format!" ); } vk::PresentModeKHR pickPresentMode( std::vector const & presentModes ) { vk::PresentModeKHR pickedMode = vk::PresentModeKHR::eFifo; for ( const auto & presentMode : presentModes ) { if ( presentMode == vk::PresentModeKHR::eMailbox ) { pickedMode = presentMode; break; } if ( presentMode == vk::PresentModeKHR::eImmediate ) { pickedMode = presentMode; } } return pickedMode; } vk::SurfaceFormatKHR pickSurfaceFormat( std::vector const & formats ) { assert( !formats.empty() ); vk::SurfaceFormatKHR pickedFormat = formats[0]; if ( formats.size() == 1 ) { if ( formats[0].format == vk::Format::eUndefined ) { pickedFormat.format = vk::Format::eB8G8R8A8Unorm; pickedFormat.colorSpace = vk::ColorSpaceKHR::eSrgbNonlinear; } } else { // request several formats, the first found will be used vk::Format requestedFormats[] = { vk::Format::eB8G8R8A8Unorm, vk::Format::eR8G8B8A8Unorm, vk::Format::eB8G8R8Unorm, vk::Format::eR8G8B8Unorm }; vk::ColorSpaceKHR requestedColorSpace = vk::ColorSpaceKHR::eSrgbNonlinear; for ( size_t i = 0; i < sizeof( requestedFormats ) / sizeof( requestedFormats[0] ); i++ ) { vk::Format requestedFormat = requestedFormats[i]; auto it = std::find_if( formats.begin(), formats.end(), [requestedFormat, requestedColorSpace]( vk::SurfaceFormatKHR const & f ) { return ( f.format == requestedFormat ) && ( f.colorSpace == requestedColorSpace ); } ); if ( it != formats.end() ) { pickedFormat = *it; break; } } } assert( pickedFormat.colorSpace == vk::ColorSpaceKHR::eSrgbNonlinear ); return pickedFormat; } void setImageLayout( vk::CommandBuffer const & commandBuffer, vk::Image image, vk::Format format, vk::ImageLayout oldImageLayout, vk::ImageLayout newImageLayout ) { vk::AccessFlags sourceAccessMask; switch ( oldImageLayout ) { case vk::ImageLayout::eTransferDstOptimal: sourceAccessMask = vk::AccessFlagBits::eTransferWrite; break; case vk::ImageLayout::ePreinitialized: sourceAccessMask = vk::AccessFlagBits::eHostWrite; break; case vk::ImageLayout::eGeneral: // sourceAccessMask is empty case vk::ImageLayout::eUndefined: break; default: assert( false ); break; } vk::PipelineStageFlags sourceStage; switch ( oldImageLayout ) { case vk::ImageLayout::eGeneral: case vk::ImageLayout::ePreinitialized: sourceStage = vk::PipelineStageFlagBits::eHost; break; case vk::ImageLayout::eTransferDstOptimal: sourceStage = vk::PipelineStageFlagBits::eTransfer; break; case vk::ImageLayout::eUndefined: sourceStage = vk::PipelineStageFlagBits::eTopOfPipe; break; default: assert( false ); break; } vk::AccessFlags destinationAccessMask; switch ( newImageLayout ) { case vk::ImageLayout::eColorAttachmentOptimal: destinationAccessMask = vk::AccessFlagBits::eColorAttachmentWrite; break; case vk::ImageLayout::eDepthStencilAttachmentOptimal: destinationAccessMask = vk::AccessFlagBits::eDepthStencilAttachmentRead | vk::AccessFlagBits::eDepthStencilAttachmentWrite; break; case vk::ImageLayout::eGeneral: // empty destinationAccessMask case vk::ImageLayout::ePresentSrcKHR: break; case vk::ImageLayout::eShaderReadOnlyOptimal: destinationAccessMask = vk::AccessFlagBits::eShaderRead; break; case vk::ImageLayout::eTransferSrcOptimal: destinationAccessMask = vk::AccessFlagBits::eTransferRead; break; case vk::ImageLayout::eTransferDstOptimal: destinationAccessMask = vk::AccessFlagBits::eTransferWrite; break; default: assert( false ); break; } vk::PipelineStageFlags destinationStage; switch ( newImageLayout ) { case vk::ImageLayout::eColorAttachmentOptimal: destinationStage = vk::PipelineStageFlagBits::eColorAttachmentOutput; break; case vk::ImageLayout::eDepthStencilAttachmentOptimal: destinationStage = vk::PipelineStageFlagBits::eEarlyFragmentTests; break; case vk::ImageLayout::eGeneral: destinationStage = vk::PipelineStageFlagBits::eHost; break; case vk::ImageLayout::ePresentSrcKHR: destinationStage = vk::PipelineStageFlagBits::eBottomOfPipe; break; case vk::ImageLayout::eShaderReadOnlyOptimal: destinationStage = vk::PipelineStageFlagBits::eFragmentShader; break; case vk::ImageLayout::eTransferDstOptimal: case vk::ImageLayout::eTransferSrcOptimal: destinationStage = vk::PipelineStageFlagBits::eTransfer; break; default: assert( false ); break; } vk::ImageAspectFlags aspectMask; if ( newImageLayout == vk::ImageLayout::eDepthStencilAttachmentOptimal ) { aspectMask = vk::ImageAspectFlagBits::eDepth; if ( format == vk::Format::eD32SfloatS8Uint || format == vk::Format::eD24UnormS8Uint ) { aspectMask |= vk::ImageAspectFlagBits::eStencil; } } else { aspectMask = vk::ImageAspectFlagBits::eColor; } vk::ImageSubresourceRange imageSubresourceRange( aspectMask, 0, 1, 0, 1 ); vk::ImageMemoryBarrier imageMemoryBarrier( sourceAccessMask, destinationAccessMask, oldImageLayout, newImageLayout, VK_QUEUE_FAMILY_IGNORED, VK_QUEUE_FAMILY_IGNORED, image, imageSubresourceRange ); return commandBuffer.pipelineBarrier( sourceStage, destinationStage, {}, nullptr, nullptr, imageMemoryBarrier ); } void submitAndWait( vk::Device const & device, vk::Queue const & queue, vk::CommandBuffer const & commandBuffer ) { vk::Fence fence = device.createFence( vk::FenceCreateInfo() ); queue.submit( vk::SubmitInfo( 0, nullptr, nullptr, 1, &commandBuffer ), fence ); while ( vk::Result::eTimeout == device.waitForFences( fence, VK_TRUE, vk::su::FenceTimeout ) ) ; device.destroyFence( fence ); } void updateDescriptorSets( vk::Device const & device, vk::DescriptorSet const & descriptorSet, std::vector> const & bufferData, vk::su::TextureData const & textureData, uint32_t bindingOffset ) { std::vector bufferInfos; bufferInfos.reserve( bufferData.size() ); std::vector writeDescriptorSets; writeDescriptorSets.reserve( bufferData.size() + 1 ); uint32_t dstBinding = bindingOffset; for ( auto const & bd : bufferData ) { bufferInfos.emplace_back( std::get<1>( bd ), 0, std::get<2>( bd ) ); writeDescriptorSets.emplace_back( descriptorSet, dstBinding++, 0, 1, std::get<0>( bd ), nullptr, &bufferInfos.back(), &std::get<3>( bd ) ); } vk::DescriptorImageInfo imageInfo( textureData.sampler, textureData.imageData->imageView, vk::ImageLayout::eShaderReadOnlyOptimal ); writeDescriptorSets.emplace_back( descriptorSet, dstBinding, 0, vk::DescriptorType::eCombinedImageSampler, imageInfo, nullptr, nullptr ); device.updateDescriptorSets( writeDescriptorSets, nullptr ); } void updateDescriptorSets( vk::Device const & device, vk::DescriptorSet const & descriptorSet, std::vector> const & bufferData, std::vector const & textureData, uint32_t bindingOffset ) { std::vector bufferInfos; bufferInfos.reserve( bufferData.size() ); std::vector writeDescriptorSets; writeDescriptorSets.reserve( bufferData.size() + ( textureData.empty() ? 0 : 1 ) ); uint32_t dstBinding = bindingOffset; for ( auto const & bd : bufferData ) { bufferInfos.emplace_back( std::get<1>( bd ), 0, std::get<2>( bd ) ); writeDescriptorSets.emplace_back( descriptorSet, dstBinding++, 0, 1, std::get<0>( bd ), nullptr, &bufferInfos.back(), &std::get<3>( bd ) ); } std::vector imageInfos; if ( !textureData.empty() ) { imageInfos.reserve( textureData.size() ); for ( auto const & td : textureData ) { imageInfos.emplace_back( td.sampler, td.imageData->imageView, vk::ImageLayout::eShaderReadOnlyOptimal ); } writeDescriptorSets.emplace_back( descriptorSet, dstBinding, 0, checked_cast( imageInfos.size() ), vk::DescriptorType::eCombinedImageSampler, imageInfos.data(), nullptr, nullptr ); } device.updateDescriptorSets( writeDescriptorSets, nullptr ); } BufferData::BufferData( vk::PhysicalDevice const & physicalDevice, vk::Device const & device, vk::DeviceSize size, vk::BufferUsageFlags usage, vk::MemoryPropertyFlags propertyFlags ) #if !defined( NDEBUG ) : m_size( size ), m_usage( usage ), m_propertyFlags( propertyFlags ) #endif { buffer = device.createBuffer( vk::BufferCreateInfo( vk::BufferCreateFlags(), size, usage ) ); deviceMemory = vk::su::allocateDeviceMemory( device, physicalDevice.getMemoryProperties(), device.getBufferMemoryRequirements( buffer ), propertyFlags ); device.bindBufferMemory( buffer, deviceMemory, 0 ); } DepthBufferData::DepthBufferData( vk::PhysicalDevice const & physicalDevice, vk::Device const & device, vk::Format format, vk::Extent2D const & extent ) : ImageData( physicalDevice, device, format, extent, vk::ImageTiling::eOptimal, vk::ImageUsageFlagBits::eDepthStencilAttachment, vk::ImageLayout::eUndefined, vk::MemoryPropertyFlagBits::eDeviceLocal, vk::ImageAspectFlagBits::eDepth ) { } ImageData::ImageData( vk::PhysicalDevice const & physicalDevice, vk::Device const & device, vk::Format format_, vk::Extent2D const & extent, vk::ImageTiling tiling, vk::ImageUsageFlags usage, vk::ImageLayout initialLayout, vk::MemoryPropertyFlags memoryProperties, vk::ImageAspectFlags aspectMask ) : format( format_ ) { vk::ImageCreateInfo imageCreateInfo( vk::ImageCreateFlags(), vk::ImageType::e2D, format, vk::Extent3D( extent, 1 ), 1, 1, vk::SampleCountFlagBits::e1, tiling, usage | vk::ImageUsageFlagBits::eSampled, vk::SharingMode::eExclusive, {}, initialLayout ); image = device.createImage( imageCreateInfo ); deviceMemory = vk::su::allocateDeviceMemory( device, physicalDevice.getMemoryProperties(), device.getImageMemoryRequirements( image ), memoryProperties ); device.bindImageMemory( image, deviceMemory, 0 ); vk::ImageViewCreateInfo imageViewCreateInfo( {}, image, vk::ImageViewType::e2D, format, {}, { aspectMask, 0, 1, 0, 1 } ); imageView = device.createImageView( imageViewCreateInfo ); } SurfaceData::SurfaceData( vk::Instance const & instance, std::string const & windowName, vk::Extent2D const & extent_ ) : extent( extent_ ), window( vk::su::createWindow( windowName, extent ) ) { VkSurfaceKHR _surface; VkResult err = glfwCreateWindowSurface( static_cast( instance ), window.handle, nullptr, &_surface ); if ( err != VK_SUCCESS ) throw std::runtime_error( "Failed to create window!" ); surface = vk::SurfaceKHR( _surface ); } SwapChainData::SwapChainData( vk::PhysicalDevice const & physicalDevice, vk::Device const & device, vk::SurfaceKHR const & surface, vk::Extent2D const & extent, vk::ImageUsageFlags usage, vk::SwapchainKHR const & oldSwapChain, uint32_t graphicsQueueFamilyIndex, uint32_t presentQueueFamilyIndex ) { vk::SurfaceFormatKHR surfaceFormat = vk::su::pickSurfaceFormat( physicalDevice.getSurfaceFormatsKHR( surface ) ); colorFormat = surfaceFormat.format; vk::SurfaceCapabilitiesKHR surfaceCapabilities = physicalDevice.getSurfaceCapabilitiesKHR( surface ); vk::Extent2D swapchainExtent; if ( surfaceCapabilities.currentExtent.width == std::numeric_limits::max() ) { // If the surface size is undefined, the size is set to the size of the images requested. swapchainExtent.width = clamp( extent.width, surfaceCapabilities.minImageExtent.width, surfaceCapabilities.maxImageExtent.width ); swapchainExtent.height = clamp( extent.height, surfaceCapabilities.minImageExtent.height, surfaceCapabilities.maxImageExtent.height ); } else { // If the surface size is defined, the swap chain size must match swapchainExtent = surfaceCapabilities.currentExtent; } vk::SurfaceTransformFlagBitsKHR preTransform = ( surfaceCapabilities.supportedTransforms & vk::SurfaceTransformFlagBitsKHR::eIdentity ) ? vk::SurfaceTransformFlagBitsKHR::eIdentity : surfaceCapabilities.currentTransform; vk::CompositeAlphaFlagBitsKHR compositeAlpha = ( surfaceCapabilities.supportedCompositeAlpha & vk::CompositeAlphaFlagBitsKHR::ePreMultiplied ) ? vk::CompositeAlphaFlagBitsKHR::ePreMultiplied : ( surfaceCapabilities.supportedCompositeAlpha & vk::CompositeAlphaFlagBitsKHR::ePostMultiplied ) ? vk::CompositeAlphaFlagBitsKHR::ePostMultiplied : ( surfaceCapabilities.supportedCompositeAlpha & vk::CompositeAlphaFlagBitsKHR::eInherit ) ? vk::CompositeAlphaFlagBitsKHR::eInherit : vk::CompositeAlphaFlagBitsKHR::eOpaque; vk::PresentModeKHR presentMode = vk::su::pickPresentMode( physicalDevice.getSurfacePresentModesKHR( surface ) ); vk::SwapchainCreateInfoKHR swapChainCreateInfo( {}, surface, vk::su::clamp( 3u, surfaceCapabilities.minImageCount, surfaceCapabilities.maxImageCount ), colorFormat, surfaceFormat.colorSpace, swapchainExtent, 1, usage, vk::SharingMode::eExclusive, {}, preTransform, compositeAlpha, presentMode, true, oldSwapChain ); if ( graphicsQueueFamilyIndex != presentQueueFamilyIndex ) { uint32_t queueFamilyIndices[2] = { graphicsQueueFamilyIndex, presentQueueFamilyIndex }; // 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::SharingMode::eConcurrent swapChainCreateInfo.imageSharingMode = vk::SharingMode::eConcurrent; swapChainCreateInfo.queueFamilyIndexCount = 2; swapChainCreateInfo.pQueueFamilyIndices = queueFamilyIndices; } swapChain = device.createSwapchainKHR( swapChainCreateInfo ); images = device.getSwapchainImagesKHR( swapChain ); imageViews.reserve( images.size() ); vk::ImageViewCreateInfo imageViewCreateInfo( {}, {}, vk::ImageViewType::e2D, colorFormat, {}, { vk::ImageAspectFlagBits::eColor, 0, 1, 0, 1 } ); for ( auto image : images ) { imageViewCreateInfo.image = image; imageViews.push_back( device.createImageView( imageViewCreateInfo ) ); } } CheckerboardImageGenerator::CheckerboardImageGenerator( std::array const & rgb0, std::array const & rgb1 ) : m_rgb0( rgb0 ), m_rgb1( rgb1 ) { } void CheckerboardImageGenerator::operator()( void * data, vk::Extent2D & extent ) const { // Checkerboard of 16x16 pixel squares uint8_t * pImageMemory = static_cast( data ); for ( uint32_t row = 0; row < extent.height; row++ ) { for ( uint32_t col = 0; col < extent.width; col++ ) { std::array const & rgb = ( ( ( row & 0x10 ) == 0 ) ^ ( ( col & 0x10 ) == 0 ) ) ? m_rgb1 : m_rgb0; pImageMemory[0] = rgb[0]; pImageMemory[1] = rgb[1]; pImageMemory[2] = rgb[2]; pImageMemory[3] = 255; pImageMemory += 4; } } } MonochromeImageGenerator::MonochromeImageGenerator( std::array const & rgb ) : m_rgb( rgb ) {} void MonochromeImageGenerator::operator()( void * data, vk::Extent2D const & extent ) const { // fill in with the monochrome color unsigned char * pImageMemory = static_cast( data ); for ( uint32_t row = 0; row < extent.height; row++ ) { for ( uint32_t col = 0; col < extent.width; col++ ) { pImageMemory[0] = m_rgb[0]; pImageMemory[1] = m_rgb[1]; pImageMemory[2] = m_rgb[2]; pImageMemory[3] = 255; pImageMemory += 4; } } } PixelsImageGenerator::PixelsImageGenerator( vk::Extent2D const & extent, size_t channels, unsigned char const * pixels ) : m_extent( extent ), m_channels( channels ), m_pixels( pixels ) { assert( m_channels == 4 ); } void PixelsImageGenerator::operator()( void * data, vk::Extent2D const & extent ) const { assert( extent == m_extent ); memcpy( data, m_pixels, extent.width * extent.height * m_channels ); } TextureData::TextureData( vk::PhysicalDevice const & physicalDevice, vk::Device const & device, vk::Extent2D const & extent_, vk::ImageUsageFlags usageFlags, vk::FormatFeatureFlags formatFeatureFlags, bool anisotropyEnable, bool forceStaging ) : format( vk::Format::eR8G8B8A8Unorm ), extent( extent_ ) { vk::FormatProperties formatProperties = physicalDevice.getFormatProperties( format ); formatFeatureFlags |= vk::FormatFeatureFlagBits::eSampledImage; needsStaging = forceStaging || ( ( formatProperties.linearTilingFeatures & formatFeatureFlags ) != formatFeatureFlags ); vk::ImageTiling imageTiling; vk::ImageLayout initialLayout; vk::MemoryPropertyFlags requirements; if ( needsStaging ) { assert( ( formatProperties.optimalTilingFeatures & formatFeatureFlags ) == formatFeatureFlags ); stagingBufferData = std::unique_ptr( new BufferData( physicalDevice, device, extent.width * extent.height * 4, vk::BufferUsageFlagBits::eTransferSrc ) ); imageTiling = vk::ImageTiling::eOptimal; usageFlags |= vk::ImageUsageFlagBits::eTransferDst; initialLayout = vk::ImageLayout::eUndefined; } else { imageTiling = vk::ImageTiling::eLinear; initialLayout = vk::ImageLayout::ePreinitialized; requirements = vk::MemoryPropertyFlagBits::eHostCoherent | vk::MemoryPropertyFlagBits::eHostVisible; } imageData = std::unique_ptr( new ImageData( physicalDevice, device, format, extent, imageTiling, usageFlags | vk::ImageUsageFlagBits::eSampled, initialLayout, requirements, vk::ImageAspectFlagBits::eColor ) ); sampler = device.createSampler( vk::SamplerCreateInfo( vk::SamplerCreateFlags(), vk::Filter::eLinear, vk::Filter::eLinear, vk::SamplerMipmapMode::eLinear, vk::SamplerAddressMode::eRepeat, vk::SamplerAddressMode::eRepeat, vk::SamplerAddressMode::eRepeat, 0.0f, anisotropyEnable, 16.0f, false, vk::CompareOp::eNever, 0.0f, 0.0f, vk::BorderColor::eFloatOpaqueBlack ) ); } UUID::UUID( uint8_t const data[VK_UUID_SIZE] ) { memcpy( m_data, data, VK_UUID_SIZE * sizeof( uint8_t ) ); } WindowData::WindowData( GLFWwindow * wnd, std::string const & name, vk::Extent2D const & extent ) : handle{ wnd }, name{ name }, extent{ extent } {} WindowData::WindowData( WindowData && other ) : handle{}, name{}, extent{} { std::swap( handle, other.handle ); std::swap( name, other.name ); std::swap( extent, other.extent ); } WindowData::~WindowData() noexcept { glfwDestroyWindow( handle ); } WindowData createWindow( std::string const & windowName, vk::Extent2D const & extent ) { struct glfwContext { glfwContext() { glfwInit(); glfwSetErrorCallback( []( int error, const char * msg ) { std::cerr << "glfw: " << "(" << error << ") " << msg << std::endl; } ); } ~glfwContext() { glfwTerminate(); } }; static auto glfwCtx = glfwContext(); (void)glfwCtx; glfwWindowHint( GLFW_CLIENT_API, GLFW_NO_API ); GLFWwindow * window = glfwCreateWindow( extent.width, extent.height, windowName.c_str(), nullptr, nullptr ); return WindowData( window, windowName, extent ); } vk::DebugUtilsMessengerCreateInfoEXT makeDebugUtilsMessengerCreateInfoEXT() { return { {}, vk::DebugUtilsMessageSeverityFlagBitsEXT::eWarning | vk::DebugUtilsMessageSeverityFlagBitsEXT::eError, vk::DebugUtilsMessageTypeFlagBitsEXT::eGeneral | vk::DebugUtilsMessageTypeFlagBitsEXT::ePerformance | vk::DebugUtilsMessageTypeFlagBitsEXT::eValidation, &vk::su::debugUtilsMessengerCallback }; } #if defined( NDEBUG ) vk::StructureChain #else vk::StructureChain #endif makeInstanceCreateInfoChain( vk::ApplicationInfo const & applicationInfo, std::vector const & layers, std::vector const & extensions ) { #if defined( NDEBUG ) // in non-debug mode just use the InstanceCreateInfo for instance creation vk::StructureChain instanceCreateInfo( { {}, &applicationInfo, layers, extensions } ); #else // in debug mode, addionally use the debugUtilsMessengerCallback in instance creation! vk::DebugUtilsMessageSeverityFlagsEXT severityFlags( vk::DebugUtilsMessageSeverityFlagBitsEXT::eWarning | vk::DebugUtilsMessageSeverityFlagBitsEXT::eError ); vk::DebugUtilsMessageTypeFlagsEXT messageTypeFlags( vk::DebugUtilsMessageTypeFlagBitsEXT::eGeneral | vk::DebugUtilsMessageTypeFlagBitsEXT::ePerformance | vk::DebugUtilsMessageTypeFlagBitsEXT::eValidation ); vk::StructureChain instanceCreateInfo( { {}, &applicationInfo, layers, extensions }, { {}, severityFlags, messageTypeFlags, &vk::su::debugUtilsMessengerCallback } ); #endif return instanceCreateInfo; } } // namespace su } // namespace vk std::ostream & operator<<( std::ostream & os, vk::su::UUID const & uuid ) { os << std::setfill( '0' ) << std::hex; for ( uint32_t j = 0; j < VK_UUID_SIZE; ++j ) { os << std::setw( 2 ) << static_cast( uuid.m_data[j] ); if ( j == 3 || j == 5 || j == 7 || j == 9 ) { std::cout << '-'; } } os << std::setfill( ' ' ) << std::dec; return os; }