package images4

import (
	"path"
	"testing"
)

func testPropSimilar(fA, fB string, isSimilar bool,
	t *testing.T) {
	p := path.Join("testdata", "proportions")
	imgA, err := Open(path.Join(p, fA))
	if err != nil {
		t.Error("Error opening image:", err)
	}
	imgB, err := Open(path.Join(p, fB))
	if err != nil {
		t.Error("Error opening image:", err)
	}
	iconA := Icon(imgA)
	iconB := Icon(imgB)

	if isSimilar == true {
		if !propSimilar(iconA, iconB) {
			t.Errorf("Expecting similarity of %v to %v.", fA, fB)
		}
	}
	if isSimilar == false {
		if propSimilar(iconA, iconB) {
			t.Errorf("Expecting non-similarity of %v to %v.", fA, fB)
		}
	}
}

func TestPropSimilar(t *testing.T) {
	testPropSimilar("100x130.png", "100x124.png", true, t)
	testPropSimilar("100x130.png", "100x122.png", false, t)
	testPropSimilar("130x100.png", "260x200.png", true, t)
	testPropSimilar("200x200.png", "260x200.png", false, t)
	testPropSimilar("130x100.png", "124x100.png", true, t)
	testPropSimilar("130x100.png", "122x100.png", false, t)
	testPropSimilar("130x100.png", "130x100.png", true, t)
	testPropSimilar("100x130.png", "130x100.png", false, t)
	testPropSimilar("124x100.png", "260x200.png", true, t)
	testPropSimilar("122x100.png", "260x200.png", false, t)
	testPropSimilar("100x124.png", "100x130.png", true, t)
}

func testEucSimilar(fA, fB string, isSimilar bool,
	t *testing.T) {
	p := path.Join("testdata", "euclidean")
	imgA, err := Open(path.Join(p, fA))
	if err != nil {
		t.Error("Error opening image:", err)
	}
	iconA := Icon(imgA)
	imgB, err := Open(path.Join(p, fB))
	if err != nil {
		t.Error("Error opening image:", err)
	}
	iconB := Icon(imgB)
	if isSimilar == true {
		if !eucSimilar(iconA, iconB) {
			t.Errorf("Expecting similarity of %v to %v.", fA, fB)
		}
	}
	if isSimilar == false {
		if eucSimilar(iconA, iconB) {
			t.Errorf("Expecting non-similarity of %v to %v.", fA, fB)
		}
	}
}

func TestEucSimilar(t *testing.T) {
	testEucSimilar("large.jpg", "distorted.jpg", true, t)
	testEucSimilar("large.jpg", "flipped.jpg", false, t)
	testEucSimilar("large.jpg", "small.jpg", true, t)
	testEucSimilar("small.gif", "small.jpg", true, t) // GIF test.
	testEucSimilar("uniform-black.png", "uniform-black.png", true, t)
	testEucSimilar("uniform-black.png", "uniform-white.png", false, t)
	testEucSimilar("uniform-green.png", "uniform-green.png", true, t)
	testEucSimilar("uniform-green.png", "uniform-white.png", false, t)
	testEucSimilar("uniform-white.png", "uniform-white.png", true, t)
}

func TestSimilar90270(t *testing.T) {

	img0, _ := Open(path.Join("testdata", "rotate", "0.jpg"))
	img90, _ := Open(path.Join("testdata", "rotate", "90.jpg"))
	img180, _ := Open(path.Join("testdata", "rotate", "180.jpg"))
	img270, _ := Open(path.Join("testdata", "rotate", "270.jpg"))

	icon0 := Icon(img0)
	icon90 := Icon(img90)
	icon180 := Icon(img180)
	icon270 := Icon(img270)

	if !Similar90270(icon0, icon90) {
		t.Errorf("0.jpg must be similar to 90.jpg")
	}
	if Similar90270(icon0, icon180) {
		t.Errorf("0.jpg must be NOT similar to 180.jpg")
	}

	if !Similar90270(icon0, icon270) {
		t.Errorf("0.jpg must be similar to 270.jpg")
	}

	if !Similar90270(icon90, icon180) {
		t.Errorf("90.jpg must be similar to 180.jpg")
	}

	if Similar90270(icon90, icon270) {
		t.Errorf("90.jpg must be NOT similar to 270.jpg")
	}
}