xref: /aosp_15_r20/external/skia/tests/sksl/compute/DesaturateFunction.wgsl (revision c8dee2aa9b3f27cf6c858bd81872bdeb2c07ed17)
1*c8dee2aaSAndroid Build Coastguard Workerdiagnostic(off, derivative_uniformity);
2*c8dee2aaSAndroid Build Coastguard Workerdiagnostic(off, chromium.unreachable_code);
3*c8dee2aaSAndroid Build Coastguard Workerstruct CSIn {
4*c8dee2aaSAndroid Build Coastguard Worker  @builtin(global_invocation_id) sk_GlobalInvocationID: vec3<u32>,
5*c8dee2aaSAndroid Build Coastguard Worker};
6*c8dee2aaSAndroid Build Coastguard Worker@group(0) @binding(0) var src: texture_2d<f32>;
7*c8dee2aaSAndroid Build Coastguard Worker@group(0) @binding(1) var dest: texture_storage_2d<rgba32float, write>;
8*c8dee2aaSAndroid Build Coastguard Workerfn desaturate_vTT(_stageIn: CSIn, src: texture_2d<f32>, dest: texture_storage_2d<rgba32float, write>) {
9*c8dee2aaSAndroid Build Coastguard Worker  {
10*c8dee2aaSAndroid Build Coastguard Worker    let _skTemp0 = _stageIn.sk_GlobalInvocationID.xy;
11*c8dee2aaSAndroid Build Coastguard Worker    var color: vec4<f32> = textureLoad(src, _skTemp0, 0);
12*c8dee2aaSAndroid Build Coastguard Worker    let _skTemp1 = dot(color.xyz, vec3<f32>(0.22, 0.67, 0.11));
13*c8dee2aaSAndroid Build Coastguard Worker    color = vec4<f32>((vec3<f32>(_skTemp1)), color.w);
14*c8dee2aaSAndroid Build Coastguard Worker    textureStore(dest, _stageIn.sk_GlobalInvocationID.xy, color);
15*c8dee2aaSAndroid Build Coastguard Worker  }
16*c8dee2aaSAndroid Build Coastguard Worker}
17*c8dee2aaSAndroid Build Coastguard Workerfn _skslMain(_stageIn: CSIn) {
18*c8dee2aaSAndroid Build Coastguard Worker  {
19*c8dee2aaSAndroid Build Coastguard Worker    let _skTemp2 = textureDimensions(src);
20*c8dee2aaSAndroid Build Coastguard Worker    let _skTemp3 = textureDimensions(src);
21*c8dee2aaSAndroid Build Coastguard Worker    if (_stageIn.sk_GlobalInvocationID.x < _skTemp2.x) && (_stageIn.sk_GlobalInvocationID.y < _skTemp3.y) {
22*c8dee2aaSAndroid Build Coastguard Worker      {
23*c8dee2aaSAndroid Build Coastguard Worker        desaturate_vTT(_stageIn, src, dest);
24*c8dee2aaSAndroid Build Coastguard Worker      }
25*c8dee2aaSAndroid Build Coastguard Worker    }
26*c8dee2aaSAndroid Build Coastguard Worker  }
27*c8dee2aaSAndroid Build Coastguard Worker}
28*c8dee2aaSAndroid Build Coastguard Worker@compute @workgroup_size(16, 16, 1) fn main(_stageIn: CSIn) {
29*c8dee2aaSAndroid Build Coastguard Worker  _skslMain(_stageIn);
30*c8dee2aaSAndroid Build Coastguard Worker}
31