xref: /aosp_15_r20/external/mesa3d/src/freedreno/vulkan/tu_sampler.cc (revision 6104692788411f58d303aa86923a9ff6ecaded22)
1 /*
2  * Copyright © 2016 Red Hat.
3  * Copyright © 2016 Bas Nieuwenhuizen
4  * SPDX-License-Identifier: MIT
5  *
6  * based in part on anv driver which is:
7  * Copyright © 2015 Intel Corporation
8  */
9 
10 #include "tu_sampler.h"
11 
12 #include "tu_device.h"
13 #include "tu_util.h"
14 
15 VKAPI_ATTR VkResult VKAPI_CALL
tu_CreateSampler(VkDevice _device,const VkSamplerCreateInfo * pCreateInfo,const VkAllocationCallbacks * pAllocator,VkSampler * pSampler)16 tu_CreateSampler(VkDevice _device,
17                  const VkSamplerCreateInfo *pCreateInfo,
18                  const VkAllocationCallbacks *pAllocator,
19                  VkSampler *pSampler)
20 {
21    VK_FROM_HANDLE(tu_device, device, _device);
22    struct tu_sampler *sampler;
23 
24    assert(pCreateInfo->sType == VK_STRUCTURE_TYPE_SAMPLER_CREATE_INFO);
25 
26    sampler = (struct tu_sampler *) vk_sampler_create(
27       &device->vk, pCreateInfo, pAllocator, sizeof(*sampler));
28    if (!sampler)
29       return vk_error(device, VK_ERROR_OUT_OF_HOST_MEMORY);
30 
31    const struct VkSamplerYcbcrConversionInfo *ycbcr_conversion =
32       vk_find_struct_const(pCreateInfo->pNext,  SAMPLER_YCBCR_CONVERSION_INFO);
33    /* for non-custom border colors, the VK enum is translated directly to an offset in
34     * the border color buffer. custom border colors are located immediately after the
35     * builtin colors, and thus an offset of TU_BORDER_COLOR_BUILTIN is added.
36     */
37    uint32_t border_color = (unsigned) pCreateInfo->borderColor;
38    if (vk_border_color_is_custom(pCreateInfo->borderColor)) {
39       mtx_lock(&device->mutex);
40       border_color = BITSET_FFS(device->custom_border_color) - 1;
41       assert(border_color < TU_BORDER_COLOR_COUNT);
42       BITSET_CLEAR(device->custom_border_color, border_color);
43       mtx_unlock(&device->mutex);
44 
45       VkClearColorValue color = sampler->vk.border_color_value;
46       if (sampler->vk.format == VK_FORMAT_D24_UNORM_S8_UINT &&
47           pCreateInfo->borderColor == VK_BORDER_COLOR_INT_CUSTOM_EXT &&
48           device->use_z24uint_s8uint) {
49          /* When sampling stencil using the special Z24UINT_S8UINT format, the
50           * border color is in the second component. Note: if
51           * customBorderColorWithoutFormat is enabled, we may miss doing this
52           * here if the format isn't specified, which is why we don't use that
53           * format.
54           */
55          color.uint32[1] = color.uint32[0];
56       }
57 
58       tu6_pack_border_color(
59          &device->global_bo_map->bcolor[border_color], &color,
60          pCreateInfo->borderColor == VK_BORDER_COLOR_INT_CUSTOM_EXT);
61       border_color += TU_BORDER_COLOR_BUILTIN;
62    }
63 
64    unsigned aniso = pCreateInfo->anisotropyEnable ?
65       util_last_bit(MIN2((uint32_t)pCreateInfo->maxAnisotropy >> 1, 8)) : 0;
66    bool miplinear = (pCreateInfo->mipmapMode == VK_SAMPLER_MIPMAP_MODE_LINEAR);
67    float min_lod = CLAMP(pCreateInfo->minLod, 0.0f, 4095.0f / 256.0f);
68    float max_lod = CLAMP(pCreateInfo->maxLod, 0.0f, 4095.0f / 256.0f);
69 
70    sampler->descriptor[0] =
71       COND(miplinear, A6XX_TEX_SAMP_0_MIPFILTER_LINEAR_NEAR) |
72       A6XX_TEX_SAMP_0_XY_MAG(tu6_tex_filter(pCreateInfo->magFilter, aniso)) |
73       A6XX_TEX_SAMP_0_XY_MIN(tu6_tex_filter(pCreateInfo->minFilter, aniso)) |
74       A6XX_TEX_SAMP_0_ANISO((enum a6xx_tex_aniso) aniso) |
75       A6XX_TEX_SAMP_0_WRAP_S(tu6_tex_wrap(pCreateInfo->addressModeU)) |
76       A6XX_TEX_SAMP_0_WRAP_T(tu6_tex_wrap(pCreateInfo->addressModeV)) |
77       A6XX_TEX_SAMP_0_WRAP_R(tu6_tex_wrap(pCreateInfo->addressModeW)) |
78       A6XX_TEX_SAMP_0_LOD_BIAS(pCreateInfo->mipLodBias);
79    sampler->descriptor[1] =
80       COND(pCreateInfo->flags & VK_SAMPLER_CREATE_NON_SEAMLESS_CUBE_MAP_BIT_EXT,
81            A6XX_TEX_SAMP_1_CUBEMAPSEAMLESSFILTOFF) |
82       COND(pCreateInfo->unnormalizedCoordinates, A6XX_TEX_SAMP_1_UNNORM_COORDS) |
83       A6XX_TEX_SAMP_1_MIN_LOD(min_lod) |
84       A6XX_TEX_SAMP_1_MAX_LOD(max_lod) |
85       COND(pCreateInfo->compareEnable,
86            A6XX_TEX_SAMP_1_COMPARE_FUNC(tu6_compare_func(pCreateInfo->compareOp)));
87    sampler->descriptor[2] = A6XX_TEX_SAMP_2_BCOLOR(border_color);
88    sampler->descriptor[3] = 0;
89 
90    if (sampler->vk.reduction_mode != VK_SAMPLER_REDUCTION_MODE_WEIGHTED_AVERAGE) {
91       sampler->descriptor[2] |= A6XX_TEX_SAMP_2_REDUCTION_MODE(
92          tu6_reduction_mode(sampler->vk.reduction_mode));
93    }
94 
95    sampler->vk.ycbcr_conversion = ycbcr_conversion ?
96       vk_ycbcr_conversion_from_handle(ycbcr_conversion->conversion) : NULL;
97 
98    if (sampler->vk.ycbcr_conversion &&
99        sampler->vk.ycbcr_conversion->state.chroma_filter == VK_FILTER_LINEAR) {
100       sampler->descriptor[2] |= A6XX_TEX_SAMP_2_CHROMA_LINEAR;
101    }
102 
103    /* TODO:
104     * A6XX_TEX_SAMP_1_MIPFILTER_LINEAR_FAR disables mipmapping, but vk has no NONE mipfilter?
105     */
106 
107    *pSampler = tu_sampler_to_handle(sampler);
108 
109    return VK_SUCCESS;
110 }
111 
112 VKAPI_ATTR void VKAPI_CALL
tu_DestroySampler(VkDevice _device,VkSampler _sampler,const VkAllocationCallbacks * pAllocator)113 tu_DestroySampler(VkDevice _device,
114                   VkSampler _sampler,
115                   const VkAllocationCallbacks *pAllocator)
116 {
117    VK_FROM_HANDLE(tu_device, device, _device);
118    VK_FROM_HANDLE(tu_sampler, sampler, _sampler);
119    uint32_t border_color;
120 
121    if (!sampler)
122       return;
123 
124    border_color = (sampler->descriptor[2] & A6XX_TEX_SAMP_2_BCOLOR__MASK) >> A6XX_TEX_SAMP_2_BCOLOR__SHIFT;
125    if (border_color >= TU_BORDER_COLOR_BUILTIN) {
126       border_color -= TU_BORDER_COLOR_BUILTIN;
127       /* if the sampler had a custom border color, free it. TODO: no lock */
128       mtx_lock(&device->mutex);
129       assert(!BITSET_TEST(device->custom_border_color, border_color));
130       BITSET_SET(device->custom_border_color, border_color);
131       mtx_unlock(&device->mutex);
132    }
133 
134    vk_sampler_destroy(&device->vk, pAllocator, &sampler->vk);
135 }
136