import sys
import os.path

import numpy as np
from numpy.testing import *
from numpy.testing.decorators import skipif

from skimage import data_dir
from skimage.io import ImageCollection, MultiImage
from skimage.io.collection import alphanumeric_key
from skimage.io import Image as ioImage


try:
    from PIL import Image
except ImportError:
    PIL_available = False
else:
    PIL_available = True

if sys.version_info[0] > 2:
    basestring = str

class TestAlphanumericKey():
    def setUp(self):
        self.test_string = 'z23a'
        self.test_str_result = ['z', 23, 'a']
        self.filenames = ['f9.10.png', 'f9.9.png', 'f10.10.png', 'f10.9.png',
            'e9.png', 'e10.png', 'em.png']
        self.sorted_filenames = \
            ['e9.png', 'e10.png', 'em.png', 'f9.9.png', 'f9.10.png',
            'f10.9.png', 'f10.10.png']

    def test_string_split(self):
        assert_equal(alphanumeric_key(self.test_string), self.test_str_result)

    def test_string_sort(self):
        sorted_filenames = sorted(self.filenames, key=alphanumeric_key)
        assert_equal(sorted_filenames, self.sorted_filenames)


class TestImageCollection():
    pattern = [os.path.join(data_dir, pic) for pic in ['camera.png',
                                                       'color.png']]
    pattern_matched = [os.path.join(data_dir, pic) for pic in 
                                                    ['camera.png', 'moon.png']]

    def setUp(self):
        self.collection = ImageCollection(self.pattern)
        self.collection_matched = ImageCollection(self.pattern_matched)

    def test_len(self):
        assert len(self.collection) == 2

    def test_getitem(self):
        num = len(self.collection)
        for i in range(-num, num):
            assert type(self.collection[i]) is ioImage
        assert_array_almost_equal(self.collection[0],
                                  self.collection[-num])

        #assert_raises expects a callable, hence this do-very-little func
        def return_img(n):
            return self.collection[n]
        assert_raises(IndexError, return_img, num)
        assert_raises(IndexError, return_img, -num - 1)

    def test_slicing(self):
        assert type(self.collection[:]) is ImageCollection
        assert len(self.collection[:]) == 2
        assert len(self.collection[:1]) == 1
        assert len(self.collection[1:]) == 1
        assert_array_almost_equal(self.collection[0], self.collection[:1][0])
        assert_array_almost_equal(self.collection[1], self.collection[1:][0])
        assert_array_almost_equal(self.collection[1], self.collection[::-1][0])
        assert_array_almost_equal(self.collection[0], self.collection[::-1][1])

    def test_files_property(self):
        assert isinstance(self.collection.files, list)

        def set_files(f):
            self.collection.files = f
        assert_raises(AttributeError, set_files, 'newfiles')

    def test_custom_load(self):
        load_pattern = [(1, 'one'), (2, 'two')]

        def load_fn(x):
            return x

        ic = ImageCollection(load_pattern, load_func=load_fn)
        assert_equal(ic[1], (2, 'two'))

    def test_concatenate(self):
        ar = self.collection_matched.concatenate()
        assert_equal(ar.shape, (len(self.collection_matched),) + 
                                self.collection[0].shape)
        assert_raises(ValueError, self.collection.concatenate)


class TestMultiImage():

    def setUp(self):
        # This multipage TIF file was created with imagemagick:
        # convert im1.tif im2.tif -adjoin multipage.tif
        if PIL_available:
            self.img = MultiImage(os.path.join(data_dir, 'multipage.tif'))

    @skipif(not PIL_available)
    def test_len(self):
        assert len(self.img) == 2

    @skipif(not PIL_available)
    def test_getitem(self):
        num = len(self.img)
        for i in range(-num, num):
            assert type(self.img[i]) is np.ndarray
        assert_array_almost_equal(self.img[0],
                                  self.img[-num])

        #assert_raises expects a callable, hence this do-very-little func
        def return_img(n):
            return self.img[n]
        assert_raises(IndexError, return_img, num)
        assert_raises(IndexError, return_img, -num - 1)

    @skipif(not PIL_available)
    def test_files_property(self):
        assert isinstance(self.img.filename, basestring)

        def set_filename(f):
            self.img.filename = f
        assert_raises(AttributeError, set_filename, 'newfile')

    @skipif(not PIL_available)
    def test_conserve_memory_property(self):
        assert isinstance(self.img.conserve_memory, bool)

        def set_mem(val):
            self.img.conserve_memory = val
        assert_raises(AttributeError, set_mem, True)

    @skipif(not PIL_available)
    def test_concatenate(self):
        ar = self.img.concatenate()
        assert_equal(ar.shape, (len(self.img),) + 
                                self.img[0].shape)


if __name__ == "__main__":
    run_module_suite()