# coding=utf-8
__author__ = 'snomy'


import os
import shutil
import Image
import ImageFilter
import zipfile
import sys


# ---------------------- 函数 --------------------------
def isFullBlack(xy, image):
    point_cnt = 0
    xy_left = (xy[0]-1, xy[1]+0)
    xy_right = (xy[0]+1, xy[1]+0)
    xy_up = (xy[0]+0, xy[1]-1)
    xy_bottom = (xy[0]+0, xy[1]+1)


    lst = [xy, xy_left, xy_right, xy_up, xy_bottom]
    for i in lst:
        xxx = i[0]
        yyy = i[1]
        if i[0] < 0:
            xxx = 0
        if i[0] >= image.size[0]:
            xxx =  xy[0]
        if i[1] < 0:
            yyy = 0
        if i[1] >= image.size[1]:
            yyy = xy[1]


        if image.getpixel((xxx, yyy)) != (0, 0, 0, 255):
            point_cnt += 1
    return point_cnt


# ---------------------- 程序执行 --------------------------
# 步进多少像数分割图片
MOVE_PIX = 32


# 判断logo.png是否存在，不存在退出
if not os.path.exists("logo.png"):
    print "logo.png is not existent"
    sys.exit()


# 判断guang.png是否存在，不存在退出
if not os.path.exists("guang.png"):
    print "guang.png is not existent"
    sys.exit()


# 1. 输入每秒显示几张
if len(sys.argv) >= 2:
    pics_per_second = eval(sys.argv[1])
    if pics_per_second <= 0:
        print "too small"
        sys.exit()
    if pics_per_second > 60:
        print "too large"
        sys.exit()
    if 1<= pics_per_second <= 60:
        print pics_per_second, "frame/s"
else:
    while True:
        pics_per_second = input("frame/s: ")
        if pics_per_second <= 0:
            print "too small"
        if pics_per_second > 60:
            print "too large"
        if 1<= pics_per_second <= 60:
            break


Im = Image.open("logo.png")
print "logo.png"
print Im.mode, Im.size, Im.format


Im_guang = Image.open("guang.png")
print "guang.png"
print Im_guang.mode, Im_guang.size, Im_guang.format


# 先删除文件夹
if os.path.exists("part0"):
    shutil.rmtree("part0")
# 创建文件夹
os.mkdir("part0")


# 判断文件是否存在
if os.path.exists("desc.txt"):
    os.remove("desc.txt")


# 2. 生成 desc.txt 文件
hDescFile = open("desc.txt", mode='w')
hDescFile.write("%d %d %d\n" % (Im.size[0], Im.size[1], pics_per_second))
hDescFile.write("p 0 0 part0\n")
hDescFile.close()


# 2.5 判断是否存在掩膜图片，没有则生成掩膜图片，有则使用现有的
img_mask = Image.new('RGBA', Im.size)
if not os.path.exists("mask.png"):
    print "mask.png is not existent"
    img_mask.paste(Im, (0, 0))
    for height in range(img_mask.size[1]):
        for width in range(img_mask.size[0]):
            #if img_mask.getpixel((width, height)) != (0, 0, 0, 255):
            cnt = isFullBlack((width, height), Im)
            if cnt == 5:
                img_mask.putpixel((width, height), ((0, 0, 0, 0)))
            elif cnt == 4:
                img_mask.putpixel((width, height), ((0, 0, 0, 50)))
            elif cnt == 3:
                img_mask.putpixel((width, height), ((0, 0, 0, 100)))
            elif cnt == 2:
                img_mask.putpixel((width, height), ((0, 0, 0, 150)))
            elif cnt == 1:
                img_mask.putpixel((width, height), ((0, 0, 0, 200)))
    #img_mask.save("img_mask.png")
else:
    print "use mask.png"
    img_mask = Image.open("mask.png")
    print img_mask.mode, img_mask.size, img_mask.format


# 3. 生成 每张图片
index = 0
while True:
    if MOVE_PIX*index > Im.size[0]:
        break
    new_img = Image.new('RGB', Im.size)
    new_img.paste(Im, (0, 0))
    new_img.paste(Im_guang, (MOVE_PIX*index, 0), Im_guang)
    new_img.paste(img_mask, (0, 0), img_mask)
    save_path = "part0/%04d.png" % index
    print save_path
    new_img.save(save_path, "PNG")
    index += 1
#end while


# 4. 生成zip文件
if os.path.exists("bootanimation.zip"):
    os.remove("bootanimation.zip")


zpfd = zipfile.ZipFile("bootanimation.zip", mode='w', compression=zipfile.ZIP_STORED)
zpfd.write("desc.txt")
startdir = "part0"
for dirpath, dirnames, filenames in os.walk(startdir):
    for filename in filenames:
        zpfd.write(os.path.join(dirpath, filename))
zpfd.close()
print " -------- make bootanimation.zip OK --------"


# 5. 清理文件
if os.path.exists("part0"):
    shutil.rmtree("part0")


if os.path.exists("desc.txt"):
    os.remove("desc.txt")