syndilights/v2/backend/processing/multiple_tests/network.py

37 lines
1.1 KiB
Python

import jarray
import time
import struct
from data_generator import image2data
def TCPConfigure(cl_class, server, port):
return cl_class(this, server, port)
def receive_config(socket):
max_height = jarray.zeros(4, "b")
max_width = jarray.zeros(4, "b")
max_framerate = jarray.zeros(4, "b")
while True:
available_bytes = socket.available()
if available_bytes > 0:
break
time.sleep(1)
socket.readBytes(max_height)
socket.readBytes(max_width)
socket.readBytes(max_framerate)
max_height = struct.unpack("<i", max_height)[0]
max_width = struct.unpack("<i", max_width)[0]
max_framerate = struct.unpack("<i", max_framerate)[0]
return max_height, max_width, max_framerate
def send_config(socket, height, width, framerate):
socket.write(struct.pack('<i', height))
socket.write(struct.pack('<i', width))
socket.write(struct.pack('<i', framerate))
def send_TCP(socket, data, long_line, type):
if socket is None:
# Dummy mode
return
image2data(data, type, long_line)
socket.write(data)