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

   1 #include <metal_stdlib>
   2 #include <simd/simd.h>
   3 using namespace metal;
   4 struct Uniforms {
   5     half4 colorGreen;
   6     half4 colorRed;
   7 };
   8 struct Inputs {
   9 };
  10 struct Outputs {
  11     half4 sk_FragColor [[color(0)]];
  12 };
  13
  14 thread bool operator==(const half2x4 left, const half2x4 right);
  15 thread bool operator!=(const half2x4 left, const half2x4 right);
  16
  17 thread bool operator==(const half4x2 left, const half4x2 right);
  18 thread bool operator!=(const half4x2 left, const half4x2 right);
  19
  20 thread bool operator==(const float4x3 left, const float4x3 right);
  21 thread bool operator!=(const float4x3 left, const float4x3 right);
  22
  23 template <typename T, int C, int R>
  24 matrix<T, C, R> matrixCompMult(matrix<T, C, R> a, const matrix<T, C, R> b) {
  25     for (int c = 0; c < C; ++c) {
  26         a[c] *= b[c];
  27     }
  28     return a;
  29 }
  30 half4x2 half4x2_from_half4_half4(half4 x0, half4 x1) {
  31     return half4x2(half2(x0.xy), half2(x0.zw), half2(x1.xy), half2(x1.zw));
  32 }
  33 thread bool operator==(const half2x4 left, const half2x4 right) {
  34     return all(left[0] == right[0]) &&
  35            all(left[1] == right[1]);
  36 }
  37 thread bool operator!=(const half2x4 left, const half2x4 right) {
  38     return !(left == right);
  39 }
  40 thread bool operator==(const half4x2 left, const half4x2 right) {
  41     return all(left[0] == right[0]) &&
  42            all(left[1] == right[1]) &&
  43            all(left[2] == right[2]) &&
  44            all(left[3] == right[3]);
  45 }
  46 thread bool operator!=(const half4x2 left, const half4x2 right) {
  47     return !(left == right);
  48 }
  49 thread bool operator==(const float4x3 left, const float4x3 right) {
  50     return all(left[0] == right[0]) &&
  51            all(left[1] == right[1]) &&
  52            all(left[2] == right[2]) &&
  53            all(left[3] == right[3]);
  54 }
  55 thread bool operator!=(const float4x3 left, const float4x3 right) {
  56     return !(left == right);
  57 }
  58 fragment Outputs fragmentMain(Inputs _in [[stage_in]], constant Uniforms& _uniforms [[buffer(0)]], bool _frontFacing [[front_facing]], float4 _fragCoord [[position]]) {
  59     Outputs _out;
  60     (void)_out;
  61     half2x4 h24 = matrixCompMult(half2x4(half4(9.0h, 9.0h, 9.0h, 9.0h), half4(9.0h, 9.0h, 9.0h, 9.0h)), half2x4(_uniforms.colorRed, _uniforms.colorGreen));
  62     half4x2 h42 = matrixCompMult(half4x2(half2(1.0h, 2.0h), half2(3.0h, 4.0h), half2(5.0h, 6.0h), half2(7.0h, 8.0h)), half4x2_from_half4_half4(_uniforms.colorRed, _uniforms.colorGreen));
  63     float4x3 f43 = float4x3(float3(12.0, 22.0, 30.0), float3(36.0, 40.0, 42.0), float3(42.0, 40.0, 36.0), float3(30.0, 22.0, 12.0));
  64     _out.sk_FragColor = (h24 == half2x4(half4(9.0h, 0.0h, 0.0h, 9.0h), half4(0.0h, 9.0h, 0.0h, 9.0h)) && h42 == half4x2(half2(1.0h, 0.0h), half2(0.0h, 4.0h), half2(0.0h, 6.0h), half2(0.0h, 8.0h))) && f43 == float4x3(float3(12.0, 22.0, 30.0), float3(36.0, 40.0, 42.0), float3(42.0, 40.0, 36.0), float3(30.0, 22.0, 12.0)) ? _uniforms.colorGreen : _uniforms.colorRed;
  65     return _out;
  66 }