xref: /aosp_15_r20/external/mesa3d/src/gallium/drivers/zink/zink_format.c (revision 6104692788411f58d303aa86923a9ff6ecaded22)
1 #include "util/format/u_format.h"
2 #include "zink_format.h"
3 #include "util/u_math.h"
4 
5 enum pipe_format
zink_decompose_vertex_format(enum pipe_format format)6 zink_decompose_vertex_format(enum pipe_format format)
7 {
8    const struct util_format_description *desc = util_format_description(format);
9    unsigned first_non_void = util_format_get_first_non_void_channel(format);
10    enum pipe_format new_format;
11    assert(first_non_void == 0);
12    if (!desc->is_array)
13       return PIPE_FORMAT_NONE;
14    if (desc->is_unorm) {
15       enum pipe_format unorm_formats[] = {
16          PIPE_FORMAT_R8_UNORM,
17          PIPE_FORMAT_R16_UNORM,
18          PIPE_FORMAT_R32_UNORM
19       };
20       return unorm_formats[desc->channel[first_non_void].size >> 4];
21    } else if (desc->is_snorm) {
22       enum pipe_format snorm_formats[] = {
23          PIPE_FORMAT_R8_SNORM,
24          PIPE_FORMAT_R16_SNORM,
25          PIPE_FORMAT_R32_SNORM
26       };
27       return snorm_formats[desc->channel[first_non_void].size >> 4];
28    } else {
29       enum pipe_format uint_formats[][3] = {
30          {PIPE_FORMAT_R8_USCALED, PIPE_FORMAT_R16_USCALED, PIPE_FORMAT_R32_USCALED},
31          {PIPE_FORMAT_R8_UINT, PIPE_FORMAT_R16_UINT, PIPE_FORMAT_R32_UINT},
32       };
33       enum pipe_format sint_formats[][3] = {
34          {PIPE_FORMAT_R8_SSCALED, PIPE_FORMAT_R16_SSCALED, PIPE_FORMAT_R32_SSCALED},
35          {PIPE_FORMAT_R8_SINT, PIPE_FORMAT_R16_SINT, PIPE_FORMAT_R32_SINT},
36       };
37       switch (desc->channel[first_non_void].type) {
38       case UTIL_FORMAT_TYPE_UNSIGNED:
39          return uint_formats[desc->channel[first_non_void].pure_integer][desc->channel[first_non_void].size >> 4];
40       case UTIL_FORMAT_TYPE_SIGNED:
41          return sint_formats[desc->channel[first_non_void].pure_integer][desc->channel[first_non_void].size >> 4];
42       case UTIL_FORMAT_TYPE_FLOAT:
43          return desc->channel[first_non_void].size == 16 ? PIPE_FORMAT_R16_FLOAT : PIPE_FORMAT_R32_FLOAT;
44          break;
45       default:
46          return PIPE_FORMAT_NONE;
47       }
48    }
49    return new_format;
50 }
51 
52 bool
zink_format_is_red_alpha(enum pipe_format format)53 zink_format_is_red_alpha(enum pipe_format format)
54 {
55    switch (format) {
56    case PIPE_FORMAT_R4A4_UNORM:
57    case PIPE_FORMAT_R8A8_SINT:
58    case PIPE_FORMAT_R8A8_SNORM:
59    case PIPE_FORMAT_R8A8_UINT:
60    case PIPE_FORMAT_R8A8_UNORM:
61    case PIPE_FORMAT_R16A16_SINT:
62    case PIPE_FORMAT_R16A16_SNORM:
63    case PIPE_FORMAT_R16A16_UINT:
64    case PIPE_FORMAT_R16A16_UNORM:
65    case PIPE_FORMAT_R16A16_FLOAT:
66    case PIPE_FORMAT_R32A32_SINT:
67    case PIPE_FORMAT_R32A32_UINT:
68    case PIPE_FORMAT_R32A32_FLOAT:
69       return true;
70    default: break;
71    }
72    return false;
73 }
74 
75 bool
zink_format_is_emulated_alpha(enum pipe_format format)76 zink_format_is_emulated_alpha(enum pipe_format format)
77 {
78    return util_format_is_alpha(format) ||
79           util_format_is_luminance(format) ||
80           util_format_is_luminance_alpha(format) ||
81           zink_format_is_red_alpha(format);
82 }
83 
84 static enum pipe_format
emulate_alpha(enum pipe_format format)85 emulate_alpha(enum pipe_format format)
86 {
87    if (format == PIPE_FORMAT_A8_UNORM)
88       return PIPE_FORMAT_R8_UNORM;
89    if (format == PIPE_FORMAT_A8_UINT)
90       return PIPE_FORMAT_R8_UINT;
91    if (format == PIPE_FORMAT_A8_SNORM)
92       return PIPE_FORMAT_R8_SNORM;
93    if (format == PIPE_FORMAT_A8_SINT)
94       return PIPE_FORMAT_R8_SINT;
95    if (format == PIPE_FORMAT_A16_UNORM)
96       return PIPE_FORMAT_R16_UNORM;
97    if (format == PIPE_FORMAT_A16_UINT)
98       return PIPE_FORMAT_R16_UINT;
99    if (format == PIPE_FORMAT_A16_SNORM)
100       return PIPE_FORMAT_R16_SNORM;
101    if (format == PIPE_FORMAT_A16_SINT)
102       return PIPE_FORMAT_R16_SINT;
103    if (format == PIPE_FORMAT_A16_FLOAT)
104       return PIPE_FORMAT_R16_FLOAT;
105    if (format == PIPE_FORMAT_A32_UINT)
106       return PIPE_FORMAT_R32_UINT;
107    if (format == PIPE_FORMAT_A32_SINT)
108       return PIPE_FORMAT_R32_SINT;
109    if (format == PIPE_FORMAT_A32_FLOAT)
110       return PIPE_FORMAT_R32_FLOAT;
111    return format;
112 }
113 
114 static enum pipe_format
emulate_red_alpha(enum pipe_format format)115 emulate_red_alpha(enum pipe_format format)
116 {
117    switch (format) {
118    case PIPE_FORMAT_R8A8_SINT:
119       return PIPE_FORMAT_R8G8_SINT;
120    case PIPE_FORMAT_R8A8_SNORM:
121       return PIPE_FORMAT_R8G8_SNORM;
122    case PIPE_FORMAT_R8A8_UINT:
123       return PIPE_FORMAT_R8G8_UINT;
124    case PIPE_FORMAT_R8A8_UNORM:
125       return PIPE_FORMAT_R8G8_UNORM;
126    case PIPE_FORMAT_R16A16_SINT:
127       return PIPE_FORMAT_R16G16_SINT;
128    case PIPE_FORMAT_R16A16_SNORM:
129       return PIPE_FORMAT_R16G16_SNORM;
130    case PIPE_FORMAT_R16A16_UINT:
131       return PIPE_FORMAT_R16G16_UINT;
132    case PIPE_FORMAT_R16A16_UNORM:
133       return PIPE_FORMAT_R16G16_UNORM;
134    case PIPE_FORMAT_R16A16_FLOAT:
135       return PIPE_FORMAT_R16G16_FLOAT;
136    case PIPE_FORMAT_R32A32_SINT:
137       return PIPE_FORMAT_R32G32_SINT;
138    case PIPE_FORMAT_R32A32_UINT:
139       return PIPE_FORMAT_R32G32_UINT;
140    case PIPE_FORMAT_R32A32_FLOAT:
141       return PIPE_FORMAT_R32G32_FLOAT;
142    default: break;
143    }
144    return format;
145 }
146 
147 enum pipe_format
zink_format_get_emulated_alpha(enum pipe_format format)148 zink_format_get_emulated_alpha(enum pipe_format format)
149 {
150    if (util_format_is_alpha(format))
151       return emulate_alpha(format);
152    if (util_format_is_luminance(format))
153       return util_format_luminance_to_red(format);
154    if (util_format_is_luminance_alpha(format)) {
155       if (util_format_is_srgb(format))
156          return format;
157       if (format == PIPE_FORMAT_LATC2_UNORM)
158          return PIPE_FORMAT_RGTC2_UNORM;
159       if (format == PIPE_FORMAT_LATC2_SNORM)
160          return PIPE_FORMAT_RGTC2_SNORM;
161 
162       format = util_format_luminance_to_red(format);
163    }
164 
165    return emulate_red_alpha(format);
166 }
167 
168 bool
zink_format_is_voidable_rgba_variant(enum pipe_format format)169 zink_format_is_voidable_rgba_variant(enum pipe_format format)
170 {
171    const struct util_format_description *desc = util_format_description(format);
172    unsigned chan;
173 
174    if(desc->block.width != 1 ||
175       desc->block.height != 1 ||
176       (desc->block.bits != 32 && desc->block.bits != 64 &&
177        desc->block.bits != 128))
178       return false;
179 
180    if (desc->nr_channels != 4)
181       return false;
182 
183    unsigned size = desc->channel[0].size;
184    for(chan = 0; chan < 4; ++chan) {
185       if(desc->channel[chan].size != size)
186          return false;
187    }
188 
189    return true;
190 }
191 
192 void
zink_format_clamp_channel_color(const struct util_format_description * desc,union pipe_color_union * dst,const union pipe_color_union * src,unsigned i)193 zink_format_clamp_channel_color(const struct util_format_description *desc, union pipe_color_union *dst, const union pipe_color_union *src, unsigned i)
194 {
195    int non_void = util_format_get_first_non_void_channel(desc->format);
196    unsigned channel = desc->swizzle[i];
197 
198    if (channel > PIPE_SWIZZLE_W || desc->channel[channel].type == UTIL_FORMAT_TYPE_VOID) {
199       if (non_void != -1) {
200          if (desc->channel[non_void].type == UTIL_FORMAT_TYPE_FLOAT) {
201             dst->f[i] = uif(UINT32_MAX);
202          } else {
203             if (desc->channel[non_void].normalized)
204                dst->f[i] = 1.0;
205             else if (desc->channel[non_void].type == UTIL_FORMAT_TYPE_SIGNED)
206                dst->i[i] = INT32_MAX;
207             else
208                dst->ui[i] = UINT32_MAX;
209          }
210       } else {
211          dst->ui[i] = src->ui[i];
212       }
213       return;
214    }
215 
216    switch (desc->channel[channel].type) {
217    case UTIL_FORMAT_TYPE_VOID:
218       unreachable("handled above");
219       break;
220    case UTIL_FORMAT_TYPE_SIGNED:
221       if (desc->channel[channel].normalized)
222          dst->i[i] = src->i[i];
223       else {
224          dst->i[i] = MAX2(src->i[i], -(1<<(desc->channel[channel].size - 1)));
225          dst->i[i] = MIN2(dst->i[i], (1 << (desc->channel[channel].size - 1)) - 1);
226       }
227       break;
228    case UTIL_FORMAT_TYPE_UNSIGNED:
229       if (desc->channel[channel].normalized)
230          dst->ui[i] = src->ui[i];
231       else
232          dst->ui[i] = MIN2(src->ui[i], BITFIELD_MASK(desc->channel[channel].size));
233       break;
234    case UTIL_FORMAT_TYPE_FIXED:
235    case UTIL_FORMAT_TYPE_FLOAT:
236       dst->ui[i] = src->ui[i];
237       break;
238    }
239 }
240 
241 void
zink_format_clamp_channel_srgb(const struct util_format_description * desc,union pipe_color_union * dst,const union pipe_color_union * src,unsigned i)242 zink_format_clamp_channel_srgb(const struct util_format_description *desc, union pipe_color_union *dst, const union pipe_color_union *src, unsigned i)
243 {
244    unsigned channel = desc->swizzle[i];
245    if (desc->colorspace == UTIL_FORMAT_COLORSPACE_SRGB &&
246        channel <= PIPE_SWIZZLE_W) {
247       switch (desc->channel[channel].type) {
248       case UTIL_FORMAT_TYPE_SIGNED:
249       case UTIL_FORMAT_TYPE_UNSIGNED:
250          dst->f[i] = CLAMP(src->f[i], 0.0, 1.0);
251          return;
252       default:
253          break;
254       }
255    }
256 
257    dst->ui[i] = src->ui[i];
258 }
259