# Requirements: numpy, scipy, Pillow,
from __future__ import print_function
import sys
import numpy as np
from scipy import ndimage
from PIL import Image, ImageFilter, ImageChops
import math
from os import listdir
from os.path import isdir, isfile

generate_blueprint_sprite_v = np.array([[0, 0, 0],
                            [0, 1, 0],
                            [0, 0, -1]])

generate_blueprint_sprite_h = np.array([[0, 0, 0],
                            [0, 0, 1],
                            [0, -1, 0]])


def rgb2gray(rgb):
    return np.dot(rgb[..., :3], [0.2989, 0.5870, 0.1140])

def process_image(data, outfilename, src_image):
    img = Image.fromarray(np.asarray(
        np.clip(data, 0, 255), dtype="uint8"), "L")
    dest = Image.new("RGBA", (img.width, img.height))
    src = img.load()
    dst = dest.load()

    realSrc = src_image.load()
    mask = src_image.filter(ImageFilter.GaussianBlur(10)).load()
    orig = src_image.load()

    # isWire = "wire" in outfilename
    isWire = False

    targetR = 104
    targetG = 200
    targetB = 255

    if isWire:
        targetR = 255
        targetG = 104
        targetB = 232 

    for x in range(img.width):
        for y in range(img.height):
            realpixl = realSrc[x, y]
            greyval = float(src[x, y])
            greyval = min(255.0, greyval)
            greyval = math.pow(
                min(1, float(greyval / 255.0 * 1)), 1.5) * 255.0 * 1
            greyval = max(0, greyval)
            alpha = mask[x, y][3] / 255.0 * 1

            edgeFactor = src[x, y] / 255.0
            noEdge = 1 - edgeFactor

            shadow = min(1, 1 - realpixl[3] / 255.0 - edgeFactor)
            noShadow = 1 - shadow

            dst[x, y] = (
                min(255, int((realpixl[0] / 255.0 * 0.4 + 0.6) * targetR * 1.1)),
                min(255, int((realpixl[1] / 255.0 * 0.4 + 0.6) * targetG * 1.1)),
                min(255, int((realpixl[2] / 255.0 * 0.4 + 0.6) * targetB * 1.1)),
                min(255, int(float(realpixl[3]) * (0.6 + 5 * edgeFactor))))


    dest.save(outfilename)


def generate_blueprint_sprite(infilename, outfilename):
    print("Processing", infilename)
    img = Image.open(infilename)
    img.load()
    img = img.filter(ImageFilter.GaussianBlur(0.5))

    image = rgb2gray(np.asarray(img, dtype="int32"))
    vertical = ndimage.convolve(image, generate_blueprint_sprite_v)
    horizontal = ndimage.convolve(image, generate_blueprint_sprite_h)
    output_image = np.sqrt(np.square(horizontal) + np.square(vertical))
    process_image(output_image, outfilename, img)


buildings = listdir("buildings")

for buildingId in buildings:
    if "hub" in buildingId:
        continue
    if "wire-" in buildingId:
        continue
    generate_blueprint_sprite("buildings/" + buildingId + "", "blueprints/" + buildingId + "")