forked from suyu/suyu
rescaling_pass: Fix and simplify shuffle/fragcoord pass
This commit is contained in:
parent
b3a9c8f108
commit
edb5844240
1 changed files with 20 additions and 26 deletions
|
@ -14,45 +14,39 @@
|
|||
|
||||
namespace Shader::Optimization {
|
||||
namespace {
|
||||
void VisitMark(const IR::Program& program, IR::Inst& inst) {
|
||||
void VisitMark(const IR::Program& program, const IR::Inst& inst) {
|
||||
const bool is_fragment_shader{program.stage == Stage::Fragment};
|
||||
if (!is_fragment_shader) {
|
||||
return;
|
||||
}
|
||||
switch (inst.GetOpcode()) {
|
||||
case IR::Opcode::ShuffleIndex:
|
||||
case IR::Opcode::ShuffleUp:
|
||||
case IR::Opcode::ShuffleDown:
|
||||
case IR::Opcode::ShuffleButterfly: {
|
||||
const auto try_mark = [is_fragment_shader](IR::Inst* op) {
|
||||
const IR::Attribute attr{op->Arg(0).Attribute()};
|
||||
const IR::Value shfl_arg{inst.Arg(0)};
|
||||
if (shfl_arg.IsImmediate()) {
|
||||
break;
|
||||
}
|
||||
const IR::Inst* const arg_inst{shfl_arg.InstRecursive()};
|
||||
if (arg_inst->GetOpcode() != IR::Opcode::BitCastU32F32) {
|
||||
break;
|
||||
}
|
||||
const IR::Value bitcast_arg{arg_inst->Arg(0)};
|
||||
if (bitcast_arg.IsImmediate()) {
|
||||
break;
|
||||
}
|
||||
IR::Inst* const bitcast_inst{bitcast_arg.InstRecursive()};
|
||||
if (bitcast_inst->GetOpcode() == IR::Opcode::GetAttribute) {
|
||||
const IR::Attribute attr{bitcast_inst->Arg(0).Attribute()};
|
||||
switch (attr) {
|
||||
case IR::Attribute::PositionX:
|
||||
case IR::Attribute::PositionY:
|
||||
if (is_fragment_shader) {
|
||||
op->SetFlags<u32>(0xDEADBEEF);
|
||||
}
|
||||
bitcast_inst->SetFlags<u32>(0xDEADBEEF);
|
||||
break;
|
||||
default:
|
||||
break;
|
||||
}
|
||||
};
|
||||
const IR::Value param_1{inst.Arg(0)};
|
||||
if (param_1.IsImmediate()) {
|
||||
break;
|
||||
}
|
||||
IR::Inst* op_a{param_1.InstRecursive()};
|
||||
if (op_a->GetOpcode() == IR::Opcode::GetAttribute) {
|
||||
try_mark(op_a);
|
||||
break;
|
||||
}
|
||||
if (op_a->GetOpcode() != IR::Opcode::BitCastF32U32) {
|
||||
break;
|
||||
}
|
||||
const IR::Value param_2{op_a->Arg(0)};
|
||||
if (param_2.IsImmediate()) {
|
||||
break;
|
||||
}
|
||||
IR::Inst* op_b{param_2.InstRecursive()};
|
||||
if (op_b->GetOpcode() == IR::Opcode::GetAttribute) {
|
||||
try_mark(op_b);
|
||||
}
|
||||
break;
|
||||
}
|
||||
|
|
Loading…
Reference in a new issue