#include "common/assert.h"
#include "graphics/shader/recompiler/frontend/translate/Translator.h"

namespace Libs::Graphics::ShaderRecompiler::Frontend {
namespace {

bool IsExecOrVcc(const Decoder::Operand& operand) {
	switch (operand.kind) {
		case Decoder::OperandKind::ExecLo:
		case Decoder::OperandKind::ExecHi:
		case Decoder::OperandKind::VccLo:
		case Decoder::OperandKind::VccHi: return true;
		default: return false;
	}
}

Decoder::Operand ConditionOperand(Decoder::OperandKind kind) {
	Decoder::Operand operand;
	operand.kind = kind;
	return operand;
}

} // namespace

void Translator::S_SAVEEXEC(const Decoder::Instruction& inst, IR::ValueOpcode operation,
                            bool negate_exec, bool negate_source, bool write_64) {
	if (write_64) {
		const auto exec_operand  = ConditionOperand(Decoder::OperandKind::ExecLo);
		const auto old           = ReadScalarU64(exec_operand);
		const auto bit_operation = operation == IR::ValueOpcode::LogicalAnd
		                               ? IR::ValueOpcode::BitwiseAnd32
		                               : IR::ValueOpcode::BitwiseOr32;
		const auto result = BinaryScalarU64(old, ReadScalarU64(inst.src0), operation, bit_operation,
		                                    negate_exec, negate_source, false);
		WriteScalarU64(inst.dst, old);
		WriteScalarU64(exec_operand, result);
		ir.SetScc(ScalarU64NonZero(result));
		return;
	}
	const auto old    = ir.GetExec();
	const auto src    = ReadMask(inst.src0);
	const auto lhs    = negate_exec ? ir.LogicalNot(old) : old;
	const auto rhs    = negate_source ? ir.LogicalNot(src) : src;
	const auto result = IR::U1(ir.Emit(operation, {lhs, rhs}));
	WriteMask(inst.dst, old);
	const auto mask = BallotMask(result);
	ir.SetExec(result);
	ir.SetExecMaskTag(IR::U1(IR::Value(true)));
	ir.SetExecLo(mask[0]);
	ir.SetExecHi(mask[1]);
	ir.SetScc(result);
}

void Translator::ADD_U32(const Decoder::Instruction& inst, bool vector, bool use_carry_in) {
	const auto zero     = IR::U32(IR::Value(0u));
	const auto lhs      = ReadU32(inst.src0);
	const auto rhs      = ReadU32(inst.src1);
	const auto carry_in = !use_carry_in ? zero
	                      : vector      ? ConditionBit(inst.src2)
	                                    : ConditionBit(ConditionOperand(Decoder::OperandKind::Scc));
	const auto add0     = ir.Emit(IR::ValueOpcode::IAddCarry32, {lhs, rhs});
	const auto partial  = ir.CompositeExtract(add0, 0);
	const auto carry0   = ir.CompositeExtract(add0, 1);
	const auto add1     = ir.Emit(IR::ValueOpcode::IAddCarry32, {partial, carry_in});
	const auto result   = ir.CompositeExtract(add1, 0);
	const auto carry1   = ir.CompositeExtract(add1, 1);
	const auto carry    = ir.INotEqual(ir.BitwiseOr(carry0, carry1), zero);
	WriteOperand(DestinationOperand(inst), result);
	if (vector) {
		WriteMask(inst.dst2, ir.LogicalAnd(ir.GetExec(), carry));
	} else {
		ir.SetScc(carry);
	}
}

void Translator::SUB_U32(const Decoder::Instruction& inst, bool vector, bool reverse) {
	const auto lhs    = ReadU32(reverse ? inst.src1 : inst.src0);
	const auto rhs    = ReadU32(reverse ? inst.src0 : inst.src1);
	const auto result = ir.ISub(lhs, rhs);
	const auto borrow = ir.UGreaterThan(rhs, lhs);
	WriteOperand(DestinationOperand(inst), result);
	if (vector) {
		WriteMask(inst.dst2, ir.LogicalAnd(ir.GetExec(), borrow));
	} else {
		ir.SetScc(borrow);
	}
}

void Translator::SUBB_U32(const Decoder::Instruction& inst, bool vector) {
	const auto lhs       = ReadU32(vector ? inst.src1 : inst.src0);
	const auto rhs       = ReadU32(vector ? inst.src0 : inst.src1);
	const auto borrow_in = vector ? ConditionBit(inst.src2)
	                              : ConditionBit(ConditionOperand(Decoder::OperandKind::Scc));
	const auto partial   = ir.ISub(lhs, rhs);
	const auto result    = ir.ISub(partial, borrow_in);
	const auto borrow0   = ir.UGreaterThan(rhs, lhs);
	const auto borrow1   = ir.UGreaterThan(borrow_in, partial);
	const auto borrow    = ir.LogicalOr(borrow0, borrow1);
	WriteOperand(DestinationOperand(inst), result);
	if (vector) {
		WriteMask(inst.dst2, ir.LogicalAnd(ir.GetExec(), borrow));
	} else {
		ir.SetScc(borrow);
	}
}

void Translator::S_ADD_SUB_I32(const Decoder::Instruction& inst, bool subtract) {
	const auto lhs      = ReadU32(inst.src0);
	const auto rhs      = ReadU32(inst.src1);
	const auto result   = subtract ? ir.ISub(lhs, rhs) : ir.IAdd(lhs, rhs);
	const auto shift    = IR::U32(IR::Value(31u));
	const auto lhs_sign = ir.ShiftRightLogical(lhs, shift);
	const auto rhs_sign = ir.ShiftRightLogical(rhs, shift);
	const auto out_sign = ir.ShiftRightLogical(result, shift);
	const auto inputs = subtract ? ir.INotEqual(lhs_sign, rhs_sign) : ir.IEqual(lhs_sign, rhs_sign);
	const auto changed = ir.INotEqual(lhs_sign, out_sign);
	WriteOperand(DestinationOperand(inst), result);
	ir.SetScc(ir.LogicalAnd(inputs, changed));
}

void Translator::S_LSHL_ADD_U32(const Decoder::Instruction& inst, uint32_t shift_amount) {
	const auto lhs           = ReadU32(inst.src0);
	const auto shift         = IR::U32(IR::Value(shift_amount));
	const auto rhs           = ReadU32(inst.src1);
	const auto shifted       = ir.ShiftLeftLogical(lhs, shift);
	const auto result        = ir.IAdd(shifted, rhs);
	const auto add_carry     = ir.ULessThan(result, shifted);
	const auto inverse_shift = ir.ISub(IR::U32(IR::Value(32u)), shift);
	const auto shifted_out   = ir.ShiftRightLogical(lhs, inverse_shift);
	const auto shift_carry   = ir.INotEqual(shifted_out, IR::U32(IR::Value(0u)));
	WriteOperand(DestinationOperand(inst), result);
	ir.SetScc(ir.LogicalOr(add_carry, shift_carry));
}

void Translator::ScalarMinMax32(const Decoder::Instruction& inst, IR::ValueOpcode value_opcode,
                                IR::ValueOpcode compare_opcode) {
	const auto lhs = ReadU32(inst.src0);
	const auto rhs = ReadU32(inst.src1);
	WriteOperand(DestinationOperand(inst), ir.Emit(value_opcode, {lhs, rhs}));
	ir.SetScc(IR::U1(ir.Emit(compare_opcode, {lhs, rhs})));
}

void Translator::EmitControlNop() {
	ir.Emit(IR::ValueOpcode::ControlNop);
}

void Translator::EmitWaitcnt() {
	ir.Emit(IR::ValueOpcode::Waitcnt);
}

void Translator::S_BARRIER() {
	ir.Emit(IR::ValueOpcode::Barrier);
}

void Translator::S_SENDMSG() {
	ir.Emit(IR::ValueOpcode::Sendmsg);
}

void Translator::S_TTRACEDATA() {
	ir.Emit(IR::ValueOpcode::TtraceData);
}

void Translator::S_INST_PREFETCH() {
	ir.Emit(IR::ValueOpcode::InstPrefetch);
}

void Translator::S_GETPC_B64(const Decoder::Instruction& inst) {
	const auto base    = IR::U64(ir.Emit(IR::ValueOpcode::GetShaderBase));
	const auto address = IR::U64(
	    ir.Emit(IR::ValueOpcode::IAdd64, {base, IR::Value(static_cast<uint64_t>(inst.pc) + 4u)}));
	if (inst.dst.kind == Decoder::OperandKind::Null) {
		return;
	}
	auto high = PlainOperand(inst.dst);
	switch (inst.dst.kind) {
		case Decoder::OperandKind::Sgpr:
			if (inst.dst.reg < 105u) {
				high.reg++;
			} else if (inst.dst.reg == 105u) {
				high.kind = Decoder::OperandKind::VccLo;
				high.reg  = 0u;
			} else {
				EXIT("S_GETPC_B64 destination does not name a valid scalar pair at pc 0x%08x",
				     inst.pc);
			}
			break;
		case Decoder::OperandKind::VccLo: high.kind = Decoder::OperandKind::VccHi; break;
		case Decoder::OperandKind::M0: high.kind = Decoder::OperandKind::Null; break;
		case Decoder::OperandKind::ExecLo: high.kind = Decoder::OperandKind::ExecHi; break;
		default:
			EXIT("S_GETPC_B64 destination does not name a valid scalar pair at pc 0x%08x", inst.pc);
	}
	const auto words = ExtractU64(address);
	WriteOperand(inst.dst, words[0]);
	WriteOperand(high, words[1]);
}

void Translator::S_CSELECT_B32(const Decoder::Instruction& inst) {
	const auto result = ir.Select(ir.GetScc(), ReadU32(inst.src0), ReadU32(inst.src1));
	WriteOperand(DestinationOperand(inst), result);
}

void Translator::S_CSELECT_B64(const Decoder::Instruction& inst) {
	const auto condition = ir.GetScc();
	const auto lhs       = ReadScalarU64(inst.src0);
	const auto rhs       = ReadScalarU64(inst.src1);
	WriteScalarU64(inst.dst, SelectScalarU64(condition, lhs, rhs));
}

void Translator::MOV_B32(const Decoder::Instruction& inst, bool apply_float_modifiers) {
	if (inst.src0.kind == Decoder::OperandKind::Sgpr &&
	    inst.dst.kind == Decoder::OperandKind::Sgpr && inst.src0.reg == inst.dst.reg) {
		return;
	}
	if (IsExecOrVcc(inst.src0) && IsExecOrVcc(inst.dst)) {
		if (inst.src0.kind != inst.dst.kind) {
			WriteRawU32(inst.dst, ReadRawU32(inst.src0));
		}
	} else if (apply_float_modifiers && (inst.src0.negate || inst.src0.absolute)) {
		WriteOperand(DestinationOperand(inst), ReadOperand(inst.src0, IR::Type::F32));
	} else {
		WriteOperand(DestinationOperand(inst), ReadOperand(inst.src0, IR::Type::U32));
	}
}

void Translator::S_MOV_B64(const Decoder::Instruction& inst) {
	WriteScalarU64(inst.dst, ReadScalarU64(inst.src0));
}

void Translator::S_WQM_B64(const Decoder::Instruction& inst) {
	const auto source = ReadScalarU64(inst.src0);
	const auto raw    = ExtractU64(
	    IR::U64(ir.Emit(IR::ValueOpcode::WqmU64, {ir.ConstructU64(source.raw[0], source.raw[1])})));
	const auto invocation = IR::U1(ir.Emit(IR::ValueOpcode::WqmMask, {source.invocation}));
	const auto result     = MakeScalarU64Result(raw, invocation, source.mask_valid);
	WriteScalarU64(inst.dst, result);
	ir.SetScc(ScalarU64NonZero(result));
}

void Translator::V_MOVRELS_B32(const Decoder::Instruction& inst) {
	if (inst.dst.kind != Decoder::OperandKind::Vgpr ||
	    inst.src0.kind != Decoder::OperandKind::Vgpr) {
		EXIT("V_MOVRELS_B32 requires VGPR source and destination at pc 0x%08x", inst.pc);
	}
	if (inst.dst.sdwa_sel != 6u || inst.dst.omod != 0u || inst.dst.clamp ||
	    inst.src0.sdwa_sel != 6u || inst.src0.sdwa_sext || inst.src0.negate || inst.src0.absolute ||
	    inst.src0.dpp) {
		EXIT("V_MOVRELS_B32 modifiers are not implemented at pc 0x%08x", inst.pc);
	}
	const auto base     = inst.src0.reg;
	const auto m0       = ir.BitwiseAnd(ReadU32(ConditionOperand(Decoder::OperandKind::M0)),
	                                    IR::U32(IR::Value(0xffu)));
	auto       selected = ir.GetVectorReg(static_cast<IR::VectorReg>(base));
	for (uint32_t index = base + 1u; index < current_vector_limit; index++) {
		const auto match = ir.IEqual(m0, IR::U32(IR::Value(index - base)));
		selected = ir.Select(match, ir.GetVectorReg(static_cast<IR::VectorReg>(index)), selected);
	}
	WriteOperand(DestinationOperand(inst), selected);
}

void Translator::V_MOVRELD_B32(const Decoder::Instruction& inst) {
	if (inst.dst.kind != Decoder::OperandKind::Vgpr) {
		EXIT("V_MOVRELD_B32 requires VGPR destination at pc 0x%08x", inst.pc);
	}
	if (inst.dst.sdwa_sel != 6u || inst.dst.omod != 0u || inst.dst.clamp ||
	    inst.src0.sdwa_sel != 6u || inst.src0.sdwa_sext || inst.src0.negate || inst.src0.absolute ||
	    inst.src0.dpp) {
		EXIT("V_MOVRELD_B32 modifiers are not implemented at pc 0x%08x", inst.pc);
	}
	const auto base  = inst.dst.reg;
	const auto value = ReadU32(inst.src0);
	const auto m0    = ir.BitwiseAnd(ReadU32(ConditionOperand(Decoder::OperandKind::M0)),
	                                 IR::U32(IR::Value(0xffu)));
	for (uint32_t index = base; index < current_vector_limit; index++) {
		const auto reg   = static_cast<IR::VectorReg>(index);
		const auto match = ir.IEqual(m0, IR::U32(IR::Value(index - base)));
		const auto write = ir.LogicalAnd(ir.GetExec(), match);
		ir.SetVectorReg(reg, ir.Select(write, value, ir.GetVectorReg(reg)));
	}
}

void Translator::V_READFIRSTLANE_B32(const Decoder::Instruction& inst) {
	const auto result = ir.Emit(IR::ValueOpcode::ReadFirstLane, {ReadU32(inst.src0), ir.GetExec()});
	WriteOperand(DestinationOperand(inst), result);
}

void Translator::V_READLANE_B32(const Decoder::Instruction& inst) {
	const auto lane_mask = IR::U32(IR::Value(current_wave_size == 32u ? 31u : 63u));
	const auto lane      = ir.BitwiseAnd(ReadU32(inst.src1), lane_mask);
	WriteOperand(DestinationOperand(inst),
	             ir.Emit(IR::ValueOpcode::ReadLane, {ReadU32(inst.src0), lane}));
}

void Translator::V_WRITELANE_B32(const Decoder::Instruction& inst) {
	EXIT_IF(inst.dst.kind != Decoder::OperandKind::Vgpr);
	const auto lane_mask = IR::U32(IR::Value(current_wave_size == 32u ? 31u : 63u));
	const auto reg       = static_cast<IR::VectorReg>(inst.dst.reg);
	const auto lane      = ir.BitwiseAnd(ReadU32(inst.src1), lane_mask);
	const auto result    = IR::U32(
	    ir.Emit(IR::ValueOpcode::WriteLane, {ir.GetVectorReg(reg), ReadU32(inst.src0), lane}));
	ir.SetVectorReg(reg, result);
}

void Translator::V_PERMLANE16_B32(const Decoder::Instruction& inst, bool x16) {
	const IR::PermlaneFlags flags {
	    .x16            = x16,
	    .fetch_inactive = inst.dst.op_sel,
	    .bound_control  = inst.dst.op_sel_hi,
	};
	const auto result =
	    ir.Emit(IR::ValueOpcode::Permlane16U32,
	            {ReadU32(inst.src0), ReadU32(inst.src1), ReadU32(inst.src2), ir.GetExec()}, flags);
	auto dst      = DestinationOperand(inst);
	dst.op_sel    = false;
	dst.op_sel_hi = false;
	WriteOperand(dst, result);
}

} // namespace Libs::Graphics::ShaderRecompiler::Frontend