#!/usr/bin/env python
# coding: utf-8

import socket
import cv2
import numpy as np
import sys
import time

if(len(sys.argv) != 3):
	print("Usage : {} hostname port".format(sys.argv[0]))
	print("e.g.   {} 192.168.0.39 1080".format(sys.argv[0]))
	sys.exit(-1)


cv2.namedWindow("Image")

# Create a UDP socket
sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
sock.settimeout(1.0)
host = sys.argv[1]
port = int(sys.argv[2])
server_address = (host, port)

while(True):
	try:
		sent = sock.sendto("get".encode('utf-8'), server_address)

		data, server = sock.recvfrom(65507)
		print("Fragment size : {}".format(len(data)))
		if len(data) == 4:
			# This is a message error sent back by the server
			if(data == "FAIL"):
				continue
		array = np.frombuffer(data, dtype=np.dtype('uint8'))
		img = cv2.imdecode(array, 1)
		cv2.imshow("Image", img)
	except socket.timeout:
		pass
	k = cv2.waitKey(1)
	if k == ord('q'):
		print("Asking the server to quit")
		sock.sendto("quit".encode('utf-8'), server_address)
		print("Quitting")
		break
	if k == ord('c'):
		print("Asking the server to restart")
		sock.sendto("clr".encode('utf-8'), server_address)
		print("Clear")
		time.sleep(1)