// Copyright 2014 Hajime Hoshi
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package colorm_test

import (
	"image/color"
	"math"
	"testing"

	"github.com/hajimehoshi/ebiten/v2/colorm"
)

func TestColorMInit(t *testing.T) {
	var m colorm.ColorM
	for i := range colorm.Dim - 1 {
		for j := range colorm.Dim {
			got := m.Element(i, j)
			want := 0.0
			if i == j {
				want = 1
			}
			if want != got {
				t.Errorf("m.Element(%d, %d) = %f, want %f", i, j, got, want)
			}
		}
	}

	m.SetElement(0, 0, 1)
	for i := range colorm.Dim - 1 {
		for j := range colorm.Dim {
			got := m.Element(i, j)
			want := 0.0
			if i == j {
				want = 1
			}
			if want != got {
				t.Errorf("m.Element(%d, %d) = %f, want %f", i, j, got, want)
			}
		}
	}
}

func TestColorMAssign(t *testing.T) {
	m := colorm.ColorM{}
	m.SetElement(0, 0, 1)
	m2 := m
	m.SetElement(0, 0, 0)
	got := m2.Element(0, 0)
	want := 1.0
	if want != got {
		t.Errorf("m2.Element(%d, %d) = %f, want %f", 0, 0, got, want)
	}
}

func TestColorMTranslate(t *testing.T) {
	expected := [4][5]float64{
		{1, 0, 0, 0, 0.5},
		{0, 1, 0, 0, 1.5},
		{0, 0, 1, 0, 2.5},
		{0, 0, 0, 1, 3.5},
	}
	m := colorm.ColorM{}
	m.Translate(0.5, 1.5, 2.5, 3.5)
	for i := range 4 {
		for j := range 5 {
			got := m.Element(i, j)
			want := expected[i][j]
			if want != got {
				t.Errorf("m.Element(%d, %d) = %f, want %f", i, j, got, want)
			}
		}
	}
}

func TestColorMScale(t *testing.T) {
	expected := [4][5]float64{
		{0.5, 0, 0, 0, 0},
		{0, 1.5, 0, 0, 0},
		{0, 0, 2.5, 0, 0},
		{0, 0, 0, 3.5, 0},
	}
	m := colorm.ColorM{}
	m.Scale(0.5, 1.5, 2.5, 3.5)
	for i := range 4 {
		for j := range 5 {
			got := m.Element(i, j)
			want := expected[i][j]
			if want != got {
				t.Errorf("m.Element(%d, %d) = %f, want %f", i, j, got, want)
			}
		}
	}
}

func TestColorMTranslateAndScale(t *testing.T) {
	expected := [4][5]float64{
		{1, 0, 0, 0, 0},
		{0, 1, 0, 0, 0},
		{0, 0, 1, 0, 0},
		{0, 0, 0, 0.5, 0.5},
	}
	m := colorm.ColorM{}
	m.Translate(0, 0, 0, 1)
	m.Scale(1, 1, 1, 0.5)
	for i := range 4 {
		for j := range 5 {
			got := m.Element(i, j)
			want := expected[i][j]
			if want != got {
				t.Errorf("m.Element(%d, %d) = %f, want %f", i, j, got, want)
			}
		}
	}
}

func TestColorMMonochrome(t *testing.T) {
	expected := [4][5]float64{
		{0.2990, 0.5870, 0.1140, 0, 0},
		{0.2990, 0.5870, 0.1140, 0, 0},
		{0.2990, 0.5870, 0.1140, 0, 0},
		{0, 0, 0, 1, 0},
	}
	m := colorm.ColorM{}
	m.ChangeHSV(0, 0, 1)
	for i := range 4 {
		for j := range 5 {
			got := m.Element(i, j)
			want := expected[i][j]
			if math.Abs(want-got) > 0.0001 {
				t.Errorf("m.Element(%d, %d) = %f, want %f", i, j, got, want)
			}
		}
	}
}

func TestColorMConcatSelf(t *testing.T) {
	expected := [4][5]float64{
		{30, 40, 30, 25, 30},
		{40, 54, 43, 37, 37},
		{30, 43, 51, 39, 34},
		{25, 37, 39, 46, 36},
	}
	m := colorm.ColorM{}
	for i := range 4 {
		for j := range 5 {
			m.SetElement(i, j, float64((i+j)%5+1))
		}
	}
	m.Concat(m)
	for i := range 4 {
		for j := range 5 {
			got := m.Element(i, j)
			want := expected[i][j]
			if want != got {
				t.Errorf("m.Element(%d, %d) = %f, want %f", i, j, got, want)
			}
		}
	}
}

func absDiffU32(x, y uint32) uint32 {
	if x < y {
		return y - x
	}
	return x - y
}

func TestColorMApply(t *testing.T) {
	mono := colorm.ColorM{}
	mono.ChangeHSV(0, 0, 1)

	shiny := colorm.ColorM{}
	shiny.Translate(1, 1, 1, 0)

	shift := colorm.ColorM{}
	shift.Translate(0.5, 0.5, 0.5, 0.5)

	cases := []struct {
		ColorM colorm.ColorM
		In     color.Color
		Out    color.Color
		Delta  uint32
	}{
		{
			ColorM: colorm.ColorM{},
			In:     color.RGBA{R: 1, G: 2, B: 3, A: 4},
			Out:    color.RGBA{R: 1, G: 2, B: 3, A: 4},
			Delta:  0x101,
		},
		{
			ColorM: mono,
			In:     color.NRGBA{R: 0xff, G: 0xff, B: 0xff},
			Out:    color.Transparent,
			Delta:  0x101,
		},
		{
			ColorM: mono,
			In:     color.RGBA{R: 0xff, A: 0xff},
			Out:    color.RGBA{R: 0x4c, G: 0x4c, B: 0x4c, A: 0xff},
			Delta:  0x101,
		},
		{
			ColorM: shiny,
			In:     color.RGBA{R: 0x80, G: 0x90, B: 0xa0, A: 0xb0},
			Out:    color.RGBA{R: 0xb0, G: 0xb0, B: 0xb0, A: 0xb0},
			Delta:  1,
		},
		{
			ColorM: shift,
			In:     color.RGBA{},
			Out:    color.RGBA{R: 0x40, G: 0x40, B: 0x40, A: 0x80},
			Delta:  0x101,
		},
	}
	for _, c := range cases {
		out := c.ColorM.Apply(c.In)
		r0, g0, b0, a0 := out.RGBA()
		r1, g1, b1, a1 := c.Out.RGBA()
		if absDiffU32(r0, r1) > c.Delta || absDiffU32(g0, g1) > c.Delta ||
			absDiffU32(b0, b1) > c.Delta || absDiffU32(a0, a1) > c.Delta {
			t.Errorf("%v.Apply(%v) = {%d, %d, %d, %d}, want {%d, %d, %d, %d}", c.ColorM, c.In, r0, g0, b0, a0, r1, g1, b1, a1)
		}
	}
}

// #1765
func TestColorMConcat(t *testing.T) {
	var a, b colorm.ColorM
	a.SetElement(1, 2, -1)
	a.Concat(b)
	if got, want := a.Element(1, 2), -1.0; got != want {
		t.Errorf("got: %f, want: %f", got, want)
	}
}
