mesa/src/virtio/vulkan/vn_render_pass.c
Yiwei Zhang dfcad90240 venus: rename common vk object base member to vk
This way we avoid the confusing base.base or even base.base.base when
venus uses common objects. This also aligns with the naming of the other
drivers.

Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/34083>
2025-03-15 22:14:58 +00:00

407 lines
16 KiB
C

/*
* Copyright 2019 Google LLC
* SPDX-License-Identifier: MIT
*
* based in part on anv and radv which are:
* Copyright © 2015 Intel Corporation
* Copyright © 2016 Red Hat.
* Copyright © 2016 Bas Nieuwenhuizen
*/
#include "vn_render_pass.h"
#include "venus-protocol/vn_protocol_driver_framebuffer.h"
#include "venus-protocol/vn_protocol_driver_render_pass.h"
#include "vk_format.h"
#include "vn_device.h"
#include "vn_image.h"
#define COUNT_PRESENT_SRC(atts, att_count, initial_count, final_count) \
do { \
*initial_count = 0; \
*final_count = 0; \
for (uint32_t i = 0; i < att_count; i++) { \
if (atts[i].initialLayout == VK_IMAGE_LAYOUT_PRESENT_SRC_KHR) \
(*initial_count)++; \
if (atts[i].finalLayout == VK_IMAGE_LAYOUT_PRESENT_SRC_KHR) \
(*final_count)++; \
} \
} while (false)
#define REPLACE_PRESENT_SRC(pass, atts, att_count, out_atts) \
do { \
struct vn_present_src_attachment *_acquire_atts = \
pass->present_acquire_attachments; \
struct vn_present_src_attachment *_release_atts = \
pass->present_release_attachments; \
\
memcpy(out_atts, atts, sizeof(*atts) * att_count); \
for (uint32_t i = 0; i < att_count; i++) { \
if (out_atts[i].initialLayout == VK_IMAGE_LAYOUT_PRESENT_SRC_KHR) { \
out_atts[i].initialLayout = VN_PRESENT_SRC_INTERNAL_LAYOUT; \
_acquire_atts->index = i; \
_acquire_atts++; \
} \
if (out_atts[i].finalLayout == VK_IMAGE_LAYOUT_PRESENT_SRC_KHR) { \
out_atts[i].finalLayout = VN_PRESENT_SRC_INTERNAL_LAYOUT; \
_release_atts->index = i; \
_release_atts++; \
} \
} \
} while (false)
#define INIT_SUBPASSES(_pass, _pCreateInfo) \
do { \
for (uint32_t i = 0; i < _pCreateInfo->subpassCount; i++) { \
__auto_type subpass_desc = &_pCreateInfo->pSubpasses[i]; \
struct vn_subpass *subpass = &_pass->subpasses[i]; \
\
for (uint32_t j = 0; j < subpass_desc->colorAttachmentCount; j++) { \
if (subpass_desc->pColorAttachments[j].attachment != \
VK_ATTACHMENT_UNUSED) { \
subpass->attachment_aspects |= VK_IMAGE_ASPECT_COLOR_BIT; \
break; \
} \
} \
\
if (subpass_desc->pDepthStencilAttachment && \
subpass_desc->pDepthStencilAttachment->attachment != \
VK_ATTACHMENT_UNUSED) { \
uint32_t att = \
subpass_desc->pDepthStencilAttachment->attachment; \
subpass->attachment_aspects |= \
vk_format_aspects(_pCreateInfo->pAttachments[att].format); \
} \
} \
} while (false)
static inline void
vn_render_pass_count_present_src(const VkRenderPassCreateInfo *create_info,
uint32_t *initial_count,
uint32_t *final_count)
{
if (VN_PRESENT_SRC_INTERNAL_LAYOUT == VK_IMAGE_LAYOUT_PRESENT_SRC_KHR) {
*initial_count = *final_count = 0;
return;
}
COUNT_PRESENT_SRC(create_info->pAttachments, create_info->attachmentCount,
initial_count, final_count);
}
static inline void
vn_render_pass_count_present_src2(const VkRenderPassCreateInfo2 *create_info,
uint32_t *initial_count,
uint32_t *final_count)
{
if (VN_PRESENT_SRC_INTERNAL_LAYOUT == VK_IMAGE_LAYOUT_PRESENT_SRC_KHR) {
*initial_count = *final_count = 0;
return;
}
COUNT_PRESENT_SRC(create_info->pAttachments, create_info->attachmentCount,
initial_count, final_count);
}
static inline void
vn_render_pass_replace_present_src(struct vn_render_pass *pass,
const VkRenderPassCreateInfo *create_info,
VkAttachmentDescription *out_atts)
{
REPLACE_PRESENT_SRC(pass, create_info->pAttachments,
create_info->attachmentCount, out_atts);
}
static inline void
vn_render_pass_replace_present_src2(struct vn_render_pass *pass,
const VkRenderPassCreateInfo2 *create_info,
VkAttachmentDescription2 *out_atts)
{
REPLACE_PRESENT_SRC(pass, create_info->pAttachments,
create_info->attachmentCount, out_atts);
}
static void
vn_render_pass_setup_present_src_barriers(struct vn_render_pass *pass)
{
/* TODO parse VkSubpassDependency for more accurate barriers */
for (uint32_t i = 0; i < pass->present_acquire_count; i++) {
struct vn_present_src_attachment *att =
&pass->present_acquire_attachments[i];
att->src_stage_mask = VK_PIPELINE_STAGE_ALL_COMMANDS_BIT;
att->src_access_mask = 0;
att->dst_stage_mask = VK_PIPELINE_STAGE_ALL_COMMANDS_BIT;
att->dst_access_mask =
VK_ACCESS_MEMORY_READ_BIT | VK_ACCESS_MEMORY_WRITE_BIT;
}
for (uint32_t i = 0; i < pass->present_release_count; i++) {
struct vn_present_src_attachment *att =
&pass->present_release_attachments[i];
att->src_stage_mask = VK_PIPELINE_STAGE_ALL_COMMANDS_BIT;
att->src_access_mask = VK_ACCESS_MEMORY_WRITE_BIT;
att->dst_stage_mask = VK_PIPELINE_STAGE_BOTTOM_OF_PIPE_BIT;
att->dst_access_mask = 0;
}
}
static struct vn_render_pass *
vn_render_pass_create(struct vn_device *dev,
uint32_t present_acquire_count,
uint32_t present_release_count,
uint32_t subpass_count,
const VkAllocationCallbacks *alloc)
{
uint32_t present_count = present_acquire_count + present_release_count;
struct vn_render_pass *pass;
struct vn_present_src_attachment *present_atts;
struct vn_subpass *subpasses;
VK_MULTIALLOC(ma);
vk_multialloc_add(&ma, &pass, __typeof__(*pass), 1);
vk_multialloc_add(&ma, &present_atts, __typeof__(*present_atts),
present_count);
vk_multialloc_add(&ma, &subpasses, __typeof__(*subpasses), subpass_count);
if (!vk_multialloc_zalloc(&ma, alloc, VK_SYSTEM_ALLOCATION_SCOPE_OBJECT))
return NULL;
vn_object_base_init(&pass->base, VK_OBJECT_TYPE_RENDER_PASS, &dev->base);
pass->present_count = present_count;
pass->present_acquire_count = present_acquire_count;
pass->present_release_count = present_release_count;
pass->subpass_count = subpass_count;
/* For each array pointer, set it only if its count != 0. This allows code
* elsewhere to intuitively use either condition, `foo_atts == NULL` or
* `foo_count != 0`.
*/
if (present_count)
pass->present_attachments = present_atts;
if (present_acquire_count)
pass->present_acquire_attachments = present_atts;
if (present_release_count)
pass->present_release_attachments =
present_atts + present_acquire_count;
if (subpass_count)
pass->subpasses = subpasses;
return pass;
}
/* render pass commands */
VkResult
vn_CreateRenderPass(VkDevice device,
const VkRenderPassCreateInfo *pCreateInfo,
const VkAllocationCallbacks *pAllocator,
VkRenderPass *pRenderPass)
{
struct vn_device *dev = vn_device_from_handle(device);
const VkAllocationCallbacks *alloc =
pAllocator ? pAllocator : &dev->base.vk.alloc;
uint32_t acquire_count;
uint32_t release_count;
vn_render_pass_count_present_src(pCreateInfo, &acquire_count,
&release_count);
struct vn_render_pass *pass = vn_render_pass_create(
dev, acquire_count, release_count, pCreateInfo->subpassCount, alloc);
if (!pass)
return vn_error(dev->instance, VK_ERROR_OUT_OF_HOST_MEMORY);
INIT_SUBPASSES(pass, pCreateInfo);
STACK_ARRAY(VkAttachmentDescription, attachments,
pCreateInfo->attachmentCount);
VkRenderPassCreateInfo local_pass_info;
if (pass->present_count) {
vn_render_pass_replace_present_src(pass, pCreateInfo, attachments);
vn_render_pass_setup_present_src_barriers(pass);
local_pass_info = *pCreateInfo;
local_pass_info.pAttachments = attachments;
pCreateInfo = &local_pass_info;
}
/* Store the viewMask of each subpass for query feedback */
const struct VkRenderPassMultiviewCreateInfo *multiview_info =
vk_find_struct_const(pCreateInfo->pNext,
RENDER_PASS_MULTIVIEW_CREATE_INFO);
if (multiview_info) {
for (uint32_t i = 0; i < multiview_info->subpassCount; i++)
pass->subpasses[i].view_mask = multiview_info->pViewMasks[i];
}
VkRenderPass pass_handle = vn_render_pass_to_handle(pass);
vn_async_vkCreateRenderPass(dev->primary_ring, device, pCreateInfo, NULL,
&pass_handle);
STACK_ARRAY_FINISH(attachments);
*pRenderPass = pass_handle;
return VK_SUCCESS;
}
VkResult
vn_CreateRenderPass2(VkDevice device,
const VkRenderPassCreateInfo2 *pCreateInfo,
const VkAllocationCallbacks *pAllocator,
VkRenderPass *pRenderPass)
{
struct vn_device *dev = vn_device_from_handle(device);
const VkAllocationCallbacks *alloc =
pAllocator ? pAllocator : &dev->base.vk.alloc;
uint32_t acquire_count;
uint32_t release_count;
vn_render_pass_count_present_src2(pCreateInfo, &acquire_count,
&release_count);
struct vn_render_pass *pass = vn_render_pass_create(
dev, acquire_count, release_count, pCreateInfo->subpassCount, alloc);
if (!pass)
return vn_error(dev->instance, VK_ERROR_OUT_OF_HOST_MEMORY);
INIT_SUBPASSES(pass, pCreateInfo);
STACK_ARRAY(VkAttachmentDescription2, attachments,
pCreateInfo->attachmentCount);
VkRenderPassCreateInfo2 local_pass_info;
if (pass->present_count) {
vn_render_pass_replace_present_src2(pass, pCreateInfo, attachments);
vn_render_pass_setup_present_src_barriers(pass);
local_pass_info = *pCreateInfo;
local_pass_info.pAttachments = attachments;
pCreateInfo = &local_pass_info;
}
/* Store the viewMask of each subpass for query feedback */
for (uint32_t i = 0; i < pCreateInfo->subpassCount; i++)
pass->subpasses[i].view_mask = pCreateInfo->pSubpasses[i].viewMask;
VkRenderPass pass_handle = vn_render_pass_to_handle(pass);
vn_async_vkCreateRenderPass2(dev->primary_ring, device, pCreateInfo, NULL,
&pass_handle);
STACK_ARRAY_FINISH(attachments);
*pRenderPass = pass_handle;
return VK_SUCCESS;
}
void
vn_DestroyRenderPass(VkDevice device,
VkRenderPass renderPass,
const VkAllocationCallbacks *pAllocator)
{
struct vn_device *dev = vn_device_from_handle(device);
struct vn_render_pass *pass = vn_render_pass_from_handle(renderPass);
const VkAllocationCallbacks *alloc =
pAllocator ? pAllocator : &dev->base.vk.alloc;
if (!pass)
return;
vn_async_vkDestroyRenderPass(dev->primary_ring, device, renderPass, NULL);
vn_object_base_fini(&pass->base);
vk_free(alloc, pass);
}
void
vn_GetRenderAreaGranularity(VkDevice device,
VkRenderPass renderPass,
VkExtent2D *pGranularity)
{
struct vn_device *dev = vn_device_from_handle(device);
struct vn_render_pass *pass = vn_render_pass_from_handle(renderPass);
if (!pass->granularity.width) {
vn_call_vkGetRenderAreaGranularity(dev->primary_ring, device,
renderPass, &pass->granularity);
}
*pGranularity = pass->granularity;
}
void
vn_GetRenderingAreaGranularity(VkDevice device,
const VkRenderingAreaInfo *pRenderingAreaInfo,
VkExtent2D *pGranularity)
{
struct vn_device *dev = vn_device_from_handle(device);
/* TODO per-device cache */
vn_call_vkGetRenderingAreaGranularity(dev->primary_ring, device,
pRenderingAreaInfo, pGranularity);
}
/* framebuffer commands */
VkResult
vn_CreateFramebuffer(VkDevice device,
const VkFramebufferCreateInfo *pCreateInfo,
const VkAllocationCallbacks *pAllocator,
VkFramebuffer *pFramebuffer)
{
struct vn_device *dev = vn_device_from_handle(device);
const VkAllocationCallbacks *alloc =
pAllocator ? pAllocator : &dev->base.vk.alloc;
/* Two render passes differ only in attachment image layouts are considered
* compatible. We must not use pCreateInfo->renderPass here.
*/
const bool imageless =
pCreateInfo->flags & VK_FRAMEBUFFER_CREATE_IMAGELESS_BIT;
const uint32_t view_count = imageless ? 0 : pCreateInfo->attachmentCount;
struct vn_framebuffer *fb =
vk_zalloc(alloc, sizeof(*fb) + sizeof(*fb->image_views) * view_count,
VN_DEFAULT_ALIGN, VK_SYSTEM_ALLOCATION_SCOPE_OBJECT);
if (!fb)
return vn_error(dev->instance, VK_ERROR_OUT_OF_HOST_MEMORY);
vn_object_base_init(&fb->base, VK_OBJECT_TYPE_FRAMEBUFFER, &dev->base);
fb->image_view_count = view_count;
memcpy(fb->image_views, pCreateInfo->pAttachments,
sizeof(*pCreateInfo->pAttachments) * view_count);
VkFramebuffer fb_handle = vn_framebuffer_to_handle(fb);
vn_async_vkCreateFramebuffer(dev->primary_ring, device, pCreateInfo, NULL,
&fb_handle);
*pFramebuffer = fb_handle;
return VK_SUCCESS;
}
void
vn_DestroyFramebuffer(VkDevice device,
VkFramebuffer framebuffer,
const VkAllocationCallbacks *pAllocator)
{
struct vn_device *dev = vn_device_from_handle(device);
struct vn_framebuffer *fb = vn_framebuffer_from_handle(framebuffer);
const VkAllocationCallbacks *alloc =
pAllocator ? pAllocator : &dev->base.vk.alloc;
if (!fb)
return;
vn_async_vkDestroyFramebuffer(dev->primary_ring, device, framebuffer,
NULL);
vn_object_base_fini(&fb->base);
vk_free(alloc, fb);
}