from tester import *

from PIL import Image
import os

base = os.path.join('Tests', 'images', 'bmp')


def get_files(d, ext='.bmp'):
    return [os.path.join(base,d,f) for f
            in os.listdir(os.path.join(base, d)) if ext in f]

def test_bad():
    """ These shouldn't crash/dos, but they shouldn't return anything either """
    for f in get_files('b'):
        try:
            im = Image.open(f)
            im.load()
        except Exception as msg:
            pass
            # print ("Bad Image %s: %s" %(f,msg))

def test_questionable():
    """ These shouldn't crash/dos, but its not well defined that these are in spec """
    for f in get_files('q'):
        try:
            im = Image.open(f)
            im.load()
        except Exception as msg:
            pass
            # print ("Bad Image %s: %s" %(f,msg))


def test_good():
    """ These should all work. There's a set of target files in the
    html directory that we can compare against. """

    # Target files, if they're not just replacing the extension
    file_map = { 'pal1wb.bmp': 'pal1.png',
                 'pal4rle.bmp': 'pal4.png', 
                 'pal8-0.bmp': 'pal8.png',
                 'pal8rle.bmp': 'pal8.png',
                 'pal8topdown.bmp': 'pal8.png',
                 'pal8nonsquare.bmp': 'pal8nonsquare-v.png',
                 'pal8os2.bmp': 'pal8.png',
                 'pal8os2sp.bmp': 'pal8.png',
                 'pal8os2v2.bmp': 'pal8.png',
                 'pal8os2v2-16.bmp': 'pal8.png',
                 'pal8v4.bmp': 'pal8.png',
                 'pal8v5.bmp': 'pal8.png',
                 'rgb16-565pal.bmp': 'rgb16-565.png',
                 'rgb24pal.bmp': 'rgb24.png',
                 'rgb32.bmp': 'rgb24.png',
                 'rgb32bf.bmp': 'rgb24.png'
                 }     
                 
    def get_compare(f):
        (head, name) = os.path.split(f)
        if name in file_map:
            return os.path.join(base, 'html', file_map[name])
        (name,ext) = os.path.splitext(name)
        return os.path.join(base, 'html', "%s.png"%name)
    
    for f in get_files('g'):
        try:
            im = Image.open(f)
            im.load()
            compare = Image.open(get_compare(f))
            compare.load()
            if im.mode == 'P':
                # assert image similar doesn't really work
                # with paletized image, since the palette might
                # be differently ordered for an equivalent image.
                im = im.convert('RGBA')
                compare = im.convert('RGBA')
            assert_image_similar(im, compare,5)
            
            
        except Exception as msg:
            # there are three here that are unsupported:
            unsupported = (os.path.join(base, 'g', 'rgb32bf.bmp'),
                           os.path.join(base, 'g', 'pal8rle.bmp'),
                           os.path.join(base, 'g', 'pal4rle.bmp'))
            if f not in unsupported:
                assert_true(False, "Unsupported Image %s: %s" %(f,msg))