import cv2
import zmq
import base64
import numpy as np
import sys
from sys import argv


context = zmq.Context()

module_ip = str(sys.argv[1])

print("Connecting to module…")
socket = context.socket(zmq.REQ)
socket.connect("tcp://" + module_ip + ":4444")
socket.send(str.encode(module_ip))

footage_socket = context.socket(zmq.SUB)
footage_socket.bind('tcp://*:5555')
footage_socket.setsockopt_string(zmq.SUBSCRIBE, np.unicode(''))

while True:
    try:
        frame = footage_socket.recv_string()
        img = base64.b64decode(frame)
        npimg = np.fromstring(img, dtype=np.uint8)
        source = cv2.imdecode(npimg, 1)
        cv2.imshow("Stream", source)
        cv2.waitKey(1)

    except KeyboardInterrupt:
        cv2.destroyAllWindows()
        break