global endpoints
This commit is contained in:
		| @@ -94,6 +94,8 @@ uint result_index = 0; | ||||
| uint result_vector_max_index; | ||||
| bool result_limit_reached = false; | ||||
|  | ||||
| uvec4 endpoints[2][4]; | ||||
|  | ||||
| // EncodingData helpers | ||||
| uint Encoding(EncodingData val) { | ||||
|     return bitfieldExtract(val.data, 0, 8); | ||||
| @@ -673,7 +675,7 @@ ivec4 BlueContract(int a, int r, int g, int b) { | ||||
|     return ivec4(a, (r + b) >> 1, (g + b) >> 1, b); | ||||
| } | ||||
|  | ||||
| void ComputeEndpoints(out uvec4 ep1, out uvec4 ep2, uint color_endpoint_mode, | ||||
| void ComputeEndpoints(uint ep_index, uint color_endpoint_mode, | ||||
|                       inout uint colvals_index) { | ||||
| #define READ_UINT_VALUES(N)                                                                        \ | ||||
|     uint v[N];                                                                                     \ | ||||
| @@ -692,22 +694,22 @@ void ComputeEndpoints(out uvec4 ep1, out uvec4 ep2, uint color_endpoint_mode, | ||||
|     switch (color_endpoint_mode) { | ||||
|     case 0: { | ||||
|         READ_UINT_VALUES(2) | ||||
|         ep1 = uvec4(0xFF, v[0], v[0], v[0]); | ||||
|         ep2 = uvec4(0xFF, v[1], v[1], v[1]); | ||||
|         endpoints[0][ep_index] = uvec4(0xFF, v[0], v[0], v[0]); | ||||
|         endpoints[1][ep_index] = uvec4(0xFF, v[1], v[1], v[1]); | ||||
|         break; | ||||
|     } | ||||
|     case 1: { | ||||
|         READ_UINT_VALUES(2) | ||||
|         const uint L0 = (v[0] >> 2) | (v[1] & 0xC0); | ||||
|         const uint L1 = min(L0 + (v[1] & 0x3F), 0xFFU); | ||||
|         ep1 = uvec4(0xFF, L0, L0, L0); | ||||
|         ep2 = uvec4(0xFF, L1, L1, L1); | ||||
|         endpoints[0][ep_index] = uvec4(0xFF, L0, L0, L0); | ||||
|         endpoints[1][ep_index] = uvec4(0xFF, L1, L1, L1); | ||||
|         break; | ||||
|     } | ||||
|     case 4: { | ||||
|         READ_UINT_VALUES(4) | ||||
|         ep1 = uvec4(v[2], v[0], v[0], v[0]); | ||||
|         ep2 = uvec4(v[3], v[1], v[1], v[1]); | ||||
|         endpoints[0][ep_index] = uvec4(v[2], v[0], v[0], v[0]); | ||||
|         endpoints[1][ep_index] = uvec4(v[3], v[1], v[1], v[1]); | ||||
|         break; | ||||
|     } | ||||
|     case 5: { | ||||
| @@ -718,24 +720,24 @@ void ComputeEndpoints(out uvec4 ep1, out uvec4 ep2, uint color_endpoint_mode, | ||||
|         transferred = BitTransferSigned(v[3], v[2]); | ||||
|         v[3] = transferred.x; | ||||
|         v[2] = transferred.y; | ||||
|         ep1 = ClampByte(ivec4(v[2], v[0], v[0], v[0])); | ||||
|         ep2 = ClampByte(ivec4(v[2] + v[3], v[0] + v[1], v[0] + v[1], v[0] + v[1])); | ||||
|         endpoints[0][ep_index] = ClampByte(ivec4(v[2], v[0], v[0], v[0])); | ||||
|         endpoints[1][ep_index] = ClampByte(ivec4(v[2] + v[3], v[0] + v[1], v[0] + v[1], v[0] + v[1])); | ||||
|         break; | ||||
|     } | ||||
|     case 6: { | ||||
|         READ_UINT_VALUES(4) | ||||
|         ep1 = uvec4(0xFF, (v[0] * v[3]) >> 8, (v[1] * v[3]) >> 8, (v[2] * v[3]) >> 8); | ||||
|         ep2 = uvec4(0xFF, v[0], v[1], v[2]); | ||||
|         endpoints[0][ep_index] = uvec4(0xFF, (v[0] * v[3]) >> 8, (v[1] * v[3]) >> 8, (v[2] * v[3]) >> 8); | ||||
|         endpoints[1][ep_index] = uvec4(0xFF, v[0], v[1], v[2]); | ||||
|         break; | ||||
|     } | ||||
|     case 8: { | ||||
|         READ_UINT_VALUES(6) | ||||
|         if ((v[1] + v[3] + v[5]) >= (v[0] + v[2] + v[4])) { | ||||
|             ep1 = uvec4(0xFF, v[0], v[2], v[4]); | ||||
|             ep2 = uvec4(0xFF, v[1], v[3], v[5]); | ||||
|             endpoints[0][ep_index] = uvec4(0xFF, v[0], v[2], v[4]); | ||||
|             endpoints[1][ep_index] = uvec4(0xFF, v[1], v[3], v[5]); | ||||
|         } else { | ||||
|             ep1 = uvec4(BlueContract(0xFF, int(v[1]), int(v[3]), int(v[5]))); | ||||
|             ep2 = uvec4(BlueContract(0xFF, int(v[0]), int(v[2]), int(v[4]))); | ||||
|             endpoints[0][ep_index] = uvec4(BlueContract(0xFF, int(v[1]), int(v[3]), int(v[5]))); | ||||
|             endpoints[1][ep_index] = uvec4(BlueContract(0xFF, int(v[0]), int(v[2]), int(v[4]))); | ||||
|         } | ||||
|         break; | ||||
|     } | ||||
| @@ -751,28 +753,28 @@ void ComputeEndpoints(out uvec4 ep1, out uvec4 ep2, uint color_endpoint_mode, | ||||
|         v[5] = transferred.x; | ||||
|         v[4] = transferred.y; | ||||
|         if ((v[1] + v[3] + v[5]) >= 0) { | ||||
|             ep1 = ClampByte(ivec4(0xFF, v[0], v[2], v[4])); | ||||
|             ep2 = ClampByte(ivec4(0xFF, v[0] + v[1], v[2] + v[3], v[4] + v[5])); | ||||
|             endpoints[0][ep_index] = ClampByte(ivec4(0xFF, v[0], v[2], v[4])); | ||||
|             endpoints[1][ep_index] = ClampByte(ivec4(0xFF, v[0] + v[1], v[2] + v[3], v[4] + v[5])); | ||||
|         } else { | ||||
|             ep1 = ClampByte(BlueContract(0xFF, v[0] + v[1], v[2] + v[3], v[4] + v[5])); | ||||
|             ep2 = ClampByte(BlueContract(0xFF, v[0], v[2], v[4])); | ||||
|             endpoints[0][ep_index] = ClampByte(BlueContract(0xFF, v[0] + v[1], v[2] + v[3], v[4] + v[5])); | ||||
|             endpoints[1][ep_index] = ClampByte(BlueContract(0xFF, v[0], v[2], v[4])); | ||||
|         } | ||||
|         break; | ||||
|     } | ||||
|     case 10: { | ||||
|         READ_UINT_VALUES(6) | ||||
|         ep1 = uvec4(v[4], (v[0] * v[3]) >> 8, (v[1] * v[3]) >> 8, (v[2] * v[3]) >> 8); | ||||
|         ep2 = uvec4(v[5], v[0], v[1], v[2]); | ||||
|         endpoints[0][ep_index] = uvec4(v[4], (v[0] * v[3]) >> 8, (v[1] * v[3]) >> 8, (v[2] * v[3]) >> 8); | ||||
|         endpoints[1][ep_index] = uvec4(v[5], v[0], v[1], v[2]); | ||||
|         break; | ||||
|     } | ||||
|     case 12: { | ||||
|         READ_UINT_VALUES(8) | ||||
|         if ((v[1] + v[3] + v[5]) >= (v[0] + v[2] + v[4])) { | ||||
|             ep1 = uvec4(v[6], v[0], v[2], v[4]); | ||||
|             ep2 = uvec4(v[7], v[1], v[3], v[5]); | ||||
|             endpoints[0][ep_index] = uvec4(v[6], v[0], v[2], v[4]); | ||||
|             endpoints[1][ep_index] = uvec4(v[7], v[1], v[3], v[5]); | ||||
|         } else { | ||||
|             ep1 = uvec4(BlueContract(int(v[7]), int(v[1]), int(v[3]), int(v[5]))); | ||||
|             ep2 = uvec4(BlueContract(int(v[6]), int(v[0]), int(v[2]), int(v[4]))); | ||||
|             endpoints[0][ep_index] = uvec4(BlueContract(int(v[7]), int(v[1]), int(v[3]), int(v[5]))); | ||||
|             endpoints[1][ep_index] = uvec4(BlueContract(int(v[6]), int(v[0]), int(v[2]), int(v[4]))); | ||||
|         } | ||||
|         break; | ||||
|     } | ||||
| @@ -794,18 +796,18 @@ void ComputeEndpoints(out uvec4 ep1, out uvec4 ep2, uint color_endpoint_mode, | ||||
|         v[6] = transferred.y; | ||||
|  | ||||
|         if ((v[1] + v[3] + v[5]) >= 0) { | ||||
|             ep1 = ClampByte(ivec4(v[6], v[0], v[2], v[4])); | ||||
|             ep2 = ClampByte(ivec4(v[7] + v[6], v[0] + v[1], v[2] + v[3], v[4] + v[5])); | ||||
|             endpoints[0][ep_index] = ClampByte(ivec4(v[6], v[0], v[2], v[4])); | ||||
|             endpoints[1][ep_index] = ClampByte(ivec4(v[7] + v[6], v[0] + v[1], v[2] + v[3], v[4] + v[5])); | ||||
|         } else { | ||||
|             ep1 = ClampByte(BlueContract(v[6] + v[7], v[0] + v[1], v[2] + v[3], v[4] + v[5])); | ||||
|             ep2 = ClampByte(BlueContract(v[6], v[0], v[2], v[4])); | ||||
|             endpoints[0][ep_index] = ClampByte(BlueContract(v[6] + v[7], v[0] + v[1], v[2] + v[3], v[4] + v[5])); | ||||
|             endpoints[1][ep_index] = ClampByte(BlueContract(v[6], v[0], v[2], v[4])); | ||||
|         } | ||||
|         break; | ||||
|     } | ||||
|     default: { | ||||
|         // HDR mode, or more likely a bug computing the color_endpoint_mode | ||||
|         ep1 = uvec4(0xFF, 0xFF, 0, 0); | ||||
|         ep2 = uvec4(0xFF, 0xFF, 0, 0); | ||||
|         endpoints[0][ep_index] = uvec4(0xFF, 0xFF, 0, 0); | ||||
|         endpoints[1][ep_index] = uvec4(0xFF, 0xFF, 0, 0); | ||||
|         break; | ||||
|     } | ||||
|     } | ||||
| @@ -1198,10 +1200,6 @@ void DecompressBlock(ivec3 coord) { | ||||
|             color_endpoint_mode[i] = cem; | ||||
|         } | ||||
|     } | ||||
|  | ||||
|     uvec4 endpoints0[4]; | ||||
|     uvec4 endpoints1[4]; | ||||
|     { | ||||
|         // This decode phase should at most push 32 elements into the vector | ||||
|         result_vector_max_index = 32; | ||||
|  | ||||
| @@ -1209,10 +1207,8 @@ void DecompressBlock(ivec3 coord) { | ||||
|         uint colvals_index = 0; | ||||
|         DecodeColorValues(color_endpoint_mode, num_partitions, color_data_bits); | ||||
|         for (uint i = 0; i < num_partitions; i++) { | ||||
|             ComputeEndpoints(endpoints0[i], endpoints1[i], color_endpoint_mode[i], | ||||
|                              colvals_index); | ||||
|             ComputeEndpoints(i, color_endpoint_mode[i], colvals_index); | ||||
|         } | ||||
|     } | ||||
|     color_endpoint_data = local_buff; | ||||
|     color_endpoint_data = bitfieldReverse(color_endpoint_data).wzyx; | ||||
|     const uint clear_byte_start = (weight_bits >> 3) + 1; | ||||
| @@ -1247,8 +1243,8 @@ void DecompressBlock(ivec3 coord) { | ||||
|                 local_partition = Select2DPartition(partition_index, i, j, num_partitions, | ||||
|                                                     (block_dims.y * block_dims.x) < 32); | ||||
|             } | ||||
|             const uvec4 C0 = ReplicateByteTo16(endpoints0[local_partition]); | ||||
|             const uvec4 C1 = ReplicateByteTo16(endpoints1[local_partition]); | ||||
|             const uvec4 C0 = ReplicateByteTo16(endpoints[0][local_partition]); | ||||
|             const uvec4 C1 = ReplicateByteTo16(endpoints[1][local_partition]); | ||||
|             const uint weight_offset = (j * block_dims.x + i); | ||||
|             const uint array_index = weight_offset / 4; | ||||
|             const uint vector_index = bfe(weight_offset, 0, 2); | ||||
|   | ||||
		Reference in New Issue
	
	Block a user