package x86_test

import (
	"testing"

	"git.urbach.dev/cli/q/src/cpu"
	"git.urbach.dev/cli/q/src/x86"
	"git.urbach.dev/go/assert"
)

func TestCompareRegisterNumber(t *testing.T) {
	usagePatterns := []struct {
		Register cpu.Register
		Number   int
		Code     []byte
	}{
		{x86.R0, 1, []byte{0x48, 0x83, 0xF8, 0x01}},
		{x86.R1, 1, []byte{0x48, 0x83, 0xF9, 0x01}},
		{x86.R2, 1, []byte{0x48, 0x83, 0xFA, 0x01}},
		{x86.R3, 1, []byte{0x48, 0x83, 0xFB, 0x01}},
		{x86.SP, 1, []byte{0x48, 0x83, 0xFC, 0x01}},
		{x86.R5, 1, []byte{0x48, 0x83, 0xFD, 0x01}},
		{x86.R6, 1, []byte{0x48, 0x83, 0xFE, 0x01}},
		{x86.R7, 1, []byte{0x48, 0x83, 0xFF, 0x01}},
		{x86.R8, 1, []byte{0x49, 0x83, 0xF8, 0x01}},
		{x86.R9, 1, []byte{0x49, 0x83, 0xF9, 0x01}},
		{x86.R10, 1, []byte{0x49, 0x83, 0xFA, 0x01}},
		{x86.R11, 1, []byte{0x49, 0x83, 0xFB, 0x01}},
		{x86.R12, 1, []byte{0x49, 0x83, 0xFC, 0x01}},
		{x86.R13, 1, []byte{0x49, 0x83, 0xFD, 0x01}},
		{x86.R14, 1, []byte{0x49, 0x83, 0xFE, 0x01}},
		{x86.R15, 1, []byte{0x49, 0x83, 0xFF, 0x01}},

		{x86.R0, 0x7FFFFFFF, []byte{0x48, 0x81, 0xF8, 0xFF, 0xFF, 0xFF, 0x7F}},
		{x86.R1, 0x7FFFFFFF, []byte{0x48, 0x81, 0xF9, 0xFF, 0xFF, 0xFF, 0x7F}},
		{x86.R2, 0x7FFFFFFF, []byte{0x48, 0x81, 0xFA, 0xFF, 0xFF, 0xFF, 0x7F}},
		{x86.R3, 0x7FFFFFFF, []byte{0x48, 0x81, 0xFB, 0xFF, 0xFF, 0xFF, 0x7F}},
		{x86.SP, 0x7FFFFFFF, []byte{0x48, 0x81, 0xFC, 0xFF, 0xFF, 0xFF, 0x7F}},
		{x86.R5, 0x7FFFFFFF, []byte{0x48, 0x81, 0xFD, 0xFF, 0xFF, 0xFF, 0x7F}},
		{x86.R6, 0x7FFFFFFF, []byte{0x48, 0x81, 0xFE, 0xFF, 0xFF, 0xFF, 0x7F}},
		{x86.R7, 0x7FFFFFFF, []byte{0x48, 0x81, 0xFF, 0xFF, 0xFF, 0xFF, 0x7F}},
		{x86.R8, 0x7FFFFFFF, []byte{0x49, 0x81, 0xF8, 0xFF, 0xFF, 0xFF, 0x7F}},
		{x86.R9, 0x7FFFFFFF, []byte{0x49, 0x81, 0xF9, 0xFF, 0xFF, 0xFF, 0x7F}},
		{x86.R10, 0x7FFFFFFF, []byte{0x49, 0x81, 0xFA, 0xFF, 0xFF, 0xFF, 0x7F}},
		{x86.R11, 0x7FFFFFFF, []byte{0x49, 0x81, 0xFB, 0xFF, 0xFF, 0xFF, 0x7F}},
		{x86.R12, 0x7FFFFFFF, []byte{0x49, 0x81, 0xFC, 0xFF, 0xFF, 0xFF, 0x7F}},
		{x86.R13, 0x7FFFFFFF, []byte{0x49, 0x81, 0xFD, 0xFF, 0xFF, 0xFF, 0x7F}},
		{x86.R14, 0x7FFFFFFF, []byte{0x49, 0x81, 0xFE, 0xFF, 0xFF, 0xFF, 0x7F}},
		{x86.R15, 0x7FFFFFFF, []byte{0x49, 0x81, 0xFF, 0xFF, 0xFF, 0xFF, 0x7F}},
	}

	for _, pattern := range usagePatterns {
		t.Logf("cmp %s, %x", pattern.Register, pattern.Number)
		code := x86.CompareRegisterNumber(nil, pattern.Register, pattern.Number)
		assert.DeepEqual(t, code, pattern.Code)
	}
}

func TestCompareRegisterRegister(t *testing.T) {
	usagePatterns := []struct {
		Left   cpu.Register
		Right  cpu.Register
		Length byte
		Code   []byte
	}{
		// 8-bit
		{x86.R0, x86.R1, 1, []byte{0x38, 0xC8}},
		{x86.R2, x86.R3, 1, []byte{0x38, 0xDA}},
		{x86.R6, x86.R7, 1, []byte{0x40, 0x38, 0xFE}},
		{x86.SP, x86.R5, 1, []byte{0x40, 0x38, 0xEC}},
		{x86.R0, x86.SP, 1, []byte{0x40, 0x38, 0xE0}},
		{x86.R2, x86.SP, 1, []byte{0x40, 0x38, 0xE2}},
		{x86.R8, x86.R9, 1, []byte{0x45, 0x38, 0xC8}},
		{x86.R15, x86.R8, 1, []byte{0x45, 0x38, 0xC7}},
		{x86.R0, x86.R9, 1, []byte{0x44, 0x38, 0xC8}},

		// 16-bit
		{x86.R0, x86.R1, 2, []byte{0x66, 0x39, 0xC8}},
		{x86.R2, x86.R3, 2, []byte{0x66, 0x39, 0xDA}},
		{x86.R6, x86.R7, 2, []byte{0x66, 0x39, 0xFE}},
		{x86.SP, x86.R5, 2, []byte{0x66, 0x39, 0xEC}},

		// 32-bit
		{x86.R0, x86.R1, 4, []byte{0x39, 0xC8}},
		{x86.R2, x86.R3, 4, []byte{0x39, 0xDA}},
		{x86.R6, x86.R7, 4, []byte{0x39, 0xFE}},
		{x86.SP, x86.R5, 4, []byte{0x39, 0xEC}},

		// 64-bit
		{x86.R0, x86.R15, 8, []byte{0x4C, 0x39, 0xF8}},
		{x86.R1, x86.R14, 8, []byte{0x4C, 0x39, 0xF1}},
		{x86.R2, x86.R13, 8, []byte{0x4C, 0x39, 0xEA}},
		{x86.R3, x86.R12, 8, []byte{0x4C, 0x39, 0xE3}},
		{x86.SP, x86.R11, 8, []byte{0x4C, 0x39, 0xDC}},
		{x86.R5, x86.R10, 8, []byte{0x4C, 0x39, 0xD5}},
		{x86.R6, x86.R9, 8, []byte{0x4C, 0x39, 0xCE}},
		{x86.R7, x86.R8, 8, []byte{0x4C, 0x39, 0xC7}},
		{x86.R8, x86.R7, 8, []byte{0x49, 0x39, 0xF8}},
		{x86.R9, x86.R6, 8, []byte{0x49, 0x39, 0xF1}},
		{x86.R10, x86.R5, 8, []byte{0x49, 0x39, 0xEA}},
		{x86.R11, x86.SP, 8, []byte{0x49, 0x39, 0xE3}},
		{x86.R12, x86.R3, 8, []byte{0x49, 0x39, 0xDC}},
		{x86.R13, x86.R2, 8, []byte{0x49, 0x39, 0xD5}},
		{x86.R14, x86.R1, 8, []byte{0x49, 0x39, 0xCE}},
		{x86.R15, x86.R0, 8, []byte{0x49, 0x39, 0xC7}},
	}

	for _, pattern := range usagePatterns {
		t.Logf("cmp %db %s, %s", pattern.Length, pattern.Left, pattern.Right)
		code := x86.CompareRegisterRegister(nil, pattern.Left, pattern.Right, pattern.Length)
		assert.DeepEqual(t, code, pattern.Code)
	}
}