| /* |
| * Copyright (c) 2015-2019 The Khronos Group Inc. |
| * Copyright (c) 2015-2019 Valve Corporation |
| * Copyright (c) 2015-2019 LunarG, Inc. |
| * Copyright (c) 2015-2019 Google, Inc. |
| * |
| * 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. |
| * |
| * Author: Courtney Goeltzenleuchter <[email protected]> |
| * Author: Tony Barbour <[email protected]> |
| * Author: Dave Houlton <[email protected]> |
| */ |
| |
| #include "vkrenderframework.h" |
| #include "vk_format_utils.h" |
| |
| #define ARRAY_SIZE(a) (sizeof(a) / sizeof(a[0])) |
| #define GET_DEVICE_PROC_ADDR(dev, entrypoint) \ |
| { \ |
| fp##entrypoint = (PFN_vk##entrypoint)vkGetDeviceProcAddr(dev, "vk" #entrypoint); \ |
| assert(fp##entrypoint != NULL); \ |
| } |
| |
| VkRenderFramework::VkRenderFramework() |
| : inst(VK_NULL_HANDLE), |
| m_device(NULL), |
| m_commandPool(VK_NULL_HANDLE), |
| m_commandBuffer(NULL), |
| m_renderPass(VK_NULL_HANDLE), |
| m_framebuffer(VK_NULL_HANDLE), |
| m_surface(VK_NULL_HANDLE), |
| m_swapchain(VK_NULL_HANDLE), |
| m_addRenderPassSelfDependency(false), |
| m_width(256.0), // default window width |
| m_height(256.0), // default window height |
| m_render_target_fmt(VK_FORMAT_R8G8B8A8_UNORM), |
| m_depth_stencil_fmt(VK_FORMAT_UNDEFINED), |
| m_clear_via_load_op(true), |
| m_depth_clear_color(1.0), |
| m_stencil_clear_color(0), |
| m_depthStencil(NULL), |
| m_CreateDebugReportCallback(VK_NULL_HANDLE), |
| m_DestroyDebugReportCallback(VK_NULL_HANDLE), |
| m_globalMsgCallback(VK_NULL_HANDLE), |
| m_devMsgCallback(VK_NULL_HANDLE) { |
| memset(&m_renderPassBeginInfo, 0, sizeof(m_renderPassBeginInfo)); |
| m_renderPassBeginInfo.sType = VK_STRUCTURE_TYPE_RENDER_PASS_BEGIN_INFO; |
| |
| // clear the back buffer to dark grey |
| m_clear_color.float32[0] = 0.25f; |
| m_clear_color.float32[1] = 0.25f; |
| m_clear_color.float32[2] = 0.25f; |
| m_clear_color.float32[3] = 0.0f; |
| } |
| |
| VkRenderFramework::~VkRenderFramework() { ShutdownFramework(); } |
| |
| VkPhysicalDevice VkRenderFramework::gpu() { |
| EXPECT_NE((VkInstance)0, inst); // Invalid to request gpu before instance exists |
| return objs[0]; |
| } |
| |
| // Return true if layer name is found and spec+implementation values are >= requested values |
| bool VkRenderFramework::InstanceLayerSupported(const char *name, uint32_t spec, uint32_t implementation) { |
| uint32_t layer_count = 0; |
| std::vector<VkLayerProperties> layer_props; |
| |
| VkResult res = vkEnumerateInstanceLayerProperties(&layer_count, NULL); |
| if (VK_SUCCESS != res) return false; |
| if (0 == layer_count) return false; |
| |
| layer_props.resize(layer_count); |
| res = vkEnumerateInstanceLayerProperties(&layer_count, layer_props.data()); |
| if (VK_SUCCESS != res) return false; |
| |
| for (auto &it : layer_props) { |
| if (0 == strncmp(name, it.layerName, VK_MAX_EXTENSION_NAME_SIZE)) { |
| return ((it.specVersion >= spec) && (it.implementationVersion >= implementation)); |
| } |
| } |
| return false; |
| } |
| |
| // Enable device profile as last layer on stack overriding devsim if there, or return if not available |
| bool VkRenderFramework::EnableDeviceProfileLayer() { |
| if (InstanceLayerSupported("VK_LAYER_LUNARG_device_profile_api")) { |
| if (VkTestFramework::m_devsim_layer) { |
| assert(0 == strcmp(m_instance_layer_names.back(), "VK_LAYER_LUNARG_device_simulation")); |
| m_instance_layer_names.pop_back(); |
| m_instance_layer_names.push_back("VK_LAYER_LUNARG_device_profile_api"); |
| } else { |
| m_instance_layer_names.push_back("VK_LAYER_LUNARG_device_profile_api"); |
| } |
| } else { |
| printf(" Did not find VK_LAYER_LUNARG_device_profile_api layer; skipped.\n"); |
| return false; |
| } |
| return true; |
| } |
| |
| // Return true if extension name is found and spec value is >= requested spec value |
| bool VkRenderFramework::InstanceExtensionSupported(const char *ext_name, uint32_t spec) { |
| uint32_t ext_count = 0; |
| std::vector<VkExtensionProperties> ext_props; |
| VkResult res = vkEnumerateInstanceExtensionProperties(nullptr, &ext_count, nullptr); |
| if (VK_SUCCESS != res) return false; |
| if (0 == ext_count) return false; |
| |
| ext_props.resize(ext_count); |
| res = vkEnumerateInstanceExtensionProperties(nullptr, &ext_count, ext_props.data()); |
| if (VK_SUCCESS != res) return false; |
| |
| for (auto &it : ext_props) { |
| if (0 == strncmp(ext_name, it.extensionName, VK_MAX_EXTENSION_NAME_SIZE)) { |
| return (it.specVersion >= spec); |
| } |
| } |
| return false; |
| } |
| |
| // Return true if instance exists and extension name is in the list |
| bool VkRenderFramework::InstanceExtensionEnabled(const char *ext_name) { |
| if (!inst) return false; |
| |
| bool ext_found = false; |
| for (auto ext : m_instance_extension_names) { |
| if (!strcmp(ext, ext_name)) { |
| ext_found = true; |
| break; |
| } |
| } |
| return ext_found; |
| } |
| |
| // Return true if extension name is found and spec value is >= requested spec value |
| bool VkRenderFramework::DeviceExtensionSupported(VkPhysicalDevice dev, const char *layer, const char *ext_name, uint32_t spec) { |
| if (!inst) { |
| EXPECT_NE((VkInstance)0, inst); // Complain, not cool without an instance |
| return false; |
| } |
| uint32_t ext_count = 0; |
| std::vector<VkExtensionProperties> ext_props; |
| VkResult res = vkEnumerateDeviceExtensionProperties(dev, layer, &ext_count, nullptr); |
| if (VK_SUCCESS != res) return false; |
| if (0 == ext_count) return false; |
| |
| ext_props.resize(ext_count); |
| res = vkEnumerateDeviceExtensionProperties(dev, layer, &ext_count, ext_props.data()); |
| if (VK_SUCCESS != res) return false; |
| |
| for (auto &it : ext_props) { |
| if (0 == strncmp(ext_name, it.extensionName, VK_MAX_EXTENSION_NAME_SIZE)) { |
| return (it.specVersion >= spec); |
| } |
| } |
| return false; |
| } |
| |
| // Return true if device is created and extension name is found in the list |
| bool VkRenderFramework::DeviceExtensionEnabled(const char *ext_name) { |
| if (NULL == m_device) return false; |
| |
| bool ext_found = false; |
| for (auto ext : m_device_extension_names) { |
| if (!strcmp(ext, ext_name)) { |
| ext_found = true; |
| break; |
| } |
| } |
| return ext_found; |
| } |
| |
| // WARNING: The DevSim layer can override the properties that are tested here, making the result of |
| // this function dubious when DevSim is active. |
| bool VkRenderFramework::DeviceIsMockICD() { |
| VkPhysicalDeviceProperties props = vk_testing::PhysicalDevice(gpu()).properties(); |
| if ((props.vendorID == 0xba5eba11) && (props.deviceID == 0xf005ba11) && (0 == strcmp("Vulkan Mock Device", props.deviceName))) { |
| return true; |
| } |
| return false; |
| } |
| |
| // Some tests may need to be skipped if the devsim layer is in use. |
| bool VkRenderFramework::DeviceSimulation() { return m_devsim_layer; } |
| |
| // Render into a RenderTarget and read the pixels back to see if the device can really draw. |
| // Note: This cannot be called from inside an initialized VkRenderFramework because frameworks cannot be "nested". |
| // It is best to call it before "Init()". |
| bool VkRenderFramework::DeviceCanDraw() { |
| InitFramework(NULL, NULL); |
| InitState(NULL, NULL, 0); |
| InitViewport(); |
| InitRenderTarget(); |
| |
| // Draw a triangle that covers the entire viewport. |
| char const *vsSource = |
| "#version 450\n" |
| "\n" |
| "vec2 vertices[3];\n" |
| "void main() { \n" |
| " vertices[0] = vec2(-10.0, -10.0);\n" |
| " vertices[1] = vec2( 10.0, -10.0);\n" |
| " vertices[2] = vec2( 0.0, 10.0);\n" |
| " gl_Position = vec4(vertices[gl_VertexIndex % 3], 0.0, 1.0);\n" |
| "}\n"; |
| // Draw with a solid color. |
| char const *fsSource = |
| "#version 450\n" |
| "\n" |
| "layout(location=0) out vec4 color;\n" |
| "void main() {\n" |
| " color = vec4(32.0/255.0);\n" |
| "}\n"; |
| VkShaderObj *vs = new VkShaderObj(m_device, vsSource, VK_SHADER_STAGE_VERTEX_BIT, this); |
| VkShaderObj *fs = new VkShaderObj(m_device, fsSource, VK_SHADER_STAGE_FRAGMENT_BIT, this); |
| |
| VkPipelineObj *pipe = new VkPipelineObj(m_device); |
| pipe->AddShader(vs); |
| pipe->AddShader(fs); |
| pipe->AddDefaultColorAttachment(); |
| |
| VkDescriptorSetObj *descriptorSet = new VkDescriptorSetObj(m_device); |
| descriptorSet->CreateVKDescriptorSet(m_commandBuffer); |
| |
| pipe->CreateVKPipeline(descriptorSet->GetPipelineLayout(), renderPass()); |
| |
| m_commandBuffer->begin(); |
| m_commandBuffer->BeginRenderPass(m_renderPassBeginInfo); |
| |
| vkCmdBindPipeline(m_commandBuffer->handle(), VK_PIPELINE_BIND_POINT_GRAPHICS, pipe->handle()); |
| m_commandBuffer->BindDescriptorSet(*descriptorSet); |
| |
| VkViewport viewport = m_viewports[0]; |
| VkRect2D scissors = m_scissors[0]; |
| |
| vkCmdSetViewport(m_commandBuffer->handle(), 0, 1, &viewport); |
| vkCmdSetScissor(m_commandBuffer->handle(), 0, 1, &scissors); |
| |
| vkCmdDraw(m_commandBuffer->handle(), 3, 1, 0, 0); |
| |
| m_commandBuffer->EndRenderPass(); |
| m_commandBuffer->end(); |
| |
| VkSubmitInfo submit_info = {}; |
| submit_info.sType = VK_STRUCTURE_TYPE_SUBMIT_INFO; |
| submit_info.commandBufferCount = 1; |
| submit_info.pCommandBuffers = &m_commandBuffer->handle(); |
| |
| vkQueueSubmit(m_device->m_queue, 1, &submit_info, VK_NULL_HANDLE); |
| vkQueueWaitIdle(m_device->m_queue); |
| |
| auto pixels = m_renderTargets[0]->Read(); |
| |
| delete descriptorSet; |
| delete pipe; |
| delete fs; |
| delete vs; |
| ShutdownFramework(); |
| return pixels[0][0] == 0x20202020; |
| } |
| |
| void VkRenderFramework::InitFramework(PFN_vkDebugReportCallbackEXT dbgFunction, void *userData, void *instance_pnext) { |
| // Only enable device profile layer by default if devsim is not enabled |
| if (!VkTestFramework::m_devsim_layer && InstanceLayerSupported("VK_LAYER_LUNARG_device_profile_api")) { |
| m_instance_layer_names.push_back("VK_LAYER_LUNARG_device_profile_api"); |
| } |
| |
| // Assert not already initialized |
| ASSERT_EQ((VkInstance)0, inst); |
| |
| // Remove any unsupported layer names from list |
| for (auto layer = m_instance_layer_names.begin(); layer != m_instance_layer_names.end();) { |
| if (!InstanceLayerSupported(*layer)) { |
| ADD_FAILURE() << "InitFramework(): Requested layer " << *layer << " was not found. Disabled."; |
| layer = m_instance_layer_names.erase(layer); |
| } else { |
| ++layer; |
| } |
| } |
| |
| // Remove any unsupported instance extension names from list |
| for (auto ext = m_instance_extension_names.begin(); ext != m_instance_extension_names.end();) { |
| if (!InstanceExtensionSupported(*ext)) { |
| ADD_FAILURE() << "InitFramework(): Requested extension " << *ext << " was not found. Disabled."; |
| ext = m_instance_extension_names.erase(ext); |
| } else { |
| ++ext; |
| } |
| } |
| |
| VkInstanceCreateInfo instInfo = {}; |
| VkResult U_ASSERT_ONLY err; |
| |
| instInfo.sType = VK_STRUCTURE_TYPE_INSTANCE_CREATE_INFO; |
| instInfo.pNext = instance_pnext; |
| instInfo.pApplicationInfo = &app_info; |
| instInfo.enabledLayerCount = m_instance_layer_names.size(); |
| instInfo.ppEnabledLayerNames = m_instance_layer_names.data(); |
| instInfo.enabledExtensionCount = m_instance_extension_names.size(); |
| instInfo.ppEnabledExtensionNames = m_instance_extension_names.data(); |
| |
| VkDebugReportCallbackCreateInfoEXT dbgCreateInfo; |
| if (dbgFunction) { |
| // Enable create time debug messages |
| memset(&dbgCreateInfo, 0, sizeof(dbgCreateInfo)); |
| dbgCreateInfo.sType = VK_STRUCTURE_TYPE_DEBUG_REPORT_CREATE_INFO_EXT; |
| dbgCreateInfo.flags = |
| VK_DEBUG_REPORT_ERROR_BIT_EXT | VK_DEBUG_REPORT_WARNING_BIT_EXT | VK_DEBUG_REPORT_PERFORMANCE_WARNING_BIT_EXT; |
| dbgCreateInfo.pfnCallback = dbgFunction; |
| dbgCreateInfo.pUserData = userData; |
| |
| dbgCreateInfo.pNext = instInfo.pNext; |
| instInfo.pNext = &dbgCreateInfo; |
| } |
| |
| err = vkCreateInstance(&instInfo, NULL, &this->inst); |
| ASSERT_VK_SUCCESS(err); |
| |
| err = vkEnumeratePhysicalDevices(inst, &this->gpu_count, NULL); |
| ASSERT_LE(this->gpu_count, ARRAY_SIZE(objs)) << "Too many gpus"; |
| ASSERT_VK_SUCCESS(err); |
| err = vkEnumeratePhysicalDevices(inst, &this->gpu_count, objs); |
| ASSERT_VK_SUCCESS(err); |
| ASSERT_GE(this->gpu_count, (uint32_t)1) << "No GPU available"; |
| if (dbgFunction) { |
| m_CreateDebugReportCallback = |
| (PFN_vkCreateDebugReportCallbackEXT)vkGetInstanceProcAddr(this->inst, "vkCreateDebugReportCallbackEXT"); |
| ASSERT_NE(m_CreateDebugReportCallback, (PFN_vkCreateDebugReportCallbackEXT)NULL) |
| << "Did not get function pointer for CreateDebugReportCallback"; |
| if (m_CreateDebugReportCallback) { |
| dbgCreateInfo.pNext = nullptr; // clean up from usage in CreateInstance above |
| err = m_CreateDebugReportCallback(this->inst, &dbgCreateInfo, NULL, &m_globalMsgCallback); |
| ASSERT_VK_SUCCESS(err); |
| |
| m_DestroyDebugReportCallback = |
| (PFN_vkDestroyDebugReportCallbackEXT)vkGetInstanceProcAddr(this->inst, "vkDestroyDebugReportCallbackEXT"); |
| ASSERT_NE(m_DestroyDebugReportCallback, (PFN_vkDestroyDebugReportCallbackEXT)NULL) |
| << "Did not get function pointer for DestroyDebugReportCallback"; |
| m_DebugReportMessage = (PFN_vkDebugReportMessageEXT)vkGetInstanceProcAddr(this->inst, "vkDebugReportMessageEXT"); |
| ASSERT_NE(m_DebugReportMessage, (PFN_vkDebugReportMessageEXT)NULL) |
| << "Did not get function pointer for DebugReportMessage"; |
| } |
| } |
| } |
| |
| void VkRenderFramework::ShutdownFramework() { |
| // Nothing to shut down without a VkInstance |
| if (!this->inst) return; |
| |
| delete m_commandBuffer; |
| m_commandBuffer = nullptr; |
| delete m_commandPool; |
| m_commandPool = nullptr; |
| if (m_framebuffer) vkDestroyFramebuffer(device(), m_framebuffer, NULL); |
| m_framebuffer = VK_NULL_HANDLE; |
| if (m_renderPass) vkDestroyRenderPass(device(), m_renderPass, NULL); |
| m_renderPass = VK_NULL_HANDLE; |
| |
| if (m_globalMsgCallback) m_DestroyDebugReportCallback(this->inst, m_globalMsgCallback, NULL); |
| m_globalMsgCallback = VK_NULL_HANDLE; |
| if (m_devMsgCallback) m_DestroyDebugReportCallback(this->inst, m_devMsgCallback, NULL); |
| m_devMsgCallback = VK_NULL_HANDLE; |
| |
| m_renderTargets.clear(); |
| |
| delete m_depthStencil; |
| m_depthStencil = nullptr; |
| |
| // reset the driver |
| delete m_device; |
| m_device = nullptr; |
| |
| if (this->inst) vkDestroyInstance(this->inst, NULL); |
| this->inst = (VkInstance)0; // In case we want to re-initialize |
| } |
| |
| void VkRenderFramework::GetPhysicalDeviceFeatures(VkPhysicalDeviceFeatures *features) { |
| if (NULL == m_device) { |
| VkDeviceObj *temp_device = new VkDeviceObj(0, objs[0], m_device_extension_names); |
| *features = temp_device->phy().features(); |
| delete (temp_device); |
| } else { |
| *features = m_device->phy().features(); |
| } |
| } |
| |
| void VkRenderFramework::GetPhysicalDeviceProperties(VkPhysicalDeviceProperties *props) { |
| *props = vk_testing::PhysicalDevice(gpu()).properties(); |
| } |
| |
| void VkRenderFramework::InitState(VkPhysicalDeviceFeatures *features, void *create_device_pnext, |
| const VkCommandPoolCreateFlags flags) { |
| // Remove any unsupported device extension names from list |
| for (auto ext = m_device_extension_names.begin(); ext != m_device_extension_names.end();) { |
| if (!DeviceExtensionSupported(objs[0], nullptr, *ext)) { |
| bool found = false; |
| for (auto layer = m_instance_layer_names.begin(); layer != m_instance_layer_names.end(); ++layer) { |
| if (DeviceExtensionSupported(objs[0], *layer, *ext)) { |
| found = true; |
| break; |
| } |
| } |
| if (!found) { |
| ADD_FAILURE() << "InitState(): The requested device extension " << *ext << " was not found. Disabled."; |
| ext = m_device_extension_names.erase(ext); |
| } else { |
| ++ext; |
| } |
| } else { |
| ++ext; |
| } |
| } |
| |
| m_device = new VkDeviceObj(0, objs[0], m_device_extension_names, features, create_device_pnext); |
| m_device->SetDeviceQueue(); |
| |
| m_depthStencil = new VkDepthStencilObj(m_device); |
| |
| m_render_target_fmt = VkTestFramework::GetFormat(inst, m_device); |
| |
| m_lineWidth = 1.0f; |
| |
| m_depthBiasConstantFactor = 0.0f; |
| m_depthBiasClamp = 0.0f; |
| m_depthBiasSlopeFactor = 0.0f; |
| |
| m_blendConstants[0] = 1.0f; |
| m_blendConstants[1] = 1.0f; |
| m_blendConstants[2] = 1.0f; |
| m_blendConstants[3] = 1.0f; |
| |
| m_minDepthBounds = 0.f; |
| m_maxDepthBounds = 1.f; |
| |
| m_compareMask = 0xff; |
| m_writeMask = 0xff; |
| m_reference = 0; |
| |
| m_commandPool = new VkCommandPoolObj(m_device, m_device->graphics_queue_node_index_, flags); |
| |
| m_commandBuffer = new VkCommandBufferObj(m_device, m_commandPool); |
| } |
| |
| void VkRenderFramework::InitViewport(float width, float height) { |
| VkViewport viewport; |
| VkRect2D scissor; |
| viewport.x = 0; |
| viewport.y = 0; |
| viewport.width = 1.f * width; |
| viewport.height = 1.f * height; |
| viewport.minDepth = 0.f; |
| viewport.maxDepth = 1.f; |
| m_viewports.push_back(viewport); |
| |
| scissor.extent.width = (int32_t)width; |
| scissor.extent.height = (int32_t)height; |
| scissor.offset.x = 0; |
| scissor.offset.y = 0; |
| m_scissors.push_back(scissor); |
| |
| m_width = width; |
| m_height = height; |
| } |
| |
| void VkRenderFramework::InitViewport() { InitViewport(m_width, m_height); } |
| |
| bool VkRenderFramework::InitSurface() { return InitSurface(m_width, m_height); } |
| |
| #ifdef VK_USE_PLATFORM_WIN32_KHR |
| LRESULT CALLBACK WindowProc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam) { |
| return DefWindowProc(hwnd, uMsg, wParam, lParam); |
| } |
| #endif // VK_USE_PLATFORM_WIN32_KHR |
| |
| bool VkRenderFramework::InitSurface(float width, float height) { |
| #if defined(VK_USE_PLATFORM_WIN32_KHR) |
| HINSTANCE window_instance = GetModuleHandle(nullptr); |
| const char class_name[] = "test"; |
| WNDCLASS wc = {}; |
| wc.lpfnWndProc = WindowProc; |
| wc.hInstance = window_instance; |
| wc.lpszClassName = class_name; |
| RegisterClass(&wc); |
| HWND window = CreateWindowEx(0, class_name, 0, 0, 0, 0, (int)m_width, (int)m_height, NULL, NULL, window_instance, NULL); |
| ShowWindow(window, SW_HIDE); |
| |
| VkWin32SurfaceCreateInfoKHR surface_create_info = {}; |
| surface_create_info.sType = VK_STRUCTURE_TYPE_WIN32_SURFACE_CREATE_INFO_KHR; |
| surface_create_info.hinstance = window_instance; |
| surface_create_info.hwnd = window; |
| VkResult err = vkCreateWin32SurfaceKHR(instance(), &surface_create_info, nullptr, &m_surface); |
| if (err != VK_SUCCESS) return false; |
| #endif |
| |
| #if defined(VK_USE_PLATFORM_ANDROID_KHR) && defined(VALIDATION_APK) |
| VkAndroidSurfaceCreateInfoKHR surface_create_info = {}; |
| surface_create_info.sType = VK_STRUCTURE_TYPE_ANDROID_SURFACE_CREATE_INFO_KHR; |
| surface_create_info.window = VkTestFramework::window; |
| VkResult err = vkCreateAndroidSurfaceKHR(instance(), &surface_create_info, nullptr, &m_surface); |
| if (err != VK_SUCCESS) return false; |
| #endif |
| |
| #if defined(VK_USE_PLATFORM_XLIB_KHR) |
| Display *dpy = XOpenDisplay(NULL); |
| if (dpy) { |
| int s = DefaultScreen(dpy); |
| Window window = XCreateSimpleWindow(dpy, RootWindow(dpy, s), 0, 0, (int)m_width, (int)m_height, 1, BlackPixel(dpy, s), |
| WhitePixel(dpy, s)); |
| VkXlibSurfaceCreateInfoKHR surface_create_info = {}; |
| surface_create_info.sType = VK_STRUCTURE_TYPE_XLIB_SURFACE_CREATE_INFO_KHR; |
| surface_create_info.dpy = dpy; |
| surface_create_info.window = window; |
| VkResult err = vkCreateXlibSurfaceKHR(instance(), &surface_create_info, nullptr, &m_surface); |
| if (err != VK_SUCCESS) return false; |
| } |
| #endif |
| |
| #if defined(VK_USE_PLATFORM_XCB_KHR) |
| if (m_surface == VK_NULL_HANDLE) { |
| xcb_connection_t *connection = xcb_connect(NULL, NULL); |
| if (connection) { |
| xcb_window_t window = xcb_generate_id(connection); |
| VkXcbSurfaceCreateInfoKHR surface_create_info = {}; |
| surface_create_info.sType = VK_STRUCTURE_TYPE_XCB_SURFACE_CREATE_INFO_KHR; |
| surface_create_info.connection = connection; |
| surface_create_info.window = window; |
| VkResult err = vkCreateXcbSurfaceKHR(instance(), &surface_create_info, nullptr, &m_surface); |
| if (err != VK_SUCCESS) return false; |
| } |
| } |
| #endif |
| |
| return (m_surface == VK_NULL_HANDLE) ? false : true; |
| } |
| |
| bool VkRenderFramework::InitSwapchain(VkImageUsageFlags imageUsage, VkSurfaceTransformFlagBitsKHR preTransform) { |
| if (InitSurface()) { |
| return InitSwapchain(m_surface, imageUsage, preTransform); |
| } |
| return false; |
| } |
| |
| bool VkRenderFramework::InitSwapchain(VkSurfaceKHR &surface, VkImageUsageFlags imageUsage, |
| VkSurfaceTransformFlagBitsKHR preTransform) { |
| for (size_t i = 0; i < m_device->queue_props.size(); ++i) { |
| VkBool32 presentSupport = false; |
| vkGetPhysicalDeviceSurfaceSupportKHR(m_device->phy().handle(), i, surface, &presentSupport); |
| } |
| |
| VkSurfaceCapabilitiesKHR capabilities; |
| vkGetPhysicalDeviceSurfaceCapabilitiesKHR(m_device->phy().handle(), surface, &capabilities); |
| |
| uint32_t format_count; |
| vkGetPhysicalDeviceSurfaceFormatsKHR(m_device->phy().handle(), surface, &format_count, nullptr); |
| std::vector<VkSurfaceFormatKHR> formats; |
| if (format_count != 0) { |
| formats.resize(format_count); |
| vkGetPhysicalDeviceSurfaceFormatsKHR(m_device->phy().handle(), surface, &format_count, formats.data()); |
| } |
| |
| uint32_t present_mode_count; |
| vkGetPhysicalDeviceSurfacePresentModesKHR(m_device->phy().handle(), surface, &present_mode_count, nullptr); |
| std::vector<VkPresentModeKHR> present_modes; |
| if (present_mode_count != 0) { |
| present_modes.resize(present_mode_count); |
| vkGetPhysicalDeviceSurfacePresentModesKHR(m_device->phy().handle(), surface, &present_mode_count, present_modes.data()); |
| } |
| |
| VkSwapchainCreateInfoKHR swapchain_create_info = {}; |
| swapchain_create_info.sType = VK_STRUCTURE_TYPE_SWAPCHAIN_CREATE_INFO_KHR; |
| swapchain_create_info.pNext = 0; |
| swapchain_create_info.surface = surface; |
| swapchain_create_info.minImageCount = capabilities.minImageCount; |
| swapchain_create_info.imageFormat = formats[0].format; |
| swapchain_create_info.imageColorSpace = formats[0].colorSpace; |
| swapchain_create_info.imageExtent = {capabilities.minImageExtent.width, capabilities.minImageExtent.height}; |
| swapchain_create_info.imageArrayLayers = capabilities.maxImageArrayLayers; |
| swapchain_create_info.imageUsage = imageUsage; |
| swapchain_create_info.imageSharingMode = VK_SHARING_MODE_EXCLUSIVE; |
| swapchain_create_info.preTransform = preTransform; |
| #ifdef VK_USE_PLATFORM_ANDROID_KHR |
| swapchain_create_info.compositeAlpha = VK_COMPOSITE_ALPHA_INHERIT_BIT_KHR; |
| #else |
| swapchain_create_info.compositeAlpha = VK_COMPOSITE_ALPHA_OPAQUE_BIT_KHR; |
| #endif |
| swapchain_create_info.presentMode = present_modes[0]; |
| swapchain_create_info.clipped = VK_FALSE; |
| swapchain_create_info.oldSwapchain = 0; |
| |
| VkResult err = vkCreateSwapchainKHR(device(), &swapchain_create_info, nullptr, &m_swapchain); |
| if (err != VK_SUCCESS) { |
| return false; |
| } |
| uint32_t imageCount = 0; |
| vkGetSwapchainImagesKHR(device(), m_swapchain, &imageCount, nullptr); |
| std::vector<VkImage> swapchainImages; |
| swapchainImages.resize(imageCount); |
| vkGetSwapchainImagesKHR(device(), m_swapchain, &imageCount, swapchainImages.data()); |
| return true; |
| } |
| |
| void VkRenderFramework::DestroySwapchain() { |
| if (m_swapchain != VK_NULL_HANDLE) { |
| vkDestroySwapchainKHR(device(), m_swapchain, nullptr); |
| m_swapchain = VK_NULL_HANDLE; |
| } |
| if (m_surface != VK_NULL_HANDLE) { |
| vkDestroySurfaceKHR(instance(), m_surface, nullptr); |
| m_surface = VK_NULL_HANDLE; |
| } |
| } |
| |
| void VkRenderFramework::InitRenderTarget() { InitRenderTarget(1); } |
| |
| void VkRenderFramework::InitRenderTarget(uint32_t targets) { InitRenderTarget(targets, NULL); } |
| |
| void VkRenderFramework::InitRenderTarget(VkImageView *dsBinding) { InitRenderTarget(1, dsBinding); } |
| |
| void VkRenderFramework::InitRenderTarget(uint32_t targets, VkImageView *dsBinding) { |
| std::vector<VkAttachmentDescription> attachments; |
| std::vector<VkAttachmentReference> color_references; |
| std::vector<VkImageView> bindings; |
| attachments.reserve(targets + 1); // +1 for dsBinding |
| color_references.reserve(targets); |
| bindings.reserve(targets + 1); // +1 for dsBinding |
| |
| VkAttachmentDescription att = {}; |
| att.format = m_render_target_fmt; |
| att.samples = VK_SAMPLE_COUNT_1_BIT; |
| att.loadOp = (m_clear_via_load_op) ? VK_ATTACHMENT_LOAD_OP_CLEAR : VK_ATTACHMENT_LOAD_OP_LOAD; |
| att.storeOp = VK_ATTACHMENT_STORE_OP_STORE; |
| att.stencilLoadOp = VK_ATTACHMENT_LOAD_OP_DONT_CARE; |
| att.stencilStoreOp = VK_ATTACHMENT_STORE_OP_DONT_CARE; |
| att.initialLayout = VK_IMAGE_LAYOUT_UNDEFINED; |
| att.finalLayout = VK_IMAGE_LAYOUT_COLOR_ATTACHMENT_OPTIMAL; |
| |
| VkAttachmentReference ref = {}; |
| ref.layout = VK_IMAGE_LAYOUT_COLOR_ATTACHMENT_OPTIMAL; |
| |
| m_renderPassClearValues.clear(); |
| VkClearValue clear = {}; |
| clear.color = m_clear_color; |
| |
| for (uint32_t i = 0; i < targets; i++) { |
| attachments.push_back(att); |
| |
| ref.attachment = i; |
| color_references.push_back(ref); |
| |
| m_renderPassClearValues.push_back(clear); |
| |
| std::unique_ptr<VkImageObj> img(new VkImageObj(m_device)); |
| |
| VkFormatProperties props; |
| |
| vkGetPhysicalDeviceFormatProperties(m_device->phy().handle(), m_render_target_fmt, &props); |
| |
| if (props.linearTilingFeatures & VK_FORMAT_FEATURE_COLOR_ATTACHMENT_BIT) { |
| img->Init((uint32_t)m_width, (uint32_t)m_height, 1, m_render_target_fmt, |
| VK_IMAGE_USAGE_COLOR_ATTACHMENT_BIT | VK_IMAGE_USAGE_TRANSFER_SRC_BIT | VK_IMAGE_USAGE_TRANSFER_DST_BIT, |
| VK_IMAGE_TILING_LINEAR); |
| } else if (props.optimalTilingFeatures & VK_FORMAT_FEATURE_COLOR_ATTACHMENT_BIT) { |
| img->Init((uint32_t)m_width, (uint32_t)m_height, 1, m_render_target_fmt, |
| VK_IMAGE_USAGE_COLOR_ATTACHMENT_BIT | VK_IMAGE_USAGE_TRANSFER_SRC_BIT | VK_IMAGE_USAGE_TRANSFER_DST_BIT, |
| VK_IMAGE_TILING_OPTIMAL); |
| } else { |
| FAIL() << "Neither Linear nor Optimal allowed for render target"; |
| } |
| |
| bindings.push_back(img->targetView(m_render_target_fmt)); |
| m_renderTargets.push_back(std::move(img)); |
| } |
| |
| VkSubpassDescription subpass = {}; |
| subpass.pipelineBindPoint = VK_PIPELINE_BIND_POINT_GRAPHICS; |
| subpass.flags = 0; |
| subpass.inputAttachmentCount = 0; |
| subpass.pInputAttachments = NULL; |
| subpass.colorAttachmentCount = targets; |
| subpass.pColorAttachments = color_references.data(); |
| subpass.pResolveAttachments = NULL; |
| |
| VkAttachmentReference ds_reference; |
| if (dsBinding) { |
| att.format = m_depth_stencil_fmt; |
| att.loadOp = (m_clear_via_load_op) ? VK_ATTACHMENT_LOAD_OP_CLEAR : VK_ATTACHMENT_LOAD_OP_LOAD; |
| ; |
| att.storeOp = VK_ATTACHMENT_STORE_OP_STORE; |
| att.stencilLoadOp = VK_ATTACHMENT_LOAD_OP_LOAD; |
| att.stencilStoreOp = VK_ATTACHMENT_STORE_OP_STORE; |
| att.initialLayout = VK_IMAGE_LAYOUT_DEPTH_STENCIL_ATTACHMENT_OPTIMAL; |
| att.finalLayout = VK_IMAGE_LAYOUT_DEPTH_STENCIL_ATTACHMENT_OPTIMAL; |
| attachments.push_back(att); |
| |
| clear.depthStencil.depth = m_depth_clear_color; |
| clear.depthStencil.stencil = m_stencil_clear_color; |
| m_renderPassClearValues.push_back(clear); |
| |
| bindings.push_back(*dsBinding); |
| |
| ds_reference.attachment = targets; |
| ds_reference.layout = VK_IMAGE_LAYOUT_DEPTH_STENCIL_ATTACHMENT_OPTIMAL; |
| subpass.pDepthStencilAttachment = &ds_reference; |
| } else { |
| subpass.pDepthStencilAttachment = NULL; |
| } |
| |
| subpass.preserveAttachmentCount = 0; |
| subpass.pPreserveAttachments = NULL; |
| |
| VkRenderPassCreateInfo rp_info = {}; |
| rp_info.sType = VK_STRUCTURE_TYPE_RENDER_PASS_CREATE_INFO; |
| rp_info.attachmentCount = attachments.size(); |
| rp_info.pAttachments = attachments.data(); |
| rp_info.subpassCount = 1; |
| rp_info.pSubpasses = &subpass; |
| VkSubpassDependency subpass_dep = {}; |
| if (m_addRenderPassSelfDependency) { |
| // Add a subpass self-dependency to subpass 0 of default renderPass |
| subpass_dep.srcSubpass = 0; |
| subpass_dep.dstSubpass = 0; |
| // Just using all framebuffer-space pipeline stages in order to get a reasonably large |
| // set of bits that can be used for both src & dst |
| subpass_dep.srcStageMask = VK_PIPELINE_STAGE_FRAGMENT_SHADER_BIT | VK_PIPELINE_STAGE_EARLY_FRAGMENT_TESTS_BIT | |
| VK_PIPELINE_STAGE_LATE_FRAGMENT_TESTS_BIT | VK_PIPELINE_STAGE_COLOR_ATTACHMENT_OUTPUT_BIT; |
| subpass_dep.dstStageMask = VK_PIPELINE_STAGE_FRAGMENT_SHADER_BIT | VK_PIPELINE_STAGE_EARLY_FRAGMENT_TESTS_BIT | |
| VK_PIPELINE_STAGE_LATE_FRAGMENT_TESTS_BIT | VK_PIPELINE_STAGE_COLOR_ATTACHMENT_OUTPUT_BIT; |
| // Add all of the gfx mem access bits that correlate to the fb-space pipeline stages |
| subpass_dep.srcAccessMask = VK_ACCESS_UNIFORM_READ_BIT | VK_ACCESS_INPUT_ATTACHMENT_READ_BIT | VK_ACCESS_SHADER_READ_BIT | |
| VK_ACCESS_SHADER_WRITE_BIT | VK_ACCESS_COLOR_ATTACHMENT_READ_BIT | |
| VK_ACCESS_COLOR_ATTACHMENT_WRITE_BIT | VK_ACCESS_DEPTH_STENCIL_ATTACHMENT_READ_BIT | |
| VK_ACCESS_DEPTH_STENCIL_ATTACHMENT_WRITE_BIT; |
| subpass_dep.dstAccessMask = VK_ACCESS_UNIFORM_READ_BIT | VK_ACCESS_INPUT_ATTACHMENT_READ_BIT | VK_ACCESS_SHADER_READ_BIT | |
| VK_ACCESS_SHADER_WRITE_BIT | VK_ACCESS_COLOR_ATTACHMENT_READ_BIT | |
| VK_ACCESS_COLOR_ATTACHMENT_WRITE_BIT | VK_ACCESS_DEPTH_STENCIL_ATTACHMENT_READ_BIT | |
| VK_ACCESS_DEPTH_STENCIL_ATTACHMENT_WRITE_BIT; |
| // Must include dep_by_region bit when src & dst both include framebuffer-space stages |
| subpass_dep.dependencyFlags = VK_DEPENDENCY_BY_REGION_BIT; |
| rp_info.dependencyCount = 1; |
| rp_info.pDependencies = &subpass_dep; |
| } |
| |
| vkCreateRenderPass(device(), &rp_info, NULL, &m_renderPass); |
| renderPass_info_ = rp_info; // Save away a copy for tests that need access to the render pass state |
| // Create Framebuffer and RenderPass with color attachments and any |
| // depth/stencil attachment |
| VkFramebufferCreateInfo fb_info = {}; |
| fb_info.sType = VK_STRUCTURE_TYPE_FRAMEBUFFER_CREATE_INFO; |
| fb_info.pNext = NULL; |
| fb_info.renderPass = m_renderPass; |
| fb_info.attachmentCount = bindings.size(); |
| fb_info.pAttachments = bindings.data(); |
| fb_info.width = (uint32_t)m_width; |
| fb_info.height = (uint32_t)m_height; |
| fb_info.layers = 1; |
| |
| vkCreateFramebuffer(device(), &fb_info, NULL, &m_framebuffer); |
| |
| m_renderPassBeginInfo.renderPass = m_renderPass; |
| m_renderPassBeginInfo.framebuffer = m_framebuffer; |
| m_renderPassBeginInfo.renderArea.extent.width = (int32_t)m_width; |
| m_renderPassBeginInfo.renderArea.extent.height = (int32_t)m_height; |
| m_renderPassBeginInfo.clearValueCount = m_renderPassClearValues.size(); |
| m_renderPassBeginInfo.pClearValues = m_renderPassClearValues.data(); |
| } |
| |
| void VkRenderFramework::DestroyRenderTarget() { |
| vkDestroyRenderPass(device(), m_renderPass, nullptr); |
| m_renderPass = VK_NULL_HANDLE; |
| vkDestroyFramebuffer(device(), m_framebuffer, nullptr); |
| m_framebuffer = VK_NULL_HANDLE; |
| } |
| |
| VkDeviceObj::VkDeviceObj(uint32_t id, VkPhysicalDevice obj) : vk_testing::Device(obj), id(id) { |
| init(); |
| |
| props = phy().properties(); |
| queue_props = phy().queue_properties(); |
| } |
| |
| VkDeviceObj::VkDeviceObj(uint32_t id, VkPhysicalDevice obj, std::vector<const char *> &extension_names, |
| VkPhysicalDeviceFeatures *features, void *create_device_pnext) |
| : vk_testing::Device(obj), id(id) { |
| init(extension_names, features, create_device_pnext); |
| |
| props = phy().properties(); |
| queue_props = phy().queue_properties(); |
| } |
| |
| uint32_t VkDeviceObj::QueueFamilyMatching(VkQueueFlags with, VkQueueFlags without, bool all_bits) { |
| // Find a queue family with and without desired capabilities |
| for (uint32_t i = 0; i < queue_props.size(); i++) { |
| auto flags = queue_props[i].queueFlags; |
| bool matches = all_bits ? (flags & with) == with : (flags & with) != 0; |
| if (matches && ((flags & without) == 0) && (queue_props[i].queueCount > 0)) { |
| return i; |
| } |
| } |
| return UINT32_MAX; |
| } |
| |
| void VkDeviceObj::SetDeviceQueue() { |
| ASSERT_NE(true, graphics_queues().empty()); |
| m_queue = graphics_queues()[0]->handle(); |
| } |
| |
| VkQueueObj *VkDeviceObj::GetDefaultQueue() { |
| if (graphics_queues().empty()) return nullptr; |
| return graphics_queues()[0]; |
| } |
| |
| VkQueueObj *VkDeviceObj::GetDefaultComputeQueue() { |
| if (compute_queues().empty()) return nullptr; |
| return compute_queues()[0]; |
| } |
| |
| VkDescriptorSetLayoutObj::VkDescriptorSetLayoutObj(const VkDeviceObj *device, |
| const std::vector<VkDescriptorSetLayoutBinding> &descriptor_set_bindings, |
| VkDescriptorSetLayoutCreateFlags flags, void *pNext) { |
| VkDescriptorSetLayoutCreateInfo dsl_ci = {}; |
| dsl_ci.sType = VK_STRUCTURE_TYPE_DESCRIPTOR_SET_LAYOUT_CREATE_INFO; |
| dsl_ci.pNext = pNext; |
| dsl_ci.flags = flags; |
| dsl_ci.bindingCount = static_cast<uint32_t>(descriptor_set_bindings.size()); |
| dsl_ci.pBindings = descriptor_set_bindings.data(); |
| |
| init(*device, dsl_ci); |
| } |
| |
| VkDescriptorSetObj::VkDescriptorSetObj(VkDeviceObj *device) : m_device(device), m_nextSlot(0) {} |
| |
| VkDescriptorSetObj::~VkDescriptorSetObj() { |
| if (m_set) { |
| delete m_set; |
| } |
| } |
| |
| int VkDescriptorSetObj::AppendDummy() { |
| /* request a descriptor but do not update it */ |
| VkDescriptorSetLayoutBinding binding = {}; |
| binding.descriptorType = VK_DESCRIPTOR_TYPE_STORAGE_BUFFER; |
| binding.descriptorCount = 1; |
| binding.binding = m_layout_bindings.size(); |
| binding.stageFlags = VK_SHADER_STAGE_ALL; |
| binding.pImmutableSamplers = NULL; |
| |
| m_layout_bindings.push_back(binding); |
| m_type_counts[VK_DESCRIPTOR_TYPE_STORAGE_BUFFER] += binding.descriptorCount; |
| |
| return m_nextSlot++; |
| } |
| |
| int VkDescriptorSetObj::AppendBuffer(VkDescriptorType type, VkConstantBufferObj &constantBuffer) { |
| assert(type == VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER || type == VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER_DYNAMIC || |
| type == VK_DESCRIPTOR_TYPE_STORAGE_BUFFER || type == VK_DESCRIPTOR_TYPE_STORAGE_BUFFER_DYNAMIC); |
| VkDescriptorSetLayoutBinding binding = {}; |
| binding.descriptorType = type; |
| binding.descriptorCount = 1; |
| binding.binding = m_layout_bindings.size(); |
| binding.stageFlags = VK_SHADER_STAGE_ALL; |
| binding.pImmutableSamplers = NULL; |
| |
| m_layout_bindings.push_back(binding); |
| m_type_counts[type] += binding.descriptorCount; |
| |
| m_writes.push_back(vk_testing::Device::write_descriptor_set(vk_testing::DescriptorSet(), m_nextSlot, 0, type, 1, |
| &constantBuffer.m_descriptorBufferInfo)); |
| |
| return m_nextSlot++; |
| } |
| |
| int VkDescriptorSetObj::AppendSamplerTexture(VkSamplerObj *sampler, VkTextureObj *texture) { |
| VkDescriptorSetLayoutBinding binding = {}; |
| binding.descriptorType = VK_DESCRIPTOR_TYPE_COMBINED_IMAGE_SAMPLER; |
| binding.descriptorCount = 1; |
| binding.binding = m_layout_bindings.size(); |
| binding.stageFlags = VK_SHADER_STAGE_ALL; |
| binding.pImmutableSamplers = NULL; |
| |
| m_layout_bindings.push_back(binding); |
| m_type_counts[VK_DESCRIPTOR_TYPE_COMBINED_IMAGE_SAMPLER] += binding.descriptorCount; |
| VkDescriptorImageInfo tmp = texture->DescriptorImageInfo(); |
| tmp.sampler = sampler->handle(); |
| m_imageSamplerDescriptors.push_back(tmp); |
| |
| m_writes.push_back(vk_testing::Device::write_descriptor_set(vk_testing::DescriptorSet(), m_nextSlot, 0, |
| VK_DESCRIPTOR_TYPE_COMBINED_IMAGE_SAMPLER, 1, &tmp)); |
| |
| return m_nextSlot++; |
| } |
| |
| VkPipelineLayout VkDescriptorSetObj::GetPipelineLayout() const { return m_pipeline_layout.handle(); } |
| |
| VkDescriptorSet VkDescriptorSetObj::GetDescriptorSetHandle() const { |
| if (m_set) |
| return m_set->handle(); |
| else |
| return VK_NULL_HANDLE; |
| } |
| |
| void VkDescriptorSetObj::CreateVKDescriptorSet(VkCommandBufferObj *commandBuffer) { |
| if (m_type_counts.size()) { |
| // create VkDescriptorPool |
| VkDescriptorPoolSize poolSize; |
| vector<VkDescriptorPoolSize> sizes; |
| for (auto it = m_type_counts.begin(); it != m_type_counts.end(); ++it) { |
| poolSize.descriptorCount = it->second; |
| poolSize.type = it->first; |
| sizes.push_back(poolSize); |
| } |
| VkDescriptorPoolCreateInfo pool = {}; |
| pool.sType = VK_STRUCTURE_TYPE_DESCRIPTOR_POOL_CREATE_INFO; |
| pool.poolSizeCount = sizes.size(); |
| pool.maxSets = 1; |
| pool.pPoolSizes = sizes.data(); |
| init(*m_device, pool); |
| } |
| |
| // create VkDescriptorSetLayout |
| VkDescriptorSetLayoutCreateInfo layout = {}; |
| layout.sType = VK_STRUCTURE_TYPE_DESCRIPTOR_SET_LAYOUT_CREATE_INFO; |
| layout.bindingCount = m_layout_bindings.size(); |
| layout.pBindings = m_layout_bindings.data(); |
| |
| m_layout.init(*m_device, layout); |
| vector<const vk_testing::DescriptorSetLayout *> layouts; |
| layouts.push_back(&m_layout); |
| |
| // create VkPipelineLayout |
| VkPipelineLayoutCreateInfo pipeline_layout = {}; |
| pipeline_layout.sType = VK_STRUCTURE_TYPE_PIPELINE_LAYOUT_CREATE_INFO; |
| pipeline_layout.setLayoutCount = layouts.size(); |
| pipeline_layout.pSetLayouts = NULL; |
| |
| m_pipeline_layout.init(*m_device, pipeline_layout, layouts); |
| |
| if (m_type_counts.size()) { |
| // create VkDescriptorSet |
| m_set = alloc_sets(*m_device, m_layout); |
| |
| // build the update array |
| size_t imageSamplerCount = 0; |
| for (std::vector<VkWriteDescriptorSet>::iterator it = m_writes.begin(); it != m_writes.end(); it++) { |
| it->dstSet = m_set->handle(); |
| if (it->descriptorType == VK_DESCRIPTOR_TYPE_COMBINED_IMAGE_SAMPLER) |
| it->pImageInfo = &m_imageSamplerDescriptors[imageSamplerCount++]; |
| } |
| |
| // do the updates |
| m_device->update_descriptor_sets(m_writes); |
| } |
| } |
| |
| VkRenderpassObj::VkRenderpassObj(VkDeviceObj *dev) { |
| // Create a renderPass with a single color attachment |
| VkAttachmentReference attach = {}; |
| attach.layout = VK_IMAGE_LAYOUT_GENERAL; |
| |
| VkSubpassDescription subpass = {}; |
| subpass.pColorAttachments = &attach; |
| subpass.colorAttachmentCount = 1; |
| |
| VkRenderPassCreateInfo rpci = {}; |
| rpci.subpassCount = 1; |
| rpci.pSubpasses = &subpass; |
| rpci.attachmentCount = 1; |
| |
| VkAttachmentDescription attach_desc = {}; |
| attach_desc.format = VK_FORMAT_B8G8R8A8_UNORM; |
| attach_desc.samples = VK_SAMPLE_COUNT_1_BIT; |
| attach_desc.initialLayout = VK_IMAGE_LAYOUT_UNDEFINED; |
| attach_desc.finalLayout = VK_IMAGE_LAYOUT_GENERAL; |
| |
| rpci.pAttachments = &attach_desc; |
| rpci.sType = VK_STRUCTURE_TYPE_RENDER_PASS_CREATE_INFO; |
| |
| device = dev->device(); |
| vkCreateRenderPass(device, &rpci, NULL, &m_renderpass); |
| } |
| |
| VkRenderpassObj::~VkRenderpassObj() { vkDestroyRenderPass(device, m_renderpass, NULL); } |
| |
| VkImageObj::VkImageObj(VkDeviceObj *dev) { |
| m_device = dev; |
| m_descriptorImageInfo.imageView = VK_NULL_HANDLE; |
| m_descriptorImageInfo.imageLayout = VK_IMAGE_LAYOUT_GENERAL; |
| } |
| |
| // clang-format off |
| void VkImageObj::ImageMemoryBarrier(VkCommandBufferObj *cmd_buf, VkImageAspectFlags aspect, |
| VkFlags output_mask /*= |
| VK_ACCESS_HOST_WRITE_BIT | |
| VK_ACCESS_SHADER_WRITE_BIT | |
| VK_ACCESS_COLOR_ATTACHMENT_WRITE_BIT | |
| VK_ACCESS_DEPTH_STENCIL_ATTACHMENT_WRITE_BIT | |
| VK_MEMORY_OUTPUT_COPY_BIT*/, |
| VkFlags input_mask /*= |
| VK_ACCESS_HOST_READ_BIT | |
| VK_ACCESS_INDIRECT_COMMAND_READ_BIT | |
| VK_ACCESS_INDEX_READ_BIT | |
| VK_ACCESS_VERTEX_ATTRIBUTE_READ_BIT | |
| VK_ACCESS_UNIFORM_READ_BIT | |
| VK_ACCESS_SHADER_READ_BIT | |
| VK_ACCESS_COLOR_ATTACHMENT_READ_BIT | |
| VK_ACCESS_DEPTH_STENCIL_ATTACHMENT_READ_BIT | |
| VK_MEMORY_INPUT_COPY_BIT*/, VkImageLayout image_layout, |
| VkPipelineStageFlags src_stages, VkPipelineStageFlags dest_stages, |
| uint32_t srcQueueFamilyIndex, uint32_t dstQueueFamilyIndex) { |
| // clang-format on |
| // TODO: Mali device crashing with VK_REMAINING_MIP_LEVELS |
| const VkImageSubresourceRange subresourceRange = |
| subresource_range(aspect, 0, /*VK_REMAINING_MIP_LEVELS*/ 1, 0, 1 /*VK_REMAINING_ARRAY_LAYERS*/); |
| VkImageMemoryBarrier barrier; |
| barrier = image_memory_barrier(output_mask, input_mask, Layout(), image_layout, subresourceRange, srcQueueFamilyIndex, |
| dstQueueFamilyIndex); |
| |
| VkImageMemoryBarrier *pmemory_barrier = &barrier; |
| |
| // write barrier to the command buffer |
| vkCmdPipelineBarrier(cmd_buf->handle(), src_stages, dest_stages, VK_DEPENDENCY_BY_REGION_BIT, 0, NULL, 0, NULL, 1, |
| pmemory_barrier); |
| } |
| |
| void VkImageObj::SetLayout(VkCommandBufferObj *cmd_buf, VkImageAspectFlags aspect, VkImageLayout image_layout) { |
| VkFlags src_mask, dst_mask; |
| const VkFlags all_cache_outputs = VK_ACCESS_HOST_WRITE_BIT | VK_ACCESS_SHADER_WRITE_BIT | VK_ACCESS_COLOR_ATTACHMENT_WRITE_BIT | |
| VK_ACCESS_DEPTH_STENCIL_ATTACHMENT_WRITE_BIT | VK_ACCESS_TRANSFER_WRITE_BIT; |
| const VkFlags all_cache_inputs = VK_ACCESS_HOST_READ_BIT | VK_ACCESS_INDIRECT_COMMAND_READ_BIT | VK_ACCESS_INDEX_READ_BIT | |
| VK_ACCESS_VERTEX_ATTRIBUTE_READ_BIT | VK_ACCESS_UNIFORM_READ_BIT | VK_ACCESS_SHADER_READ_BIT | |
| VK_ACCESS_COLOR_ATTACHMENT_READ_BIT | VK_ACCESS_DEPTH_STENCIL_ATTACHMENT_READ_BIT | |
| VK_ACCESS_MEMORY_READ_BIT; |
| |
| if (image_layout == m_descriptorImageInfo.imageLayout) { |
| return; |
| } |
| |
| switch (image_layout) { |
| case VK_IMAGE_LAYOUT_TRANSFER_SRC_OPTIMAL: |
| if (m_descriptorImageInfo.imageLayout == VK_IMAGE_LAYOUT_COLOR_ATTACHMENT_OPTIMAL) |
| src_mask = VK_ACCESS_COLOR_ATTACHMENT_WRITE_BIT; |
| else |
| src_mask = VK_ACCESS_TRANSFER_WRITE_BIT; |
| dst_mask = VK_ACCESS_SHADER_READ_BIT | VK_ACCESS_TRANSFER_READ_BIT; |
| break; |
| |
| case VK_IMAGE_LAYOUT_TRANSFER_DST_OPTIMAL: |
| if (m_descriptorImageInfo.imageLayout == VK_IMAGE_LAYOUT_COLOR_ATTACHMENT_OPTIMAL) |
| src_mask = VK_ACCESS_COLOR_ATTACHMENT_WRITE_BIT; |
| else if (m_descriptorImageInfo.imageLayout == VK_IMAGE_LAYOUT_SHADER_READ_ONLY_OPTIMAL) |
| src_mask = VK_ACCESS_INPUT_ATTACHMENT_READ_BIT; |
| else |
| src_mask = VK_ACCESS_TRANSFER_WRITE_BIT; |
| dst_mask = VK_ACCESS_TRANSFER_WRITE_BIT; |
| break; |
| |
| case VK_IMAGE_LAYOUT_SHADER_READ_ONLY_OPTIMAL: |
| if (m_descriptorImageInfo.imageLayout == VK_IMAGE_LAYOUT_TRANSFER_DST_OPTIMAL) |
| src_mask = VK_ACCESS_TRANSFER_WRITE_BIT; |
| else |
| src_mask = VK_ACCESS_INPUT_ATTACHMENT_READ_BIT; |
| dst_mask = VK_ACCESS_SHADER_READ_BIT | VK_ACCESS_MEMORY_READ_BIT; |
| break; |
| |
| case VK_IMAGE_LAYOUT_COLOR_ATTACHMENT_OPTIMAL: |
| if (m_descriptorImageInfo.imageLayout == VK_IMAGE_LAYOUT_TRANSFER_SRC_OPTIMAL) |
| src_mask = VK_ACCESS_TRANSFER_READ_BIT; |
| else |
| src_mask = 0; |
| dst_mask = VK_ACCESS_COLOR_ATTACHMENT_WRITE_BIT; |
| break; |
| |
| case VK_IMAGE_LAYOUT_DEPTH_STENCIL_ATTACHMENT_OPTIMAL: |
| dst_mask = VK_ACCESS_DEPTH_STENCIL_ATTACHMENT_WRITE_BIT; |
| src_mask = all_cache_outputs; |
| break; |
| |
| default: |
| src_mask = all_cache_outputs; |
| dst_mask = all_cache_inputs; |
| break; |
| } |
| |
| if (m_descriptorImageInfo.imageLayout == VK_IMAGE_LAYOUT_UNDEFINED) src_mask = 0; |
| |
| ImageMemoryBarrier(cmd_buf, aspect, src_mask, dst_mask, image_layout); |
| m_descriptorImageInfo.imageLayout = image_layout; |
| } |
| |
| void VkImageObj::SetLayout(VkImageAspectFlags aspect, VkImageLayout image_layout) { |
| if (image_layout == m_descriptorImageInfo.imageLayout) { |
| return; |
| } |
| |
| VkCommandPoolObj pool(m_device, m_device->graphics_queue_node_index_); |
| VkCommandBufferObj cmd_buf(m_device, &pool); |
| |
| /* Build command buffer to set image layout in the driver */ |
| cmd_buf.begin(); |
| SetLayout(&cmd_buf, aspect, image_layout); |
| cmd_buf.end(); |
| |
| cmd_buf.QueueCommandBuffer(); |
| } |
| |
| bool VkImageObj::IsCompatible(const VkImageUsageFlags usages, const VkFormatFeatureFlags features) { |
| VkFormatFeatureFlags all_feature_flags = |
| VK_FORMAT_FEATURE_SAMPLED_IMAGE_BIT | VK_FORMAT_FEATURE_STORAGE_IMAGE_BIT | VK_FORMAT_FEATURE_STORAGE_IMAGE_ATOMIC_BIT | |
| VK_FORMAT_FEATURE_UNIFORM_TEXEL_BUFFER_BIT | VK_FORMAT_FEATURE_STORAGE_TEXEL_BUFFER_BIT | |
| VK_FORMAT_FEATURE_STORAGE_TEXEL_BUFFER_ATOMIC_BIT | VK_FORMAT_FEATURE_VERTEX_BUFFER_BIT | |
| VK_FORMAT_FEATURE_COLOR_ATTACHMENT_BIT | VK_FORMAT_FEATURE_COLOR_ATTACHMENT_BLEND_BIT | |
| VK_FORMAT_FEATURE_DEPTH_STENCIL_ATTACHMENT_BIT | VK_FORMAT_FEATURE_BLIT_SRC_BIT | VK_FORMAT_FEATURE_BLIT_DST_BIT | |
| VK_FORMAT_FEATURE_SAMPLED_IMAGE_FILTER_LINEAR_BIT; |
| if (m_device->IsEnabledExtension(VK_IMG_FILTER_CUBIC_EXTENSION_NAME)) { |
| all_feature_flags |= VK_FORMAT_FEATURE_SAMPLED_IMAGE_FILTER_CUBIC_BIT_IMG; |
| } |
| |
| if (m_device->IsEnabledExtension(VK_KHR_MAINTENANCE1_EXTENSION_NAME)) { |
| all_feature_flags |= VK_FORMAT_FEATURE_TRANSFER_SRC_BIT_KHR | VK_FORMAT_FEATURE_TRANSFER_DST_BIT_KHR; |
| } |
| |
| if (m_device->IsEnabledExtension(VK_EXT_SAMPLER_FILTER_MINMAX_EXTENSION_NAME)) { |
| all_feature_flags |= VK_FORMAT_FEATURE_SAMPLED_IMAGE_FILTER_MINMAX_BIT_EXT; |
| } |
| |
| if (m_device->IsEnabledExtension(VK_KHR_SAMPLER_YCBCR_CONVERSION_EXTENSION_NAME)) { |
| all_feature_flags |= VK_FORMAT_FEATURE_MIDPOINT_CHROMA_SAMPLES_BIT_KHR | |
| VK_FORMAT_FEATURE_SAMPLED_IMAGE_YCBCR_CONVERSION_LINEAR_FILTER_BIT_KHR | |
| VK_FORMAT_FEATURE_SAMPLED_IMAGE_YCBCR_CONVERSION_SEPARATE_RECONSTRUCTION_FILTER_BIT_KHR | |
| VK_FORMAT_FEATURE_SAMPLED_IMAGE_YCBCR_CONVERSION_CHROMA_RECONSTRUCTION_EXPLICIT_BIT_KHR | |
| VK_FORMAT_FEATURE_SAMPLED_IMAGE_YCBCR_CONVERSION_CHROMA_RECONSTRUCTION_EXPLICIT_FORCEABLE_BIT_KHR | |
| VK_FORMAT_FEATURE_DISJOINT_BIT_KHR | VK_FORMAT_FEATURE_COSITED_CHROMA_SAMPLES_BIT_KHR; |
| } |
| |
| if ((features & all_feature_flags) == 0) return false; // whole format unsupported |
| |
| if ((usages & VK_IMAGE_USAGE_SAMPLED_BIT) && !(features & VK_FORMAT_FEATURE_SAMPLED_IMAGE_BIT)) return false; |
| if ((usages & VK_IMAGE_USAGE_STORAGE_BIT) && !(features & VK_FORMAT_FEATURE_STORAGE_IMAGE_BIT)) return false; |
| if ((usages & VK_IMAGE_USAGE_COLOR_ATTACHMENT_BIT) && !(features & VK_FORMAT_FEATURE_COLOR_ATTACHMENT_BIT)) return false; |
| if ((usages & VK_IMAGE_USAGE_DEPTH_STENCIL_ATTACHMENT_BIT) && !(features & VK_FORMAT_FEATURE_DEPTH_STENCIL_ATTACHMENT_BIT)) |
| return false; |
| |
| if (m_device->IsEnabledExtension(VK_KHR_MAINTENANCE1_EXTENSION_NAME)) { |
| // WORKAROUND: for DevSim not reporting extended enums, and possibly some drivers too |
| const auto all_nontransfer_feature_flags = |
| all_feature_flags ^ (VK_FORMAT_FEATURE_TRANSFER_SRC_BIT_KHR | VK_FORMAT_FEATURE_TRANSFER_DST_BIT_KHR); |
| const bool transfer_probably_supported_anyway = (features & all_nontransfer_feature_flags) > 0; |
| if (!transfer_probably_supported_anyway) { |
| if ((usages & VK_IMAGE_USAGE_TRANSFER_SRC_BIT) && !(features & VK_FORMAT_FEATURE_TRANSFER_SRC_BIT_KHR)) return false; |
| if ((usages & VK_IMAGE_USAGE_TRANSFER_DST_BIT) && !(features & VK_FORMAT_FEATURE_TRANSFER_DST_BIT_KHR)) return false; |
| } |
| } |
| |
| return true; |
| } |
| |
| void VkImageObj::InitNoLayout(uint32_t const width, uint32_t const height, uint32_t const mipLevels, VkFormat const format, |
| VkFlags const usage, VkImageTiling const requested_tiling, VkMemoryPropertyFlags const reqs, |
| const std::vector<uint32_t> *queue_families, bool memory) { |
| VkFormatProperties image_fmt; |
| VkImageTiling tiling = VK_IMAGE_TILING_OPTIMAL; |
| |
| vkGetPhysicalDeviceFormatProperties(m_device->phy().handle(), format, &image_fmt); |
| |
| if (requested_tiling == VK_IMAGE_TILING_LINEAR) { |
| if (IsCompatible(usage, image_fmt.linearTilingFeatures)) { |
| tiling = VK_IMAGE_TILING_LINEAR; |
| } else if (IsCompatible(usage, image_fmt.optimalTilingFeatures)) { |
| tiling = VK_IMAGE_TILING_OPTIMAL; |
| } else { |
| FAIL() << "VkImageObj::init() error: unsupported tiling configuration. Usage: " << std::hex << std::showbase << usage |
| << ", supported linear features: " << image_fmt.linearTilingFeatures; |
| } |
| } else if (IsCompatible(usage, image_fmt.optimalTilingFeatures)) { |
| tiling = VK_IMAGE_TILING_OPTIMAL; |
| } else if (IsCompatible(usage, image_fmt.linearTilingFeatures)) { |
| tiling = VK_IMAGE_TILING_LINEAR; |
| } else { |
| FAIL() << "VkImageObj::init() error: unsupported tiling configuration. Usage: " << std::hex << std::showbase << usage |
| << ", supported optimal features: " << image_fmt.optimalTilingFeatures; |
| } |
| |
| VkImageCreateInfo imageCreateInfo = vk_testing::Image::create_info(); |
| imageCreateInfo.imageType = VK_IMAGE_TYPE_2D; |
| imageCreateInfo.format = format; |
| imageCreateInfo.extent.width = width; |
| imageCreateInfo.extent.height = height; |
| imageCreateInfo.mipLevels = mipLevels; |
| imageCreateInfo.tiling = tiling; |
| imageCreateInfo.initialLayout = VK_IMAGE_LAYOUT_UNDEFINED; |
| |
| // Automatically set sharing mode etc. based on queue family information |
| if (queue_families && (queue_families->size() > 1)) { |
| imageCreateInfo.sharingMode = VK_SHARING_MODE_CONCURRENT; |
| imageCreateInfo.queueFamilyIndexCount = static_cast<uint32_t>(queue_families->size()); |
| imageCreateInfo.pQueueFamilyIndices = queue_families->data(); |
| } |
| |
| Layout(imageCreateInfo.initialLayout); |
| imageCreateInfo.usage = usage; |
| if (memory) |
| vk_testing::Image::init(*m_device, imageCreateInfo, reqs); |
| else |
| vk_testing::Image::init_no_mem(*m_device, imageCreateInfo); |
| } |
| |
| void VkImageObj::Init(uint32_t const width, uint32_t const height, uint32_t const mipLevels, VkFormat const format, |
| VkFlags const usage, VkImageTiling const requested_tiling, VkMemoryPropertyFlags const reqs, |
| const std::vector<uint32_t> *queue_families, bool memory) { |
| InitNoLayout(width, height, mipLevels, format, usage, requested_tiling, reqs, queue_families, memory); |
| |
| if (!initialized() || !memory) return; // We don't have a valid handle from early stage init, and thus SetLayout will fail |
| |
| VkImageLayout newLayout; |
| if (usage & VK_IMAGE_USAGE_COLOR_ATTACHMENT_BIT) |
| newLayout = VK_IMAGE_LAYOUT_COLOR_ATTACHMENT_OPTIMAL; |
| else if (usage & VK_IMAGE_USAGE_SAMPLED_BIT) |
| newLayout = VK_IMAGE_LAYOUT_SHADER_READ_ONLY_OPTIMAL; |
| else |
| newLayout = m_descriptorImageInfo.imageLayout; |
| |
| VkImageAspectFlags image_aspect = 0; |
| if (FormatIsDepthAndStencil(format)) { |
| image_aspect = VK_IMAGE_ASPECT_STENCIL_BIT | VK_IMAGE_ASPECT_DEPTH_BIT; |
| } else if (FormatIsDepthOnly(format)) { |
| image_aspect = VK_IMAGE_ASPECT_DEPTH_BIT; |
| } else if (FormatIsStencilOnly(format)) { |
| image_aspect = VK_IMAGE_ASPECT_STENCIL_BIT; |
| } else { // color |
| image_aspect = VK_IMAGE_ASPECT_COLOR_BIT; |
| } |
| SetLayout(image_aspect, newLayout); |
| } |
| |
| void VkImageObj::init(const VkImageCreateInfo *create_info) { |
| VkFormatProperties image_fmt; |
| vkGetPhysicalDeviceFormatProperties(m_device->phy().handle(), create_info->format, &image_fmt); |
| |
| switch (create_info->tiling) { |
| case VK_IMAGE_TILING_OPTIMAL: |
| if (!IsCompatible(create_info->usage, image_fmt.optimalTilingFeatures)) { |
| FAIL() << "VkImageObj::init() error: unsupported tiling configuration. Usage: " << std::hex << std::showbase |
| << create_info->usage << ", supported optimal features: " << image_fmt.optimalTilingFeatures; |
| } |
| break; |
| case VK_IMAGE_TILING_LINEAR: |
| if (!IsCompatible(create_info->usage, image_fmt.linearTilingFeatures)) { |
| FAIL() << "VkImageObj::init() error: unsupported tiling configuration. Usage: " << std::hex << std::showbase |
| << create_info->usage << ", supported linear features: " << image_fmt.linearTilingFeatures; |
| } |
| break; |
| default: |
| break; |
| } |
| Layout(create_info->initialLayout); |
| |
| vk_testing::Image::init(*m_device, *create_info, 0); |
| |
| VkImageAspectFlags image_aspect = 0; |
| if (FormatIsDepthAndStencil(create_info->format)) { |
| image_aspect = VK_IMAGE_ASPECT_STENCIL_BIT | VK_IMAGE_ASPECT_DEPTH_BIT; |
| } else if (FormatIsDepthOnly(create_info->format)) { |
| image_aspect = VK_IMAGE_ASPECT_DEPTH_BIT; |
| } else if (FormatIsStencilOnly(create_info->format)) { |
| image_aspect = VK_IMAGE_ASPECT_STENCIL_BIT; |
| } else { // color |
| image_aspect = VK_IMAGE_ASPECT_COLOR_BIT; |
| } |
| SetLayout(image_aspect, VK_IMAGE_LAYOUT_GENERAL); |
| } |
| |
| VkResult VkImageObj::CopyImage(VkImageObj &src_image) { |
| VkImageLayout src_image_layout, dest_image_layout; |
| |
| VkCommandPoolObj pool(m_device, m_device->graphics_queue_node_index_); |
| VkCommandBufferObj cmd_buf(m_device, &pool); |
| |
| /* Build command buffer to copy staging texture to usable texture */ |
| cmd_buf.begin(); |
| |
| /* TODO: Can we determine image aspect from image object? */ |
| src_image_layout = src_image.Layout(); |
| src_image.SetLayout(&cmd_buf, VK_IMAGE_ASPECT_COLOR_BIT, VK_IMAGE_LAYOUT_TRANSFER_SRC_OPTIMAL); |
| |
| dest_image_layout = (this->Layout() == VK_IMAGE_LAYOUT_UNDEFINED) ? VK_IMAGE_LAYOUT_GENERAL : this->Layout(); |
| this->SetLayout(&cmd_buf, VK_IMAGE_ASPECT_COLOR_BIT, VK_IMAGE_LAYOUT_TRANSFER_DST_OPTIMAL); |
| |
| VkImageCopy copy_region = {}; |
| copy_region.srcSubresource.aspectMask = VK_IMAGE_ASPECT_COLOR_BIT; |
| copy_region.srcSubresource.baseArrayLayer = 0; |
| copy_region.srcSubresource.mipLevel = 0; |
| copy_region.srcSubresource.layerCount = 1; |
| copy_region.srcOffset.x = 0; |
| copy_region.srcOffset.y = 0; |
| copy_region.srcOffset.z = 0; |
| copy_region.dstSubresource.aspectMask = VK_IMAGE_ASPECT_COLOR_BIT; |
| copy_region.dstSubresource.baseArrayLayer = 0; |
| copy_region.dstSubresource.mipLevel = 0; |
| copy_region.dstSubresource.layerCount = 1; |
| copy_region.dstOffset.x = 0; |
| copy_region.dstOffset.y = 0; |
| copy_region.dstOffset.z = 0; |
| copy_region.extent = src_image.extent(); |
| |
| vkCmdCopyImage(cmd_buf.handle(), src_image.handle(), src_image.Layout(), handle(), Layout(), 1, ©_region); |
| |
| src_image.SetLayout(&cmd_buf, VK_IMAGE_ASPECT_COLOR_BIT, src_image_layout); |
| |
| this->SetLayout(&cmd_buf, VK_IMAGE_ASPECT_COLOR_BIT, dest_image_layout); |
| |
| cmd_buf.end(); |
| |
| cmd_buf.QueueCommandBuffer(); |
| |
| return VK_SUCCESS; |
| } |
| |
| // Same as CopyImage, but in the opposite direction |
| VkResult VkImageObj::CopyImageOut(VkImageObj &dst_image) { |
| VkImageLayout src_image_layout, dest_image_layout; |
| |
| VkCommandPoolObj pool(m_device, m_device->graphics_queue_node_index_); |
| VkCommandBufferObj cmd_buf(m_device, &pool); |
| |
| cmd_buf.begin(); |
| |
| src_image_layout = this->Layout(); |
| this->SetLayout(&cmd_buf, VK_IMAGE_ASPECT_COLOR_BIT, VK_IMAGE_LAYOUT_TRANSFER_SRC_OPTIMAL); |
| |
| dest_image_layout = (dst_image.Layout() == VK_IMAGE_LAYOUT_UNDEFINED) ? VK_IMAGE_LAYOUT_GENERAL : dst_image.Layout(); |
| dst_image.SetLayout(&cmd_buf, VK_IMAGE_ASPECT_COLOR_BIT, VK_IMAGE_LAYOUT_TRANSFER_DST_OPTIMAL); |
| |
| VkImageCopy copy_region = {}; |
| copy_region.srcSubresource.aspectMask = VK_IMAGE_ASPECT_COLOR_BIT; |
| copy_region.srcSubresource.baseArrayLayer = 0; |
| copy_region.srcSubresource.mipLevel = 0; |
| copy_region.srcSubresource.layerCount = 1; |
| copy_region.srcOffset.x = 0; |
| copy_region.srcOffset.y = 0; |
| copy_region.srcOffset.z = 0; |
| copy_region.dstSubresource.aspectMask = VK_IMAGE_ASPECT_COLOR_BIT; |
| copy_region.dstSubresource.baseArrayLayer = 0; |
| copy_region.dstSubresource.mipLevel = 0; |
| copy_region.dstSubresource.layerCount = 1; |
| copy_region.dstOffset.x = 0; |
| copy_region.dstOffset.y = 0; |
| copy_region.dstOffset.z = 0; |
| copy_region.extent = dst_image.extent(); |
| |
| vkCmdCopyImage(cmd_buf.handle(), handle(), Layout(), dst_image.handle(), dst_image.Layout(), 1, ©_region); |
| |
| this->SetLayout(&cmd_buf, VK_IMAGE_ASPECT_COLOR_BIT, src_image_layout); |
| |
| dst_image.SetLayout(&cmd_buf, VK_IMAGE_ASPECT_COLOR_BIT, dest_image_layout); |
| |
| cmd_buf.end(); |
| |
| cmd_buf.QueueCommandBuffer(); |
| |
| return VK_SUCCESS; |
| } |
| |
| // Return 16x16 pixel block |
| std::array<std::array<uint32_t, 16>, 16> VkImageObj::Read() { |
| VkImageObj stagingImage(m_device); |
| VkMemoryPropertyFlags reqs = VK_MEMORY_PROPERTY_HOST_VISIBLE_BIT | VK_MEMORY_PROPERTY_HOST_COHERENT_BIT; |
| |
| stagingImage.Init(16, 16, 1, VK_FORMAT_B8G8R8A8_UNORM, VK_IMAGE_USAGE_TRANSFER_DST_BIT | VK_IMAGE_USAGE_TRANSFER_SRC_BIT, |
| VK_IMAGE_TILING_LINEAR, reqs); |
| stagingImage.SetLayout(VK_IMAGE_ASPECT_COLOR_BIT, VK_IMAGE_LAYOUT_GENERAL); |
| VkSubresourceLayout layout = stagingImage.subresource_layout(subresource(VK_IMAGE_ASPECT_COLOR_BIT, 0, 0)); |
| CopyImageOut(stagingImage); |
| void *data = stagingImage.MapMemory(); |
| std::array<std::array<uint32_t, 16>, 16> m = {}; |
| if (data) { |
| for (uint32_t y = 0; y < stagingImage.extent().height; y++) { |
| uint32_t *row = (uint32_t *)((char *)data + layout.rowPitch * y); |
| for (uint32_t x = 0; x < stagingImage.extent().width; x++) m[y][x] = row[x]; |
| } |
| } |
| stagingImage.UnmapMemory(); |
| return m; |
| } |
| |
| VkTextureObj::VkTextureObj(VkDeviceObj *device, uint32_t *colors) : VkImageObj(device) { |
| m_device = device; |
| const VkFormat tex_format = VK_FORMAT_B8G8R8A8_UNORM; |
| uint32_t tex_colors[2] = {0xffff0000, 0xff00ff00}; |
| void *data; |
| uint32_t x, y; |
| VkImageObj stagingImage(device); |
| VkMemoryPropertyFlags reqs = VK_MEMORY_PROPERTY_HOST_VISIBLE_BIT | VK_MEMORY_PROPERTY_HOST_COHERENT_BIT; |
| |
| stagingImage.Init(16, 16, 1, tex_format, VK_IMAGE_USAGE_TRANSFER_DST_BIT | VK_IMAGE_USAGE_TRANSFER_SRC_BIT, |
| VK_IMAGE_TILING_LINEAR, reqs); |
| VkSubresourceLayout layout = stagingImage.subresource_layout(subresource(VK_IMAGE_ASPECT_COLOR_BIT, 0, 0)); |
| |
| if (colors == NULL) colors = tex_colors; |
| |
| VkImageViewCreateInfo view = {}; |
| view.sType = VK_STRUCTURE_TYPE_IMAGE_VIEW_CREATE_INFO; |
| view.pNext = NULL; |
| view.image = VK_NULL_HANDLE; |
| view.viewType = VK_IMAGE_VIEW_TYPE_2D; |
| view.format = tex_format; |
| view.components.r = VK_COMPONENT_SWIZZLE_R; |
| view.components.g = VK_COMPONENT_SWIZZLE_G; |
| view.components.b = VK_COMPONENT_SWIZZLE_B; |
| view.components.a = VK_COMPONENT_SWIZZLE_A; |
| view.subresourceRange.aspectMask = VK_IMAGE_ASPECT_COLOR_BIT; |
| view.subresourceRange.baseMipLevel = 0; |
| view.subresourceRange.levelCount = 1; |
| view.subresourceRange.baseArrayLayer = 0; |
| view.subresourceRange.layerCount = 1; |
| |
| /* create image */ |
| Init(16, 16, 1, tex_format, VK_IMAGE_USAGE_SAMPLED_BIT | VK_IMAGE_USAGE_TRANSFER_DST_BIT, VK_IMAGE_TILING_OPTIMAL); |
| stagingImage.SetLayout(VK_IMAGE_ASPECT_COLOR_BIT, VK_IMAGE_LAYOUT_GENERAL); |
| |
| /* create image view */ |
| view.image = handle(); |
| m_textureView.init(*m_device, view); |
| m_descriptorImageInfo.imageView = m_textureView.handle(); |
| |
| data = stagingImage.MapMemory(); |
| |
| for (y = 0; y < extent().height; y++) { |
| uint32_t *row = (uint32_t *)((char *)data + layout.rowPitch * y); |
| for (x = 0; x < extent().width; x++) row[x] = colors[(x & 1) ^ (y & 1)]; |
| } |
| stagingImage.UnmapMemory(); |
| stagingImage.SetLayout(VK_IMAGE_ASPECT_COLOR_BIT, VK_IMAGE_LAYOUT_TRANSFER_SRC_OPTIMAL); |
| VkImageObj::CopyImage(stagingImage); |
| } |
| |
| VkSamplerObj::VkSamplerObj(VkDeviceObj *device) { |
| m_device = device; |
| |
| VkSamplerCreateInfo samplerCreateInfo; |
| memset(&samplerCreateInfo, 0, sizeof(samplerCreateInfo)); |
| samplerCreateInfo.sType = VK_STRUCTURE_TYPE_SAMPLER_CREATE_INFO; |
| samplerCreateInfo.magFilter = VK_FILTER_NEAREST; |
| samplerCreateInfo.minFilter = VK_FILTER_NEAREST; |
| samplerCreateInfo.mipmapMode = VK_SAMPLER_MIPMAP_MODE_NEAREST; |
| samplerCreateInfo.addressModeU = VK_SAMPLER_ADDRESS_MODE_REPEAT; |
| samplerCreateInfo.addressModeV = VK_SAMPLER_ADDRESS_MODE_REPEAT; |
| samplerCreateInfo.addressModeW = VK_SAMPLER_ADDRESS_MODE_REPEAT; |
| samplerCreateInfo.mipLodBias = 0.0; |
| samplerCreateInfo.anisotropyEnable = VK_FALSE; |
| samplerCreateInfo.maxAnisotropy = 1; |
| samplerCreateInfo.compareOp = VK_COMPARE_OP_NEVER; |
| samplerCreateInfo.minLod = 0.0; |
| samplerCreateInfo.maxLod = 0.0; |
| samplerCreateInfo.borderColor = VK_BORDER_COLOR_FLOAT_OPAQUE_WHITE; |
| samplerCreateInfo.unnormalizedCoordinates = VK_FALSE; |
| |
| init(*m_device, samplerCreateInfo); |
| } |
| |
| /* |
| * Basic ConstantBuffer constructor. Then use create methods to fill in the |
| * details. |
| */ |
| VkConstantBufferObj::VkConstantBufferObj(VkDeviceObj *device, VkBufferUsageFlags usage) { |
| m_device = device; |
| |
| memset(&m_descriptorBufferInfo, 0, sizeof(m_descriptorBufferInfo)); |
| |
| // Special case for usages outside of original limits of framework |
| if ((VK_BUFFER_USAGE_TRANSFER_SRC_BIT | VK_BUFFER_USAGE_TRANSFER_DST_BIT) != usage) { |
| init_no_mem(*m_device, create_info(0, usage)); |
| } |
| } |
| |
| VkConstantBufferObj::VkConstantBufferObj(VkDeviceObj *device, VkDeviceSize allocationSize, const void *data, |
| VkBufferUsageFlags usage) { |
| m_device = device; |
| |
| memset(&m_descriptorBufferInfo, 0, sizeof(m_descriptorBufferInfo)); |
| |
| VkMemoryPropertyFlags reqs = VK_MEMORY_PROPERTY_HOST_VISIBLE_BIT | VK_MEMORY_PROPERTY_HOST_COHERENT_BIT; |
| |
| if ((VK_BUFFER_USAGE_TRANSFER_SRC_BIT | VK_BUFFER_USAGE_TRANSFER_DST_BIT) == usage) { |
| init_as_src_and_dst(*m_device, allocationSize, reqs); |
| } else { |
| init(*m_device, create_info(allocationSize, usage), reqs); |
| } |
| |
| void *pData = memory().map(); |
| memcpy(pData, data, static_cast<size_t>(allocationSize)); |
| memory().unmap(); |
| |
| /* |
| * Constant buffers are going to be used as vertex input buffers |
| * or as shader uniform buffers. So, we'll create the shaderbuffer |
| * descriptor here so it's ready if needed. |
| */ |
| this->m_descriptorBufferInfo.buffer = handle(); |
| this->m_descriptorBufferInfo.offset = 0; |
| this->m_descriptorBufferInfo.range = allocationSize; |
| } |
| |
| VkPipelineShaderStageCreateInfo const &VkShaderObj::GetStageCreateInfo() const { return m_stage_info; } |
| |
| VkShaderObj::VkShaderObj(VkDeviceObj *device, const char *shader_code, VkShaderStageFlagBits stage, VkRenderFramework *framework, |
| char const *name, bool debug, VkSpecializationInfo *specInfo) { |
| VkResult U_ASSERT_ONLY err = VK_SUCCESS; |
| std::vector<unsigned int> spv; |
| VkShaderModuleCreateInfo moduleCreateInfo; |
| |
| m_device = device; |
| m_stage_info.sType = VK_STRUCTURE_TYPE_PIPELINE_SHADER_STAGE_CREATE_INFO; |
| m_stage_info.pNext = nullptr; |
| m_stage_info.flags = 0; |
| m_stage_info.stage = stage; |
| m_stage_info.module = VK_NULL_HANDLE; |
| m_stage_info.pName = name; |
| m_stage_info.pSpecializationInfo = specInfo; |
| |
| moduleCreateInfo.sType = VK_STRUCTURE_TYPE_SHADER_MODULE_CREATE_INFO; |
| moduleCreateInfo.pNext = nullptr; |
| |
| framework->GLSLtoSPV(stage, shader_code, spv, debug); |
| moduleCreateInfo.pCode = spv.data(); |
| moduleCreateInfo.codeSize = spv.size() * sizeof(unsigned int); |
| moduleCreateInfo.flags = 0; |
| |
| err = init_try(*m_device, moduleCreateInfo); |
| m_stage_info.module = handle(); |
| assert(VK_SUCCESS == err); |
| } |
| |
| VkShaderObj::VkShaderObj(VkDeviceObj *device, const std::string spv_source, VkShaderStageFlagBits stage, |
| VkRenderFramework *framework, char const *name, VkSpecializationInfo *specInfo) { |
| VkResult U_ASSERT_ONLY err = VK_SUCCESS; |
| std::vector<unsigned int> spv; |
| VkShaderModuleCreateInfo moduleCreateInfo; |
| |
| m_device = device; |
| m_stage_info.sType = VK_STRUCTURE_TYPE_PIPELINE_SHADER_STAGE_CREATE_INFO; |
| m_stage_info.pNext = nullptr; |
| m_stage_info.flags = 0; |
| m_stage_info.stage = stage; |
| m_stage_info.module = VK_NULL_HANDLE; |
| m_stage_info.pName = name; |
| m_stage_info.pSpecializationInfo = specInfo; |
| |
| moduleCreateInfo.sType = VK_STRUCTURE_TYPE_SHADER_MODULE_CREATE_INFO; |
| moduleCreateInfo.pNext = nullptr; |
| |
| framework->ASMtoSPV(SPV_ENV_VULKAN_1_0, 0, spv_source.data(), spv); |
| moduleCreateInfo.pCode = spv.data(); |
| moduleCreateInfo.codeSize = spv.size() * sizeof(unsigned int); |
| moduleCreateInfo.flags = 0; |
| |
| err = init_try(*m_device, moduleCreateInfo); |
| m_stage_info.module = handle(); |
| assert(VK_SUCCESS == err); |
| } |
| |
| VkPipelineLayoutObj::VkPipelineLayoutObj(VkDeviceObj *device, |
| const std::vector<const VkDescriptorSetLayoutObj *> &descriptor_layouts, |
| const std::vector<VkPushConstantRange> &push_constant_ranges) { |
| VkPipelineLayoutCreateInfo pl_ci = {}; |
| pl_ci.sType = VK_STRUCTURE_TYPE_PIPELINE_LAYOUT_CREATE_INFO; |
| pl_ci.pushConstantRangeCount = static_cast<uint32_t>(push_constant_ranges.size()); |
| pl_ci.pPushConstantRanges = push_constant_ranges.data(); |
| |
| auto descriptor_layouts_unwrapped = MakeTestbindingHandles<const vk_testing::DescriptorSetLayout>(descriptor_layouts); |
| |
| init(*device, pl_ci, descriptor_layouts_unwrapped); |
| } |
| |
| void VkPipelineLayoutObj::Reset() { *this = VkPipelineLayoutObj(); } |
| |
| VkPipelineObj::VkPipelineObj(VkDeviceObj *device) { |
| m_device = device; |
| |
| m_vi_state.sType = VK_STRUCTURE_TYPE_PIPELINE_VERTEX_INPUT_STATE_CREATE_INFO; |
| m_vi_state.pNext = nullptr; |
| m_vi_state.flags = 0; |
| m_vi_state.vertexBindingDescriptionCount = 0; |
| m_vi_state.pVertexBindingDescriptions = nullptr; |
| m_vi_state.vertexAttributeDescriptionCount = 0; |
| m_vi_state.pVertexAttributeDescriptions = nullptr; |
| |
| m_ia_state.sType = VK_STRUCTURE_TYPE_PIPELINE_INPUT_ASSEMBLY_STATE_CREATE_INFO; |
| m_ia_state.pNext = nullptr; |
| m_ia_state.flags = 0; |
| m_ia_state.topology = VK_PRIMITIVE_TOPOLOGY_TRIANGLE_LIST; |
| m_ia_state.primitiveRestartEnable = VK_FALSE; |
| |
| m_te_state = nullptr; |
| |
| m_vp_state.sType = VK_STRUCTURE_TYPE_PIPELINE_VIEWPORT_STATE_CREATE_INFO; |
| m_vp_state.pNext = VK_NULL_HANDLE; |
| m_vp_state.flags = 0; |
| m_vp_state.viewportCount = 1; |
| m_vp_state.scissorCount = 1; |
| m_vp_state.pViewports = nullptr; |
| m_vp_state.pScissors = nullptr; |
| |
| m_rs_state.sType = VK_STRUCTURE_TYPE_PIPELINE_RASTERIZATION_STATE_CREATE_INFO; |
| m_rs_state.pNext = &m_line_state; |
| m_rs_state.flags = 0; |
| m_rs_state.depthClampEnable = VK_FALSE; |
| m_rs_state.rasterizerDiscardEnable = VK_FALSE; |
| m_rs_state.polygonMode = VK_POLYGON_MODE_FILL; |
| m_rs_state.cullMode = VK_CULL_MODE_BACK_BIT; |
| m_rs_state.frontFace = VK_FRONT_FACE_CLOCKWISE; |
| m_rs_state.depthBiasEnable = VK_FALSE; |
| m_rs_state.depthBiasConstantFactor = 0.0f; |
| m_rs_state.depthBiasClamp = 0.0f; |
| m_rs_state.depthBiasSlopeFactor = 0.0f; |
| m_rs_state.lineWidth = 1.0f; |
| |
| m_line_state.sType = VK_STRUCTURE_TYPE_PIPELINE_RASTERIZATION_LINE_STATE_CREATE_INFO_EXT; |
| m_line_state.pNext = nullptr; |
| m_line_state.lineRasterizationMode = VK_LINE_RASTERIZATION_MODE_DEFAULT_EXT; |
| m_line_state.stippledLineEnable = VK_FALSE; |
| m_line_state.lineStippleFactor = 0; |
| m_line_state.lineStipplePattern = 0; |
| |
| m_ms_state.sType = VK_STRUCTURE_TYPE_PIPELINE_MULTISAMPLE_STATE_CREATE_INFO; |
| m_ms_state.pNext = nullptr; |
| m_ms_state.flags = 0; |
| m_ms_state.rasterizationSamples = VK_SAMPLE_COUNT_1_BIT; |
| m_ms_state.sampleShadingEnable = VK_FALSE; |
| m_ms_state.minSampleShading = 0.0f; |
| m_ms_state.pSampleMask = nullptr; |
| m_ms_state.alphaToCoverageEnable = VK_FALSE; |
| m_ms_state.alphaToOneEnable = VK_FALSE; |
| |
| m_ds_state = nullptr; |
| |
| memset(&m_cb_state, 0, sizeof(m_cb_state)); |
| m_cb_state.sType = VK_STRUCTURE_TYPE_PIPELINE_COLOR_BLEND_STATE_CREATE_INFO; |
| m_cb_state.blendConstants[0] = 1.0f; |
| m_cb_state.blendConstants[1] = 1.0f; |
| m_cb_state.blendConstants[2] = 1.0f; |
| m_cb_state.blendConstants[3] = 1.0f; |
| |
| memset(&m_pd_state, 0, sizeof(m_pd_state)); |
| } |
| |
| void VkPipelineObj::AddShader(VkShaderObj *shader) { m_shaderStages.push_back(shader->GetStageCreateInfo()); } |
| |
| void VkPipelineObj::AddShader(VkPipelineShaderStageCreateInfo const &createInfo) { m_shaderStages.push_back(createInfo); } |
| |
| void VkPipelineObj::AddVertexInputAttribs(VkVertexInputAttributeDescription *vi_attrib, uint32_t count) { |
| m_vi_state.pVertexAttributeDescriptions = vi_attrib; |
| m_vi_state.vertexAttributeDescriptionCount = count; |
| } |
| |
| void VkPipelineObj::AddVertexInputBindings(VkVertexInputBindingDescription *vi_binding, uint32_t count) { |
| m_vi_state.pVertexBindingDescriptions = vi_binding; |
| m_vi_state.vertexBindingDescriptionCount = count; |
| } |
| |
| void VkPipelineObj::AddColorAttachment(uint32_t binding, const VkPipelineColorBlendAttachmentState &att) { |
| if (binding + 1 > m_colorAttachments.size()) { |
| m_colorAttachments.resize(binding + 1); |
| } |
| m_colorAttachments[binding] = att; |
| } |
| |
| void VkPipelineObj::SetDepthStencil(const VkPipelineDepthStencilStateCreateInfo *ds_state) { m_ds_state = ds_state; } |
| |
| void VkPipelineObj::SetViewport(const vector<VkViewport> viewports) { |
| m_viewports = viewports; |
| // If we explicitly set a null viewport, pass it through to create info |
| // but preserve viewportCount because it musn't change |
| if (m_viewports.size() == 0) { |
| m_vp_state.pViewports = nullptr; |
| } |
| } |
| |
| void VkPipelineObj::SetScissor(const vector<VkRect2D> scissors) { |
| m_scissors = scissors; |
| // If we explicitly set a null scissor, pass it through to create info |
| // but preserve scissorCount because it musn't change |
| if (m_scissors.size() == 0) { |
| m_vp_state.pScissors = nullptr; |
| } |
| } |
| |
| void VkPipelineObj::MakeDynamic(VkDynamicState state) { |
| /* Only add a state once */ |
| for (auto it = m_dynamic_state_enables.begin(); it != m_dynamic_state_enables.end(); it++) { |
| if ((*it) == state) return; |
| } |
| m_dynamic_state_enables.push_back(state); |
| } |
| |
| void VkPipelineObj::SetMSAA(const VkPipelineMultisampleStateCreateInfo *ms_state) { m_ms_state = *ms_state; } |
| |
| void VkPipelineObj::SetInputAssembly(const VkPipelineInputAssemblyStateCreateInfo *ia_state) { m_ia_state = *ia_state; } |
| |
| void VkPipelineObj::SetRasterization(const VkPipelineRasterizationStateCreateInfo *rs_state) { |
| m_rs_state = *rs_state; |
| m_rs_state.pNext = &m_line_state; |
| } |
| |
| void VkPipelineObj::SetTessellation(const VkPipelineTessellationStateCreateInfo *te_state) { m_te_state = te_state; } |
| |
| void VkPipelineObj::SetLineState(const VkPipelineRasterizationLineStateCreateInfoEXT *line_state) { m_line_state = *line_state; } |
| |
| void VkPipelineObj::InitGraphicsPipelineCreateInfo(VkGraphicsPipelineCreateInfo *gp_ci) { |
| gp_ci->stageCount = m_shaderStages.size(); |
| gp_ci->pStages = m_shaderStages.size() ? m_shaderStages.data() : nullptr; |
| |
| m_vi_state.sType = VK_STRUCTURE_TYPE_PIPELINE_VERTEX_INPUT_STATE_CREATE_INFO; |
| gp_ci->pVertexInputState = &m_vi_state; |
| |
| m_ia_state.sType = VK_STRUCTURE_TYPE_PIPELINE_INPUT_ASSEMBLY_STATE_CREATE_INFO; |
| gp_ci->pInputAssemblyState = &m_ia_state; |
| |
| gp_ci->sType = VK_STRUCTURE_TYPE_GRAPHICS_PIPELINE_CREATE_INFO; |
| gp_ci->pNext = NULL; |
| gp_ci->flags = 0; |
| |
| m_cb_state.attachmentCount = m_colorAttachments.size(); |
| m_cb_state.pAttachments = m_colorAttachments.data(); |
| |
| if (m_viewports.size() > 0) { |
| m_vp_state.viewportCount = m_viewports.size(); |
| m_vp_state.pViewports = m_viewports.data(); |
| } else { |
| MakeDynamic(VK_DYNAMIC_STATE_VIEWPORT); |
| } |
| |
| if (m_scissors.size() > 0) { |
| m_vp_state.scissorCount = m_scissors.size(); |
| m_vp_state.pScissors = m_scissors.data(); |
| } else { |
| MakeDynamic(VK_DYNAMIC_STATE_SCISSOR); |
| } |
| |
| memset(&m_pd_state, 0, sizeof(m_pd_state)); |
| if (m_dynamic_state_enables.size() > 0) { |
| m_pd_state.sType = VK_STRUCTURE_TYPE_PIPELINE_DYNAMIC_STATE_CREATE_INFO; |
| m_pd_state.dynamicStateCount = m_dynamic_state_enables.size(); |
| m_pd_state.pDynamicStates = m_dynamic_state_enables.data(); |
| gp_ci->pDynamicState = &m_pd_state; |
| } |
| |
| gp_ci->subpass = 0; |
| gp_ci->pViewportState = &m_vp_state; |
| gp_ci->pRasterizationState = &m_rs_state; |
| gp_ci->pMultisampleState = &m_ms_state; |
| gp_ci->pDepthStencilState = m_ds_state; |
| gp_ci->pColorBlendState = &m_cb_state; |
| gp_ci->pTessellationState = m_te_state; |
| } |
| |
| VkResult VkPipelineObj::CreateVKPipeline(VkPipelineLayout layout, VkRenderPass render_pass, VkGraphicsPipelineCreateInfo *gp_ci) { |
| VkGraphicsPipelineCreateInfo info = {}; |
| |
| // if not given a CreateInfo, create and initialize a local one. |
| if (gp_ci == nullptr) { |
| gp_ci = &info; |
| InitGraphicsPipelineCreateInfo(gp_ci); |
| } |
| |
| gp_ci->layout = layout; |
| gp_ci->renderPass = render_pass; |
| |
| return init_try(*m_device, *gp_ci); |
| } |
| |
| VkCommandBufferObj::VkCommandBufferObj(VkDeviceObj *device, VkCommandPoolObj *pool, VkCommandBufferLevel level, VkQueueObj *queue) { |
| m_device = device; |
| if (queue) { |
| m_queue = queue; |
| } else { |
| m_queue = m_device->GetDefaultQueue(); |
| } |
| assert(m_queue); |
| |
| auto create_info = vk_testing::CommandBuffer::create_info(pool->handle()); |
| create_info.level = level; |
| init(*device, create_info); |
| } |
| |
| void VkCommandBufferObj::PipelineBarrier(VkPipelineStageFlags src_stages, VkPipelineStageFlags dest_stages, |
| VkDependencyFlags dependencyFlags, uint32_t memoryBarrierCount, |
| const VkMemoryBarrier *pMemoryBarriers, uint32_t bufferMemoryBarrierCount, |
| const VkBufferMemoryBarrier *pBufferMemoryBarriers, uint32_t imageMemoryBarrierCount, |
| const VkImageMemoryBarrier *pImageMemoryBarriers) { |
| vkCmdPipelineBarrier(handle(), src_stages, dest_stages, dependencyFlags, memoryBarrierCount, pMemoryBarriers, |
| bufferMemoryBarrierCount, pBufferMemoryBarriers, imageMemoryBarrierCount, pImageMemoryBarriers); |
| } |
| |
| void VkCommandBufferObj::ClearAllBuffers(const vector<std::unique_ptr<VkImageObj>> &color_objs, VkClearColorValue clear_color, |
| VkDepthStencilObj *depth_stencil_obj, float depth_clear_value, |
| uint32_t stencil_clear_value) { |
| // whatever we want to do, we do it to the whole buffer |
| VkImageSubresourceRange subrange = {}; |
| // srRange.aspectMask to be set later |
| subrange.baseMipLevel = 0; |
| // TODO: Mali device crashing with VK_REMAINING_MIP_LEVELS |
| subrange.levelCount = 1; // VK_REMAINING_MIP_LEVELS; |
| subrange.baseArrayLayer = 0; |
| // TODO: Mesa crashing with VK_REMAINING_ARRAY_LAYERS |
| subrange.layerCount = 1; // VK_REMAINING_ARRAY_LAYERS; |
| |
| const VkImageLayout clear_layout = VK_IMAGE_LAYOUT_TRANSFER_DST_OPTIMAL; |
| |
| for (const auto &color_obj : color_objs) { |
| subrange.aspectMask = VK_IMAGE_ASPECT_COLOR_BIT; |
| color_obj->Layout(VK_IMAGE_LAYOUT_UNDEFINED); |
| color_obj->SetLayout(this, subrange.aspectMask, clear_layout); |
| ClearColorImage(color_obj->image(), clear_layout, &clear_color, 1, &subrange); |
| } |
| |
| if (depth_stencil_obj && depth_stencil_obj->Initialized()) { |
| subrange.aspectMask = VK_IMAGE_ASPECT_DEPTH_BIT | VK_IMAGE_ASPECT_STENCIL_BIT; |
| if (FormatIsDepthOnly(depth_stencil_obj->format())) subrange.aspectMask = VK_IMAGE_ASPECT_DEPTH_BIT; |
| if (FormatIsStencilOnly(depth_stencil_obj->format())) subrange.aspectMask = VK_IMAGE_ASPECT_STENCIL_BIT; |
| |
| depth_stencil_obj->Layout(VK_IMAGE_LAYOUT_UNDEFINED); |
| depth_stencil_obj->SetLayout(this, subrange.aspectMask, clear_layout); |
| |
| VkClearDepthStencilValue clear_value = {depth_clear_value, stencil_clear_value}; |
| ClearDepthStencilImage(depth_stencil_obj->handle(), clear_layout, &clear_value, 1, &subrange); |
| } |
| } |
| |
| void VkCommandBufferObj::FillBuffer(VkBuffer buffer, VkDeviceSize offset, VkDeviceSize fill_size, uint32_t data) { |
| vkCmdFillBuffer(handle(), buffer, offset, fill_size, data); |
| } |
| |
| void VkCommandBufferObj::UpdateBuffer(VkBuffer buffer, VkDeviceSize dstOffset, VkDeviceSize dataSize, const void *pData) { |
| vkCmdUpdateBuffer(handle(), buffer, dstOffset, dataSize, pData); |
| } |
| |
| void VkCommandBufferObj::CopyImage(VkImage srcImage, VkImageLayout srcImageLayout, VkImage dstImage, VkImageLayout dstImageLayout, |
| uint32_t regionCount, const VkImageCopy *pRegions) { |
| vkCmdCopyImage(handle(), srcImage, srcImageLayout, dstImage, dstImageLayout, regionCount, pRegions); |
| } |
| |
| void VkCommandBufferObj::ResolveImage(VkImage srcImage, VkImageLayout srcImageLayout, VkImage dstImage, |
| VkImageLayout dstImageLayout, uint32_t regionCount, const VkImageResolve *pRegions) { |
| vkCmdResolveImage(handle(), srcImage, srcImageLayout, dstImage, dstImageLayout, regionCount, pRegions); |
| } |
| |
| void VkCommandBufferObj::ClearColorImage(VkImage image, VkImageLayout imageLayout, const VkClearColorValue *pColor, |
| uint32_t rangeCount, const VkImageSubresourceRange *pRanges) { |
| vkCmdClearColorImage(handle(), image, imageLayout, pColor, rangeCount, pRanges); |
| } |
| |
| void VkCommandBufferObj::ClearDepthStencilImage(VkImage image, VkImageLayout imageLayout, const VkClearDepthStencilValue *pColor, |
| uint32_t rangeCount, const VkImageSubresourceRange *pRanges) { |
| vkCmdClearDepthStencilImage(handle(), image, imageLayout, pColor, rangeCount, pRanges); |
| } |
| |
| void VkCommandBufferObj::BuildAccelerationStructure(VkAccelerationStructureObj *as, VkBuffer scratchBuffer) { |
| BuildAccelerationStructure(as, scratchBuffer, VK_NULL_HANDLE); |
| } |
| |
| void VkCommandBufferObj::BuildAccelerationStructure(VkAccelerationStructureObj *as, VkBuffer scratchBuffer, VkBuffer instanceData) { |
| PFN_vkCmdBuildAccelerationStructureNV vkCmdBuildAccelerationStructureNV = |
| (PFN_vkCmdBuildAccelerationStructureNV)vkGetDeviceProcAddr(as->dev(), "vkCmdBuildAccelerationStructureNV"); |
| assert(vkCmdBuildAccelerationStructureNV != nullptr); |
| |
| vkCmdBuildAccelerationStructureNV(handle(), &as->info(), instanceData, 0, VK_FALSE, as->handle(), VK_NULL_HANDLE, scratchBuffer, |
| 0); |
| } |
| |
| void VkCommandBufferObj::PrepareAttachments(const vector<std::unique_ptr<VkImageObj>> &color_atts, |
| VkDepthStencilObj *depth_stencil_att) { |
| for (const auto &color_att : color_atts) { |
| color_att->SetLayout(this, VK_IMAGE_ASPECT_COLOR_BIT, VK_IMAGE_LAYOUT_COLOR_ATTACHMENT_OPTIMAL); |
| } |
| |
| if (depth_stencil_att && depth_stencil_att->Initialized()) { |
| VkImageAspectFlags aspect = VK_IMAGE_ASPECT_DEPTH_BIT | VK_IMAGE_ASPECT_STENCIL_BIT; |
| if (FormatIsDepthOnly(depth_stencil_att->Format())) aspect = VK_IMAGE_ASPECT_DEPTH_BIT; |
| if (FormatIsStencilOnly(depth_stencil_att->Format())) aspect = VK_IMAGE_ASPECT_STENCIL_BIT; |
| |
| depth_stencil_att->SetLayout(this, aspect, VK_IMAGE_LAYOUT_DEPTH_STENCIL_ATTACHMENT_OPTIMAL); |
| } |
| } |
| |
| void VkCommandBufferObj::BeginRenderPass(const VkRenderPassBeginInfo &info) { |
| vkCmdBeginRenderPass(handle(), &info, VK_SUBPASS_CONTENTS_INLINE); |
| } |
| |
| void VkCommandBufferObj::EndRenderPass() { vkCmdEndRenderPass(handle()); } |
| |
| void VkCommandBufferObj::SetViewport(uint32_t firstViewport, uint32_t viewportCount, const VkViewport *pViewports) { |
| vkCmdSetViewport(handle(), firstViewport, viewportCount, pViewports); |
| } |
| |
| void VkCommandBufferObj::SetStencilReference(VkStencilFaceFlags faceMask, uint32_t reference) { |
| vkCmdSetStencilReference(handle(), faceMask, reference); |
| } |
| |
| void VkCommandBufferObj::DrawIndexed(uint32_t indexCount, uint32_t instanceCount, uint32_t firstIndex, int32_t vertexOffset, |
| uint32_t firstInstance) { |
| vkCmdDrawIndexed(handle(), indexCount, instanceCount, firstIndex, vertexOffset, firstInstance); |
| } |
| |
| void VkCommandBufferObj::Draw(uint32_t vertexCount, uint32_t instanceCount, uint32_t firstVertex, uint32_t firstInstance) { |
| vkCmdDraw(handle(), vertexCount, instanceCount, firstVertex, firstInstance); |
| } |
| |
| void VkCommandBufferObj::QueueCommandBuffer(bool checkSuccess) { |
| VkFenceObj nullFence; |
| QueueCommandBuffer(nullFence, checkSuccess); |
| } |
| |
| void VkCommandBufferObj::QueueCommandBuffer(const VkFenceObj &fence, bool checkSuccess) { |
| VkResult err = VK_SUCCESS; |
| |
| err = m_queue->submit(*this, fence, checkSuccess); |
| if (checkSuccess) { |
| ASSERT_VK_SUCCESS(err); |
| } |
| |
| err = m_queue->wait(); |
| if (checkSuccess) { |
| ASSERT_VK_SUCCESS(err); |
| } |
| |
| // TODO: Determine if we really want this serialization here |
| // Wait for work to finish before cleaning up. |
| vkDeviceWaitIdle(m_device->device()); |
| } |
| |
| void VkCommandBufferObj::BindDescriptorSet(VkDescriptorSetObj &descriptorSet) { |
| VkDescriptorSet set_obj = descriptorSet.GetDescriptorSetHandle(); |
| |
| // bind pipeline, vertex buffer (descriptor set) and WVP (dynamic buffer view) |
| if (set_obj) { |
| vkCmdBindDescriptorSets(handle(), VK_PIPELINE_BIND_POINT_GRAPHICS, descriptorSet.GetPipelineLayout(), 0, 1, &set_obj, 0, |
| NULL); |
| } |
| } |
| |
| void VkCommandBufferObj::BindIndexBuffer(VkBufferObj *indexBuffer, VkDeviceSize offset, VkIndexType indexType) { |
| vkCmdBindIndexBuffer(handle(), indexBuffer->handle(), offset, indexType); |
| } |
| |
| void VkCommandBufferObj::BindVertexBuffer(VkConstantBufferObj *vertexBuffer, VkDeviceSize offset, uint32_t binding) { |
| vkCmdBindVertexBuffers(handle(), binding, 1, &vertexBuffer->handle(), &offset); |
| } |
| |
| VkCommandPoolObj::VkCommandPoolObj(VkDeviceObj *device, uint32_t queue_family_index, VkCommandPoolCreateFlags flags) { |
| init(*device, vk_testing::CommandPool::create_info(queue_family_index, flags)); |
| } |
| |
| bool VkDepthStencilObj::Initialized() { return m_initialized; } |
| VkDepthStencilObj::VkDepthStencilObj(VkDeviceObj *device) : VkImageObj(device) { m_initialized = false; } |
| |
| VkImageView *VkDepthStencilObj::BindInfo() { return &m_attachmentBindInfo; } |
| |
| VkFormat VkDepthStencilObj::Format() const { return this->m_depth_stencil_fmt; } |
| |
| void VkDepthStencilObj::Init(VkDeviceObj *device, int32_t width, int32_t height, VkFormat format, VkImageUsageFlags usage) { |
| VkImageViewCreateInfo view_info = {}; |
| |
| m_device = device; |
| m_initialized = true; |
| m_depth_stencil_fmt = format; |
| |
| /* create image */ |
| VkImageObj::Init(width, height, 1, m_depth_stencil_fmt, usage, VK_IMAGE_TILING_OPTIMAL); |
| |
| VkImageAspectFlags aspect = VK_IMAGE_ASPECT_STENCIL_BIT | VK_IMAGE_ASPECT_DEPTH_BIT; |
| if (FormatIsDepthOnly(format)) |
| aspect = VK_IMAGE_ASPECT_DEPTH_BIT; |
| else if (FormatIsStencilOnly(format)) |
| aspect = VK_IMAGE_ASPECT_STENCIL_BIT; |
| |
| SetLayout(aspect, VK_IMAGE_LAYOUT_DEPTH_STENCIL_ATTACHMENT_OPTIMAL); |
| |
| view_info.sType = VK_STRUCTURE_TYPE_IMAGE_VIEW_CREATE_INFO; |
| view_info.pNext = NULL; |
| view_info.image = VK_NULL_HANDLE; |
| view_info.subresourceRange.aspectMask = aspect; |
| view_info.subresourceRange.baseMipLevel = 0; |
| view_info.subresourceRange.levelCount = 1; |
| view_info.subresourceRange.baseArrayLayer = 0; |
| view_info.subresourceRange.layerCount = 1; |
| view_info.flags = 0; |
| view_info.format = m_depth_stencil_fmt; |
| view_info.image = handle(); |
| view_info.viewType = VK_IMAGE_VIEW_TYPE_2D; |
| m_imageView.init(*m_device, view_info); |
| |
| m_attachmentBindInfo = m_imageView.handle(); |
| } |
