yair/gst-plugin-linescan
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

opencv optional (100fps max)

Browse Source
This commit is contained in:
yair 2025-11-14 18:10:52 +02:00
parent 25f32bf8e9
commit fa3dbdef38
1 changed files with 57 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
scripts/recv_raw_rolling.py
View File

@ -11,13 +11,28 @@
# scripts/go/main.go - Significantly faster, handles 200+ fps easily # scripts/go/main.go - Significantly faster, handles 200+ fps easily
# Build: Run scripts/build_go_receiver.ps1 (Windows) or scripts/build_go_receiver.sh (Linux/macOS) # Build: Run scripts/build_go_receiver.ps1 (Windows) or scripts/build_go_receiver.sh (Linux/macOS)
# See scripts/go/README.md for setup instructions # See scripts/go/README.md for setup instructions
#
# Usage:
# python recv_raw_rolling.py # With OpenCV display (default)
# python recv_raw_rolling.py --no-display # Stats only, no display (max performance)
import socket import socket
import numpy as np import numpy as np
import cv2
import time import time
import argparse
from collections import deque from collections import deque
# Parse command-line arguments
parser = argparse.ArgumentParser(description='Receive raw column stream via UDP')
parser.add_argument('--no-display', action='store_true',
help='Disable OpenCV display for maximum performance (stats only)')
args = parser.parse_args()
# Import OpenCV only if display is enabled
ENABLE_DISPLAY = not args.no_display
if ENABLE_DISPLAY:
import cv2
# Debug flag - set to True to see frame reception details # Debug flag - set to True to see frame reception details
DEBUG = False DEBUG = False
@ -30,7 +45,10 @@ STATUS_INTERVAL = 100 # Print status every N frames
# Rotation mode - set to rotate incoming data before display # Rotation mode - set to rotate incoming data before display
# Options: None, cv2.ROTATE_90_CLOCKWISE, cv2.ROTATE_90_COUNTERCLOCKWISE, cv2.ROTATE_180 # Options: None, cv2.ROTATE_90_CLOCKWISE, cv2.ROTATE_90_COUNTERCLOCKWISE, cv2.ROTATE_180
ROTATION = cv2.ROTATE_90_COUNTERCLOCKWISE # Rotate rows to columns if ENABLE_DISPLAY:
ROTATION = cv2.ROTATE_90_COUNTERCLOCKWISE # Rotate rows to columns
else:
ROTATION = None
# Stream parameters (match your GStreamer sender) # Stream parameters (match your GStreamer sender)
COLUMN_WIDTH = 4 # Width from 200fps-2456x4pix-cw.ini COLUMN_WIDTH = 4 # Width from 200fps-2456x4pix-cw.ini
@ -51,15 +69,19 @@ sock.setsockopt(socket.SOL_SOCKET, socket.SO_RCVBUF, 16777216) # 16MB buffer
sock.bind((UDP_IP, UDP_PORT)) sock.bind((UDP_IP, UDP_PORT))
print(f"Receiving raw {COLUMN_WIDTH}x{COLUMN_HEIGHT} RGB columns on UDP port {UDP_PORT}") print(f"Receiving raw {COLUMN_WIDTH}x{COLUMN_HEIGHT} RGB columns on UDP port {UDP_PORT}")
print(f"Display width: {DISPLAY_WIDTH} pixels (rolling)") if ENABLE_DISPLAY:
print(f"Display: ENABLED - Rolling display ({DISPLAY_WIDTH}x{DISPLAY_HEIGHT})")
else:
print(f"Display: DISABLED - Stats only mode (max performance)")
if DEBUG: if DEBUG:
print(f"Expected frame size: {FRAME_SIZE} bytes") print(f"Expected frame size: {FRAME_SIZE} bytes")
cv2.namedWindow("Rolling Column Stream", cv2.WINDOW_NORMAL) # Initialize display if enabled
if ENABLE_DISPLAY:
cv2.namedWindow("Rolling Column Stream", cv2.WINDOW_NORMAL)
rolling_buffer = np.zeros((DISPLAY_HEIGHT, DISPLAY_WIDTH, CHANNELS), dtype=np.uint8)
current_column = 0
# Initialize the rolling buffer
rolling_buffer = np.zeros((DISPLAY_HEIGHT, DISPLAY_WIDTH, CHANNELS), dtype=np.uint8)
current_column = 0
frame_count = 0 frame_count = 0
# Line drop detection state # Line drop detection state
@ -111,6 +133,7 @@ while True:
print(status) print(status)
if ENABLE_DISPLAY:
# Parse the incoming data # Parse the incoming data
frame = np.frombuffer(data, dtype=np.uint8).reshape((COLUMN_WIDTH, COLUMN_HEIGHT, CHANNELS)) frame = np.frombuffer(data, dtype=np.uint8).reshape((COLUMN_WIDTH, COLUMN_HEIGHT, CHANNELS))
@ -134,5 +157,9 @@ while True:
if cv2.waitKey(1) == 27: # ESC to quit if cv2.waitKey(1) == 27: # ESC to quit
break break
else:
# No display mode - just validate the data can be reshaped
frame = np.frombuffer(data, dtype=np.uint8).reshape((COLUMN_WIDTH, COLUMN_HEIGHT, CHANNELS))
cv2.destroyAllWindows() if ENABLE_DISPLAY:
cv2.destroyAllWindows()