#!/usr/bin/env python

from maxasc import *
import anuga.utilities.system_tools as aust

import exceptions
class TestError(exceptions.Exception): pass
import unittest

import sys
import os
import re
import glob

HEADER_SIZE = 6

# pattern string used to split multimax data
SpacesPatternString = ' +'

# generate 're' pattern for 'any number of spaces'
SpacesPattern = re.compile(SpacesPatternString)


def FilesEqual(file1, file2):
    """
    Compare two ASC files for equality.
    """
    
    def alltrue(a, b):
        return a and b

    def do_list(prefix, l):
        print prefix, '#'*100
        for (i, x) in enumerate(l):
            print '%05d: %s\t' % (i, str(x)),
            if (i+1) % 5 == 0:
                print
        print
        print
    
    # get both files into memory
    fd = open(file1, 'r')
    data1 = fd.readlines()
    fd.close()
    fd = open(file2, 'r')
    data2 = fd.readlines()
    fd.close()

    # check we have same number of lines in each file
    if len(data1) != len(data2):
        print '# lines differs: len(data1)=%d, len(data2)=%d' % (len(data1), len(data2))
        return False
    
    # read header lines, check identical
    for i in range(HEADER_SIZE):
        if data1[i] != data2[i]:
            print 'headers differ:'
            print data1[i]
            print data2[i]
            return False

    # read data lines, check same *values*
    for line_num in range(HEADER_SIZE, len(data1)):
        d1 = SpacesPattern.split(data1[line_num].strip())
        d2 = SpacesPattern.split(data2[line_num].strip())

        if len(d1) != len(d2):
            print '# columns differs, len(d1)=%d, len(d2)=%d on line %d' % (len(d1), len(d2), line_num)
            return False

        fd1 = [float(value) for value in d1]
        fd2 = [float(value) for value in d2]

        vec = map(lambda a,b: a==b, fd1, fd2)
        
        if not reduce(lambda a,b: a and b, vec):
            print 'line number = %d (out of %d)' % (line_num, len(data1))
            do_list('fd1', fd1)
            do_list('fd2', fd2)
            do_list('vec', vec)
            return False
    
    return True


class Test_MaxAsc(unittest.TestCase):
    def tearDown(self):
        # delete all output files
        files = glob.glob('*.out.asc')
        for file in files:
            os.remove(file)

    def test_unequal_lines(self):
        test_path = aust.get_pathname_from_package('anuga.lib.maxasc')
        in_file = os.path.join(test_path, 'test1.asc')
        expected_file = os.path.join(test_path, 'test1_bad_num_lines.asc')
        self.failUnlessRaises(RuntimeError, MaxAsc,
                              'test.out.asc',
                              [in_file, expected_file])

    def test_headers_differ(self):
        test_path = aust.get_pathname_from_package('anuga.lib.maxasc')
        in_file = os.path.join(test_path, 'test1.asc')
        expected_file = os.path.join(test_path, 'test1_bad_num_lines.asc')
        self.failUnlessRaises(RuntimeError, MaxAsc,
                              'test.out.asc',
                              [in_file, expected_file])

    def test_wrong_num_columns(self):
        test_path = aust.get_pathname_from_package('anuga.lib.maxasc')
        in_file = os.path.join(test_path, 'test1.asc')
        expected_file = os.path.join(test_path, 'test1_wrong_num_columns.asc')
        self.failUnlessRaises(RuntimeError, MaxAsc,
                              'test.out.asc',
                              [in_file, expected_file])

    def test_same_input_equals_output1(self):
        test_path = aust.get_pathname_from_package('anuga.lib.maxasc')
        in_file = os.path.join(test_path, 'test1.asc')
        MaxAsc('test1.out.asc', [in_file])
        self.failUnless(FilesEqual('test1.out.asc', in_file))

    def test_same_input_equals_bigA(self):
        test_path = aust.get_pathname_from_package('anuga.lib.maxasc')
        in_file = os.path.join(test_path, 'perthAll_stage_250m.asc')
        MaxAsc('perth.out.asc', [in_file])
        self.failUnless(FilesEqual('perth.out.asc', in_file))

    def test_same_input_equals_bigB(self):
        test_path = aust.get_pathname_from_package('anuga.lib.maxasc')
        in_file = os.path.join(test_path, 'perthAll_stage_250m_all.asc')
        MaxAsc('perth.out.asc', [in_file])
        self.failUnless(FilesEqual('perth.out.asc', in_file))

    def test_same_input_equals_bigC(self):
        test_path = aust.get_pathname_from_package('anuga.lib.maxasc')
        in_file = os.path.join(test_path, 'perthAll_stage_original.asc')
        MaxAsc('perth.out.asc', [in_file])
        self.failUnless(FilesEqual('perth.out.asc', in_file))

    def test_same_input_equals_big3(self):
        test_path = aust.get_pathname_from_package('anuga.lib.maxasc')
        in_file = os.path.join(test_path, 'perthAll_stage_250m.asc')
        MaxAsc('perth.out.asc', [in_file, in_file, in_file])
        self.failUnless(FilesEqual('perth.out.asc', in_file))

    def test_same_input_equals_outputN(self):
        test_path = aust.get_pathname_from_package('anuga.lib.maxasc')
        in_file = os.path.join(test_path, 'test1.asc')
        MaxAsc('test1.out.asc', [in_file] * 30)
        self.failUnless(FilesEqual('test1.out.asc', in_file))

    def test_different_input2(self):
        test_path = aust.get_pathname_from_package('anuga.lib.maxasc')
        in_file = os.path.join(test_path, 'test1.asc')
        in_file2 = os.path.join(test_path, 'test2.asc')
        expected_file = os.path.join(test_path, 'test2.expected.asc')
        MaxAsc('test2.out.asc', [in_file, in_file2])
        self.failUnless(FilesEqual('test2.out.asc', expected_file))

    def test_different_input3(self):
        test_path = aust.get_pathname_from_package('anuga.lib.maxasc')
        in_file = os.path.join(test_path, 'test1.asc')
        in_file2 = os.path.join(test_path, 'test2.asc')
        in_file3 = os.path.join(test_path, 'test3.asc')
        expected_file = os.path.join(test_path, 'test3.expected.asc')
        MaxAsc('test3.out.asc', [in_file, in_file2, in_file3])
        self.failUnless(FilesEqual('test3.out.asc', expected_file))

if __name__ == '__main__':
    suite = unittest.makeSuite(Test_MaxAsc,'test')
    runner = unittest.TextTestRunner(verbosity=1)
    runner.run(suite)