8ed23eb917
These are not very interesting right now, because the in and out types boil down to the same thing (int/int, float/float). When half- precision types are enabled, these helpers will be more useful. They will return an array which casts each element from int-to-short or float-to-half (or vice versa). Change-Id: Ida716ddd27d370ba33fd23f17a1b07fa5a201e40 Bug: skia:12339 Reviewed-on: https://skia-review.googlesource.com/c/skia/+/463337 Commit-Queue: John Stiles <johnstiles@google.com> Reviewed-by: Brian Osman <brianosman@google.com>
88 lines
3.0 KiB
Metal
88 lines
3.0 KiB
Metal
#include <metal_stdlib>
|
|
#include <simd/simd.h>
|
|
using namespace metal;
|
|
struct Uniforms {
|
|
float4 colorGreen;
|
|
float4 colorRed;
|
|
};
|
|
struct Inputs {
|
|
};
|
|
struct Outputs {
|
|
float4 sk_FragColor [[color(0)]];
|
|
};
|
|
|
|
template <typename T1, typename T2, size_t N>
|
|
bool operator==(thread const array<T1, N>& left, thread const array<T2, N>& right);
|
|
template <typename T1, typename T2, size_t N>
|
|
bool operator!=(thread const array<T1, N>& left, thread const array<T2, N>& right);
|
|
|
|
thread bool operator==(const float2x2 left, const float2x2 right);
|
|
thread bool operator!=(const float2x2 left, const float2x2 right);
|
|
|
|
template <size_t N>
|
|
array<float, N> array_of_float_from_float(thread const array<float, N>& x) {
|
|
array<float, N> result;
|
|
for (int i = 0; i < N; ++i) {
|
|
result[i] = float(x[i]);
|
|
}
|
|
return result;
|
|
}
|
|
|
|
template <size_t N>
|
|
array<int3, N> array_of_int3_from_int3(thread const array<int3, N>& x) {
|
|
array<int3, N> result;
|
|
for (int i = 0; i < N; ++i) {
|
|
result[i] = int3(x[i]);
|
|
}
|
|
return result;
|
|
}
|
|
|
|
template <size_t N>
|
|
array<float2x2, N> array_of_float2x2_from_float2x2(thread const array<float2x2, N>& x) {
|
|
array<float2x2, N> result;
|
|
for (int i = 0; i < N; ++i) {
|
|
result[i] = float2x2(x[i]);
|
|
}
|
|
return result;
|
|
}
|
|
|
|
template <typename T1, typename T2, size_t N>
|
|
bool operator==(thread const array<T1, N>& left, thread const array<T2, N>& right) {
|
|
for (size_t index = 0; index < N; ++index) {
|
|
if (!all(left[index] == right[index])) {
|
|
return false;
|
|
}
|
|
}
|
|
return true;
|
|
}
|
|
|
|
template <typename T1, typename T2, size_t N>
|
|
bool operator!=(thread const array<T1, N>& left, thread const array<T2, N>& right) {
|
|
return !(left == right);
|
|
}
|
|
thread bool operator==(const float2x2 left, const float2x2 right) {
|
|
return all(left[0] == right[0]) &&
|
|
all(left[1] == right[1]);
|
|
}
|
|
thread bool operator!=(const float2x2 left, const float2x2 right) {
|
|
return !(left == right);
|
|
}
|
|
fragment Outputs fragmentMain(Inputs _in [[stage_in]], constant Uniforms& _uniforms [[buffer(0)]], bool _frontFacing [[front_facing]], float4 _fragCoord [[position]]) {
|
|
Outputs _out;
|
|
(void)_out;
|
|
array<float, 4> f = array<float, 4>{1.0, 2.0, 3.0, 4.0};
|
|
array<float, 4> h = array_of_float_from_float(f);
|
|
f = array_of_float_from_float(h);
|
|
h = array_of_float_from_float(f);
|
|
array<int3, 3> i3 = array<int3, 3>{int3(1), int3(2), int3(3)};
|
|
array<int3, 3> s3 = array_of_int3_from_int3(i3);
|
|
i3 = array_of_int3_from_int3(s3);
|
|
s3 = array_of_int3_from_int3(i3);
|
|
array<float2x2, 2> h2x2 = array<float2x2, 2>{float2x2(float2(1.0, 2.0), float2(3.0, 4.0)), float2x2(float2(5.0, 6.0), float2(7.0, 8.0))};
|
|
array<float2x2, 2> f2x2 = array_of_float2x2_from_float2x2(h2x2);
|
|
f2x2 = array_of_float2x2_from_float2x2(h2x2);
|
|
h2x2 = array_of_float2x2_from_float2x2(f2x2);
|
|
_out.sk_FragColor = (f == array_of_float_from_float(h) && i3 == array_of_int3_from_int3(s3)) && f2x2 == array_of_float2x2_from_float2x2(h2x2) ? _uniforms.colorGreen : _uniforms.colorRed;
|
|
return _out;
|
|
}
|