From 3f678f6d6edf987d3386ae9f8cbfdbda0e56d8a6 Mon Sep 17 00:00:00 2001 From: Laurent Date: Sat, 27 Dec 2025 11:13:36 +0100 Subject: [PATCH] Calm before disaster --- rpi/main.py | 29 +++++++++++++++++++++++++++-- 1 file changed, 27 insertions(+), 2 deletions(-) diff --git a/rpi/main.py b/rpi/main.py index eb500f77..d8ddb946 100644 --- a/rpi/main.py +++ b/rpi/main.py @@ -1,15 +1,38 @@ #!/usr/bin/env python3 +import cv2 from hardware.screen.screen import Screen from hardware.rfid.reader import RfidReader +from services.detection_service import DetectionService from services.mqtt_service import MQTTService screen = Screen() -service = MQTTService("127.0.0.1", 1883) +mqtt_service = MQTTService("127.0.0.1", 1883) +detection_service = DetectionService() def publish(chip_id : int) : screen.displayMessage(str(chip_id)) - service.publish("app", "/board-mate/rfid/scan", str(chip_id), 0) + mqtt_service.publish("app", "/board-mate/rfid/scan", str(chip_id), 0) + +def detect() : + cap = cv2.VideoCapture(0) + cap.set(cv2.CAP_PROP_FRAME_WIDTH, 640) + cap.set(cv2.CAP_PROP_FRAME_HEIGHT, 480) + cap.set(cv2.CAP_PROP_FPS, 30) + + print("Initialized") + if not cap.isOpened(): + print("Error: Could not open camera") + exit() + + cv2.namedWindow("Predictions", cv2.WINDOW_NORMAL) + + while True: + ret, frame = cap.read() + detection_service.get_fen(frame) + if not ret: + print("Error: Failed to grab frame") + break if __name__ == "__main__": screen.enableBackground() @@ -18,6 +41,8 @@ if __name__ == "__main__": reader.subscribe(publish) reader.start() + detect() + while True: pass