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

package testblas

import (
	"testing"

	"gonum.org/v1/gonum/blas"
)

var zher2TestCases = []struct {
	alpha      complex128
	incX, incY int
	x          []complex128
	y          []complex128
	a          []complex128

	want []complex128
}{
	{
		alpha: 1 + 2i,
		incX:  1,
		incY:  1,
	},
	{
		alpha: 1 + 2i,
		incX:  1,
		incY:  1,
		x: []complex128{
			-6 + 2i,
			-2 - 4i,
			0 + 0i,
			0 + 7i,
		},
		y: []complex128{
			2 - 5i,
			0 + 0i,
			-8 - 9i,
			6 + 6i,
		},
		a: []complex128{
			2 + 0i, -9 + 7i, 3 + 11i, 10 - 1i,
			-9 - 7i, 16 + 0i, -5 + 2i, -7 - 5i,
			3 - 11i, -5 - 2i, 14 + 0i, 2 - 1i,
			10 + 1i, -7 + 5i, 2 + 1i, 18 + 0i,
		},
		want: []complex128{
			62 + 0i, 43 - 7i, 173 + 1i, -173 + 55i,
			43 + 7i, 16 + 0i, 19 + 120i, -19 - 89i,
			173 - 1i, 19 - 120i, 14 + 0i, 51 + 181i,
			-173 - 55i, -19 + 89i, 51 - 181i, -66 + 0i,
		},
	},
	{
		alpha: 1 + 2i,
		incX:  1,
		incY:  1,
		x: []complex128{
			-6 + 2i,
			-2 - 4i,
			0 + 0i,
			0 + 7i,
		},
		y: []complex128{
			2 - 5i,
			-8 - 9i,
			0 + 0i,
			6 + 6i,
		},
		a: []complex128{
			2 + 0i, -9 + 7i, 3 + 11i, 10 - 1i,
			-9 - 7i, 16 + 0i, -5 + 2i, -7 - 5i,
			3 - 11i, -5 - 2i, 14 + 0i, 2 - 1i,
			10 + 1i, -7 + 5i, 2 + 1i, 18 + 0i,
		},
		want: []complex128{
			62 + 0i, 213 - 17i, 3 + 11i, -173 + 55i,
			213 + 17i, 64 + 0i, -5 + 2i, 30 + 93i,
			3 - 11i, -5 - 2i, 14 + 0i, 2 - 1i,
			-173 - 55i, 30 - 93i, 2 + 1i, -66 + 0i,
		},
	},
	{
		alpha: 1 + 2i,
		incX:  2,
		incY:  4,
		x: []complex128{
			-6 + 2i,
			-2 - 4i,
			0 + 0i,
			0 + 7i,
		},
		y: []complex128{
			2 - 5i,
			-8 - 9i,
			0 + 0i,
			6 + 6i,
		},
		a: []complex128{
			2 + 0i, -9 + 7i, 3 + 11i, 10 - 1i,
			-9 - 7i, 16 + 0i, -5 + 2i, -7 - 5i,
			3 - 11i, -5 - 2i, 14 + 0i, 2 - 1i,
			10 + 1i, -7 + 5i, 2 + 1i, 18 + 0i,
		},
		want: []complex128{
			62 + 0i, 213 - 17i, 3 + 11i, -173 + 55i,
			213 + 17i, 64 + 0i, -5 + 2i, 30 + 93i,
			3 - 11i, -5 - 2i, 14 + 0i, 2 - 1i,
			-173 - 55i, 30 - 93i, 2 + 1i, -66 + 0i,
		},
	},
	{
		alpha: 1 + 2i,
		incX:  3,
		incY:  7,
		x: []complex128{
			-6 + 2i,
			-2 - 4i,
			0 + 0i,
			0 + 7i,
		},
		y: []complex128{
			2 - 5i,
			0 + 0i,
			-8 - 9i,
			6 + 6i,
		},
		a: []complex128{
			2 + 0i, -9 + 7i, 3 + 11i, 10 - 1i,
			-9 - 7i, 16 + 0i, -5 + 2i, -7 - 5i,
			3 - 11i, -5 - 2i, 14 + 0i, 2 - 1i,
			10 + 1i, -7 + 5i, 2 + 1i, 18 + 0i,
		},
		want: []complex128{
			62 + 0i, 43 - 7i, 173 + 1i, -173 + 55i,
			43 + 7i, 16 + 0i, 19 + 120i, -19 - 89i,
			173 - 1i, 19 - 120i, 14 + 0i, 51 + 181i,
			-173 - 55i, -19 + 89i, 51 - 181i, -66 + 0i,
		},
	},
	{
		alpha: 1 + 2i,
		incX:  -3,
		incY:  7,
		x: []complex128{
			0 + 7i,
			0 + 0i,
			-2 - 4i,
			-6 + 2i,
		},
		y: []complex128{
			2 - 5i,
			0 + 0i,
			-8 - 9i,
			6 + 6i,
		},
		a: []complex128{
			2 + 0i, -9 + 7i, 3 + 11i, 10 - 1i,
			-9 - 7i, 16 + 0i, -5 + 2i, -7 - 5i,
			3 - 11i, -5 - 2i, 14 + 0i, 2 - 1i,
			10 + 1i, -7 + 5i, 2 + 1i, 18 + 0i,
		},
		want: []complex128{
			62 + 0i, 43 - 7i, 173 + 1i, -173 + 55i,
			43 + 7i, 16 + 0i, 19 + 120i, -19 - 89i,
			173 - 1i, 19 - 120i, 14 + 0i, 51 + 181i,
			-173 - 55i, -19 + 89i, 51 - 181i, -66 + 0i,
		},
	},
	{
		alpha: 1 + 2i,
		incX:  3,
		incY:  -7,
		x: []complex128{
			-6 + 2i,
			-2 - 4i,
			0 + 0i,
			0 + 7i,
		},
		y: []complex128{
			6 + 6i,
			-8 - 9i,
			0 + 0i,
			2 - 5i,
		},
		a: []complex128{
			2 + 0i, -9 + 7i, 3 + 11i, 10 - 1i,
			-9 - 7i, 16 + 0i, -5 + 2i, -7 - 5i,
			3 - 11i, -5 - 2i, 14 + 0i, 2 - 1i,
			10 + 1i, -7 + 5i, 2 + 1i, 18 + 0i,
		},
		want: []complex128{
			62 + 0i, 43 - 7i, 173 + 1i, -173 + 55i,
			43 + 7i, 16 + 0i, 19 + 120i, -19 - 89i,
			173 - 1i, 19 - 120i, 14 + 0i, 51 + 181i,
			-173 - 55i, -19 + 89i, 51 - 181i, -66 + 0i,
		},
	},
	{
		alpha: 1 + 2i,
		incX:  -3,
		incY:  -7,
		x: []complex128{
			0 + 7i,
			0 + 0i,
			-2 - 4i,
			-6 + 2i,
		},
		y: []complex128{
			6 + 6i,
			-8 - 9i,
			0 + 0i,
			2 - 5i,
		},
		a: []complex128{
			2 + 0i, -9 + 7i, 3 + 11i, 10 - 1i,
			-9 - 7i, 16 + 0i, -5 + 2i, -7 - 5i,
			3 - 11i, -5 - 2i, 14 + 0i, 2 - 1i,
			10 + 1i, -7 + 5i, 2 + 1i, 18 + 0i,
		},
		want: []complex128{
			62 + 0i, 43 - 7i, 173 + 1i, -173 + 55i,
			43 + 7i, 16 + 0i, 19 + 120i, -19 - 89i,
			173 - 1i, 19 - 120i, 14 + 0i, 51 + 181i,
			-173 - 55i, -19 + 89i, 51 - 181i, -66 + 0i,
		},
	},
	{
		alpha: 0,
		incX:  1,
		incY:  1,
		x: []complex128{
			-6 + 2i,
			-2 - 4i,
			0 + 0i,
			0 + 7i,
		},
		y: []complex128{
			2 - 5i,
			0 + 0i,
			-8 - 9i,
			6 + 6i,
		},
		a: []complex128{
			2 + 0i, -9 + 7i, 3 + 11i, 10 - 1i,
			-9 - 7i, 16 + 0i, -5 + 2i, -7 - 5i,
			3 - 11i, -5 - 2i, 14 + 0i, 2 - 1i,
			10 + 1i, -7 + 5i, 2 + 1i, 18 + 0i,
		},
		want: []complex128{
			2 + 0i, -9 + 7i, 3 + 11i, 10 - 1i,
			-9 - 7i, 16 + 0i, -5 + 2i, -7 - 5i,
			3 - 11i, -5 - 2i, 14 + 0i, 2 - 1i,
			10 + 1i, -7 + 5i, 2 + 1i, 18 + 0i,
		},
	},
}

type Zher2er interface {
	Zher2(uplo blas.Uplo, n int, alpha complex128, x []complex128, incX int, y []complex128, incY int, a []complex128, lda int)
}

func Zher2Test(t *testing.T, impl Zher2er) {
	for tc, test := range zher2TestCases {
		n := len(test.x)
		incX := test.incX
		incY := test.incY
		for _, uplo := range []blas.Uplo{blas.Lower, blas.Upper} {
			for _, lda := range []int{max(1, n), n + 11} {
				x := makeZVector(test.x, incX)
				xCopy := make([]complex128, len(x))
				copy(xCopy, x)

				y := makeZVector(test.y, incY)
				yCopy := make([]complex128, len(y))
				copy(yCopy, y)

				a := makeZGeneral(test.a, n, n, lda)
				want := makeZGeneral(test.want, n, n, lda)

				if uplo == blas.Upper {
					for i := 0; i < n; i++ {
						for j := 0; j < i; j++ {
							a[i*lda+j] = znan
							want[i*lda+j] = znan
						}
					}
				} else {
					for i := 0; i < n; i++ {
						for j := i + 1; j < n; j++ {
							a[i*lda+j] = znan
							want[i*lda+j] = znan
						}
					}
				}

				impl.Zher2(uplo, n, test.alpha, x, incX, y, incY, a, lda)

				if !zsame(x, xCopy) {
					t.Errorf("Case %v (uplo=%v,incX=%v,incY=%v,lda=%v: unexpected modification of x", tc, uplo, incX, incY, lda)
				}
				if !zsame(y, yCopy) {
					t.Errorf("Case %v (uplo=%v,incX=%v,incY=%v,lda=%v: unexpected modification of y", tc, uplo, incX, incY, lda)
				}
				if !zsame(want, a) {
					t.Errorf("Case %v (uplo=%v,incX=%v,incY=%v,lda=%v: unexpected result\nwant: %v\ngot:  %v", tc, uplo, incX, incY, lda, want, a)
				}
			}
		}
	}
}