#!/usr/bin/python                                                                                 
# -*- coding: iso-8859-1 -*-                                                                      

'''
    Implementation of random background adding to a specific image

    Author: Guillaume Sicard
'''

import sys, os, random, fnmatch
import Image, numpy           

class AddBackground():
    def __init__(self, threshold = 128):
        self.h = 32
        self.w = 32
        self.threshold = threshold;
        self.bg_image_dir = '/data/lisa/data/ift6266h10/image_net/'
        self.pattern = '*.JPEG'
        self.image_files = fnmatch.filter(os.listdir(self.bg_image_dir),self.pattern)

    # get threshold value
    def get_settings_names(self):
        return [str(self.threshold)]

    # no need, except for testmod.py
    def regenerate_parameters(self, complexity):
        value = random.gauss(0, 0.5*complexity)
        return [value]

    # load an image
    def load_image(self,filename):
        image = Image.open(filename).convert('L')
        image = numpy.asarray(image)
        image = (image / 255.0).astype(numpy.float32)
        return image

    # save an image
    def save_image(self,array, filename):
        image = (array * 255.0).astype('int')
        image = Image.fromarray(image)
        if (filename != ''):
            image.save(filename)
        else:
            image.show()

    # make a random 32x32 crop of an image
    def rand_crop(self,image):
        i_w, i_h = image.shape
        x, y = random.randint(0, i_w - self.w), random.randint(0, i_h - self.h)
        return image[x:x + self.w, y:y + self.h]

    # select a random background image from "bg_image_dir" and crops it
    def rand_bg_image(self):
        i = random.randint(0, len(self.image_files) - 1)

        image = self.load_image(self.bg_image_dir + self.image_files[i])
        self.bg_image = self.rand_crop(image)

    # set "bg_image" as background to "image", based on a pixels threshold
    def set_bg(self,image):
        b = (image < self.threshold / 255.0).astype(numpy.float32)
        return b * self.bg_image + ( 1 - b) * image

    # transform an image file and return an array
    def transform_image_from_file(self, filename):
        self.rand_bg_image()
        image = self.load_image(filename)
        image = self.set_bg(image)
        return image

    # standard array to array transform
    def transform_image(self, image):
        self.rand_bg_image()
        image = self.set_bg(image)
        return image

    # test method
    def test(self,filename):
        import time

        sys.stdout.write('Starting addBackground test : loading image')
        sys.stdout.flush()

        image = self.load_image(filename)

        t = 0
        n = 500
        for i in range(n):
            t0 =  time.time()
            image2 = self.transform_image(image)
            t = ( i * t + (time.time() - t0) ) / (i + 1)
            sys.stdout.write('.')
            sys.stdout.flush()
            
        print "Done!\nAverage time : " + str(1000 * t) + " ms"

if __name__ == '__main__':

    myAddBackground = AddBackground()
    myAddBackground.test('./images/0-LiberationSans-Italic.ttf.jpg')