#!/usr/bin/env ruby
# -*- mode: ruby; coding: utf-8-unix -*- 
require 'test/unit'
require 'opencv'
require 'date'
require File.expand_path(File.dirname(__FILE__)) + '/helper'

include OpenCV

# Tests for OpenCV::EigenFaces
class TestEigenFaces < OpenCVTestCase
  def setup
    @eigenfaces = EigenFaces.new

    @eigenfaces_trained = EigenFaces.new
    img = CvMat.load(FILENAME_LENA256x256, CV_LOAD_IMAGE_GRAYSCALE)
    @eigenfaces_trained.train([img], [1])
  end

  def test_initialize
    [EigenFaces.new, EigenFaces.new(1), EigenFaces.new(1, 99999)].each { |ef|
      assert_equal(EigenFaces, ef.class)
    }

    assert_raise(TypeError) {
      EigenFaces.new(DUMMY_OBJ)
    }

    assert_raise(TypeError) {
      EigenFaces.new(1, DUMMY_OBJ)
    }
  end

  def test_train
    img = CvMat.load(FILENAME_LENA256x256, CV_LOAD_IMAGE_GRAYSCALE)
    assert_nil(@eigenfaces.train([img], [1]))

    assert_raise(TypeError) {
      @eigenfaces.train(DUMMY_OBJ, [1])
    }

    assert_raise(TypeError) {
      @eigenfaces.train([img], DUMMY_OBJ)
    }
  end

  def test_predict
    img = CvMat.load(FILENAME_LENA256x256, CV_LOAD_IMAGE_GRAYSCALE)
    label = 1
    @eigenfaces.train([img], [label])
    assert_equal(label, @eigenfaces.predict(img))

    assert_raise(TypeError) {
      @eigenfaces.predict(DUMMY_OBJ)
    }
  end

  def test_predict_with_confidence
    img = CvMat.load(FILENAME_LENA256x256, CV_LOAD_IMAGE_GRAYSCALE)
    label = 1
    @eigenfaces.train([img], [label])
    lbl, conf = @eigenfaces.predict_with_confidence(img)
    assert_equal(label, lbl)
    assert_equal(0.0, conf)

    assert_raise(TypeError) {
      @eigenfaces.predict_with_confidence(DUMMY_OBJ)
    }
  end

  def test_save
    img = CvMat.load(FILENAME_LENA256x256, CV_LOAD_IMAGE_GRAYSCALE)
    label = 1
    @eigenfaces.train([img], [label])
    filename = "eigenfaces_save-#{DateTime.now.strftime('%Y%m%d%H%M%S')}.xml"
    begin
      @eigenfaces.save(filename)
      assert(File.exist? filename)
    ensure
      File.delete filename
    end
    assert_raise(TypeError) {
      @eigenfaces.save(DUMMY_OBJ)
    }
  end

  def test_load
    assert_nothing_raised {
      @eigenfaces.load('eigenfaces_save.xml')
    }
    assert_raise(TypeError) {
      @eigenfaces.load(DUMMY_OBJ)
    }
  end

  def test_name
    assert_equal('FaceRecognizer.Eigenfaces', @eigenfaces.name)
  end

  def test_get_mat
    mat = @eigenfaces_trained.get_mat('eigenvalues')
    assert_not_nil(mat)
    assert_equal(CvMat, mat.class)

    assert_raise(TypeError) {
      @eigenfaces_trained.get_mat(DUMMY_OBJ)
    }
  end
end