108 lines
3.7 KiB
Python
108 lines
3.7 KiB
Python
|
#%%
|
||
|
|
import zmq
|
||
|
|
import time
|
||
|
|
import math
|
||
|
|
|
||
|
|
from communication.communicate_datapacket import DataPacket
|
||
|
|
from communication.test_class_animate import TestClassAnimate
|
||
|
|
|
||
|
|
|
||
|
|
def sender(url, sender_processing, max_processing=math.inf, VERBOSE=False):
|
||
|
|
|
||
|
|
context = zmq.Context()
|
||
|
|
|
||
|
|
socket = context.socket(zmq.REQ)
|
||
|
|
with socket.connect(url):
|
||
|
|
|
||
|
|
try:
|
||
|
|
cur_processing = 0
|
||
|
|
running = True
|
||
|
|
data_in = None
|
||
|
|
while running:
|
||
|
|
|
||
|
|
# MAIN JOB: the following function performs heavy
|
||
|
|
# data processing and sends a result to the receiving process
|
||
|
|
data_out = sender_processing.update(data_in)
|
||
|
|
packet_out = DataPacket(data=data_out, message="sender")
|
||
|
|
socket.send_pyobj(packet_out)
|
||
|
|
|
||
|
|
# CONFIRMATION: Waits indefinitely until receiving a
|
||
|
|
# confirmation data packet. Can be stopped by pressing
|
||
|
|
# CTRL-C, which is handled below as exception
|
||
|
|
if VERBOSE:
|
||
|
|
print("SENDER: Sent, waiting for receiver confirmation...")
|
||
|
|
confirm = False
|
||
|
|
while not confirm:
|
||
|
|
try:
|
||
|
|
packet_in = socket.recv_pyobj(zmq.NOBLOCK)
|
||
|
|
confirm = True
|
||
|
|
except zmq.error.Again:
|
||
|
|
pass
|
||
|
|
if not packet_in.message == "receiver":
|
||
|
|
if VERBOSE:
|
||
|
|
print("SENDER: Something went wrong, exiting!")
|
||
|
|
running = False
|
||
|
|
else:
|
||
|
|
if VERBOSE:
|
||
|
|
print("SENDER: Receiver confirmed")
|
||
|
|
data_in = packet_in.data
|
||
|
|
if data_in:
|
||
|
|
if "exit" in data_in.keys():
|
||
|
|
if VERBOSE:
|
||
|
|
print("Exit received!")
|
||
|
|
raise KeyboardInterrupt
|
||
|
|
|
||
|
|
# CHECK: if maximum iterations reached...
|
||
|
|
cur_processing += 1
|
||
|
|
if max_processing != -1:
|
||
|
|
if cur_processing >= max_processing:
|
||
|
|
raise KeyboardInterrupt
|
||
|
|
|
||
|
|
except KeyboardInterrupt:
|
||
|
|
if VERBOSE:
|
||
|
|
print("SENDER: Someone wants to killlllll meeeeeeeeeeee!")
|
||
|
|
|
||
|
|
# CTRL-C has pressed, try to stop other process
|
||
|
|
if VERBOSE:
|
||
|
|
print("SENDER: Generating and sending packet...")
|
||
|
|
packet_out = DataPacket(message="stop")
|
||
|
|
socket.send_pyobj(packet_out)
|
||
|
|
|
||
|
|
# CONFIRMATION: Loops and waits for confirmation that data was received
|
||
|
|
if VERBOSE:
|
||
|
|
print("SENDER: Waiting for reception confirmation...")
|
||
|
|
confirm = False
|
||
|
|
while not confirm:
|
||
|
|
try:
|
||
|
|
packet_in = socket.recv_pyobj(zmq.NOBLOCK)
|
||
|
|
confirm = True
|
||
|
|
except zmq.error.Again:
|
||
|
|
pass
|
||
|
|
|
||
|
|
if not packet_in.message == "receiver":
|
||
|
|
if VERBOSE:
|
||
|
|
print("SENDER: Something went wrong, but we're exiting anyway!")
|
||
|
|
|
||
|
|
# exit gracefully...
|
||
|
|
if VERBOSE:
|
||
|
|
print("Closing socket and destroying context!")
|
||
|
|
socket.close()
|
||
|
|
context.destroy()
|
||
|
|
if VERBOSE:
|
||
|
|
print("Done with everything!")
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
|
||
|
|
VERBOSE: bool = True
|
||
|
|
url: str = "tcp://localhost:5555" # number specifies communication channel
|
||
|
|
|
||
|
|
class_processing = TestClassAnimate(show_horizontal_bar=True, wait_interval=5)
|
||
|
|
sender(
|
||
|
|
url=url, sender_processing=class_processing, max_processing=3, VERBOSE=VERBOSE
|
||
|
|
)
|
||
|
|
del class_processing
|
||
|
|
|
||
|
|
|
||
|
|
# %%
|