def install_and_import(package, package_name=None):
    if package_name is None:
        package_name = package

    import importlib
    try:
        importlib.import_module(package)
    except ImportError:
        import pip
        pip.main(['install', package_name])
    finally:
        globals()[package] = importlib.import_module(package)

install_and_import("Serial", "pyserial")
install_and_import("PIL", "pillow")
install_and_import("tqdm")
install_and_import("numpy")
install_and_import("scipy")


from serial import Serial
import serial.tools.list_ports
from tqdm import tqdm

from PIL import Image

from scipy.signal import convolve2d
import numpy as np

IMAGE_NAME2="test2.png"
IMAGE_NAME1="test1.png"

BASYS3_PID=0x6010
BASYS3_VID=0x0403

IMG_HEIGHT=256
IMG_WIDTH=256

dev=""
for port in serial.tools.list_ports.comports():
    if(port.vid==BASYS3_VID and port.pid==BASYS3_PID):
        dev=port.device

if not dev:
    raise RuntimeError("Basys 3 Not Found!")

test_n=int(input("Insert test number (1 or 2): ").strip())

if(test_n not in [1,2]):
    raise RuntimeError("Test numer must be be 1 or 2")

dev=Serial(dev,115200)

img=Image.open(IMAGE_NAME1 if test_n==1 else IMAGE_NAME2)
mat=np.asarray(img,dtype=np.uint8)

mat=mat[:,:,:3]
if(mat.max()>127):
    mat=mat//2

buff=mat.tobytes()

mat=np.sum(mat,axis=2)//3

sim_img=convolve2d(mat,[[-1,-1,-1],[-1,8,-1],[-1,-1,-1]], mode="same")

sim_img[sim_img<0]=0
sim_img[sim_img>127]=127

dev.write(b'\xff')
for i in tqdm(range(IMG_HEIGHT)):
    dev.write(buff[(i)*IMG_WIDTH*3:(i+1)*IMG_WIDTH*3])

dev.write(b'\xf1')
dev.flush()

res=dev.read(IMG_HEIGHT*IMG_WIDTH+2)

res_img=np.frombuffer(res[1:-1],dtype=np.uint8)

res_img=res_img.reshape((IMG_HEIGHT,IMG_WIDTH))


assert np.all(res_img==sim_img), "Image Mismatch!"

im=Image.fromarray(np.uint8(res_img))
im.show()