mirror of
https://git.suyu.dev/suyu/suyu.git
synced 2024-11-02 13:02:44 +01:00
fix: more msl errors
This commit is contained in:
parent
055112b739
commit
49f60adeb2
8 changed files with 139 additions and 70 deletions
|
@ -155,7 +155,7 @@ void EmitCode(EmitContext& ctx, const IR::Program& program) {
|
||||||
break;
|
break;
|
||||||
case IR::AbstractSyntaxNode::Type::Return:
|
case IR::AbstractSyntaxNode::Type::Return:
|
||||||
case IR::AbstractSyntaxNode::Type::Unreachable:
|
case IR::AbstractSyntaxNode::Type::Unreachable:
|
||||||
ctx.Add("return;");
|
ctx.Add("return __out;");
|
||||||
break;
|
break;
|
||||||
case IR::AbstractSyntaxNode::Type::Loop:
|
case IR::AbstractSyntaxNode::Type::Loop:
|
||||||
ctx.Add("for(;;){{");
|
ctx.Add("for(;;){{");
|
||||||
|
|
|
@ -37,7 +37,7 @@ std::string_view OutputVertexIndex(EmitContext& ctx) {
|
||||||
|
|
||||||
std::string ChooseCbuf(EmitContext& ctx, const IR::Value& binding, std::string_view index) {
|
std::string ChooseCbuf(EmitContext& ctx, const IR::Value& binding, std::string_view index) {
|
||||||
if (binding.IsImmediate()) {
|
if (binding.IsImmediate()) {
|
||||||
return fmt::format("{}_cbuf{}[{}]", ctx.stage_name, binding.U32(), index);
|
return fmt::format("{}_cbuf{}.data[{}]", ctx.stage_name, binding.U32(), index);
|
||||||
} else {
|
} else {
|
||||||
const auto binding_var{ctx.var_alloc.Consume(binding)};
|
const auto binding_var{ctx.var_alloc.Consume(binding)};
|
||||||
return fmt::format("GetCbufIndirect({},{})", binding_var, index);
|
return fmt::format("GetCbufIndirect({},{})", binding_var, index);
|
||||||
|
@ -151,7 +151,7 @@ void EmitGetCbufU32x2(EmitContext& ctx, IR::Inst& inst, const IR::Value& binding
|
||||||
const IR::Value& offset) {
|
const IR::Value& offset) {
|
||||||
const auto cast{ctx.profile.has_gl_cbuf_ftou_bug ? "" : "as_type<uint>"};
|
const auto cast{ctx.profile.has_gl_cbuf_ftou_bug ? "" : "as_type<uint>"};
|
||||||
if (offset.IsImmediate()) {
|
if (offset.IsImmediate()) {
|
||||||
const auto cbuf{fmt::format("{}_cbuf{}", ctx.stage_name, binding.U32())};
|
const auto cbuf{fmt::format("{}_cbuf{}.data", ctx.stage_name, binding.U32())};
|
||||||
static constexpr u32 cbuf_size{0x10000};
|
static constexpr u32 cbuf_size{0x10000};
|
||||||
const u32 u32_offset{offset.U32()};
|
const u32 u32_offset{offset.U32()};
|
||||||
const s32 signed_offset{static_cast<s32>(offset.U32())};
|
const s32 signed_offset{static_cast<s32>(offset.U32())};
|
||||||
|
@ -213,7 +213,7 @@ void EmitGetAttribute(EmitContext& ctx, IR::Inst& inst, IR::Attribute attr,
|
||||||
case IR::Attribute::PositionY:
|
case IR::Attribute::PositionY:
|
||||||
case IR::Attribute::PositionZ:
|
case IR::Attribute::PositionZ:
|
||||||
case IR::Attribute::PositionW: {
|
case IR::Attribute::PositionW: {
|
||||||
ctx.AddF32("{}={}.{};", inst, "__out.position", swizzle);
|
ctx.AddF32("{}={}.{};", inst, "__in.position", swizzle);
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
case IR::Attribute::PointSpriteS:
|
case IR::Attribute::PointSpriteS:
|
||||||
|
@ -225,10 +225,10 @@ void EmitGetAttribute(EmitContext& ctx, IR::Inst& inst, IR::Attribute attr,
|
||||||
ctx.AddF32("{}=gl_TessCoord.{};", inst, swizzle);
|
ctx.AddF32("{}=gl_TessCoord.{};", inst, swizzle);
|
||||||
break;
|
break;
|
||||||
case IR::Attribute::InstanceId:
|
case IR::Attribute::InstanceId:
|
||||||
ctx.AddF32("{}=as_type<float>(gl_InstanceID);", inst);
|
ctx.AddF32("{}=as_type<float>(iid);", inst);
|
||||||
break;
|
break;
|
||||||
case IR::Attribute::VertexId:
|
case IR::Attribute::VertexId:
|
||||||
ctx.AddF32("{}=as_type<float>(gl_VertexID);", inst);
|
ctx.AddF32("{}=as_type<float>(vid);", inst);
|
||||||
break;
|
break;
|
||||||
case IR::Attribute::FrontFace:
|
case IR::Attribute::FrontFace:
|
||||||
ctx.AddF32("{}=as_type<float>(gl_FrontFacing?-1:0);", inst);
|
ctx.AddF32("{}=as_type<float>(gl_FrontFacing?-1:0);", inst);
|
||||||
|
@ -253,10 +253,10 @@ void EmitGetAttributeU32(EmitContext& ctx, IR::Inst& inst, IR::Attribute attr, s
|
||||||
ctx.AddU32("{}=uint(gl_PrimitiveID);", inst);
|
ctx.AddU32("{}=uint(gl_PrimitiveID);", inst);
|
||||||
break;
|
break;
|
||||||
case IR::Attribute::InstanceId:
|
case IR::Attribute::InstanceId:
|
||||||
ctx.AddU32("{}=uint(gl_InstanceID);", inst);
|
ctx.AddU32("{}=uint(iid);", inst);
|
||||||
break;
|
break;
|
||||||
case IR::Attribute::VertexId:
|
case IR::Attribute::VertexId:
|
||||||
ctx.AddU32("{}=uint(gl_VertexID);", inst);
|
ctx.AddU32("{}=uint(vid);", inst);
|
||||||
break;
|
break;
|
||||||
case IR::Attribute::BaseInstance:
|
case IR::Attribute::BaseInstance:
|
||||||
ctx.AddU32("{}=uint(gl_BaseInstance);", inst);
|
ctx.AddU32("{}=uint(gl_BaseInstance);", inst);
|
||||||
|
@ -396,7 +396,7 @@ void EmitSetPatch(EmitContext& ctx, IR::Patch patch, std::string_view value) {
|
||||||
|
|
||||||
void EmitSetFragColor(EmitContext& ctx, u32 index, u32 component, std::string_view value) {
|
void EmitSetFragColor(EmitContext& ctx, u32 index, u32 component, std::string_view value) {
|
||||||
const char swizzle{"xyzw"[component]};
|
const char swizzle{"xyzw"[component]};
|
||||||
ctx.Add("frag_color{}.{}={};", index, swizzle, value);
|
ctx.Add("__out.color{}.{}={};", index, swizzle, value);
|
||||||
}
|
}
|
||||||
|
|
||||||
void EmitSetSampleMask(EmitContext& ctx, std::string_view value) {
|
void EmitSetSampleMask(EmitContext& ctx, std::string_view value) {
|
||||||
|
|
|
@ -12,7 +12,7 @@ void EmitJoin(EmitContext&) {
|
||||||
}
|
}
|
||||||
|
|
||||||
void EmitDemoteToHelperInvocation(EmitContext& ctx) {
|
void EmitDemoteToHelperInvocation(EmitContext& ctx) {
|
||||||
ctx.Add("discard;");
|
ctx.Add("discard_fragment();");
|
||||||
}
|
}
|
||||||
|
|
||||||
} // namespace Shader::Backend::MSL
|
} // namespace Shader::Backend::MSL
|
||||||
|
|
|
@ -18,6 +18,26 @@ std::string Texture(EmitContext& ctx, const IR::TextureInstInfo& info, const IR:
|
||||||
return fmt::format("tex{}{}", def.binding, index_offset);
|
return fmt::format("tex{}{}", def.binding, index_offset);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
void TransformTextureCoords(const IR::TextureInstInfo& info, std::string_view& coords) {
|
||||||
|
// If the texture is array, the last component of the coords is the array index
|
||||||
|
if (info.type == TextureType::ColorArray1D) {
|
||||||
|
coords = fmt::format("{}.x,{}.y", coords, coords);
|
||||||
|
}
|
||||||
|
if (info.type == TextureType::ColorArray2D) {
|
||||||
|
coords = fmt::format("{}.xy,{}.z", coords, coords);
|
||||||
|
}
|
||||||
|
if (info.type == TextureType::ColorArrayCube) {
|
||||||
|
coords = fmt::format("{}.xyz,{}.w", coords, coords);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string Sampler(EmitContext& ctx, const IR::TextureInstInfo& info, const IR::Value& index) {
|
||||||
|
const auto def{info.type == TextureType::Buffer ? ctx.texture_buffers.at(info.descriptor_index)
|
||||||
|
: ctx.textures.at(info.descriptor_index)};
|
||||||
|
const auto index_offset{def.count > 1 ? fmt::format("[{}]", ctx.var_alloc.Consume(index)) : ""};
|
||||||
|
return fmt::format("samp{}{}", def.binding, index_offset);
|
||||||
|
}
|
||||||
|
|
||||||
std::string Image(EmitContext& ctx, const IR::TextureInstInfo& info, const IR::Value& index) {
|
std::string Image(EmitContext& ctx, const IR::TextureInstInfo& info, const IR::Value& index) {
|
||||||
const auto def{info.type == TextureType::Buffer ? ctx.image_buffers.at(info.descriptor_index)
|
const auto def{info.type == TextureType::Buffer ? ctx.image_buffers.at(info.descriptor_index)
|
||||||
: ctx.images.at(info.descriptor_index)};
|
: ctx.images.at(info.descriptor_index)};
|
||||||
|
@ -160,6 +180,7 @@ std::string ImageGatherSubpixelOffset(const IR::TextureInstInfo& info, std::stri
|
||||||
}
|
}
|
||||||
} // Anonymous namespace
|
} // Anonymous namespace
|
||||||
|
|
||||||
|
// TODO
|
||||||
void EmitImageSampleImplicitLod(EmitContext& ctx, IR::Inst& inst, const IR::Value& index,
|
void EmitImageSampleImplicitLod(EmitContext& ctx, IR::Inst& inst, const IR::Value& index,
|
||||||
std::string_view coords, std::string_view bias_lc,
|
std::string_view coords, std::string_view bias_lc,
|
||||||
const IR::Value& offset) {
|
const IR::Value& offset) {
|
||||||
|
@ -168,6 +189,8 @@ void EmitImageSampleImplicitLod(EmitContext& ctx, IR::Inst& inst, const IR::Valu
|
||||||
throw NotImplementedException("EmitImageSampleImplicitLod Lod clamp samples");
|
throw NotImplementedException("EmitImageSampleImplicitLod Lod clamp samples");
|
||||||
}
|
}
|
||||||
const auto texture{Texture(ctx, info, index)};
|
const auto texture{Texture(ctx, info, index)};
|
||||||
|
TransformTextureCoords(info, coords);
|
||||||
|
const auto sampler{Sampler(ctx, info, index)};
|
||||||
const auto bias{info.has_bias ? fmt::format(",{}", bias_lc) : ""};
|
const auto bias{info.has_bias ? fmt::format(",{}", bias_lc) : ""};
|
||||||
const auto texel{ctx.var_alloc.Define(inst, MslVarType::F32x4)};
|
const auto texel{ctx.var_alloc.Define(inst, MslVarType::F32x4)};
|
||||||
const auto sparse_inst{PrepareSparse(inst)};
|
const auto sparse_inst{PrepareSparse(inst)};
|
||||||
|
@ -186,9 +209,10 @@ void EmitImageSampleImplicitLod(EmitContext& ctx, IR::Inst& inst, const IR::Valu
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
if (ctx.stage == Stage::Fragment) {
|
if (ctx.stage == Stage::Fragment) {
|
||||||
ctx.Add("{}=texture({},{}{});", texel, texture, coords, bias);
|
// TODO: is the array index correct
|
||||||
|
ctx.Add("{}={}.sample({},{}{});", texel, texture, sampler, coords, bias);
|
||||||
} else {
|
} else {
|
||||||
ctx.Add("{}=textureLod({},{},0.0);", texel, texture, coords);
|
ctx.Add("{}={}.sampleLod({},{},0.0);", texel, texture, sampler, coords);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return;
|
return;
|
||||||
|
@ -213,6 +237,7 @@ void EmitImageSampleExplicitLod(EmitContext& ctx, IR::Inst& inst, const IR::Valu
|
||||||
throw NotImplementedException("EmitImageSampleExplicitLod Lod clamp samples");
|
throw NotImplementedException("EmitImageSampleExplicitLod Lod clamp samples");
|
||||||
}
|
}
|
||||||
const auto texture{Texture(ctx, info, index)};
|
const auto texture{Texture(ctx, info, index)};
|
||||||
|
TransformTextureCoords(info, coords);
|
||||||
const auto texel{ctx.var_alloc.Define(inst, MslVarType::F32x4)};
|
const auto texel{ctx.var_alloc.Define(inst, MslVarType::F32x4)};
|
||||||
const auto sparse_inst{PrepareSparse(inst)};
|
const auto sparse_inst{PrepareSparse(inst)};
|
||||||
const bool supports_sparse{ctx.profile.support_gl_sparse_textures};
|
const bool supports_sparse{ctx.profile.support_gl_sparse_textures};
|
||||||
|
@ -254,6 +279,7 @@ void EmitImageSampleDrefImplicitLod(EmitContext& ctx, IR::Inst& inst, const IR::
|
||||||
throw NotImplementedException("EmitImageSampleDrefImplicitLod Lod clamp samples");
|
throw NotImplementedException("EmitImageSampleDrefImplicitLod Lod clamp samples");
|
||||||
}
|
}
|
||||||
const auto texture{Texture(ctx, info, index)};
|
const auto texture{Texture(ctx, info, index)};
|
||||||
|
TransformTextureCoords(info, coords);
|
||||||
const auto bias{info.has_bias ? fmt::format(",{}", bias_lc) : ""};
|
const auto bias{info.has_bias ? fmt::format(",{}", bias_lc) : ""};
|
||||||
const bool needs_shadow_ext{NeedsShadowLodExt(info.type)};
|
const bool needs_shadow_ext{NeedsShadowLodExt(info.type)};
|
||||||
const auto cast{needs_shadow_ext ? "float4" : "float3"};
|
const auto cast{needs_shadow_ext ? "float4" : "float3"};
|
||||||
|
@ -309,6 +335,7 @@ void EmitImageSampleDrefExplicitLod(EmitContext& ctx, IR::Inst& inst, const IR::
|
||||||
throw NotImplementedException("EmitImageSampleDrefExplicitLod Lod clamp samples");
|
throw NotImplementedException("EmitImageSampleDrefExplicitLod Lod clamp samples");
|
||||||
}
|
}
|
||||||
const auto texture{Texture(ctx, info, index)};
|
const auto texture{Texture(ctx, info, index)};
|
||||||
|
TransformTextureCoords(info, coords);
|
||||||
const bool needs_shadow_ext{NeedsShadowLodExt(info.type)};
|
const bool needs_shadow_ext{NeedsShadowLodExt(info.type)};
|
||||||
const bool use_grad{!ctx.profile.support_gl_texture_shadow_lod && needs_shadow_ext};
|
const bool use_grad{!ctx.profile.support_gl_texture_shadow_lod && needs_shadow_ext};
|
||||||
const auto cast{needs_shadow_ext ? "float3" : "float3"};
|
const auto cast{needs_shadow_ext ? "float3" : "float3"};
|
||||||
|
@ -348,6 +375,7 @@ void EmitImageGather(EmitContext& ctx, IR::Inst& inst, const IR::Value& index,
|
||||||
std::string_view coords, const IR::Value& offset, const IR::Value& offset2) {
|
std::string_view coords, const IR::Value& offset, const IR::Value& offset2) {
|
||||||
const auto info{inst.Flags<IR::TextureInstInfo>()};
|
const auto info{inst.Flags<IR::TextureInstInfo>()};
|
||||||
const auto texture{Texture(ctx, info, index)};
|
const auto texture{Texture(ctx, info, index)};
|
||||||
|
TransformTextureCoords(info, coords);
|
||||||
const auto texel{ctx.var_alloc.Define(inst, MslVarType::F32x4)};
|
const auto texel{ctx.var_alloc.Define(inst, MslVarType::F32x4)};
|
||||||
const auto sparse_inst{PrepareSparse(inst)};
|
const auto sparse_inst{PrepareSparse(inst)};
|
||||||
const bool supports_sparse{ctx.profile.support_gl_sparse_textures};
|
const bool supports_sparse{ctx.profile.support_gl_sparse_textures};
|
||||||
|
@ -402,6 +430,7 @@ void EmitImageGatherDref(EmitContext& ctx, IR::Inst& inst, const IR::Value& inde
|
||||||
std::string_view dref) {
|
std::string_view dref) {
|
||||||
const auto info{inst.Flags<IR::TextureInstInfo>()};
|
const auto info{inst.Flags<IR::TextureInstInfo>()};
|
||||||
const auto texture{Texture(ctx, info, index)};
|
const auto texture{Texture(ctx, info, index)};
|
||||||
|
TransformTextureCoords(info, coords);
|
||||||
const auto texel{ctx.var_alloc.Define(inst, MslVarType::F32x4)};
|
const auto texel{ctx.var_alloc.Define(inst, MslVarType::F32x4)};
|
||||||
const auto sparse_inst{PrepareSparse(inst)};
|
const auto sparse_inst{PrepareSparse(inst)};
|
||||||
const bool supports_sparse{ctx.profile.support_gl_sparse_textures};
|
const bool supports_sparse{ctx.profile.support_gl_sparse_textures};
|
||||||
|
@ -459,6 +488,7 @@ void EmitImageFetch(EmitContext& ctx, IR::Inst& inst, const IR::Value& index,
|
||||||
throw NotImplementedException("EmitImageFetch Lod clamp samples");
|
throw NotImplementedException("EmitImageFetch Lod clamp samples");
|
||||||
}
|
}
|
||||||
const auto texture{Texture(ctx, info, index)};
|
const auto texture{Texture(ctx, info, index)};
|
||||||
|
TransformTextureCoords(info, coords);
|
||||||
const auto sparse_inst{PrepareSparse(inst)};
|
const auto sparse_inst{PrepareSparse(inst)};
|
||||||
const auto texel{ctx.var_alloc.Define(inst, MslVarType::F32x4)};
|
const auto texel{ctx.var_alloc.Define(inst, MslVarType::F32x4)};
|
||||||
const bool supports_sparse{ctx.profile.support_gl_sparse_textures};
|
const bool supports_sparse{ctx.profile.support_gl_sparse_textures};
|
||||||
|
@ -552,6 +582,7 @@ void EmitImageGradient(EmitContext& ctx, IR::Inst& inst, const IR::Value& index,
|
||||||
throw NotImplementedException("EmitImageGradient offset");
|
throw NotImplementedException("EmitImageGradient offset");
|
||||||
}
|
}
|
||||||
const auto texture{Texture(ctx, info, index)};
|
const auto texture{Texture(ctx, info, index)};
|
||||||
|
TransformTextureCoords(info, coords);
|
||||||
const auto texel{ctx.var_alloc.Define(inst, MslVarType::F32x4)};
|
const auto texel{ctx.var_alloc.Define(inst, MslVarType::F32x4)};
|
||||||
const bool multi_component{info.num_derivatives > 1 || info.has_lod_clamp};
|
const bool multi_component{info.num_derivatives > 1 || info.has_lod_clamp};
|
||||||
const auto derivatives_vec{ctx.var_alloc.Consume(derivatives)};
|
const auto derivatives_vec{ctx.var_alloc.Consume(derivatives)};
|
||||||
|
|
|
@ -6,6 +6,7 @@
|
||||||
#include "shader_recompiler/backend/msl/msl_emit_context.h"
|
#include "shader_recompiler/backend/msl/msl_emit_context.h"
|
||||||
#include "shader_recompiler/frontend/ir/program.h"
|
#include "shader_recompiler/frontend/ir/program.h"
|
||||||
#include "shader_recompiler/profile.h"
|
#include "shader_recompiler/profile.h"
|
||||||
|
|
||||||
#include "shader_recompiler/runtime_info.h"
|
#include "shader_recompiler/runtime_info.h"
|
||||||
|
|
||||||
namespace Shader::Backend::MSL {
|
namespace Shader::Backend::MSL {
|
||||||
|
@ -18,6 +19,8 @@ char Swizzle(size_t offset) {
|
||||||
return "xyzw"[CbufIndex(offset)];
|
return "xyzw"[CbufIndex(offset)];
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// TODO
|
||||||
|
/*
|
||||||
std::string_view InterpDecorator(Interpolation interp) {
|
std::string_view InterpDecorator(Interpolation interp) {
|
||||||
switch (interp) {
|
switch (interp) {
|
||||||
case Interpolation::Smooth:
|
case Interpolation::Smooth:
|
||||||
|
@ -29,8 +32,10 @@ std::string_view InterpDecorator(Interpolation interp) {
|
||||||
}
|
}
|
||||||
throw InvalidArgument("Invalid interpolation {}", interp);
|
throw InvalidArgument("Invalid interpolation {}", interp);
|
||||||
}
|
}
|
||||||
|
*/
|
||||||
|
|
||||||
// TODO
|
// TODO
|
||||||
|
/*
|
||||||
std::string_view InputArrayDecorator(Stage stage) {
|
std::string_view InputArrayDecorator(Stage stage) {
|
||||||
switch (stage) {
|
switch (stage) {
|
||||||
case Stage::Geometry:
|
case Stage::Geometry:
|
||||||
|
@ -41,6 +46,7 @@ std::string_view InputArrayDecorator(Stage stage) {
|
||||||
return "";
|
return "";
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
*/
|
||||||
|
|
||||||
// TODO
|
// TODO
|
||||||
std::string OutputDecorator(Stage stage, u32 size) {
|
std::string OutputDecorator(Stage stage, u32 size) {
|
||||||
|
@ -208,13 +214,13 @@ std::string_view ImageFormatString(ImageFormat format) {
|
||||||
|
|
||||||
std::string_view ImageAccessQualifier(bool is_written, bool is_read) {
|
std::string_view ImageAccessQualifier(bool is_written, bool is_read) {
|
||||||
if (is_written && is_read) {
|
if (is_written && is_read) {
|
||||||
return "access::read, access::write";
|
return ",access::read,access::write";
|
||||||
}
|
}
|
||||||
if (is_written) {
|
if (is_written) {
|
||||||
return "access::write";
|
return ",access::write";
|
||||||
}
|
}
|
||||||
if (is_read) {
|
if (is_read) {
|
||||||
return "access::read";
|
return ",access::read";
|
||||||
}
|
}
|
||||||
return "";
|
return "";
|
||||||
}
|
}
|
||||||
|
@ -258,71 +264,98 @@ EmitContext::EmitContext(IR::Program& program, Bindings& bindings, const Profile
|
||||||
// SetupOutPerVertex(*this, header);
|
// SetupOutPerVertex(*this, header);
|
||||||
// SetupInPerVertex(*this, header);
|
// SetupInPerVertex(*this, header);
|
||||||
|
|
||||||
|
// Stage input
|
||||||
|
bool has_stage_input{};
|
||||||
|
header += "struct __Input {\n";
|
||||||
|
if (stage == Stage::Fragment) {
|
||||||
|
header += "float4 position [[position]];\n";
|
||||||
|
has_stage_input = true;
|
||||||
|
}
|
||||||
for (size_t index = 0; index < IR::NUM_GENERICS; ++index) {
|
for (size_t index = 0; index < IR::NUM_GENERICS; ++index) {
|
||||||
if (!info.loads.Generic(index) || !runtime_info.previous_stage_stores.Generic(index)) {
|
if (!info.loads.Generic(index) || !runtime_info.previous_stage_stores.Generic(index)) {
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
const auto qualifier{uses_geometry_passthrough ? "passthrough"
|
DefineStageInOut(index, program.invocations, true);
|
||||||
: fmt::format("location={}", index)};
|
has_stage_input = true;
|
||||||
header += fmt::format("layout({}){}in vec4 in_attr{}{};", qualifier,
|
|
||||||
InterpDecorator(info.interpolation[index]), index,
|
|
||||||
InputArrayDecorator(stage));
|
|
||||||
}
|
}
|
||||||
for (size_t index = 0; index < info.uses_patches.size(); ++index) {
|
for (size_t index = 0; index < info.uses_patches.size(); ++index) {
|
||||||
|
// TODO: what is this
|
||||||
if (!info.uses_patches[index]) {
|
if (!info.uses_patches[index]) {
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
const auto qualifier{stage == Stage::TessellationControl ? "out" : "in"};
|
// TODO: implement
|
||||||
header += fmt::format("layout(location={})patch {} vec4 patch{};", index, qualifier, index);
|
}
|
||||||
|
header += "};\n";
|
||||||
|
|
||||||
|
if (has_stage_input) {
|
||||||
|
input_str = "__Input __in [[stage_in]]";
|
||||||
|
has_at_least_one_input = true;
|
||||||
|
}
|
||||||
|
if (stage == Stage::VertexA || stage == Stage::VertexB) {
|
||||||
|
// TODO: don't always declare these
|
||||||
|
if (has_at_least_one_input)
|
||||||
|
input_str += ",";
|
||||||
|
input_str += "uint vid [[vertex_id]]";
|
||||||
|
has_at_least_one_input = true;
|
||||||
|
|
||||||
|
if (has_at_least_one_input)
|
||||||
|
input_str += ",";
|
||||||
|
input_str += "uint iid [[instance_id]]";
|
||||||
|
has_at_least_one_input = true;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Stage output
|
||||||
|
header += "struct __Output {\n";
|
||||||
|
if (stage == Stage::VertexB || stage == Stage::Geometry) {
|
||||||
|
header += "float4 position [[position]];\n";
|
||||||
}
|
}
|
||||||
if (stage == Stage::Fragment) {
|
if (stage == Stage::Fragment) {
|
||||||
for (size_t index = 0; index < info.stores_frag_color.size(); ++index) {
|
for (size_t index = 0; index < info.stores_frag_color.size(); ++index) {
|
||||||
if (!info.stores_frag_color[index] && !profile.need_declared_frag_colors) {
|
if (!info.stores_frag_color[index] && !profile.need_declared_frag_colors) {
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
header += fmt::format("layout(location={})out vec4 frag_color{};", index, index);
|
header += fmt::format("float4 color{} [[color({})]];", index, index);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
header += "struct __Output {\n";
|
|
||||||
if (stage == Stage::VertexB || stage == Stage::Geometry) {
|
|
||||||
header += "float4 position [[position]];\n";
|
|
||||||
}
|
|
||||||
for (size_t index = 0; index < IR::NUM_GENERICS; ++index) {
|
for (size_t index = 0; index < IR::NUM_GENERICS; ++index) {
|
||||||
if (info.stores.Generic(index)) {
|
if (info.stores.Generic(index)) {
|
||||||
DefineGenericOutput(index, program.invocations);
|
DefineStageInOut(index, program.invocations, false);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
header += "};\n";
|
header += "};\n";
|
||||||
bool added = DefineInputs(bindings);
|
DefineInputs(bindings);
|
||||||
if (info.uses_rescaling_uniform) {
|
if (info.uses_rescaling_uniform) {
|
||||||
if (added)
|
if (has_at_least_one_input)
|
||||||
input_str += ",";
|
input_str += ",";
|
||||||
input_str += "constant float4& scaling";
|
input_str += "constant float4& scaling";
|
||||||
added = true;
|
has_at_least_one_input = true;
|
||||||
}
|
}
|
||||||
if (info.uses_render_area) {
|
if (info.uses_render_area) {
|
||||||
if (added)
|
if (has_at_least_one_input)
|
||||||
input_str += ",";
|
input_str += ",";
|
||||||
input_str += "constant float4& render_area";
|
input_str += "constant float4& render_area";
|
||||||
added = true;
|
has_at_least_one_input = true;
|
||||||
}
|
}
|
||||||
DefineHelperFunctions();
|
DefineHelperFunctions();
|
||||||
DefineConstants();
|
DefineConstants();
|
||||||
}
|
}
|
||||||
|
|
||||||
bool EmitContext::DefineInputs(Bindings& bindings) {
|
void EmitContext::DefineInputs(Bindings& bindings) {
|
||||||
bool added{false};
|
|
||||||
|
|
||||||
// Constant buffers
|
// Constant buffers
|
||||||
for (const auto& desc : info.constant_buffer_descriptors) {
|
for (const auto& desc : info.constant_buffer_descriptors) {
|
||||||
const u32 cbuf_used_size{Common::DivCeil(info.constant_buffer_used_sizes[desc.index], 16U)};
|
const u32 cbuf_used_size{Common::DivCeil(info.constant_buffer_used_sizes[desc.index], 16U)};
|
||||||
const u32 cbuf_binding_size{info.uses_global_memory ? 0x1000U : cbuf_used_size};
|
const u32 cbuf_binding_size{info.uses_global_memory ? 0x1000U : cbuf_used_size};
|
||||||
if (added)
|
|
||||||
|
const std::string cbuf_struct_name{fmt::format("{}_cbuf{}_t", stage_name, desc.index)};
|
||||||
|
header +=
|
||||||
|
fmt::format("struct {} {{float4 data[{}];}};\n", cbuf_struct_name, cbuf_binding_size);
|
||||||
|
|
||||||
|
if (has_at_least_one_input)
|
||||||
input_str += ",";
|
input_str += ",";
|
||||||
input_str += fmt::format("constant float4& {}_cbuf{}[{}] [[buffer({})]]", stage_name,
|
input_str += fmt::format("constant {}& {}_cbuf{} [[buffer({})]]", cbuf_struct_name,
|
||||||
desc.index, cbuf_binding_size, bindings.uniform_buffer);
|
stage_name, desc.index, bindings.uniform_buffer);
|
||||||
bindings.uniform_buffer += desc.count;
|
bindings.uniform_buffer += desc.count;
|
||||||
added = true;
|
has_at_least_one_input = true;
|
||||||
}
|
}
|
||||||
|
|
||||||
// Constant buffer indirect
|
// Constant buffer indirect
|
||||||
|
@ -331,13 +364,14 @@ bool EmitContext::DefineInputs(Bindings& bindings) {
|
||||||
// Storage space buffers
|
// Storage space buffers
|
||||||
u32 index{};
|
u32 index{};
|
||||||
for (const auto& desc : info.storage_buffers_descriptors) {
|
for (const auto& desc : info.storage_buffers_descriptors) {
|
||||||
if (added)
|
if (has_at_least_one_input)
|
||||||
input_str += ",";
|
input_str += ",";
|
||||||
input_str += fmt::format("device uint& {}_ssbo{}[] [[buffer({})]]", stage_name, index,
|
const std::string address_space{desc.is_written ? "device" : "constant"};
|
||||||
bindings.storage_buffer);
|
input_str += fmt::format("{} uint* {}_ssbo{} [[buffer({})]]", address_space, stage_name,
|
||||||
bindings.storage_buffer += desc.count;
|
index, bindings.uniform_buffer);
|
||||||
|
bindings.uniform_buffer += desc.count;
|
||||||
index += desc.count;
|
index += desc.count;
|
||||||
added = true;
|
has_at_least_one_input = true;
|
||||||
}
|
}
|
||||||
|
|
||||||
// Images
|
// Images
|
||||||
|
@ -350,24 +384,24 @@ bool EmitContext::DefineInputs(Bindings& bindings) {
|
||||||
const auto qualifier{ImageAccessQualifier(desc.is_written, desc.is_read)};
|
const auto qualifier{ImageAccessQualifier(desc.is_written, desc.is_read)};
|
||||||
const auto array_decorator{desc.count > 1 ? fmt::format("[{}]", desc.count) : ""};
|
const auto array_decorator{desc.count > 1 ? fmt::format("[{}]", desc.count) : ""};
|
||||||
input_str += fmt::format("layout(binding={}{}) uniform {}uimageBuffer img{}{};",
|
input_str += fmt::format("layout(binding={}{}) uniform {}uimageBuffer img{}{};",
|
||||||
bindings.image, format, qualifier, bindings.image, array_decorator);
|
bindings.image, format, qualifier, bindings.image,
|
||||||
bindings.image += desc.count;
|
array_decorator); bindings.image += desc.count;
|
||||||
}
|
}
|
||||||
*/
|
*/
|
||||||
images.reserve(info.image_descriptors.size());
|
images.reserve(info.image_descriptors.size());
|
||||||
for (const auto& desc : info.image_descriptors) {
|
for (const auto& desc : info.image_descriptors) {
|
||||||
images.push_back({bindings.image, desc.count});
|
images.push_back({bindings.texture, desc.count});
|
||||||
// TODO: do we need format?
|
// TODO: do we need format?
|
||||||
// const auto format{ImageFormatString(desc.format)};
|
// const auto format{ImageFormatString(desc.format)};
|
||||||
const auto image_type{ImageType(desc.type)};
|
const auto image_type{ImageType(desc.type)};
|
||||||
const auto qualifier{ImageAccessQualifier(desc.is_written, desc.is_read)};
|
const auto qualifier{ImageAccessQualifier(desc.is_written, desc.is_read)};
|
||||||
const auto array_decorator{desc.count > 1 ? fmt::format("[{}]", desc.count) : ""};
|
const auto array_decorator{desc.count > 1 ? fmt::format("[{}]", desc.count) : ""};
|
||||||
if (added)
|
if (has_at_least_one_input)
|
||||||
input_str += ",";
|
input_str += ",";
|
||||||
input_str += fmt::format("{}<{}> {}_img{}{} [[texture({})]]", qualifier, image_type,
|
input_str += fmt::format("{}<float{}> img{}{} [[texture({})]]", qualifier, image_type,
|
||||||
stage_name, bindings.image, array_decorator, bindings.image);
|
bindings.texture, array_decorator, bindings.texture);
|
||||||
bindings.image += desc.count;
|
bindings.texture += desc.count;
|
||||||
added = true;
|
has_at_least_one_input = true;
|
||||||
}
|
}
|
||||||
|
|
||||||
// Textures
|
// Textures
|
||||||
|
@ -389,28 +423,26 @@ bool EmitContext::DefineInputs(Bindings& bindings) {
|
||||||
const auto texture_type{desc.is_depth ? DepthSamplerType(desc.type)
|
const auto texture_type{desc.is_depth ? DepthSamplerType(desc.type)
|
||||||
: ColorSamplerType(desc.type, desc.is_multisample)};
|
: ColorSamplerType(desc.type, desc.is_multisample)};
|
||||||
const auto array_decorator{desc.count > 1 ? fmt::format("[{}]", desc.count) : ""};
|
const auto array_decorator{desc.count > 1 ? fmt::format("[{}]", desc.count) : ""};
|
||||||
if (added)
|
if (has_at_least_one_input)
|
||||||
input_str += ",";
|
input_str += ",";
|
||||||
input_str += fmt::format("{} {}_tex{}{} [[texture({})]]", texture_type, stage_name,
|
input_str += fmt::format("{}<float> tex{}{} [[texture({})]]", texture_type,
|
||||||
bindings.texture, array_decorator, bindings.texture);
|
|
||||||
input_str += fmt::format(",sampler {}_samp{}{} [[sampler({})]]", stage_name,
|
|
||||||
bindings.texture, array_decorator, bindings.texture);
|
bindings.texture, array_decorator, bindings.texture);
|
||||||
|
input_str += fmt::format(",sampler samp{}{} [[sampler({})]]", bindings.texture,
|
||||||
|
array_decorator, bindings.texture);
|
||||||
bindings.texture += desc.count;
|
bindings.texture += desc.count;
|
||||||
added = true;
|
has_at_least_one_input = true;
|
||||||
}
|
}
|
||||||
|
|
||||||
return added;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// TODO
|
// TODO
|
||||||
void EmitContext::DefineGenericOutput(size_t index, u32 invocations) {
|
void EmitContext::DefineStageInOut(size_t index, u32 invocations, bool is_input) {
|
||||||
const auto type{fmt::format("float{}", 4)};
|
const auto type{fmt::format("float{}", 4)};
|
||||||
std::string name{fmt::format("attr{}", index)};
|
std::string name{fmt::format("attr{}", index)};
|
||||||
header += fmt::format("{} {}{} [[user(locn{})]];\n", type, name,
|
header += fmt::format("{} {}{} [[user(locn{})]];\n", type, name,
|
||||||
OutputDecorator(stage, invocations), index);
|
OutputDecorator(stage, invocations), index);
|
||||||
|
|
||||||
const GenericElementInfo element_info{
|
const GenericElementInfo element_info{
|
||||||
.name = "__out." + name,
|
.name = (is_input ? "__in." : "__out.") + name,
|
||||||
.first_element = 0,
|
.first_element = 0,
|
||||||
.num_components = 4,
|
.num_components = 4,
|
||||||
};
|
};
|
||||||
|
@ -418,6 +450,9 @@ void EmitContext::DefineGenericOutput(size_t index, u32 invocations) {
|
||||||
}
|
}
|
||||||
|
|
||||||
void EmitContext::DefineHelperFunctions() {
|
void EmitContext::DefineHelperFunctions() {
|
||||||
|
header +=
|
||||||
|
"uint bitfieldExtract(uint value, int offset, int bits) {\nreturn (value >> offset) & "
|
||||||
|
"((1 << bits) - 1);\n}\n";
|
||||||
if (info.uses_global_increment || info.uses_shared_increment) {
|
if (info.uses_global_increment || info.uses_shared_increment) {
|
||||||
header += "uint CasIncrement(uint op_a,uint op_b){return op_a>=op_b?0u:(op_a+1u);}";
|
header += "uint CasIncrement(uint op_a,uint op_b){return op_a>=op_b?0u:(op_a+1u);}";
|
||||||
}
|
}
|
||||||
|
@ -497,7 +532,7 @@ std::string EmitContext::DefineGlobalMemoryFunctions() {
|
||||||
const auto& ssbo{info.storage_buffers_descriptors[index]};
|
const auto& ssbo{info.storage_buffers_descriptors[index]};
|
||||||
const u32 size_cbuf_offset{ssbo.cbuf_offset + 8};
|
const u32 size_cbuf_offset{ssbo.cbuf_offset + 8};
|
||||||
const auto ssbo_addr{fmt::format("ssbo_addr{}", index)};
|
const auto ssbo_addr{fmt::format("ssbo_addr{}", index)};
|
||||||
const auto cbuf{fmt::format("{}_cbuf{}", stage_name, ssbo.cbuf_index)};
|
const auto cbuf{fmt::format("{}_cbuf{}.data", stage_name, ssbo.cbuf_index)};
|
||||||
std::array<std::string, 2> addr_xy;
|
std::array<std::string, 2> addr_xy;
|
||||||
std::array<std::string, 2> size_xy;
|
std::array<std::string, 2> size_xy;
|
||||||
for (size_t i = 0; i < addr_xy.size(); ++i) {
|
for (size_t i = 0; i < addr_xy.size(); ++i) {
|
||||||
|
|
|
@ -145,6 +145,7 @@ public:
|
||||||
|
|
||||||
Stage stage{};
|
Stage stage{};
|
||||||
std::string_view stage_name = "invalid";
|
std::string_view stage_name = "invalid";
|
||||||
|
bool has_at_least_one_input{};
|
||||||
|
|
||||||
std::vector<TextureImageDefinition> texture_buffers;
|
std::vector<TextureImageDefinition> texture_buffers;
|
||||||
std::vector<TextureImageDefinition> image_buffers;
|
std::vector<TextureImageDefinition> image_buffers;
|
||||||
|
@ -160,8 +161,8 @@ public:
|
||||||
|
|
||||||
private:
|
private:
|
||||||
// TODO: break down into smaller functions
|
// TODO: break down into smaller functions
|
||||||
bool DefineInputs(Bindings& bindings);
|
void DefineInputs(Bindings& bindings);
|
||||||
void DefineGenericOutput(size_t index, u32 invocations);
|
void DefineStageInOut(size_t index, u32 invocations, bool is_input);
|
||||||
void DefineHelperFunctions();
|
void DefineHelperFunctions();
|
||||||
void DefineConstants();
|
void DefineConstants();
|
||||||
std::string DefineGlobalMemoryFunctions();
|
std::string DefineGlobalMemoryFunctions();
|
||||||
|
|
|
@ -139,9 +139,9 @@ void GraphicsPipeline::MakePipeline(MTL::RenderPassDescriptor* render_pass) {
|
||||||
MTL::RenderPipelineDescriptor* pipeline_descriptor =
|
MTL::RenderPipelineDescriptor* pipeline_descriptor =
|
||||||
MTL::RenderPipelineDescriptor::alloc()->init();
|
MTL::RenderPipelineDescriptor::alloc()->init();
|
||||||
pipeline_descriptor->setVertexFunction(functions[0]);
|
pipeline_descriptor->setVertexFunction(functions[0]);
|
||||||
pipeline_descriptor->setFragmentFunction(functions[1]);
|
pipeline_descriptor->setFragmentFunction(functions[4]);
|
||||||
// pipeline_descriptor->setVertexDescriptor(vertex_descriptor);
|
// pipeline_descriptor->setVertexDescriptor(vertex_descriptor);
|
||||||
// TODO: get the attachment count from render pass descriptor
|
// TODO: get the attachment count from render pass descriptor
|
||||||
for (u32 index = 0; index < NUM_RT; index++) {
|
for (u32 index = 0; index < NUM_RT; index++) {
|
||||||
auto* render_pass_attachment = render_pass->colorAttachments()->object(index);
|
auto* render_pass_attachment = render_pass->colorAttachments()->object(index);
|
||||||
// TODO: is this the correct way to check if the attachment is valid?
|
// TODO: is this the correct way to check if the attachment is valid?
|
||||||
|
|
|
@ -268,6 +268,7 @@ std::unique_ptr<GraphicsPipeline> PipelineCache::CreateGraphicsPipeline(
|
||||||
ConvertLegacyToGeneric(program, runtime_info);
|
ConvertLegacyToGeneric(program, runtime_info);
|
||||||
const std::string code{EmitMSL(profile, runtime_info, program, binding)};
|
const std::string code{EmitMSL(profile, runtime_info, program, binding)};
|
||||||
// HACK
|
// HACK
|
||||||
|
std::cout << "SHADER INDEX: " << index - 1 << std::endl;
|
||||||
std::cout << code << std::endl;
|
std::cout << code << std::endl;
|
||||||
MTL::CompileOptions* compile_options = MTL::CompileOptions::alloc()->init();
|
MTL::CompileOptions* compile_options = MTL::CompileOptions::alloc()->init();
|
||||||
NS::Error* error = nullptr;
|
NS::Error* error = nullptr;
|
||||||
|
@ -276,12 +277,13 @@ std::unique_ptr<GraphicsPipeline> PipelineCache::CreateGraphicsPipeline(
|
||||||
if (error) {
|
if (error) {
|
||||||
LOG_ERROR(Render_Metal, "failed to create library: {}",
|
LOG_ERROR(Render_Metal, "failed to create library: {}",
|
||||||
error->description()->cString(NS::ASCIIStringEncoding));
|
error->description()->cString(NS::ASCIIStringEncoding));
|
||||||
// std::cout << error->description()->cString(NS::ASCIIStringEncoding) << std::endl;
|
// HACK
|
||||||
|
std::cout << error->description()->cString(NS::ASCIIStringEncoding) << std::endl;
|
||||||
// HACK
|
// HACK
|
||||||
throw;
|
throw;
|
||||||
}
|
}
|
||||||
|
|
||||||
functions[index] =
|
functions[index - 1] =
|
||||||
library->newFunction(NS::String::string("main_", NS::ASCIIStringEncoding));
|
library->newFunction(NS::String::string("main_", NS::ASCIIStringEncoding));
|
||||||
previous_stage = &program;
|
previous_stage = &program;
|
||||||
}
|
}
|
||||||
|
@ -326,7 +328,7 @@ std::unique_ptr<GraphicsPipeline> PipelineCache::CreateGraphicsPipeline(
|
||||||
}
|
}
|
||||||
|
|
||||||
functions[0] = library->newFunction(NS::String::string("vertexMain", NS::ASCIIStringEncoding));
|
functions[0] = library->newFunction(NS::String::string("vertexMain", NS::ASCIIStringEncoding));
|
||||||
functions[1] =
|
functions[4] =
|
||||||
library->newFunction(NS::String::string("fragmentMain", NS::ASCIIStringEncoding));
|
library->newFunction(NS::String::string("fragmentMain", NS::ASCIIStringEncoding));
|
||||||
|
|
||||||
return std::make_unique<GraphicsPipeline>(device, command_recorder, key, buffer_cache,
|
return std::make_unique<GraphicsPipeline>(device, command_recorder, key, buffer_cache,
|
||||||
|
|
Loading…
Reference in a new issue