#Path to Cairo Lib
import os
os.environ["PATH"] = r"D:\Programme\msys64\mingw64\bin" + os.pathsep + os.environ["PATH"]

#Imports
import drawSvg as draw
from PIL import Image
import requests
from io import BytesIO
import numpy as np
import webbrowser

#Global Settings
url = "https://upload.wikimedia.org/wikipedia/commons/8/8c/Glockenkelter_Stetten_i.R.jpg"
bmp_size = (3000, 2000)
scan_size = (75, 50)
submove = 2

#Image
response = requests.get(url, headers={"User-Agent": "PythonImage/0.0 (https://www.python.org/; python@noreply.org) generic-library/0.0"})
im = Image.open(BytesIO(response.content))
im = im.resize(bmp_size, Image.ANTIALIAS)
im_byte_arr = BytesIO()
im.save(im_byte_arr, format='PNG')

#Scanned Image
im = im.resize(scan_size, Image.ANTIALIAS)
pix = np.array(im)
pix = np.transpose(pix, (1, 0, 2))

def render(f):
    #Settings / Objects
    size = (620, 860)
    d = draw.Drawing(size[0], size[1], origin=(0, 0), displayInline=False)
    d.append(draw.Rectangle(0, 0, size[0], size[1], fill="#ffffff", fill_opacity=1.0))

    #Calculations
    gap = (600 / pix.shape[0], 400 / pix.shape[1])
    radius = min(gap[0], gap[1]) / 2

    x_move = (gap[0] / submove) * f
    scan_pos = f / submove

    #Image
    d.append(draw.Image(10, 10 + 30, 600, 400, data=im_byte_arr.getvalue(), mimeType="image/png", embed=True))
    d.append(draw.Rectangle(10, 10 + 30, 600, 400, stroke="#000000", stroke_width=2, fill_opacity=0.0))

    #Scanned Image
    d.append(draw.Rectangle(10, 10 + 30 + 400 + 10, 600, 400, stroke="#000000", stroke_width=2, fill_opacity=1.0))
    for x in range(pix.shape[0]):
        for y in range(pix.shape[1]):
            color = (pix[x][y]).astype(int)
            if scan_pos < x - 2: color[0] = 0
            if scan_pos < x - 1: color[1] = 0
            if scan_pos < x: color[2] = 0
            color = tuple(color)
            color = '#%02x%02x%02x' % color
            d.append(draw.Circle(10 + x * gap[0] + radius, 10 + 30 + 400 + 10 - y * gap[1] - radius + 400, radius, fill=color))
    d.append(draw.Rectangle(10, 10 + 30 + 400 + 10, 600, 400, stroke="#000000", stroke_width=2, fill_opacity=0.0))

    #Scanline
    d.append(draw.Rectangle(10 - 5 + x_move - radius, 5, 5 * gap[0], 440, stroke="#000000", stroke_width=2, fill_opacity=0.0))
    for x in range(3):
        color = '#%02x%02x%02x' % [(0, 0, 255), (0, 255, 0), (255, 0, 0)][x]
        d.append(draw.Circle(10 + gap[0] + x * gap[0] + radius - 5 + x_move - radius, 30, radius, fill=color))
        for y in range(pix.shape[1]):
            stroke = 2
            d.append(draw.Rectangle(10 + gap[0] + x * gap[0] + x * stroke - stroke - 5 + x_move - radius, 10 + 30 + y * gap[1], 2 * radius, 2 * radius, stroke=color, stroke_width=stroke, fill_opacity=0.0))
    d.append(draw.Text("Sensor", 10, 10 + gap[0] * 2 + radius - 5 + x_move - radius, 15, center=True))

    #Output
    d.setRenderSize(size[0], size[1])

    return d

#Render to Images
with draw.animate_video("video.tiff", render) as anim:
    for i in range(scan_size[0] * submove):
        anim.draw_frame(i)

#Render to SVG/PNG
#d = render(0)
#d.saveSvg("a.svg")
#d.savePng("a.png")

#View
#webbrowser.open('file://' + os.path.realpath("video.tiff"))
#input()