package opencv

import (
	"path"
	"runtime"
	"os"
	"testing"
	"syscall"
)

func TestResize(t *testing.T) {
	_, currentfile, _, _ := runtime.Caller(0)
	filename := path.Join(path.Dir(currentfile), "../images/lena.jpg")

	image := LoadImage(filename)
	if image == nil {
		t.Fatal("LoadImage fail")
	}
	defer image.Release()

	rimage := Resize(image, 10, 10, CV_INTER_LINEAR)
	if rimage == nil {
		t.Fatal("Resize fail")
	}
	defer rimage.Release()

	if rimage.Width() != 10 {
		t.Fatalf("excepted width is 10, returned %d\n", rimage.Width())
	}

	if rimage.Height() != 10 {
		t.Fatalf("excepted width is 10, returned %d\n", rimage.Height())
	}
}

func TestCrop(t *testing.T) {
	_, currentfile, _, _ := runtime.Caller(0)
	filename := path.Join(path.Dir(currentfile), "../images/lena.jpg")

	image := LoadImage(filename)
	if image == nil {
		t.Fatal("LoadImage fail")
	}
	defer image.Release()

	crop := Crop(image, 0, 0, 200, 200)
	if crop == nil {
		t.Fatal("Crop fail")
	}
	defer crop.Release()

	if crop.Width() != 200 {
		t.Fatalf("excepted width is 200, returned %d\n", crop.Width())
	}

	if crop.Height() != 200 {
		t.Fatalf("excepted width is 200, returned %d\n", crop.Height())
	}
}

func TestFindContours(t *testing.T) {
	_, currentfile, _, _ := runtime.Caller(0)
	filename := path.Join(path.Dir(currentfile), "../images/pic5.png")

	image := LoadImage(filename)
	if image == nil {
		t.Fatal("LoadImage fail")
	}
	defer image.Release()

	grayscale := CreateImage(image.Width(), image.Height(), IPL_DEPTH_8U, 1)
	CvtColor(image, grayscale, CV_BGR2GRAY)
	defer grayscale.Release()


	edges := CreateImage(grayscale.Width(), grayscale.Height(), grayscale.Depth(), grayscale.Channels())
	defer edges.Release()
	Canny(grayscale, edges, 50, 200, 3)

	contourType := CreateContourType()
	seq := contourType.FindContours(edges)
	defer seq.Release()

	contours := CreateImage(grayscale.Width(), grayscale.Height(), grayscale.Depth(), grayscale.Channels())
	white := NewScalar(255, 255, 255, 0)
	contours.Set(white)

	black := NewScalar(0, 0, 0, 0)
	red := NewScalar(0, 255, 0, 0)

	for ; seq != nil; seq = (*Seq)(seq.h_next) {
		DrawContours(contours, seq, red, black, 0, 2, 8, Point{0, 0})
	}

	filename = path.Join(path.Dir(currentfile), "../images/pic5_contours.png")
	// Uncomment this code to create the test image "../images/shapes_contours.png"
	// It is part of the repo, and what this test compares against
	//
	//SaveImage(filename), contours, 0)

	tempfilename := path.Join(os.TempDir(), "pic5_contours.png")
	defer syscall.Unlink(tempfilename)
	SaveImage(tempfilename, contours, 0)

	// Compare actual image with expected image
	same, err := BinaryCompare(filename, tempfilename)
	if err != nil {
		t.Fatal(err)
	}
	if !same {
		t.Error("Expected contour file != actual contour file")
	}

}