// Copyright 2026 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

//go:build goexperiment.simd && arm64

package simd_test

import (
	"simd/archsimd"
	"testing"
)

func TestAdd(t *testing.T) {
	testFloat32x4Binary(t, archsimd.Float32x4.Add, addSlice[float32])
	testFloat64x2Binary(t, archsimd.Float64x2.Add, addSlice[float64])

	testInt8x16Binary(t, archsimd.Int8x16.Add, addSlice[int8])
	testInt16x8Binary(t, archsimd.Int16x8.Add, addSlice[int16])
	testInt32x4Binary(t, archsimd.Int32x4.Add, addSlice[int32])
	testInt64x2Binary(t, archsimd.Int64x2.Add, addSlice[int64])

	testUint8x16Binary(t, archsimd.Uint8x16.Add, addSlice[uint8])
	testUint16x8Binary(t, archsimd.Uint16x8.Add, addSlice[uint16])
	testUint32x4Binary(t, archsimd.Uint32x4.Add, addSlice[uint32])
	testUint64x2Binary(t, archsimd.Uint64x2.Add, addSlice[uint64])
}

func TestMul(t *testing.T) {
	testFloat32x4Binary(t, archsimd.Float32x4.Mul, mulSlice[float32])
	testFloat64x2Binary(t, archsimd.Float64x2.Mul, mulSlice[float64])

	testInt8x16Binary(t, archsimd.Int8x16.Mul, mulSlice[int8])
	testInt16x8Binary(t, archsimd.Int16x8.Mul, mulSlice[int16])
	testInt32x4Binary(t, archsimd.Int32x4.Mul, mulSlice[int32])

	testUint8x16Binary(t, archsimd.Uint8x16.Mul, mulSlice[uint8])
	testUint16x8Binary(t, archsimd.Uint16x8.Mul, mulSlice[uint16])
	testUint32x4Binary(t, archsimd.Uint32x4.Mul, mulSlice[uint32])
}

func TestSub(t *testing.T) {
	testFloat32x4Binary(t, archsimd.Float32x4.Sub, subSlice[float32])
	testFloat64x2Binary(t, archsimd.Float64x2.Sub, subSlice[float64])

	testInt8x16Binary(t, archsimd.Int8x16.Sub, subSlice[int8])
	testInt16x8Binary(t, archsimd.Int16x8.Sub, subSlice[int16])
	testInt32x4Binary(t, archsimd.Int32x4.Sub, subSlice[int32])
	testInt64x2Binary(t, archsimd.Int64x2.Sub, subSlice[int64])

	testUint8x16Binary(t, archsimd.Uint8x16.Sub, subSlice[uint8])
	testUint16x8Binary(t, archsimd.Uint16x8.Sub, subSlice[uint16])
	testUint32x4Binary(t, archsimd.Uint32x4.Sub, subSlice[uint32])
	testUint64x2Binary(t, archsimd.Uint64x2.Sub, subSlice[uint64])
}

func TestDiv(t *testing.T) {
	testFloat32x4Binary(t, archsimd.Float32x4.Div, divSlice[float32])
	testFloat64x2Binary(t, archsimd.Float64x2.Div, divSlice[float64])
}

func TestMax(t *testing.T) {
	testFloat32x4Binary(t, archsimd.Float32x4.Max, maxSlice[float32])
	testFloat64x2Binary(t, archsimd.Float64x2.Max, maxSlice[float64])

	testInt8x16Binary(t, archsimd.Int8x16.Max, maxSlice[int8])
	testInt16x8Binary(t, archsimd.Int16x8.Max, maxSlice[int16])
	testInt32x4Binary(t, archsimd.Int32x4.Max, maxSlice[int32])
	testUint8x16Binary(t, archsimd.Uint8x16.Max, maxSlice[uint8])
	testUint16x8Binary(t, archsimd.Uint16x8.Max, maxSlice[uint16])
	testUint32x4Binary(t, archsimd.Uint32x4.Max, maxSlice[uint32])
}

func TestMin(t *testing.T) {
	testFloat32x4Binary(t, archsimd.Float32x4.Min, minSlice[float32])
	testFloat64x2Binary(t, archsimd.Float64x2.Min, minSlice[float64])

	testInt8x16Binary(t, archsimd.Int8x16.Min, minSlice[int8])
	testInt16x8Binary(t, archsimd.Int16x8.Min, minSlice[int16])
	testInt32x4Binary(t, archsimd.Int32x4.Min, minSlice[int32])
	testUint8x16Binary(t, archsimd.Uint8x16.Min, minSlice[uint8])
	testUint16x8Binary(t, archsimd.Uint16x8.Min, minSlice[uint16])
	testUint32x4Binary(t, archsimd.Uint32x4.Min, minSlice[uint32])
}

func TestConcatEven(t *testing.T) {
	testInt8x16Binary(t, archsimd.Int8x16.ConcatEven, deinterleaveSlice[int8](128, false))
	testInt16x8Binary(t, archsimd.Int16x8.ConcatEven, deinterleaveSlice[int16](128, false))
	testInt32x4Binary(t, archsimd.Int32x4.ConcatEven, deinterleaveSlice[int32](128, false))
	testInt64x2Binary(t, archsimd.Int64x2.ConcatEven, deinterleaveSlice[int64](128, false))
	testUint8x16Binary(t, archsimd.Uint8x16.ConcatEven, deinterleaveSlice[uint8](128, false))
	testUint16x8Binary(t, archsimd.Uint16x8.ConcatEven, deinterleaveSlice[uint16](128, false))
	testUint32x4Binary(t, archsimd.Uint32x4.ConcatEven, deinterleaveSlice[uint32](128, false))
	testUint64x2Binary(t, archsimd.Uint64x2.ConcatEven, deinterleaveSlice[uint64](128, false))
}

func TestConcatOdd(t *testing.T) {
	testInt8x16Binary(t, archsimd.Int8x16.ConcatOdd, deinterleaveSlice[int8](128, true))
	testInt16x8Binary(t, archsimd.Int16x8.ConcatOdd, deinterleaveSlice[int16](128, true))
	testInt32x4Binary(t, archsimd.Int32x4.ConcatOdd, deinterleaveSlice[int32](128, true))
	testInt64x2Binary(t, archsimd.Int64x2.ConcatOdd, deinterleaveSlice[int64](128, true))
	testUint8x16Binary(t, archsimd.Uint8x16.ConcatOdd, deinterleaveSlice[uint8](128, true))
	testUint16x8Binary(t, archsimd.Uint16x8.ConcatOdd, deinterleaveSlice[uint16](128, true))
	testUint32x4Binary(t, archsimd.Uint32x4.ConcatOdd, deinterleaveSlice[uint32](128, true))
	testUint64x2Binary(t, archsimd.Uint64x2.ConcatOdd, deinterleaveSlice[uint64](128, true))
}

func TestInterleaveEven(t *testing.T) {
	testInt8x16Binary(t, archsimd.Int8x16.InterleaveEven, transposeSlice[int8](128, false))
	testInt16x8Binary(t, archsimd.Int16x8.InterleaveEven, transposeSlice[int16](128, false))
	testInt32x4Binary(t, archsimd.Int32x4.InterleaveEven, transposeSlice[int32](128, false))
	testInt64x2Binary(t, archsimd.Int64x2.InterleaveEven, transposeSlice[int64](128, false))
	testUint8x16Binary(t, archsimd.Uint8x16.InterleaveEven, transposeSlice[uint8](128, false))
	testUint16x8Binary(t, archsimd.Uint16x8.InterleaveEven, transposeSlice[uint16](128, false))
	testUint32x4Binary(t, archsimd.Uint32x4.InterleaveEven, transposeSlice[uint32](128, false))
	testUint64x2Binary(t, archsimd.Uint64x2.InterleaveEven, transposeSlice[uint64](128, false))
}

func TestInterleaveOdd(t *testing.T) {
	testInt8x16Binary(t, archsimd.Int8x16.InterleaveOdd, transposeSlice[int8](128, true))
	testInt16x8Binary(t, archsimd.Int16x8.InterleaveOdd, transposeSlice[int16](128, true))
	testInt32x4Binary(t, archsimd.Int32x4.InterleaveOdd, transposeSlice[int32](128, true))
	testInt64x2Binary(t, archsimd.Int64x2.InterleaveOdd, transposeSlice[int64](128, true))
	testUint8x16Binary(t, archsimd.Uint8x16.InterleaveOdd, transposeSlice[uint8](128, true))
	testUint16x8Binary(t, archsimd.Uint16x8.InterleaveOdd, transposeSlice[uint16](128, true))
	testUint32x4Binary(t, archsimd.Uint32x4.InterleaveOdd, transposeSlice[uint32](128, true))
	testUint64x2Binary(t, archsimd.Uint64x2.InterleaveOdd, transposeSlice[uint64](128, true))
}

func TestInterleaveLoARM64(t *testing.T) {
	testInt8x16Binary(t, archsimd.Int8x16.InterleaveLo, interleaveSlice[int8](128, false))
	testInt16x8Binary(t, archsimd.Int16x8.InterleaveLo, interleaveSlice[int16](128, false))
	testInt32x4Binary(t, archsimd.Int32x4.InterleaveLo, interleaveSlice[int32](128, false))
	testInt64x2Binary(t, archsimd.Int64x2.InterleaveLo, interleaveSlice[int64](128, false))
	testUint8x16Binary(t, archsimd.Uint8x16.InterleaveLo, interleaveSlice[uint8](128, false))
	testUint16x8Binary(t, archsimd.Uint16x8.InterleaveLo, interleaveSlice[uint16](128, false))
	testUint32x4Binary(t, archsimd.Uint32x4.InterleaveLo, interleaveSlice[uint32](128, false))
	testUint64x2Binary(t, archsimd.Uint64x2.InterleaveLo, interleaveSlice[uint64](128, false))
}

func TestInterleaveHiARM64(t *testing.T) {
	testInt8x16Binary(t, archsimd.Int8x16.InterleaveHi, interleaveSlice[int8](128, true))
	testInt16x8Binary(t, archsimd.Int16x8.InterleaveHi, interleaveSlice[int16](128, true))
	testInt32x4Binary(t, archsimd.Int32x4.InterleaveHi, interleaveSlice[int32](128, true))
	testInt64x2Binary(t, archsimd.Int64x2.InterleaveHi, interleaveSlice[int64](128, true))
	testUint8x16Binary(t, archsimd.Uint8x16.InterleaveHi, interleaveSlice[uint8](128, true))
	testUint16x8Binary(t, archsimd.Uint16x8.InterleaveHi, interleaveSlice[uint16](128, true))
	testUint32x4Binary(t, archsimd.Uint32x4.InterleaveHi, interleaveSlice[uint32](128, true))
	testUint64x2Binary(t, archsimd.Uint64x2.InterleaveHi, interleaveSlice[uint64](128, true))
}

func TestGetElem(t *testing.T) {
	// Int8x16
	{
		a := []int8{1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16}
		v := archsimd.LoadInt8x16(a)
		if e := v.GetElem(2); e != a[2] {
			t.Errorf("Int8x16.GetElem(2) = %d, want %d", e, a[2])
		}
	}
	// Int16x8
	{
		a := []int16{10, 20, 30, 40, 50, 60, 70, 80}
		v := archsimd.LoadInt16x8(a)
		if e := v.GetElem(3); e != a[3] {
			t.Errorf("Int16x8.GetElem(3) = %d, want %d", e, a[3])
		}
	}
	// Int32x4
	{
		a := []int32{100, 200, 300, 400}
		v := archsimd.LoadInt32x4(a)
		if e := v.GetElem(1); e != a[1] {
			t.Errorf("Int32x4.GetElem(1) = %d, want %d", e, a[1])
		}
	}
	// Int64x2
	{
		a := []int64{1000, 2000}
		v := archsimd.LoadInt64x2(a)
		if e := v.GetElem(0); e != a[0] {
			t.Errorf("Int64x2.GetElem(0) = %d, want %d", e, a[0])
		}
	}
	// Uint8x16
	{
		a := []uint8{1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16}
		v := archsimd.LoadUint8x16(a)
		if e := v.GetElem(5); e != a[5] {
			t.Errorf("Uint8x16.GetElem(5) = %d, want %d", e, a[5])
		}
	}
	// Uint16x8
	{
		a := []uint16{100, 200, 300, 400, 500, 600, 700, 800}
		v := archsimd.LoadUint16x8(a)
		if e := v.GetElem(7); e != a[7] {
			t.Errorf("Uint16x8.GetElem(7) = %d, want %d", e, a[7])
		}
	}
	// Uint32x4
	{
		a := []uint32{1000, 2000, 3000, 4000}
		v := archsimd.LoadUint32x4(a)
		if e := v.GetElem(2); e != a[2] {
			t.Errorf("Uint32x4.GetElem(2) = %d, want %d", e, a[2])
		}
	}
	// Uint64x2
	{
		a := []uint64{10000, 20000}
		v := archsimd.LoadUint64x2(a)
		if e := v.GetElem(1); e != a[1] {
			t.Errorf("Uint64x2.GetElem(1) = %d, want %d", e, a[1])
		}
	}
	// Float32x4
	{
		a := []float32{1.0, 2.0, 3.0, 4.0}
		v := archsimd.LoadFloat32x4(a)
		if e := v.GetElem(3); e != a[3] {
			t.Errorf("Float32x4.GetElem(3) = %f, want %f", e, a[3])
		}
	}
	// Float64x2
	{
		a := []float64{10.5, 20.5}
		v := archsimd.LoadFloat64x2(a)
		if e := v.GetElem(0); e != a[0] {
			t.Errorf("Float64x2.GetElem(0) = %f, want %f", e, a[0])
		}
	}
}

func TestSetElem(t *testing.T) {
	// Int8x16
	{
		a := []int8{1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16}
		v := archsimd.LoadInt8x16(a)
		v = v.SetElem(3, int8(99))
		a[3] = 99
		b := make([]int8, 16)
		v.Store(b)
		checkSlices(t, b, a)
	}
	// Int16x8
	{
		a := []int16{10, 20, 30, 40, 50, 60, 70, 80}
		v := archsimd.LoadInt16x8(a)
		v = v.SetElem(5, int16(123))
		a[5] = 123
		b := make([]int16, 8)
		v.Store(b)
		checkSlices(t, b, a)
	}
	// Int32x4
	{
		a := []int32{100, 200, 300, 400}
		v := archsimd.LoadInt32x4(a)
		v = v.SetElem(2, int32(999))
		a[2] = 999
		b := make([]int32, 4)
		v.Store(b)
		checkSlices(t, b, a)
	}
	// Int64x2
	{
		a := []int64{1000, 2000}
		v := archsimd.LoadInt64x2(a)
		v = v.SetElem(1, int64(5555))
		a[1] = 5555
		b := make([]int64, 2)
		v.Store(b)
		checkSlices(t, b, a)
	}
	// Uint8x16
	{
		a := []uint8{1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16}
		v := archsimd.LoadUint8x16(a)
		v = v.SetElem(7, uint8(200))
		a[7] = 200
		b := make([]uint8, 16)
		v.Store(b)
		checkSlices(t, b, a)
	}
	// Uint16x8
	{
		a := []uint16{100, 200, 300, 400, 500, 600, 700, 800}
		v := archsimd.LoadUint16x8(a)
		v = v.SetElem(0, uint16(1111))
		a[0] = 1111
		b := make([]uint16, 8)
		v.Store(b)
		checkSlices(t, b, a)
	}
	// Uint32x4
	{
		a := []uint32{1000, 2000, 3000, 4000}
		v := archsimd.LoadUint32x4(a)
		v = v.SetElem(3, uint32(9999))
		a[3] = 9999
		b := make([]uint32, 4)
		v.Store(b)
		checkSlices(t, b, a)
	}
	// Uint64x2
	{
		a := []uint64{10000, 20000}
		v := archsimd.LoadUint64x2(a)
		v = v.SetElem(0, uint64(55555))
		a[0] = 55555
		b := make([]uint64, 2)
		v.Store(b)
		checkSlices(t, b, a)
	}
	// Float32x4
	{
		a := []float32{1.0, 2.0, 3.0, 4.0}
		v := archsimd.LoadFloat32x4(a)
		v = v.SetElem(1, float32(42.5))
		a[1] = 42.5
		b := make([]float32, 4)
		v.Store(b)
		checkSlices(t, b, a)
	}
	// Float64x2
	{
		a := []float64{10.5, 20.5}
		v := archsimd.LoadFloat64x2(a)
		v = v.SetElem(0, float64(99.9))
		a[0] = 99.9
		b := make([]float64, 2)
		v.Store(b)
		checkSlices(t, b, a)
	}
}