rescaling_pass: Fix and simplify shuffle/fragcoord pass
This commit is contained in:
		@@ -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;
 | 
			
		||||
    }
 | 
			
		||||
 
 | 
			
		||||
		Reference in New Issue
	
	Block a user