Merge pull request #3237 from ReinUsesLisp/vk-shader-decompiler

vk_shader_decompiler: Misc changes
This commit is contained in:
Fernando Sahmkow 2019-12-22 12:36:56 -04:00 committed by GitHub
commit 3dc585d011
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 52 additions and 41 deletions

2
externals/sirit vendored

@ -1 +1 @@
Subproject commit 12f40a80324d7c154f19f25c448a5ce27d38cd18 Subproject commit 9f4d057aa28c4e9509bdc767afb27b4aee303b7e

View File

@ -543,7 +543,7 @@ private:
} }
for (u32 rt = 0; rt < static_cast<u32>(frag_colors.size()); ++rt) { for (u32 rt = 0; rt < static_cast<u32>(frag_colors.size()); ++rt) {
if (!IsRenderTargetUsed(rt)) { if (!specialization.enabled_rendertargets[rt]) {
continue; continue;
} }
@ -1555,26 +1555,11 @@ private:
Expression Texture(Operation operation) { Expression Texture(Operation operation) {
const auto& meta = std::get<MetaTexture>(operation.GetMeta()); const auto& meta = std::get<MetaTexture>(operation.GetMeta());
UNIMPLEMENTED_IF(!meta.aoffi.empty());
const bool can_implicit = stage == ShaderType::Fragment; const bool can_implicit = stage == ShaderType::Fragment;
const Id sampler = GetTextureSampler(operation); const Id sampler = GetTextureSampler(operation);
const Id coords = GetCoordinates(operation, Type::Float); const Id coords = GetCoordinates(operation, Type::Float);
if (meta.depth_compare) {
// Depth sampling
UNIMPLEMENTED_IF(meta.bias);
const Id dref = AsFloat(Visit(meta.depth_compare));
if (can_implicit) {
return {OpImageSampleDrefImplicitLod(t_float, sampler, coords, dref, {}),
Type::Float};
} else {
return {OpImageSampleDrefExplicitLod(t_float, sampler, coords, dref,
spv::ImageOperandsMask::Lod, v_float_zero),
Type::Float};
}
}
std::vector<Id> operands; std::vector<Id> operands;
spv::ImageOperandsMask mask{}; spv::ImageOperandsMask mask{};
if (meta.bias) { if (meta.bias) {
@ -1582,13 +1567,36 @@ private:
operands.push_back(AsFloat(Visit(meta.bias))); operands.push_back(AsFloat(Visit(meta.bias)));
} }
if (!can_implicit) {
mask = mask | spv::ImageOperandsMask::Lod;
operands.push_back(v_float_zero);
}
if (!meta.aoffi.empty()) {
mask = mask | spv::ImageOperandsMask::Offset;
operands.push_back(GetOffsetCoordinates(operation));
}
if (meta.depth_compare) {
// Depth sampling
UNIMPLEMENTED_IF(meta.bias);
const Id dref = AsFloat(Visit(meta.depth_compare));
if (can_implicit) {
return {
OpImageSampleDrefImplicitLod(t_float, sampler, coords, dref, mask, operands),
Type::Float};
} else {
return {
OpImageSampleDrefExplicitLod(t_float, sampler, coords, dref, mask, operands),
Type::Float};
}
}
Id texture; Id texture;
if (can_implicit) { if (can_implicit) {
texture = OpImageSampleImplicitLod(t_float4, sampler, coords, mask, operands); texture = OpImageSampleImplicitLod(t_float4, sampler, coords, mask, operands);
} else { } else {
texture = OpImageSampleExplicitLod(t_float4, sampler, coords, texture = OpImageSampleExplicitLod(t_float4, sampler, coords, mask, operands);
mask | spv::ImageOperandsMask::Lod, v_float_zero,
operands);
} }
return GetTextureElement(operation, texture, Type::Float); return GetTextureElement(operation, texture, Type::Float);
} }
@ -1601,7 +1609,8 @@ private:
const Id lod = AsFloat(Visit(meta.lod)); const Id lod = AsFloat(Visit(meta.lod));
spv::ImageOperandsMask mask = spv::ImageOperandsMask::Lod; spv::ImageOperandsMask mask = spv::ImageOperandsMask::Lod;
std::vector<Id> operands; std::vector<Id> operands{lod};
if (!meta.aoffi.empty()) { if (!meta.aoffi.empty()) {
mask = mask | spv::ImageOperandsMask::Offset; mask = mask | spv::ImageOperandsMask::Offset;
operands.push_back(GetOffsetCoordinates(operation)); operands.push_back(GetOffsetCoordinates(operation));
@ -1609,11 +1618,10 @@ private:
if (meta.sampler.IsShadow()) { if (meta.sampler.IsShadow()) {
const Id dref = AsFloat(Visit(meta.depth_compare)); const Id dref = AsFloat(Visit(meta.depth_compare));
return { return {OpImageSampleDrefExplicitLod(t_float, sampler, coords, dref, mask, operands),
OpImageSampleDrefExplicitLod(t_float, sampler, coords, dref, mask, lod, operands),
Type::Float}; Type::Float};
} }
const Id texture = OpImageSampleExplicitLod(t_float4, sampler, coords, mask, lod, operands); const Id texture = OpImageSampleExplicitLod(t_float4, sampler, coords, mask, operands);
return GetTextureElement(operation, texture, Type::Float); return GetTextureElement(operation, texture, Type::Float);
} }
@ -1722,7 +1730,7 @@ private:
const std::vector grad = {dx, dy}; const std::vector grad = {dx, dy};
static constexpr auto mask = spv::ImageOperandsMask::Grad; static constexpr auto mask = spv::ImageOperandsMask::Grad;
const Id texture = OpImageSampleImplicitLod(t_float4, sampler, coords, mask, grad); const Id texture = OpImageSampleExplicitLod(t_float4, sampler, coords, mask, grad);
return GetTextureElement(operation, texture, Type::Float); return GetTextureElement(operation, texture, Type::Float);
} }
@ -1833,7 +1841,7 @@ private:
} }
void PreExit() { void PreExit() {
if (stage == ShaderType::Vertex) { if (stage == ShaderType::Vertex && specialization.ndc_minus_one_to_one) {
const u32 position_index = out_indices.position.value(); const u32 position_index = out_indices.position.value();
const Id z_pointer = AccessElement(t_out_float, out_vertex, position_index, 2U); const Id z_pointer = AccessElement(t_out_float, out_vertex, position_index, 2U);
const Id w_pointer = AccessElement(t_out_float, out_vertex, position_index, 3U); const Id w_pointer = AccessElement(t_out_float, out_vertex, position_index, 3U);
@ -1860,12 +1868,18 @@ private:
// rendertargets/components are skipped in the register assignment. // rendertargets/components are skipped in the register assignment.
u32 current_reg = 0; u32 current_reg = 0;
for (u32 rt = 0; rt < Maxwell::NumRenderTargets; ++rt) { for (u32 rt = 0; rt < Maxwell::NumRenderTargets; ++rt) {
if (!specialization.enabled_rendertargets[rt]) {
// Skip rendertargets that are not enabled
continue;
}
// TODO(Subv): Figure out how dual-source blending is configured in the Switch. // TODO(Subv): Figure out how dual-source blending is configured in the Switch.
for (u32 component = 0; component < 4; ++component) { for (u32 component = 0; component < 4; ++component) {
const Id pointer = AccessElement(t_out_float, frag_colors.at(rt), component);
if (header.ps.IsColorComponentOutputEnabled(rt, component)) { if (header.ps.IsColorComponentOutputEnabled(rt, component)) {
OpStore(AccessElement(t_out_float, frag_colors.at(rt), component), OpStore(pointer, SafeGetRegister(current_reg));
SafeGetRegister(current_reg));
++current_reg; ++current_reg;
} else {
OpStore(pointer, component == 3 ? v_float_one : v_float_zero);
} }
} }
} }
@ -1995,15 +2009,6 @@ private:
return DeclareBuiltIn(builtin, spv::StorageClass::Input, type, std::move(name)); return DeclareBuiltIn(builtin, spv::StorageClass::Input, type, std::move(name));
} }
bool IsRenderTargetUsed(u32 rt) const {
for (u32 component = 0; component < 4; ++component) {
if (header.ps.IsColorComponentOutputEnabled(rt, component)) {
return true;
}
}
return false;
}
template <typename... Args> template <typename... Args>
Id AccessElement(Id pointer_type, Id composite, Args... elements_) { Id AccessElement(Id pointer_type, Id composite, Args... elements_) {
std::vector<Id> members; std::vector<Id> members;
@ -2567,7 +2572,7 @@ public:
const Id target = decomp.Constant(decomp.t_uint, expr.value); const Id target = decomp.Constant(decomp.t_uint, expr.value);
Id gpr = decomp.OpLoad(decomp.t_float, decomp.registers.at(expr.gpr)); Id gpr = decomp.OpLoad(decomp.t_float, decomp.registers.at(expr.gpr));
gpr = decomp.OpBitcast(decomp.t_uint, gpr); gpr = decomp.OpBitcast(decomp.t_uint, gpr);
return decomp.OpLogicalEqual(decomp.t_uint, gpr, target); return decomp.OpIEqual(decomp.t_bool, gpr, target);
} }
Id Visit(const Expr& node) { Id Visit(const Expr& node) {
@ -2637,11 +2642,11 @@ public:
const Id loop_label = decomp.OpLabel(); const Id loop_label = decomp.OpLabel();
const Id endloop_label = decomp.OpLabel(); const Id endloop_label = decomp.OpLabel();
const Id loop_start_block = decomp.OpLabel(); const Id loop_start_block = decomp.OpLabel();
const Id loop_end_block = decomp.OpLabel(); const Id loop_continue_block = decomp.OpLabel();
current_loop_exit = endloop_label; current_loop_exit = endloop_label;
decomp.OpBranch(loop_label); decomp.OpBranch(loop_label);
decomp.AddLabel(loop_label); decomp.AddLabel(loop_label);
decomp.OpLoopMerge(endloop_label, loop_end_block, spv::LoopControlMask::MaskNone); decomp.OpLoopMerge(endloop_label, loop_continue_block, spv::LoopControlMask::MaskNone);
decomp.OpBranch(loop_start_block); decomp.OpBranch(loop_start_block);
decomp.AddLabel(loop_start_block); decomp.AddLabel(loop_start_block);
ASTNode current = ast.nodes.GetFirst(); ASTNode current = ast.nodes.GetFirst();
@ -2649,6 +2654,8 @@ public:
Visit(current); Visit(current);
current = current->GetNext(); current = current->GetNext();
} }
decomp.OpBranch(loop_continue_block);
decomp.AddLabel(loop_continue_block);
ExprDecompiler expr_parser{decomp}; ExprDecompiler expr_parser{decomp};
const Id condition = expr_parser.Visit(ast.condition); const Id condition = expr_parser.Visit(ast.condition);
decomp.OpBranchConditional(condition, loop_label, endloop_label); decomp.OpBranchConditional(condition, loop_label, endloop_label);

View File

@ -94,6 +94,7 @@ struct Specialization final {
Maxwell::PrimitiveTopology primitive_topology{}; Maxwell::PrimitiveTopology primitive_topology{};
std::optional<float> point_size{}; std::optional<float> point_size{};
std::array<Maxwell::VertexAttribute::Type, Maxwell::NumVertexAttributes> attribute_types{}; std::array<Maxwell::VertexAttribute::Type, Maxwell::NumVertexAttributes> attribute_types{};
bool ndc_minus_one_to_one{};
// Tessellation specific // Tessellation specific
struct { struct {
@ -101,6 +102,9 @@ struct Specialization final {
Maxwell::TessellationSpacing spacing{}; Maxwell::TessellationSpacing spacing{};
bool clockwise{}; bool clockwise{};
} tessellation; } tessellation;
// Fragment specific
std::bitset<8> enabled_rendertargets;
}; };
// Old gcc versions don't consider this trivially copyable. // Old gcc versions don't consider this trivially copyable.
// static_assert(std::is_trivially_copyable_v<Specialization>); // static_assert(std::is_trivially_copyable_v<Specialization>);