mirror of
https://github.com/encounter/dawn-cmake.git
synced 2025-08-04 19:25:47 +00:00
f24b37e122
Add `out` parameters to expression and type generators. Use the new helper classes in TextGenerator. Cleans up bad formatting. Prepares the writer generating 'pre' statements, required for atomics. If-else statements are generated slightly differently. This is done so that 'pre' statements for the else conditions are scoped correctly. This is identical to the HLSL writer. Bug tint:892 Change-Id: I4c6e96c90673ba30898b3682bf3198497d63a2d4 Reviewed-on: https://dawn-review.googlesource.com/c/tint/+/56067 Kokoro: Kokoro <noreply+kokoro@google.com> Reviewed-by: James Price <jrprice@google.com> Commit-Queue: Ben Clayton <bclayton@google.com>
49 lines
2.2 KiB
Plaintext
49 lines
2.2 KiB
Plaintext
#include <metal_stdlib>
|
|
|
|
using namespace metal;
|
|
struct Uniforms {
|
|
/* 0x0000 */ uint dstTextureFlipY;
|
|
/* 0x0004 */ uint channelCount;
|
|
/* 0x0008 */ packed_uint2 srcCopyOrigin;
|
|
/* 0x0010 */ packed_uint2 dstCopyOrigin;
|
|
/* 0x0018 */ packed_uint2 copySize;
|
|
};
|
|
struct OutputBuf {
|
|
/* 0x0000 */ uint result[1];
|
|
};
|
|
|
|
bool aboutEqual(float value, float expect) {
|
|
return (fabs((value - expect)) < 0.001f);
|
|
}
|
|
|
|
kernel void tint_symbol(texture2d<float, access::sample> tint_symbol_2 [[texture(0)]], texture2d<float, access::sample> tint_symbol_3 [[texture(1)]], uint3 GlobalInvocationID [[thread_position_in_grid]], constant Uniforms& uniforms [[buffer(3)]], device OutputBuf& output [[buffer(2)]]) {
|
|
int2 const srcSize = int2(tint_symbol_2.get_width(), tint_symbol_2.get_height());
|
|
int2 const dstSize = int2(tint_symbol_3.get_width(), tint_symbol_3.get_height());
|
|
uint2 const dstTexCoord = uint2(GlobalInvocationID.xy);
|
|
float4 const nonCoveredColor = float4(0.0f, 1.0f, 0.0f, 1.0f);
|
|
bool success = true;
|
|
if (((((dstTexCoord.x < uniforms.dstCopyOrigin.x) || (dstTexCoord.y < uniforms.dstCopyOrigin.y)) || (dstTexCoord.x >= (uniforms.dstCopyOrigin.x + uniforms.copySize.x))) || (dstTexCoord.y >= (uniforms.dstCopyOrigin.y + uniforms.copySize.y)))) {
|
|
success = (success && all((tint_symbol_3.read(uint2(int2(dstTexCoord)), 0) == nonCoveredColor)));
|
|
} else {
|
|
uint2 srcTexCoord = ((dstTexCoord - uniforms.dstCopyOrigin) + uniforms.srcCopyOrigin);
|
|
if ((uniforms.dstTextureFlipY == 1u)) {
|
|
srcTexCoord.y = ((uint(srcSize.y) - srcTexCoord.y) - 1u);
|
|
}
|
|
float4 const srcColor = tint_symbol_2.read(uint2(int2(srcTexCoord)), 0);
|
|
float4 const dstColor = tint_symbol_3.read(uint2(int2(dstTexCoord)), 0);
|
|
if ((uniforms.channelCount == 2u)) {
|
|
success = ((success && aboutEqual(dstColor.r, srcColor.r)) && aboutEqual(dstColor.g, srcColor.g));
|
|
} else {
|
|
success = ((((success && aboutEqual(dstColor.r, srcColor.r)) && aboutEqual(dstColor.g, srcColor.g)) && aboutEqual(dstColor.b, srcColor.b)) && aboutEqual(dstColor.a, srcColor.a));
|
|
}
|
|
}
|
|
uint const outputIndex = ((GlobalInvocationID.y * uint(dstSize.x)) + GlobalInvocationID.x);
|
|
if (success) {
|
|
output.result[outputIndex] = 1u;
|
|
} else {
|
|
output.result[outputIndex] = 0u;
|
|
}
|
|
return;
|
|
}
|
|
|