xref: /aosp_15_r20/external/skia/resources/sksl/intrinsics/Step.sksl (revision c8dee2aa9b3f27cf6c858bd81872bdeb2c07ed17)

uniform half4 testInputs;
uniform half4 colorGreen, colorRed;

half4 main(float2 coords) {
    const half4 constVal   = half4(-1.25, 0, 0.75, 2.25);
    const half4 constGreen = half4(0, 1, 0, 1);
    const half4 constRed   = half4(1, 0, 0, 1);
    half4 expectedA = half4(0, 0, 1, 1);
    half4 expectedB = half4(1, 1, 0, 0);
    half4 expectedC = half4(0, 1, 1, 1);

    return (step(0.5, testInputs.x)                == expectedA.x     &&
            step(0.5, testInputs.xy)               == expectedA.xy    &&
            step(0.5, testInputs.xyz)              == expectedA.xyz   &&
            step(0.5, testInputs.xyzw)             == expectedA.xyzw  &&
            step(0.5, constVal.x)                  == expectedA.x     &&
            step(0.5, constVal.xy)                 == expectedA.xy    &&
            step(0.5, constVal.xyz)                == expectedA.xyz   &&
            step(0.5, constVal.xyzw)               == expectedA.xyzw  &&
            step(testInputs.x,    constGreen.x)    == expectedB.x     &&
            step(testInputs.xy,   constGreen.xy)   == expectedB.xy    &&
            step(testInputs.xyz,  constGreen.xyz)  == expectedB.xyz   &&
            step(testInputs.xyzw, constGreen.xyzw) == expectedB.xyzw  &&
            step(constVal.x,      constGreen.x)    == expectedB.x     &&
            step(constVal.xy,     constGreen.xy)   == expectedB.xy    &&
            step(constVal.xyz,    constGreen.xyz)  == expectedB.xyz   &&
            step(constVal.xyzw,   constGreen.xyzw) == expectedB.xyzw  &&
            step(colorRed.x,      colorGreen.x)    == expectedC.x     &&
            step(colorRed.xy,     colorGreen.xy)   == expectedC.xy    &&
            step(colorRed.xyz,    colorGreen.xyz)  == expectedC.xyz   &&
            step(colorRed.xyzw,   colorGreen.xyzw) == expectedC.xyzw  &&
            step(constRed.x,      constGreen.x)    == expectedC.x     &&
            step(constRed.xy,     constGreen.xy)   == expectedC.xy    &&
            step(constRed.xyz,    constGreen.xyz)  == expectedC.xyz   &&
            step(constRed.xyzw,   constGreen.xyzw) == expectedC.xyzw)
            ? colorGreen : colorRed;
}