45 lines
1.3 KiB
Python
45 lines
1.3 KiB
Python
|
from threading import Thread
|
||
|
|
from queue import Queue
|
||
|
|
import time
|
||
|
|
|
||
|
|
from config import SERVER_IP, SERVER_PORT
|
||
|
|
from communication import create_client_socket, receive_image_and_emotions
|
||
|
|
from robot import Robot
|
||
|
|
from own_robot import OwnRobot
|
||
|
|
|
||
|
|
from cli import print_connect, print_disconnect
|
||
|
|
|
||
|
|
|
||
|
|
def update_robot(queue: Queue, robot: Robot):
|
||
|
|
while True:
|
||
|
|
photo, emotions = queue.get()
|
||
|
|
with queue.mutex:
|
||
|
|
queue.queue.clear()
|
||
|
|
robot.update(photo, emotions)
|
||
|
|
time.sleep(robot.update_interval)
|
||
|
|
|
||
|
|
|
||
|
|
def update_from_server(queue: Queue):
|
||
|
|
while True:
|
||
|
|
client_socket = create_client_socket()
|
||
|
|
try:
|
||
|
|
client_socket.connect((SERVER_IP, SERVER_PORT))
|
||
|
|
print_connect()
|
||
|
|
while True:
|
||
|
|
result = receive_image_and_emotions(client_socket)
|
||
|
|
if result is not None:
|
||
|
|
queue.put(result)
|
||
|
|
except Exception: #Normally only if disconnect EOFError
|
||
|
|
print_disconnect()
|
||
|
|
client_socket.close()
|
||
|
|
time.sleep(1)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
data_queue = Queue()
|
||
|
|
own_robot = OwnRobot()
|
||
|
|
update_from_server_thread = Thread(target=update_from_server, args=(data_queue,))
|
||
|
|
update_robot_thread = Thread(target=update_robot, args=(data_queue, own_robot))
|
||
|
|
update_from_server_thread.start()
|
||
|
|
update_robot_thread.start()
|