1 files changed, 83 insertions, 0 deletions
diff --git a/src/tcp_async.py b/src/tcp_async.py
new file mode 100644
index 0000000..05cadd7
--- /dev/null
+++ b/src/tcp_async.py
@@ -0,0 +1,83 @@
+"""Tcp client for asynchronous uhd message tcp port"""
+
+import threading
+import Queue
+import time
+import socket
+
+class _TcpAsyncClient(threading.Thread):
+    """Thead for message polling"""
+    queue = Queue.Queue()
+    q_quit = Queue.Queue()
+
+    ip_address = None
+    port = None
+    BUFFER_SIZE = 1
+
+    def __init__(self, ip_address, port):
+        super(_TcpAsyncClient, self).__init__()
+        self.ip_address = ip_address
+        self.port = port
+
+    def __exit__(self):
+        self.stop()
+
+    def run(self):
+        """connect and poll messages to queue"""
+
+        #Establish connection
+        sock = None
+        print("Connecting to asynchronous uhd message tcp port " + str(self.port))
+        while self.q_quit.empty():
+            try:
+                sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+                sock.connect((self.ip_address, self.port))
+                break
+            except socket.error:
+                print("connecting to asynchronous uhd message tcp port " + str(self.port))
+                #traceback.print_exc()
+                sock.close()
+                time.sleep(0.5)
+        print("Connected to asynchronous uhd message tcp port " + str(self.port))
+
+        #Read messages
+        sock.settimeout(1)
+        while self.q_quit.empty():
+            try:
+                data = sock.recv(self.BUFFER_SIZE)
+                self.queue.put(data)
+            except socket.timeout:
+                pass
+
+        sock.close()
+
+    def stop(self):
+        """stop thread"""
+        print("stop tcp_async uhd message tcp thread")
+        self.q_quit.put("end")
+
+
+class UhdAsyncMsg(object):
+    """Creates a thread to connect to the asynchronous uhd messages tcp port"""
+
+    def __init__(self, ip_address = "127.0.0.1", port = 47010):
+        self.tcpa = _TcpAsyncClient(ip_address, port)
+        self.tcpa.start()
+
+    def __exit__(self):
+        self.tcpa.stop()
+
+    def stop(self):
+        """stop tcp thread"""
+        self.tcpa.stop()
+
+    def get_res(self):
+        """get received messages as string of integer"""
+        out = ""
+        while not self.tcpa.queue.empty():
+            out += str(ord(self.tcpa.queue.get()))
+        return out
+
+    def has_msg(self):
+        """Checks if one or more messages were received and empties the message queue"""
+        return self.get_res() != ""