lavapipe: Handle multiple planes in GetDescriptorEXT

Fixes: a13a07d ("lavapipe: add descriptor sets bindings for planar images")
Reviewed-by: Mike Blumenkrantz <michael.blumenkrantz@gmail.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/28697>
This commit is contained in:
Konstantin Seurer
2024-04-11 19:49:29 +02:00
committed by Marge Bot
parent c48d6097cd
commit 8f5fb4e095

View File

@@ -1092,20 +1092,28 @@ VKAPI_ATTR void VKAPI_CALL lvp_GetDescriptorEXT(
if (info && info->imageView) {
LVP_FROM_HANDLE(lvp_image_view, iview, info->imageView);
lp_jit_texture_from_pipe(&desc->texture, iview->planes[0].sv);
desc->functions = iview->planes[0].texture_handle->functions;
unsigned plane_count = iview->plane_count;
if (info->sampler) {
LVP_FROM_HANDLE(lvp_sampler, sampler, info->sampler);
desc->sampler = sampler->desc.sampler;
desc->texture.sampler_index = sampler->desc.texture.sampler_index;
} else {
lp_jit_sampler_from_pipe(&desc->sampler, &sampler);
desc->texture.sampler_index = 0;
for (unsigned p = 0; p < plane_count; p++) {
lp_jit_texture_from_pipe(&desc[p].texture, iview->planes[p].sv);
desc[p].functions = iview->planes[p].texture_handle->functions;
if (info->sampler) {
LVP_FROM_HANDLE(lvp_sampler, sampler, info->sampler);
desc[p].sampler = sampler->desc.sampler;
desc[p].texture.sampler_index = sampler->desc.texture.sampler_index;
} else {
lp_jit_sampler_from_pipe(&desc->sampler, &sampler);
desc[p].texture.sampler_index = 0;
}
}
} else {
desc->functions = device->null_texture_handle->functions;
desc->texture.sampler_index = 0;
unsigned plane_count = size / sizeof(struct lp_descriptor);
for (unsigned p = 0; p < plane_count; p++) {
desc[p].functions = device->null_texture_handle->functions;
desc[p].texture.sampler_index = 0;
}
}
break;
@@ -1114,11 +1122,20 @@ VKAPI_ATTR void VKAPI_CALL lvp_GetDescriptorEXT(
case VK_DESCRIPTOR_TYPE_SAMPLED_IMAGE: {
if (pCreateInfo->data.pSampledImage && pCreateInfo->data.pSampledImage->imageView) {
LVP_FROM_HANDLE(lvp_image_view, iview, pCreateInfo->data.pSampledImage->imageView);
lp_jit_texture_from_pipe(&desc->texture, iview->planes[0].sv);
desc->functions = iview->planes[0].texture_handle->functions;
unsigned plane_count = iview->plane_count;
for (unsigned p = 0; p < plane_count; p++) {
lp_jit_texture_from_pipe(&desc[p].texture, iview->planes[p].sv);
desc[p].functions = iview->planes[p].texture_handle->functions;
}
} else {
desc->functions = device->null_texture_handle->functions;
desc->texture.sampler_index = 0;
unsigned plane_count = size / sizeof(struct lp_descriptor);
for (unsigned p = 0; p < plane_count; p++) {
desc[p].functions = device->null_texture_handle->functions;
desc[p].texture.sampler_index = 0;
}
}
break;
}
@@ -1128,10 +1145,18 @@ VKAPI_ATTR void VKAPI_CALL lvp_GetDescriptorEXT(
case VK_DESCRIPTOR_TYPE_INPUT_ATTACHMENT: {
if (pCreateInfo->data.pStorageImage && pCreateInfo->data.pStorageImage->imageView) {
LVP_FROM_HANDLE(lvp_image_view, iview, pCreateInfo->data.pStorageImage->imageView);
lp_jit_image_from_pipe(&desc->image, &iview->planes[0].iv);
desc->functions = iview->planes[0].image_handle->functions;
unsigned plane_count = iview->plane_count;
for (unsigned p = 0; p < plane_count; p++) {
lp_jit_image_from_pipe(&desc[p].image, &iview->planes[p].iv);
desc[p].functions = iview->planes[p].image_handle->functions;
}
} else {
desc->functions = device->null_image_handle->functions;
unsigned plane_count = size / sizeof(struct lp_descriptor);
for (unsigned p = 0; p < plane_count; p++)
desc[p].functions = device->null_image_handle->functions;
}
break;
}