xref: /aosp_15_r20/external/skia/resources/sksl/shared/Matrices.sksl (revision c8dee2aa9b3f27cf6c858bd81872bdeb2c07ed17)

uniform half4 colorGreen, colorRed;

bool test_float() {
    bool ok = true;

    float2x2 m1 = float2x2(float4(1, 2, 3, 4));
    ok = ok && (m1 == float2x2(1, 2, 3, 4));

    // This generates {5, 0, 0, 5} on some Radeon GPUs.
//    float2x2 m2 = float2x2(float4(5));
//    ok = ok && (m2 == float2x2(5, 5, 5, 5));

    float2x2 m3 = float2x2(m1);
    ok = ok && (m3 == float2x2(1, 2, 3, 4));

    float2x2 m4 = float2x2(6);
    ok = ok && (m4 == float2x2(6, 0, 0, 6));

    m3 *= m4;
    ok = ok && (m3 == float2x2(6, 12, 18, 24));

    float2x2 m5 = float2x2(m1[1][1]);
    ok = ok && (m5 == float2x2(4, 0, 0, 4));

    m1 += m5;
    ok = ok && (m1 == float2x2(5, 2, 3, 8));

    float2x2 m7 = float2x2(5, float3(6, 7, 8));
    ok = ok && (m7 == float2x2(5, 6, 7, 8));

    float3x3 m9 = float3x3(9);
    ok = ok && (m9 == float3x3(9, 0, 0, 0, 9, 0, 0, 0, 9));

    float4x4 m10 = float4x4(11);
    ok = ok && (m10 == float4x4(11, 0, 0, 0, 0, 11, 0, 0, 0, 0, 11, 0, 0, 0, 0, 11));

    float4x4 m11 = float4x4(20, 20, 20, 20, 20, 20, 20, 20, 20, 20, 20, 20, 20, 20, 20, 20);
    m11 -= m10;
    ok = ok && (m11 == float4x4(9, 20, 20, 20, 20, 9, 20, 20, 20, 20, 9, 20, 20, 20, 20, 9));

    return ok;
}

bool test_half() {
    bool ok = true;

    half2x2 m1 = half2x2(half4(1, 2, 3, 4));
    ok = ok && (m1 == half2x2(1, 2, 3, 4));

    // This generates {5, 0, 0, 5} on some Radeon GPUs.
//    half2x2 m2 = half2x2(half4(5));
//    ok = ok && (m2 == half2x2(5, 5, 5, 5));

    half2x2 m3 = half2x2(m1);
    ok = ok && (m3 == half2x2(1, 2, 3, 4));

    half2x2 m4 = half2x2(6);
    ok = ok && (m4 == half2x2(6, 0, 0, 6));

    m3 *= m4;
    ok = ok && (m3 == half2x2(6, 12, 18, 24));

    half2x2 m5 = half2x2(m1[1][1]);
    ok = ok && (m5 == half2x2(4, 0, 0, 4));

    m1 += m5;
    ok = ok && (m1 == half2x2(5, 2, 3, 8));

    half2x2 m7 = half2x2(5, half3(6, 7, 8));
    ok = ok && (m7 == half2x2(5, 6, 7, 8));

    half3x3 m9 = half3x3(9);
    ok = ok && (m9 == half3x3(9, 0, 0, 0, 9, 0, 0, 0, 9));

    half4x4 m10 = half4x4(11);
    ok = ok && (m10 == half4x4(11, 0, 0, 0, 0, 11, 0, 0, 0, 0, 11, 0, 0, 0, 0, 11));

    half4x4 m11 = half4x4(20, 20, 20, 20, 20, 20, 20, 20, 20, 20, 20, 20, 20, 20, 20, 20);
    m11 -= m10;
    ok = ok && (m11 == half4x4(9, 20, 20, 20, 20, 9, 20, 20, 20, 20, 9, 20, 20, 20, 20, 9));

    return ok;
}

bool test_comma() {
    float2x2 x, y;
    return (x = float2x2(1, 2, 3, 4),
            y = 0.5 * float2x2(2, 4, 6, 8),
            x == y);
}

half4 main(float2 coords) {
    return test_float() && test_half() && test_comma() ? colorGreen : colorRed;
}