mirror of
https://github.com/encounter/dawn-cmake.git
synced 2025-07-14 17:16:01 +00:00
a5d73ce965
This is a major reworking of this transform. The old transform code was getting unwieldy, with part of the complication coming from the handling of multiple return statements. By generating a wrapper function instead, we can avoid a lot of this complexity. The original entry point function is stripped of all shader IO attributes (as well as `stage` and `workgroup_size`), but the body is left unmodified. A new entry point wrapper function is introduced which calls the original function, packing/unpacking the shader inputs as necessary, and propagates the result to the corresponding shader outputs. The new code has been refactored to use a state object with the different parts of the transform split into separate functions, which makes it much more manageable. Fixed: tint:1076 Bug: tint:920 Change-Id: I3490a0ea7a3509a4e198ce730e476516649d8d96 Reviewed-on: https://dawn-review.googlesource.com/c/tint/+/60521 Auto-Submit: James Price <jrprice@google.com> Kokoro: Kokoro <noreply+kokoro@google.com> Commit-Queue: James Price <jrprice@google.com> Reviewed-by: Ben Clayton <bclayton@google.com>
63 lines
2.2 KiB
Plaintext
63 lines
2.2 KiB
Plaintext
#include <metal_stdlib>
|
|
|
|
using namespace metal;
|
|
struct Uniforms {
|
|
/* 0x0000 */ packed_float2 u_scale;
|
|
/* 0x0008 */ packed_float2 u_offset;
|
|
};
|
|
struct VertexOutputs {
|
|
float2 texcoords;
|
|
float4 position;
|
|
};
|
|
struct tint_symbol {
|
|
float2 texcoords [[user(locn0)]];
|
|
float4 position [[position]];
|
|
};
|
|
struct tint_array_wrapper {
|
|
float2 arr[3];
|
|
};
|
|
struct tint_symbol_2 {
|
|
float2 texcoord [[user(locn0)]];
|
|
};
|
|
struct tint_symbol_3 {
|
|
float4 value [[color(0)]];
|
|
};
|
|
|
|
VertexOutputs vs_main_inner(constant Uniforms& uniforms, uint VertexIndex) {
|
|
tint_array_wrapper texcoord = {.arr={float2(-0.5f, 0.0f), float2(1.5f, 0.0f), float2(0.5f, 2.0f)}};
|
|
VertexOutputs output = {};
|
|
output.position = float4(((texcoord.arr[VertexIndex] * 2.0f) - float2(1.0f, 1.0f)), 0.0f, 1.0f);
|
|
bool flipY = (uniforms.u_scale.y < 0.0f);
|
|
if (flipY) {
|
|
output.texcoords = ((((texcoord.arr[VertexIndex] * uniforms.u_scale) + uniforms.u_offset) * float2(1.0f, -1.0f)) + float2(0.0f, 1.0f));
|
|
} else {
|
|
output.texcoords = ((((texcoord.arr[VertexIndex] * float2(1.0f, -1.0f)) + float2(0.0f, 1.0f)) * uniforms.u_scale) + uniforms.u_offset);
|
|
}
|
|
return output;
|
|
}
|
|
|
|
vertex tint_symbol vs_main(uint VertexIndex [[vertex_id]], constant Uniforms& uniforms [[buffer(0)]]) {
|
|
VertexOutputs const inner_result = vs_main_inner(uniforms, VertexIndex);
|
|
tint_symbol wrapper_result = {};
|
|
wrapper_result.texcoords = inner_result.texcoords;
|
|
wrapper_result.position = inner_result.position;
|
|
return wrapper_result;
|
|
}
|
|
|
|
float4 fs_main_inner(float2 texcoord, texture2d<float, access::sample> tint_symbol_4, sampler tint_symbol_5) {
|
|
float2 clampedTexcoord = clamp(texcoord, float2(0.0f, 0.0f), float2(1.0f, 1.0f));
|
|
if (!(all((clampedTexcoord == texcoord)))) {
|
|
discard_fragment();
|
|
}
|
|
float4 srcColor = tint_symbol_4.sample(tint_symbol_5, texcoord);
|
|
return srcColor;
|
|
}
|
|
|
|
fragment tint_symbol_3 fs_main(texture2d<float, access::sample> tint_symbol_6 [[texture(2)]], sampler tint_symbol_7 [[sampler(1)]], tint_symbol_2 tint_symbol_1 [[stage_in]]) {
|
|
float4 const inner_result_1 = fs_main_inner(tint_symbol_1.texcoord, tint_symbol_6, tint_symbol_7);
|
|
tint_symbol_3 wrapper_result_1 = {};
|
|
wrapper_result_1.value = inner_result_1;
|
|
return wrapper_result_1;
|
|
}
|
|
|