reference/shaders-msl/tese/read-tess-level-in-func.raw-tess-in.msl2.tese

   1 #pragma clang diagnostic ignored "-Wmissing-prototypes"
   2 #pragma clang diagnostic ignored "-Wmissing-braces"
   3
   4 #include <metal_stdlib>
   5 #include <simd/simd.h>
   6
   7 using namespace metal;
   8
   9 template<typename T, size_t Num>
  10 struct spvUnsafeArray
  11 {
  12     T elements[Num ? Num : 1];
  13
  14     thread T& operator [] (size_t pos) thread
  15     {
  16         return elements[pos];
  17     }
  18     constexpr const thread T& operator [] (size_t pos) const thread
  19     {
  20         return elements[pos];
  21     }
  22
  23     device T& operator [] (size_t pos) device
  24     {
  25         return elements[pos];
  26     }
  27     constexpr const device T& operator [] (size_t pos) const device
  28     {
  29         return elements[pos];
  30     }
  31
  32     constexpr const constant T& operator [] (size_t pos) const constant
  33     {
  34         return elements[pos];
  35     }
  36
  37     threadgroup T& operator [] (size_t pos) threadgroup
  38     {
  39         return elements[pos];
  40     }
  41     constexpr const threadgroup T& operator [] (size_t pos) const threadgroup
  42     {
  43         return elements[pos];
  44     }
  45 };
  46
  47 struct main0_out
  48 {
  49     float4 gl_Position [[position]];
  50 };
  51
  52 static inline __attribute__((always_inline))
  53 float4 read_tess_levels(thread spvUnsafeArray<float, 4>& gl_TessLevelOuter, thread spvUnsafeArray<float, 2>& gl_TessLevelInner)
  54 {
  55     return float4(gl_TessLevelOuter[0], gl_TessLevelOuter[1], gl_TessLevelOuter[2], gl_TessLevelOuter[3]) + float2(gl_TessLevelInner[0], gl_TessLevelInner[1]).xyxy;
  56 }
  57
  58 [[ patch(triangle, 0) ]] vertex main0_out main0(uint gl_PrimitiveID [[patch_id]], const device MTLTriangleTessellationFactorsHalf* spvTessLevel [[buffer(26)]])
  59 {
  60     main0_out out = {};
  61     spvUnsafeArray<float, 4> gl_TessLevelOuter = {};
  62     spvUnsafeArray<float, 2> gl_TessLevelInner = {};
  63     gl_TessLevelOuter[0] = spvTessLevel[gl_PrimitiveID].edgeTessellationFactor[0];
  64     gl_TessLevelOuter[1] = spvTessLevel[gl_PrimitiveID].edgeTessellationFactor[1];
  65     gl_TessLevelOuter[2] = spvTessLevel[gl_PrimitiveID].edgeTessellationFactor[2];
  66     gl_TessLevelInner[0] = spvTessLevel[gl_PrimitiveID].insideTessellationFactor;
  67     out.gl_Position = read_tess_levels(gl_TessLevelOuter, gl_TessLevelInner);
  68     return out;
  69 }
  70