package arm_test

import (
	"testing"

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

func TestLoad(t *testing.T) {
	usagePatterns := []struct {
		Destination cpu.Register
		Base        cpu.Register
		Offset      cpu.Register
		Scale       arm.Scale
		Length      byte
		Code        uint32
	}{
		{arm.X0, arm.X1, arm.X2, arm.Scale1, 8, 0xF8626820},
		{arm.X0, arm.X1, arm.X2, arm.Scale1, 4, 0xB8626820},
		{arm.X0, arm.X1, arm.X2, arm.Scale1, 2, 0x78626820},
		{arm.X0, arm.X1, arm.X2, arm.Scale1, 1, 0x38626820},
		{arm.X0, arm.X1, arm.X2, arm.ScaleLength, 8, 0xF8627820},
	}

	for _, pattern := range usagePatterns {
		t.Logf("ldr %s, [%s, #%s] %db", pattern.Destination, pattern.Base, pattern.Offset, pattern.Length)
		code := arm.Load(pattern.Destination, pattern.Base, pattern.Offset, pattern.Scale, pattern.Length)
		assert.Equal(t, code, pattern.Code)
	}
}

func TestLoadSigned(t *testing.T) {
	usagePatterns := []struct {
		Destination cpu.Register
		Base        cpu.Register
		Offset      cpu.Register
		Scale       arm.Scale
		Length      byte
		Code        uint32
	}{
		{arm.X0, arm.X1, arm.X2, arm.Scale1, 4, 0xB8A26820},
		{arm.X0, arm.X1, arm.X2, arm.Scale1, 2, 0x78A26820},
		{arm.X0, arm.X1, arm.X2, arm.Scale1, 1, 0x38A26820},
	}

	for _, pattern := range usagePatterns {
		t.Logf("ldrs %s, [%s, #%s] %db", pattern.Destination, pattern.Base, pattern.Offset, pattern.Length)
		code := arm.LoadSigned(pattern.Destination, pattern.Base, pattern.Offset, pattern.Scale, pattern.Length)
		assert.Equal(t, code, pattern.Code)
	}
}

func TestLoadFixedOffset(t *testing.T) {
	usagePatterns := []struct {
		Destination cpu.Register
		Base        cpu.Register
		Mode        arm.AddressMode
		Offset      int
		Length      byte
		Code        uint32
	}{
		{arm.X0, arm.X1, arm.UnscaledImmediate, -8, 1, 0x385F8020},
		{arm.X1, arm.X0, arm.UnscaledImmediate, -8, 1, 0x385F8001},
		{arm.X0, arm.X1, arm.UnscaledImmediate, -8, 2, 0x785F8020},
		{arm.X1, arm.X0, arm.UnscaledImmediate, -8, 2, 0x785F8001},
		{arm.X0, arm.X1, arm.UnscaledImmediate, -8, 4, 0xB85F8020},
		{arm.X1, arm.X0, arm.UnscaledImmediate, -8, 4, 0xB85F8001},
		{arm.X0, arm.X1, arm.UnscaledImmediate, -8, 8, 0xF85F8020},
		{arm.X1, arm.X0, arm.UnscaledImmediate, -8, 8, 0xF85F8001},
		{arm.X2, arm.X1, arm.UnscaledImmediate, -8, 8, 0xF85F8022},
		{arm.X2, arm.X1, arm.UnscaledImmediate, 0, 8, 0xF8400022},
		{arm.X2, arm.X1, arm.UnscaledImmediate, 8, 8, 0xF8408022},
		{arm.X2, arm.X1, arm.UnscaledImmediate, -256, 8, 0xF8500022},
		{arm.X2, arm.X1, arm.UnscaledImmediate, 255, 8, 0xF84FF022},

		{arm.X0, arm.SP, arm.PostIndex, 16, 8, 0xF84107E0},
		{arm.X1, arm.SP, arm.PostIndex, 16, 8, 0xF84107E1},
		{arm.X2, arm.SP, arm.PostIndex, 16, 8, 0xF84107E2},
	}

	for _, pattern := range usagePatterns {
		t.Logf("ldur %s, [%s, %d] %db", pattern.Destination, pattern.Base, pattern.Offset, pattern.Length)
		code := arm.LoadFixedOffset(pattern.Destination, pattern.Base, pattern.Mode, pattern.Offset, pattern.Length)
		assert.Equal(t, code, pattern.Code)
	}
}

func TestLoadFixedOffsetSigned(t *testing.T) {
	usagePatterns := []struct {
		Destination cpu.Register
		Base        cpu.Register
		Mode        arm.AddressMode
		Offset      int
		Length      byte
		Code        uint32
	}{
		{arm.X0, arm.X1, arm.UnscaledImmediate, -8, 1, 0x389F8020},
		{arm.X0, arm.X1, arm.UnscaledImmediate, -8, 2, 0x789F8020},
		{arm.X0, arm.X1, arm.UnscaledImmediate, -8, 4, 0xB89F8020},
	}

	for _, pattern := range usagePatterns {
		t.Logf("ldurs %s, [%s, %d] %db", pattern.Destination, pattern.Base, pattern.Offset, pattern.Length)
		code := arm.LoadFixedOffsetSigned(pattern.Destination, pattern.Base, pattern.Mode, pattern.Offset, pattern.Length)
		assert.Equal(t, code, pattern.Code)
	}
}

func TestLoadFixedOffsetScaled(t *testing.T) {
	usagePatterns := []struct {
		Destination cpu.Register
		Base        cpu.Register
		Mode        arm.AddressMode
		Offset      uint
		Length      byte
		Code        uint32
	}{
		{arm.X1, arm.X0, arm.UnscaledImmediate, 0, 8, 0xF9400001},
		{arm.X1, arm.X0, arm.UnscaledImmediate, 1, 8, 0xF9400401},
		{arm.X1, arm.X0, arm.UnscaledImmediate, 4095, 8, 0xF97FFC01},
		{arm.X1, arm.X0, arm.UnscaledImmediate, 0, 1, 0x39400001},
		{arm.X1, arm.X0, arm.UnscaledImmediate, 0, 2, 0x79400001},
		{arm.X1, arm.X0, arm.UnscaledImmediate, 0, 4, 0xB9400001},
	}

	for _, pattern := range usagePatterns {
		t.Logf("ldr %s, [%s, %d] %db", pattern.Destination, pattern.Base, pattern.Offset*uint(pattern.Length), pattern.Length)
		code := arm.LoadFixedOffsetScaled(pattern.Destination, pattern.Base, pattern.Mode, pattern.Offset, pattern.Length)
		assert.Equal(t, code, pattern.Code)
	}
}

func TestLoadFixedOffsetScaledSigned(t *testing.T) {
	usagePatterns := []struct {
		Destination cpu.Register
		Base        cpu.Register
		Mode        arm.AddressMode
		Offset      uint
		Length      byte
		Code        uint32
	}{
		{arm.X1, arm.X0, arm.UnscaledImmediate, 0, 1, 0x39800001},
		{arm.X1, arm.X0, arm.UnscaledImmediate, 0, 2, 0x79800001},
		{arm.X1, arm.X0, arm.UnscaledImmediate, 0, 4, 0xB9800001},
	}

	for _, pattern := range usagePatterns {
		t.Logf("ldrs %s, [%s, %d] %db", pattern.Destination, pattern.Base, pattern.Offset*uint(pattern.Length), pattern.Length)
		code := arm.LoadFixedOffsetScaledSigned(pattern.Destination, pattern.Base, pattern.Mode, pattern.Offset, pattern.Length)
		assert.Equal(t, code, pattern.Code)
	}
}