SPIRV-Cross/reference/shaders-msl-no-opt/packing/matrix-4x4-std430.comp

89 lines
2.4 KiB
Plaintext
Raw Normal View History

2019-07-22 09:23:06 +00:00
#pragma clang diagnostic ignored "-Wmissing-prototypes"
#include <metal_stdlib>
#include <simd/simd.h>
using namespace metal;
struct SSBOCol
{
float4x4 col_major0;
float4x4 col_major1;
};
struct SSBORow
{
float4x4 row_major0;
float4x4 row_major1;
};
2019-09-24 22:13:04 +00:00
constant uint3 gl_WorkGroupSize [[maybe_unused]] = uint3(1u);
2019-09-17 19:11:19 +00:00
static inline __attribute__((always_inline))
void load_store_to_variable_col_major(device SSBOCol& v_29)
2019-07-22 09:23:06 +00:00
{
float4x4 loaded = v_29.col_major0;
v_29.col_major1 = loaded;
}
2019-09-17 19:11:19 +00:00
static inline __attribute__((always_inline))
void load_store_to_variable_row_major(device SSBORow& v_41)
2019-07-22 09:23:06 +00:00
{
float4x4 loaded = transpose(v_41.row_major0);
v_41.row_major0 = transpose(loaded);
}
2019-09-17 19:11:19 +00:00
static inline __attribute__((always_inline))
void copy_col_major_to_col_major(device SSBOCol& v_29)
2019-07-22 09:23:06 +00:00
{
v_29.col_major0 = v_29.col_major1;
}
2019-09-17 19:11:19 +00:00
static inline __attribute__((always_inline))
void copy_col_major_to_row_major(device SSBOCol& v_29, device SSBORow& v_41)
2019-07-22 09:23:06 +00:00
{
v_41.row_major0 = transpose(v_29.col_major0);
}
2019-09-17 19:11:19 +00:00
static inline __attribute__((always_inline))
void copy_row_major_to_col_major(device SSBOCol& v_29, device SSBORow& v_41)
2019-07-22 09:23:06 +00:00
{
v_29.col_major0 = transpose(v_41.row_major0);
}
2019-09-17 19:11:19 +00:00
static inline __attribute__((always_inline))
void copy_row_major_to_row_major(device SSBORow& v_41)
2019-07-22 09:23:06 +00:00
{
v_41.row_major0 = v_41.row_major1;
}
2019-09-17 19:11:19 +00:00
static inline __attribute__((always_inline))
void copy_columns(device SSBOCol& v_29, device SSBORow& v_41)
2019-07-22 09:23:06 +00:00
{
v_29.col_major0[1] = float4(v_41.row_major0[0][1], v_41.row_major0[1][1], v_41.row_major0[2][1], v_41.row_major0[3][1]);
v_41.row_major0[0][1] = v_29.col_major0[1].x;
v_41.row_major0[1][1] = v_29.col_major0[1].y;
v_41.row_major0[2][1] = v_29.col_major0[1].z;
v_41.row_major0[3][1] = v_29.col_major0[1].w;
}
2019-09-17 19:11:19 +00:00
static inline __attribute__((always_inline))
void copy_elements(device SSBOCol& v_29, device SSBORow& v_41)
2019-07-22 09:23:06 +00:00
{
((device float*)&v_29.col_major0[0])[1u] = ((device float*)&v_41.row_major0[1u])[0];
((device float*)&v_41.row_major0[1u])[0] = ((device float*)&v_29.col_major0[0])[1u];
2019-07-22 09:23:06 +00:00
}
kernel void main0(device SSBOCol& v_29 [[buffer(0)]], device SSBORow& v_41 [[buffer(1)]])
{
load_store_to_variable_col_major(v_29);
load_store_to_variable_row_major(v_41);
copy_col_major_to_col_major(v_29);
copy_col_major_to_row_major(v_29, v_41);
copy_row_major_to_col_major(v_29, v_41);
copy_row_major_to_row_major(v_41);
copy_columns(v_29, v_41);
copy_elements(v_29, v_41);
}