1 /* Copyright (c) 2017-2023 Hans-Kristian Arntzen
2 *
3 * SPDX-License-Identifier: MIT
4 */
5
6 #include <assert.h>
7 #include <stdbool.h>
8
9 #include "radv_meta.h"
10 #include "sid.h"
11 #include "vk_common_entrypoints.h"
12 #include "vk_format.h"
13
14 static void
decode_astc(struct radv_cmd_buffer * cmd_buffer,struct radv_image_view * src_iview,struct radv_image_view * dst_iview,VkImageLayout layout,const VkOffset3D * offset,const VkExtent3D * extent)15 decode_astc(struct radv_cmd_buffer *cmd_buffer, struct radv_image_view *src_iview, struct radv_image_view *dst_iview,
16 VkImageLayout layout, const VkOffset3D *offset, const VkExtent3D *extent)
17 {
18 struct radv_device *device = radv_cmd_buffer_device(cmd_buffer);
19 struct radv_meta_state *state = &device->meta_state;
20 struct vk_texcompress_astc_write_descriptor_set write_desc_set;
21 VkFormat format = src_iview->image->vk.format;
22 int blk_w = vk_format_get_blockwidth(format);
23 int blk_h = vk_format_get_blockheight(format);
24
25 vk_texcompress_astc_fill_write_descriptor_sets(state->astc_decode, &write_desc_set,
26 radv_image_view_to_handle(src_iview), layout,
27 radv_image_view_to_handle(dst_iview), format);
28 radv_meta_push_descriptor_set(cmd_buffer, VK_PIPELINE_BIND_POINT_COMPUTE, state->astc_decode->p_layout, 0,
29 VK_TEXCOMPRESS_ASTC_WRITE_DESC_SET_COUNT, write_desc_set.descriptor_set);
30
31 VkPipeline pipeline =
32 vk_texcompress_astc_get_decode_pipeline(&device->vk, &state->alloc, state->astc_decode, state->cache, format);
33 if (pipeline == VK_NULL_HANDLE)
34 return;
35
36 radv_CmdBindPipeline(radv_cmd_buffer_to_handle(cmd_buffer), VK_PIPELINE_BIND_POINT_COMPUTE, pipeline);
37
38 bool is_3Dimage = (src_iview->image->vk.image_type == VK_IMAGE_TYPE_3D) ? true : false;
39 int push_constants[5] = {offset->x / blk_w, offset->y / blk_h, extent->width + offset->x, extent->height + offset->y,
40 is_3Dimage};
41 vk_common_CmdPushConstants(radv_cmd_buffer_to_handle(cmd_buffer), device->meta_state.etc_decode.pipeline_layout,
42 VK_SHADER_STAGE_COMPUTE_BIT, 0, 20, push_constants);
43
44 struct radv_dispatch_info info = {
45 .blocks[0] = DIV_ROUND_UP(extent->width, blk_w * 2),
46 .blocks[1] = DIV_ROUND_UP(extent->height, blk_h * 2),
47 .blocks[2] = extent->depth,
48 .offsets[0] = 0,
49 .offsets[1] = 0,
50 .offsets[2] = offset->z,
51 .unaligned = 0,
52 };
53 radv_compute_dispatch(cmd_buffer, &info);
54 }
55
56 static VkImageViewType
get_view_type(const struct radv_image * image)57 get_view_type(const struct radv_image *image)
58 {
59 switch (image->vk.image_type) {
60 case VK_IMAGE_TYPE_2D:
61 return VK_IMAGE_VIEW_TYPE_2D_ARRAY;
62 case VK_IMAGE_TYPE_3D:
63 return VK_IMAGE_VIEW_TYPE_3D;
64 default:
65 unreachable("bad VkImageViewType");
66 }
67 }
68
69 static void
image_view_init(struct radv_device * device,struct radv_image * image,VkFormat format,VkImageAspectFlags aspectMask,uint32_t baseMipLevel,uint32_t baseArrayLayer,uint32_t layerCount,struct radv_image_view * iview)70 image_view_init(struct radv_device *device, struct radv_image *image, VkFormat format, VkImageAspectFlags aspectMask,
71 uint32_t baseMipLevel, uint32_t baseArrayLayer, uint32_t layerCount, struct radv_image_view *iview)
72 {
73 VkImageViewCreateInfo iview_create_info = {
74 .sType = VK_STRUCTURE_TYPE_IMAGE_VIEW_CREATE_INFO,
75 .image = radv_image_to_handle(image),
76 .viewType = get_view_type(image),
77 .format = format,
78 .subresourceRange =
79 {
80 .aspectMask = aspectMask,
81 .baseMipLevel = baseMipLevel,
82 .levelCount = 1,
83 .baseArrayLayer = 0,
84 .layerCount = baseArrayLayer + layerCount,
85 },
86 };
87
88 radv_image_view_init(iview, device, &iview_create_info, NULL);
89 }
90
91 void
radv_meta_decode_astc(struct radv_cmd_buffer * cmd_buffer,struct radv_image * image,VkImageLayout layout,const VkImageSubresourceLayers * subresource,VkOffset3D offset,VkExtent3D extent)92 radv_meta_decode_astc(struct radv_cmd_buffer *cmd_buffer, struct radv_image *image, VkImageLayout layout,
93 const VkImageSubresourceLayers *subresource, VkOffset3D offset, VkExtent3D extent)
94 {
95 struct radv_device *device = radv_cmd_buffer_device(cmd_buffer);
96 struct radv_meta_saved_state saved_state;
97 radv_meta_save(&saved_state, cmd_buffer,
98 RADV_META_SAVE_COMPUTE_PIPELINE | RADV_META_SAVE_CONSTANTS | RADV_META_SAVE_DESCRIPTORS |
99 RADV_META_SUSPEND_PREDICATING);
100
101 uint32_t base_slice = radv_meta_get_iview_layer(image, subresource, &offset);
102 uint32_t slice_count = image->vk.image_type == VK_IMAGE_TYPE_3D
103 ? extent.depth
104 : vk_image_subresource_layer_count(&image->vk, subresource);
105
106 extent = vk_image_sanitize_extent(&image->vk, extent);
107 offset = vk_image_sanitize_offset(&image->vk, offset);
108
109 struct radv_image_view src_iview, dst_iview;
110 image_view_init(device, image, VK_FORMAT_R32G32B32A32_UINT, VK_IMAGE_ASPECT_COLOR_BIT, subresource->mipLevel,
111 subresource->baseArrayLayer, vk_image_subresource_layer_count(&image->vk, subresource), &src_iview);
112 image_view_init(device, image, VK_FORMAT_R8G8B8A8_UINT, VK_IMAGE_ASPECT_PLANE_1_BIT, subresource->mipLevel,
113 subresource->baseArrayLayer, vk_image_subresource_layer_count(&image->vk, subresource), &dst_iview);
114
115 VkExtent3D extent_copy = {
116 .width = extent.width,
117 .height = extent.height,
118 .depth = slice_count,
119 };
120 decode_astc(cmd_buffer, &src_iview, &dst_iview, layout, &(VkOffset3D){offset.x, offset.y, base_slice}, &extent_copy);
121
122 radv_image_view_finish(&src_iview);
123 radv_image_view_finish(&dst_iview);
124
125 radv_meta_restore(&saved_state, cmd_buffer);
126 }
127