88 lines
1.8 KiB
Python
88 lines
1.8 KiB
Python
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()
|