2021-06-19 08:18:50 +00:00
|
|
|
#include <metal_stdlib>
|
|
|
|
|
|
|
|
using namespace metal;
|
|
|
|
struct Uniforms {
|
|
|
|
/* 0x0000 */ uint dstTextureFlipY;
|
|
|
|
/* 0x0004 */ uint channelCount;
|
2021-08-23 21:45:23 +00:00
|
|
|
/* 0x0008 */ uint2 srcCopyOrigin;
|
|
|
|
/* 0x0010 */ uint2 dstCopyOrigin;
|
|
|
|
/* 0x0018 */ uint2 copySize;
|
2021-06-19 08:18:50 +00:00
|
|
|
};
|
|
|
|
struct OutputBuf {
|
|
|
|
/* 0x0000 */ uint result[1];
|
|
|
|
};
|
|
|
|
|
|
|
|
bool aboutEqual(float value, float expect) {
|
2021-06-28 15:30:57 +00:00
|
|
|
return (fabs((value - expect)) < 0.001f);
|
2021-06-19 08:18:50 +00:00
|
|
|
}
|
|
|
|
|
2021-08-04 22:15:28 +00:00
|
|
|
void tint_symbol_inner(constant Uniforms& uniforms, device OutputBuf& output, uint3 GlobalInvocationID, texture2d<float, access::sample> tint_symbol_1, texture2d<float, access::sample> tint_symbol_2) {
|
|
|
|
int2 const srcSize = int2(tint_symbol_1.get_width(), tint_symbol_1.get_height());
|
|
|
|
int2 const dstSize = int2(tint_symbol_2.get_width(), tint_symbol_2.get_height());
|
2021-10-20 16:12:33 +00:00
|
|
|
uint2 const dstTexCoord = uint2(uint3(GlobalInvocationID).xy);
|
2021-06-19 08:18:50 +00:00
|
|
|
float4 const nonCoveredColor = float4(0.0f, 1.0f, 0.0f, 1.0f);
|
|
|
|
bool success = true;
|
2021-10-20 16:12:33 +00:00
|
|
|
if (((((dstTexCoord[0] < uniforms.dstCopyOrigin[0]) || (dstTexCoord[1] < uniforms.dstCopyOrigin[1])) || (dstTexCoord[0] >= (uniforms.dstCopyOrigin[0] + uniforms.copySize[0]))) || (dstTexCoord[1] >= (uniforms.dstCopyOrigin[1] + uniforms.copySize[1])))) {
|
2021-08-04 22:15:28 +00:00
|
|
|
success = (success && all((tint_symbol_2.read(uint2(int2(dstTexCoord)), 0) == nonCoveredColor)));
|
2021-06-19 08:18:50 +00:00
|
|
|
} else {
|
|
|
|
uint2 srcTexCoord = ((dstTexCoord - uniforms.dstCopyOrigin) + uniforms.srcCopyOrigin);
|
|
|
|
if ((uniforms.dstTextureFlipY == 1u)) {
|
2021-10-20 16:12:33 +00:00
|
|
|
srcTexCoord[1] = ((uint(srcSize[1]) - srcTexCoord[1]) - 1u);
|
2021-06-19 08:18:50 +00:00
|
|
|
}
|
2021-08-04 22:15:28 +00:00
|
|
|
float4 const srcColor = tint_symbol_1.read(uint2(int2(srcTexCoord)), 0);
|
|
|
|
float4 const dstColor = tint_symbol_2.read(uint2(int2(dstTexCoord)), 0);
|
2021-06-19 08:18:50 +00:00
|
|
|
if ((uniforms.channelCount == 2u)) {
|
2021-10-20 16:12:33 +00:00
|
|
|
success = ((success && aboutEqual(dstColor[0], srcColor[0])) && aboutEqual(dstColor[1], srcColor[1]));
|
2021-06-19 08:18:50 +00:00
|
|
|
} else {
|
2021-10-20 16:12:33 +00:00
|
|
|
success = ((((success && aboutEqual(dstColor[0], srcColor[0])) && aboutEqual(dstColor[1], srcColor[1])) && aboutEqual(dstColor[2], srcColor[2])) && aboutEqual(dstColor[3], srcColor[3]));
|
2021-06-19 08:18:50 +00:00
|
|
|
}
|
|
|
|
}
|
2021-10-20 16:12:33 +00:00
|
|
|
uint const outputIndex = ((GlobalInvocationID[1] * uint(dstSize[0])) + GlobalInvocationID[0]);
|
2021-06-19 08:18:50 +00:00
|
|
|
if (success) {
|
|
|
|
output.result[outputIndex] = 1u;
|
|
|
|
} else {
|
|
|
|
output.result[outputIndex] = 0u;
|
|
|
|
}
|
2021-08-04 22:15:28 +00:00
|
|
|
}
|
|
|
|
|
2021-08-12 19:47:20 +00:00
|
|
|
kernel void tint_symbol(texture2d<float, access::sample> tint_symbol_3 [[texture(0)]], texture2d<float, access::sample> tint_symbol_4 [[texture(1)]], uint3 GlobalInvocationID [[thread_position_in_grid]], constant Uniforms& uniforms [[buffer(0)]], device OutputBuf& output [[buffer(1)]]) {
|
2021-08-04 22:15:28 +00:00
|
|
|
tint_symbol_inner(uniforms, output, GlobalInvocationID, tint_symbol_3, tint_symbol_4);
|
2021-06-19 08:18:50 +00:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
|