xref: /aosp_15_r20/external/skia/tests/sksl/shared/SampleMask.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 FSIn {
4*c8dee2aaSAndroid Build Coastguard Worker  @builtin(sample_mask) sk_SampleMaskIn: u32,
5*c8dee2aaSAndroid Build Coastguard Worker};
6*c8dee2aaSAndroid Build Coastguard Workerstruct FSOut {
7*c8dee2aaSAndroid Build Coastguard Worker  @location(0) sk_FragColor: vec4<f32>,
8*c8dee2aaSAndroid Build Coastguard Worker  @builtin(sample_mask) sk_SampleMask: u32,
9*c8dee2aaSAndroid Build Coastguard Worker};
10*c8dee2aaSAndroid Build Coastguard Workerfn samplemaskin_as_color_h4(_stageIn: FSIn) -> vec4<f32> {
11*c8dee2aaSAndroid Build Coastguard Worker  {
12*c8dee2aaSAndroid Build Coastguard Worker    return vec4<f32>(f32(_stageIn.sk_SampleMaskIn));
13*c8dee2aaSAndroid Build Coastguard Worker  }
14*c8dee2aaSAndroid Build Coastguard Worker}
15*c8dee2aaSAndroid Build Coastguard Workerfn clear_samplemask_v(_stageOut: ptr<function, FSOut>) {
16*c8dee2aaSAndroid Build Coastguard Worker  {
17*c8dee2aaSAndroid Build Coastguard Worker    (*_stageOut).sk_SampleMask = 0u;
18*c8dee2aaSAndroid Build Coastguard Worker  }
19*c8dee2aaSAndroid Build Coastguard Worker}
20*c8dee2aaSAndroid Build Coastguard Workerfn reset_samplemask_v(_stageIn: FSIn, _stageOut: ptr<function, FSOut>) {
21*c8dee2aaSAndroid Build Coastguard Worker  {
22*c8dee2aaSAndroid Build Coastguard Worker    (*_stageOut).sk_SampleMask = _stageIn.sk_SampleMaskIn;
23*c8dee2aaSAndroid Build Coastguard Worker  }
24*c8dee2aaSAndroid Build Coastguard Worker}
25*c8dee2aaSAndroid Build Coastguard Workerfn _skslMain(_stageIn: FSIn, _stageOut: ptr<function, FSOut>) {
26*c8dee2aaSAndroid Build Coastguard Worker  {
27*c8dee2aaSAndroid Build Coastguard Worker    clear_samplemask_v(_stageOut);
28*c8dee2aaSAndroid Build Coastguard Worker    reset_samplemask_v(_stageIn, _stageOut);
29*c8dee2aaSAndroid Build Coastguard Worker    (*_stageOut).sk_SampleMask = 4294967295u;
30*c8dee2aaSAndroid Build Coastguard Worker    let _skTemp0 = samplemaskin_as_color_h4(_stageIn);
31*c8dee2aaSAndroid Build Coastguard Worker    (*_stageOut).sk_FragColor = _skTemp0 * 0.00390625;
32*c8dee2aaSAndroid Build Coastguard Worker  }
33*c8dee2aaSAndroid Build Coastguard Worker}
34*c8dee2aaSAndroid Build Coastguard Worker@fragment fn main(_stageIn: FSIn) -> FSOut {
35*c8dee2aaSAndroid Build Coastguard Worker  var _stageOut: FSOut;
36*c8dee2aaSAndroid Build Coastguard Worker  _skslMain(_stageIn, &_stageOut);
37*c8dee2aaSAndroid Build Coastguard Worker  return _stageOut;
38*c8dee2aaSAndroid Build Coastguard Worker}
39