SPIRV-Cross/reference/shaders-msl-no-opt/packing/matrix-3x4-scalar.comp

100 lines
3.8 KiB
Plaintext
Raw Normal View History

#pragma clang diagnostic ignored "-Wmissing-prototypes"
#include <metal_stdlib>
#include <simd/simd.h>
using namespace metal;
typedef packed_float3 packed_rm_float3x4[4];
struct SSBOCol
{
float3x4 col_major0;
float3x4 col_major1;
};
struct SSBORow
{
packed_rm_float3x4 row_major0;
packed_rm_float3x4 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& _29)
{
float3x4 loaded = _29.col_major0;
_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& _41)
{
float3x4 loaded = transpose(float4x3(float3(_41.row_major0[0]), float3(_41.row_major0[1]), float3(_41.row_major0[2]), float3(_41.row_major0[3])));
_41.row_major0[0] = float3(loaded[0][0], loaded[1][0], loaded[2][0]);
_41.row_major0[1] = float3(loaded[0][1], loaded[1][1], loaded[2][1]);
_41.row_major0[2] = float3(loaded[0][2], loaded[1][2], loaded[2][2]);
_41.row_major0[3] = float3(loaded[0][3], loaded[1][3], loaded[2][3]);
}
2019-09-17 19:11:19 +00:00
static inline __attribute__((always_inline))
void copy_col_major_to_col_major(device SSBOCol& _29)
{
_29.col_major0 = _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& _29, device SSBORow& _41)
{
_41.row_major0[0] = float3(_29.col_major0[0][0], _29.col_major0[1][0], _29.col_major0[2][0]);
_41.row_major0[1] = float3(_29.col_major0[0][1], _29.col_major0[1][1], _29.col_major0[2][1]);
_41.row_major0[2] = float3(_29.col_major0[0][2], _29.col_major0[1][2], _29.col_major0[2][2]);
_41.row_major0[3] = float3(_29.col_major0[0][3], _29.col_major0[1][3], _29.col_major0[2][3]);
}
2019-09-17 19:11:19 +00:00
static inline __attribute__((always_inline))
void copy_row_major_to_col_major(device SSBOCol& _29, device SSBORow& _41)
{
_29.col_major0 = transpose(float4x3(float3(_41.row_major0[0]), float3(_41.row_major0[1]), float3(_41.row_major0[2]), float3(_41.row_major0[3])));
}
2019-09-17 19:11:19 +00:00
static inline __attribute__((always_inline))
void copy_row_major_to_row_major(device SSBORow& _41)
{
_41.row_major0[0] = float4x3(float3(_41.row_major1[0]), float3(_41.row_major1[1]), float3(_41.row_major1[2]), float3(_41.row_major1[3]))[0];
_41.row_major0[1] = float4x3(float3(_41.row_major1[0]), float3(_41.row_major1[1]), float3(_41.row_major1[2]), float3(_41.row_major1[3]))[1];
_41.row_major0[2] = float4x3(float3(_41.row_major1[0]), float3(_41.row_major1[1]), float3(_41.row_major1[2]), float3(_41.row_major1[3]))[2];
_41.row_major0[3] = float4x3(float3(_41.row_major1[0]), float3(_41.row_major1[1]), float3(_41.row_major1[2]), float3(_41.row_major1[3]))[3];
}
2019-09-17 19:11:19 +00:00
static inline __attribute__((always_inline))
void copy_columns(device SSBOCol& _29, device SSBORow& _41)
{
_29.col_major0[1] = float4(_41.row_major0[0][1], _41.row_major0[1][1], _41.row_major0[2][1], _41.row_major0[3][1]);
((device float*)&_41.row_major0[0])[1] = _29.col_major0[1].x;
((device float*)&_41.row_major0[1])[1] = _29.col_major0[1].y;
((device float*)&_41.row_major0[2])[1] = _29.col_major0[1].z;
((device float*)&_41.row_major0[3])[1] = _29.col_major0[1].w;
}
2019-09-17 19:11:19 +00:00
static inline __attribute__((always_inline))
void copy_elements(device SSBOCol& _29, device SSBORow& _41)
{
((device float*)&_29.col_major0[0])[1u] = _41.row_major0[1u][0];
_41.row_major0[1u][0] = ((device float*)&_29.col_major0[0])[1u];
}
kernel void main0(device SSBOCol& _29 [[buffer(0)]], device SSBORow& _41 [[buffer(1)]])
{
load_store_to_variable_col_major(_29);
load_store_to_variable_row_major(_41);
copy_col_major_to_col_major(_29);
copy_col_major_to_row_major(_29, _41);
copy_row_major_to_col_major(_29, _41);
copy_row_major_to_row_major(_41);
copy_columns(_29, _41);
copy_elements(_29, _41);
}