xref: /aosp_15_r20/external/skia/tests/sksl/shared/RectangleTexture.metal (revision c8dee2aa9b3f27cf6c858bd81872bdeb2c07ed17)
1*c8dee2aaSAndroid Build Coastguard Worker#include <metal_stdlib>
2*c8dee2aaSAndroid Build Coastguard Worker#include <simd/simd.h>
3*c8dee2aaSAndroid Build Coastguard Worker#ifdef __clang__
4*c8dee2aaSAndroid Build Coastguard Worker#pragma clang diagnostic ignored "-Wall"
5*c8dee2aaSAndroid Build Coastguard Worker#endif
6*c8dee2aaSAndroid Build Coastguard Workerusing namespace metal;
7*c8dee2aaSAndroid Build Coastguard Worker
8*c8dee2aaSAndroid Build Coastguard Workerstruct sampler2D {
9*c8dee2aaSAndroid Build Coastguard Worker    texture2d<half> tex;
10*c8dee2aaSAndroid Build Coastguard Worker    sampler smp;
11*c8dee2aaSAndroid Build Coastguard Worker};
12*c8dee2aaSAndroid Build Coastguard Workerhalf4 sample(sampler2D i, float2 p, float b=0) { return i.tex.sample(i.smp, p, bias(b)); }
13*c8dee2aaSAndroid Build Coastguard Workerhalf4 sample(sampler2D i, float3 p, float b=0) { return i.tex.sample(i.smp, p.xy / p.z, bias(b)); }
14*c8dee2aaSAndroid Build Coastguard Workerhalf4 sampleLod(sampler2D i, float2 p, float lod) { return i.tex.sample(i.smp, p, level(lod)); }
15*c8dee2aaSAndroid Build Coastguard Workerhalf4 sampleLod(sampler2D i, float3 p, float lod) {
16*c8dee2aaSAndroid Build Coastguard Worker    return i.tex.sample(i.smp, p.xy / p.z, level(lod));
17*c8dee2aaSAndroid Build Coastguard Worker}
18*c8dee2aaSAndroid Build Coastguard Workerhalf4 sampleGrad(sampler2D i, float2 p, float2 dPdx, float2 dPdy) {
19*c8dee2aaSAndroid Build Coastguard Worker    return i.tex.sample(i.smp, p, gradient2d(dPdx, dPdy));
20*c8dee2aaSAndroid Build Coastguard Worker}
21*c8dee2aaSAndroid Build Coastguard Worker
22*c8dee2aaSAndroid Build Coastguard Workerstruct Inputs {
23*c8dee2aaSAndroid Build Coastguard Worker};
24*c8dee2aaSAndroid Build Coastguard Workerstruct Outputs {
25*c8dee2aaSAndroid Build Coastguard Worker    half4 sk_FragColor [[color(0)]];
26*c8dee2aaSAndroid Build Coastguard Worker};
27*c8dee2aaSAndroid Build Coastguard Workerstruct Globals {
28*c8dee2aaSAndroid Build Coastguard Worker    sampler2D test2D;
29*c8dee2aaSAndroid Build Coastguard Worker    sampler2D test2DRect;
30*c8dee2aaSAndroid Build Coastguard Worker};
31*c8dee2aaSAndroid Build Coastguard Workerfragment Outputs fragmentMain(Inputs _in [[stage_in]], texture2d<half> test2D_Tex [[texture(0)]], sampler test2D_Smplr [[sampler(0)]], texture2d<half> test2DRect_Tex [[texture(1)]], sampler test2DRect_Smplr [[sampler(1)]], bool _frontFacing [[front_facing]], float4 _fragCoord [[position]]) {
32*c8dee2aaSAndroid Build Coastguard Worker    Globals _globals{{test2D_Tex, test2D_Smplr}, {test2DRect_Tex, test2DRect_Smplr}};
33*c8dee2aaSAndroid Build Coastguard Worker    (void)_globals;
34*c8dee2aaSAndroid Build Coastguard Worker    Outputs _out;
35*c8dee2aaSAndroid Build Coastguard Worker    (void)_out;
36*c8dee2aaSAndroid Build Coastguard Worker    _out.sk_FragColor = sample(_globals.test2D, float2(0.5));
37*c8dee2aaSAndroid Build Coastguard Worker    _out.sk_FragColor = sample(_globals.test2DRect, float2(0.5));
38*c8dee2aaSAndroid Build Coastguard Worker    _out.sk_FragColor = sample(_globals.test2DRect, float3(0.5));
39*c8dee2aaSAndroid Build Coastguard Worker    return _out;
40*c8dee2aaSAndroid Build Coastguard Worker}
41