PickleRick/DESD
1
0
Fork 0
Code Issues 2 Pull Requests Releases Activity

Refactor bram_writer and test script: improve code readability, update package installation method, and enhance image processing logic

Browse Source
This commit is contained in:
Davide Cavagnola
2025-04-17 01:24:18 +02:00
parent f363f09506
commit 9bf8c21957
2 changed files with 198 additions and 155 deletions
Download Patch File Download Diff File Expand all files Collapse all files

80
LAB2/test/test.py
View File

@@ -1,87 +1,89 @@
import sys
import subprocess
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])
subprocess.check_call([sys.executable, "-m", "pip", "install", package_name])
finally:
globals()[package] = importlib.import_module(package)
install_and_import("Serial", "pyserial")
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"
IMAGE_NAME2 = "test2.png"
IMAGE_NAME1 = "test1.png"
BASYS3_PID=0x6010
BASYS3_VID=0x0403
BASYS3_PID = 0x6010
BASYS3_VID = 0x0403
IMG_HEIGHT=256
IMG_WIDTH=256
IMG_HEIGHT = 256
IMG_WIDTH = 256
dev=""
dev = ""
for port in serial.tools.list_ports.comports():
if(port.vid==BASYS3_VID and port.pid==BASYS3_PID):
dev=port.device
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())
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")
if test_n not in [1, 2]:
raise RuntimeError("Test number must be 1 or 2")
dev=Serial(dev,115200)
dev = Serial(dev, 115200)
img=Image.open(IMAGE_NAME1 if test_n==1 else IMAGE_NAME2)
mat=np.asarray(img,dtype=np.uint8)
img = Image.open(IMAGE_NAME1 if test_n == 1 else IMAGE_NAME2)
if img.mode != "RGB":
img = img.convert("RGB")
mat = np.asarray(img, dtype=np.uint8)
mat=mat[:,:,:3]
if(mat.max()>127):
mat=mat//2
mat = mat[:, :, :3]
if mat.max() > 127:
mat = mat // 2
buff=mat.tobytes()
buff = mat.tobytes()
mat=np.sum(mat,axis=2)//3
mat_gray = np.sum(mat, axis=2) // 3
sim_img=convolve2d(mat,[[-1,-1,-1],[-1,8,-1],[-1,-1,-1]], mode="same")
sim_img = convolve2d(mat_gray, [[-1, -1, -1], [-1, 8, -1], [-1, -1, -1]], mode="same")
sim_img[sim_img<0]=0
sim_img[sim_img>127]=127
sim_img[sim_img < 0] = 0
sim_img[sim_img > 127] = 127
sim_img = sim_img.astype(np.uint8)
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(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 = dev.read(IMG_HEIGHT * IMG_WIDTH + 2)
res_img=np.frombuffer(res[1:-1],dtype=np.uint8)
res_img = np.frombuffer(res[1:-1], dtype=np.uint8)
res_img = res_img.reshape((IMG_HEIGHT, IMG_WIDTH))
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 = Image.fromarray(res_img)
im.show()
if np.all(res_img != sim_img):
print("Image Mismatch!")
dev.close()