submodule/skia/tests/sksl/intrinsics/OuterProduct.metal

   1 #include <metal_stdlib>
   2 #include <simd/simd.h>
   3 using namespace metal;
   4 struct Uniforms {
   5     half4 colorGreen;
   6     half4 colorRed;
   7     float2x2 testMatrix2x2;
   8     float3x3 testMatrix3x3;
   9     half4 testInputs;
  10 };
  11 struct Inputs {
  12 };
  13 struct Outputs {
  14     half4 sk_FragColor [[color(0)]];
  15 };
  16
  17 thread bool operator==(const float2x2 left, const float2x2 right);
  18 thread bool operator!=(const float2x2 left, const float2x2 right);
  19
  20 thread bool operator==(const float3x3 left, const float3x3 right);
  21 thread bool operator!=(const float3x3 left, const float3x3 right);
  22
  23 thread bool operator==(const float3x2 left, const float3x2 right);
  24 thread bool operator!=(const float3x2 left, const float3x2 right);
  25
  26 thread bool operator==(const float4x4 left, const float4x4 right);
  27 thread bool operator!=(const float4x4 left, const float4x4 right);
  28
  29 thread bool operator==(const float2x4 left, const float2x4 right);
  30 thread bool operator!=(const float2x4 left, const float2x4 right);
  31
  32 thread bool operator==(const float4x2 left, const float4x2 right);
  33 thread bool operator!=(const float4x2 left, const float4x2 right);
  34 thread bool operator==(const float2x2 left, const float2x2 right) {
  35     return all(left[0] == right[0]) &&
  36            all(left[1] == right[1]);
  37 }
  38 thread bool operator!=(const float2x2 left, const float2x2 right) {
  39     return !(left == right);
  40 }
  41
  42 template <typename T, int C, int R>
  43 matrix<T, C, R> outerProduct(const vec<T, R> a, const vec<T, C> b) {
  44     matrix<T, C, R> result;
  45     for (int c = 0; c < C; ++c) {
  46         result[c] = a * b[c];
  47     }
  48     return result;
  49 }
  50 thread bool operator==(const float3x3 left, const float3x3 right) {
  51     return all(left[0] == right[0]) &&
  52            all(left[1] == right[1]) &&
  53            all(left[2] == right[2]);
  54 }
  55 thread bool operator!=(const float3x3 left, const float3x3 right) {
  56     return !(left == right);
  57 }
  58 thread bool operator==(const float3x2 left, const float3x2 right) {
  59     return all(left[0] == right[0]) &&
  60            all(left[1] == right[1]) &&
  61            all(left[2] == right[2]);
  62 }
  63 thread bool operator!=(const float3x2 left, const float3x2 right) {
  64     return !(left == right);
  65 }
  66 thread bool operator==(const float4x4 left, const float4x4 right) {
  67     return all(left[0] == right[0]) &&
  68            all(left[1] == right[1]) &&
  69            all(left[2] == right[2]) &&
  70            all(left[3] == right[3]);
  71 }
  72 thread bool operator!=(const float4x4 left, const float4x4 right) {
  73     return !(left == right);
  74 }
  75 thread bool operator==(const float2x4 left, const float2x4 right) {
  76     return all(left[0] == right[0]) &&
  77            all(left[1] == right[1]);
  78 }
  79 thread bool operator!=(const float2x4 left, const float2x4 right) {
  80     return !(left == right);
  81 }
  82 thread bool operator==(const float4x2 left, const float4x2 right) {
  83     return all(left[0] == right[0]) &&
  84            all(left[1] == right[1]) &&
  85            all(left[2] == right[2]) &&
  86            all(left[3] == right[3]);
  87 }
  88 thread bool operator!=(const float4x2 left, const float4x2 right) {
  89     return !(left == right);
  90 }
  91 fragment Outputs fragmentMain(Inputs _in [[stage_in]], constant Uniforms& _uniforms [[buffer(0)]], bool _frontFacing [[front_facing]], float4 _fragCoord [[position]]) {
  92     Outputs _out;
  93     (void)_out;
  94     const float2 c12 = float2(1.0, 2.0);
  95     _out.sk_FragColor = ((((outerProduct(_uniforms.testMatrix2x2[0], _uniforms.testMatrix2x2[1]) == float2x2(float2(3.0, 6.0), float2(4.0, 8.0)) && outerProduct(_uniforms.testMatrix3x3[0], _uniforms.testMatrix3x3[1]) == float3x3(float3(4.0, 8.0, 12.0), float3(5.0, 10.0, 15.0), float3(6.0, 12.0, 18.0))) && outerProduct(_uniforms.testMatrix2x2[0], _uniforms.testMatrix3x3[1]) == float3x2(float2(4.0, 8.0), float2(5.0, 10.0), float2(6.0, 12.0))) && float4x4(outerProduct(_uniforms.testInputs, half4(1.0h, 0.0h, 0.0h, 2.0h))) == float4x4(float4(-1.25, 0.0, 0.75, 2.25), float4(0.0, 0.0, 0.0, 0.0), float4(0.0, 0.0, 0.0, 0.0), float4(-2.5, 0.0, 1.5, 4.5))) && outerProduct(float4(_uniforms.testInputs), c12) == float2x4(float4(-1.25, 0.0, 0.75, 2.25), float4(-2.5, 0.0, 1.5, 4.5))) && outerProduct(c12, float4(_uniforms.testInputs)) == float4x2(float2(-1.25, -2.5), float2(0.0, 0.0), float2(0.75, 1.5), float2(2.25, 4.5)) ? _uniforms.colorGreen : _uniforms.colorRed;
  96     return _out;
  97 }