xref: /aosp_15_r20/external/skia/tests/sksl/compute/Raytrace.wgsl (revision c8dee2aa9b3f27cf6c858bd81872bdeb2c07ed17)
1diagnostic(off, derivative_uniformity);
2diagnostic(off, chromium.unreachable_code);
3struct CSIn {
4  @builtin(global_invocation_id) sk_GlobalInvocationID: vec3<u32>,
5};
6@group(0) @binding(0) var dest: texture_storage_2d<rgba32float, write>;
7fn _skslMain(_stageIn: CSIn) {
8  {
9    var pixel: vec4<f32> = vec4<f32>(0.0, 0.0, 0.0, 1.0);
10    const max_x: f32 = 5.0;
11    const max_y: f32 = 5.0;
12    let _skTemp0 = textureDimensions(dest);
13    let _skTemp1 = textureDimensions(dest);
14    let x: f32 = f32(_stageIn.sk_GlobalInvocationID.x * 2u - _skTemp0.x) / f32(_skTemp1.x);
15    let _skTemp2 = textureDimensions(dest);
16    let _skTemp3 = textureDimensions(dest);
17    let y: f32 = f32(_stageIn.sk_GlobalInvocationID.y * 2u - _skTemp2.y) / f32(_skTemp3.y);
18    const ray_origin: vec3<f32> = vec3<f32>(0.0, 0.0, -1.0);
19    let ray_target: vec3<f32> = vec3<f32>(x * max_x, y * max_y, 0.0);
20    const sphere_center: vec3<f32> = vec3<f32>(0.0, 0.0, -10.0);
21    const sphere_radius: f32 = 1.0;
22    let t_minus_c: vec3<f32> = ray_target - sphere_center;
23    let _skTemp4 = dot(ray_origin, t_minus_c);
24    let b: f32 = _skTemp4;
25    let _skTemp5 = dot(t_minus_c, t_minus_c);
26    let c: f32 = _skTemp5 - sphere_radius * sphere_radius;
27    let bsqmc: f32 = b * b - c;
28    if bsqmc >= 0.0 {
29      {
30        pixel = vec4<f32>(0.4, 0.4, 1.0, 1.0);
31      }
32    }
33    textureStore(dest, _stageIn.sk_GlobalInvocationID.xy, pixel);
34  }
35}
36@compute @workgroup_size(16, 16, 1) fn main(_stageIn: CSIn) {
37  _skslMain(_stageIn);
38}
39