Merge pull request #3237 from ReinUsesLisp/vk-shader-decompiler
vk_shader_decompiler: Misc changes
This commit is contained in:
commit
3dc585d011
3 changed files with 52 additions and 41 deletions
2
externals/sirit
vendored
2
externals/sirit
vendored
|
@ -1 +1 @@
|
||||||
Subproject commit 12f40a80324d7c154f19f25c448a5ce27d38cd18
|
Subproject commit 9f4d057aa28c4e9509bdc767afb27b4aee303b7e
|
|
@ -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);
|
||||||
|
|
|
@ -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>);
|
||||||
|
|
Loading…
Reference in a new issue