// Copyright 2018 yuzu Emulator Project
// Licensed under GPLv2 or any later version
// Refer to the license.txt file included.

#include "common/assert.h"
#include "common/common_types.h"
#include "video_core/engines/shader_bytecode.h"
#include "video_core/shader/node_helper.h"
#include "video_core/shader/shader_ir.h"

namespace VideoCommon::Shader {

using Tegra::Shader::Instruction;
using Tegra::Shader::OpCode;

namespace {
constexpr u64 NUM_PROGRAMMABLE_PREDICATES = 7;
}

u32 ShaderIR::DecodeRegisterSetPredicate(NodeBlock& bb, u32 pc) {
    const Instruction instr = {program_code[pc]};
    const auto opcode = OpCode::Decode(instr);

    UNIMPLEMENTED_IF(instr.p2r_r2p.mode != Tegra::Shader::R2pMode::Pr);

    const Node apply_mask = [&] {
        switch (opcode->get().GetId()) {
        case OpCode::Id::R2P_IMM:
        case OpCode::Id::P2R_IMM:
            return Immediate(static_cast<u32>(instr.p2r_r2p.immediate_mask));
        default:
            UNREACHABLE();
            return Immediate(0);
        }
    }();

    const auto offset = static_cast<u32>(instr.p2r_r2p.byte) * 8;

    switch (opcode->get().GetId()) {
    case OpCode::Id::R2P_IMM: {
        const Node mask = GetRegister(instr.gpr8);

        for (u64 pred = 0; pred < NUM_PROGRAMMABLE_PREDICATES; ++pred) {
            const auto shift = static_cast<u32>(pred);

            const Node apply_compare = BitfieldExtract(apply_mask, shift, 1);
            const Node condition =
                Operation(OperationCode::LogicalUNotEqual, apply_compare, Immediate(0));

            const Node value_compare = BitfieldExtract(mask, offset + shift, 1);
            const Node value =
                Operation(OperationCode::LogicalUNotEqual, value_compare, Immediate(0));

            const Node code = Operation(OperationCode::LogicalAssign, GetPredicate(pred), value);
            bb.push_back(Conditional(condition, {code}));
        }
        break;
    }
    case OpCode::Id::P2R_IMM: {
        Node value = Immediate(0);
        for (u64 pred = 0; pred < NUM_PROGRAMMABLE_PREDICATES; ++pred) {
            Node bit = Operation(OperationCode::Select, GetPredicate(pred), Immediate(1U << pred),
                                 Immediate(0));
            value = Operation(OperationCode::UBitwiseOr, std::move(value), std::move(bit));
        }
        value = Operation(OperationCode::UBitwiseAnd, std::move(value), apply_mask);
        value = BitfieldInsert(GetRegister(instr.gpr8), std::move(value), offset, 8);

        SetRegister(bb, instr.gpr0, std::move(value));
        break;
    }
    default:
        UNIMPLEMENTED_MSG("Unhandled P2R/R2R instruction: {}", opcode->get().GetName());
        break;
    }

    return pc;
}

} // namespace VideoCommon::Shader