#!/usr/bin/env python2
# -*- coding: utf-8 -*-
"""
Test the epipolar curve code, at least for simple cases.

Created on Thu Mar 23 16:12:21 2017

@author: yosef
"""

import unittest
import numpy as np

from optv.calibration import Calibration
from optv.parameters import ControlParams, VolumeParams
from optv.epipolar import epipolar_curve

class TestEpipolarCurve(unittest.TestCase):
    def test_two_cameras(self):
        ori_tmpl = r'testing_fodder/calibration/sym_cam{cam_num}.tif.ori'
        add_file = r'testing_fodder/calibration/cam1.tif.addpar'
        
        orig_cal = Calibration()
        orig_cal.from_file(ori_tmpl.format(cam_num=1), add_file)
        proj_cal = Calibration()
        proj_cal.from_file(ori_tmpl.format(cam_num=3), add_file)
        
        # reorient cams:
        orig_cal.set_angles(np.r_[0., -np.pi/4., 0.])
        proj_cal.set_angles(np.r_[0., 3*np.pi/4., 0.])
        
        cpar = ControlParams(4)
        cpar.read_control_par("testing_fodder/corresp/control.par")
        sens_size = cpar.get_image_size()
        
        vpar = VolumeParams()
        vpar.read_volume_par("testing_fodder/corresp/criteria.par")
        vpar.set_Zmin_lay([-10, -10])
        vpar.set_Zmax_lay([10, 10])
        
        mult_params = cpar.get_multimedia_params()
        mult_params.set_n1(1.)
        mult_params.set_layers(np.array([1.]), np.array([1.]))
        mult_params.set_n3(1.)
        
        # Central point translates to central point because cameras point 
        # directly at each other.
        mid = np.r_[sens_size]/2.
        line = epipolar_curve(mid, orig_cal, proj_cal, 5, cpar, vpar)
        self.failUnless(np.all(abs(line - mid) < 1e-6))
        
        # An equatorial point draws a latitude.
        line = epipolar_curve(
            mid - np.r_[100., 0.], orig_cal, proj_cal, 5, cpar, vpar)
        np.testing.assert_array_equal(np.argsort(line[:,0]), np.arange(5)[::-1])
        self.failUnless(np.all(abs(line[:,1] - mid[1]) < 1e-6))