[XeSL] Metal Shading Language definitions
This commit is contained in:
@@ -9,52 +9,55 @@
|
||||
|
||||
#include "texture_load.xesli"
|
||||
|
||||
xesl_entry
|
||||
xesl_writeTypedStorageBuffer(xesl_uint4, xe_texture_load_dest, set=0,
|
||||
binding=0, u0, space0)
|
||||
xesl_writeTypedStorageBuffer_declare(xesl_uint4, xe_texture_load_dest, set=0,
|
||||
binding=0, u0, space0)
|
||||
xesl_typedStorageBuffer_declare(xesl_uint4, xe_texture_load_source, set=1,
|
||||
binding=0, t0, space0)
|
||||
xesl_entry_bindings_begin_compute
|
||||
XE_TEXTURE_LOAD_CONSTANT_BUFFER_BINDING
|
||||
xesl_entry_binding_next
|
||||
xesl_typedStorageBuffer(xesl_uint4, xe_texture_load_source, set=1, binding=0,
|
||||
t0, space0)
|
||||
xesl_entry_bindings_end_local_size(kXeTextureLoadGroupSizeX,
|
||||
kXeTextureLoadGroupSizeY, 1)
|
||||
xesl_input_global_invocation_id
|
||||
xesl_entry_signature_end
|
||||
xesl_writeTypedStorageBuffer_binding(xesl_uint4, xe_texture_load_dest,
|
||||
buffer(1))
|
||||
xesl_entry_binding_next
|
||||
xesl_typedStorageBuffer_binding(xesl_uint4, xe_texture_load_source, buffer(2))
|
||||
xesl_entry_bindings_end_inputs_begin_compute
|
||||
xesl_entry_input_globalInvocationID
|
||||
xesl_entry_inputs_end_code_begin_compute
|
||||
// 1 thread = 4 DXT3A-as-1111 blocks to 16x4 16bpp texels passed through an
|
||||
// externally provided
|
||||
// `uint4 XE_TEXTURE_LOAD_DXT3A_AS_1_1_1_1_TO_16BPP(uint2 halfblocks)`
|
||||
// conversion function.
|
||||
XeTextureLoadInfo load_info = XeTextureLoadGetInfo(
|
||||
xesl_function_call_constantBuffer(xe_texture_load_constants));
|
||||
xesl_uint3 block_index = xesl_GlobalInvocationID << xesl_uint3(2u, 0u, 0u);
|
||||
xesl_dont_flatten
|
||||
if (any(xesl_greaterThanEqual(block_index.xy,
|
||||
XeTextureLoadSizeBlocks().xy))) {
|
||||
if (any(xesl_greaterThanEqual(block_index.xy, load_info.size_blocks.xy))) {
|
||||
return;
|
||||
}
|
||||
xesl_uint3 texel_index_host = block_index << xesl_uint3(2u, 2u, 0u);
|
||||
uint blocks_pitch_host = XeTextureLoadHostPitch();
|
||||
uint height_texels = XeTextureLoadHeightTexels();
|
||||
uint block_offset_host = uint(
|
||||
(XeTextureHostLinearOffset(xesl_int3(texel_index_host), blocks_pitch_host,
|
||||
height_texels, 2u) +
|
||||
XeTextureLoadHostOffset()) >> 4u);
|
||||
uint elements_pitch_host = blocks_pitch_host >> 4u;
|
||||
(XeTextureHostLinearOffset(xesl_int3(texel_index_host),
|
||||
load_info.host_pitch, load_info.height_texels,
|
||||
2u) +
|
||||
load_info.host_offset) >> 4u);
|
||||
uint elements_pitch_host = load_info.host_pitch >> 4u;
|
||||
uint block_offset_guest =
|
||||
XeTextureLoadGuestBlockOffset(block_index, 8u, 3u) >> 4u;
|
||||
uint endian = XeTextureLoadEndian32();
|
||||
XeTextureLoadGuestBlockOffset(load_info, block_index, 8u, 3u) >> 4u;
|
||||
xesl_uint4 blocks_01 = XeEndianSwap32(
|
||||
xesl_typedStorageBufferLoad(xe_texture_load_source, block_offset_guest),
|
||||
endian);
|
||||
load_info.endian_32);
|
||||
// Odd 2 blocks = even 2 blocks + 32 bytes when tiled.
|
||||
block_offset_guest += XeTextureLoadIsTiled() ? 2u : 1u;
|
||||
block_offset_guest += load_info.is_tiled ? 2u : 1u;
|
||||
xesl_uint4 blocks_23 = XeEndianSwap32(
|
||||
xesl_typedStorageBufferLoad(xe_texture_load_source, block_offset_guest),
|
||||
endian);
|
||||
load_info.endian_32);
|
||||
xesl_writeTypedStorageBufferStore(
|
||||
xe_texture_load_dest, block_offset_host,
|
||||
XE_TEXTURE_LOAD_DXT3A_AS_1_1_1_1_TO_16BPP(blocks_01.xz));
|
||||
xesl_writeTypedStorageBufferStore(
|
||||
xe_texture_load_dest, block_offset_host + 1u,
|
||||
XE_TEXTURE_LOAD_DXT3A_AS_1_1_1_1_TO_16BPP(blocks_23.xz));
|
||||
xesl_dont_flatten if (++texel_index_host.y < height_texels) {
|
||||
xesl_dont_flatten if (++texel_index_host.y < load_info.height_texels) {
|
||||
block_offset_host += elements_pitch_host;
|
||||
xesl_uint4 high_halfblocks = xesl_uint4(blocks_01.xz, blocks_23.xz) >> 16u;
|
||||
xesl_writeTypedStorageBufferStore(
|
||||
@@ -63,7 +66,7 @@ xesl_entry_signature_end
|
||||
xesl_writeTypedStorageBufferStore(
|
||||
xe_texture_load_dest, block_offset_host + 1u,
|
||||
XE_TEXTURE_LOAD_DXT3A_AS_1_1_1_1_TO_16BPP(high_halfblocks.zw));
|
||||
xesl_dont_flatten if (++texel_index_host.y < height_texels) {
|
||||
xesl_dont_flatten if (++texel_index_host.y < load_info.height_texels) {
|
||||
block_offset_host += elements_pitch_host;
|
||||
xesl_writeTypedStorageBufferStore(
|
||||
xe_texture_load_dest, block_offset_host,
|
||||
@@ -71,7 +74,7 @@ xesl_entry_signature_end
|
||||
xesl_writeTypedStorageBufferStore(
|
||||
xe_texture_load_dest, block_offset_host + 1u,
|
||||
XE_TEXTURE_LOAD_DXT3A_AS_1_1_1_1_TO_16BPP(blocks_23.yw));
|
||||
xesl_dont_flatten if (++texel_index_host.y < height_texels) {
|
||||
xesl_dont_flatten if (++texel_index_host.y < load_info.height_texels) {
|
||||
block_offset_host += elements_pitch_host;
|
||||
high_halfblocks = xesl_uint4(blocks_01.yw, blocks_23.yw) >> 16u;
|
||||
xesl_writeTypedStorageBufferStore(
|
||||
@@ -83,4 +86,4 @@ xesl_entry_signature_end
|
||||
}
|
||||
}
|
||||
}
|
||||
xesl_entry_end
|
||||
xesl_entry_code_end_compute
|
||||
|
||||
Reference in New Issue
Block a user