Do changes for live analysis

2 files changed, 11 insertions, 4 deletions

diff --git a/src/SendDictTcp.py b/src/SendDictTcp.py
index 562e36a..d852fd6 100644
--- a/src/SendDictTcp.py
+++ b/src/SendDictTcp.py
@@ -1,3 +1,4 @@
+import time
 import socket
 import json
 
@@ -5,9 +6,15 @@ class SendDictTcp(object):
     def __init__(self, host, port):
         self.host = host
         self.port = port
-        self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
-        self.sock.connect((self.host, self.port))
-        self.buffer_size = 1024
+        while True:
+            try:
+                self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+                self.sock.connect((self.host, self.port))
+                self.buffer_size = 1024
+                break
+            except:
+                print("Waiting for connecetion to %s:%d" %(self.host, self.port))
+                time.sleep(1)
 
     def send(self, msg):
         self.sock.send(json.dumps(msg))
diff --git a/src/dab_util.py b/src/dab_util.py
index fbd1cc6..fa52015 100644
--- a/src/dab_util.py
+++ b/src/dab_util.py
@@ -11,7 +11,7 @@ c["frame_2048000"]=c["frame_ms"] * 2048
 c["sym_8192000"]=96./76*8192
 c["sym_2048000"]=96./76*2048
 
-def calc_fft(signal, fft_size = 1024, sampling_rate = 1, plot = False):
+def calc_fft(signal, fft_size = 65536, sampling_rate = 8192000, plot = False):
     """returns one numpy array for the frequencies and one for the corresponding fft"""
     signal_spectrum = np.fft.fftshift(np.fft.fft(signal, fft_size))
     freqs = np.fft.fftshift(np.fft.fftfreq(fft_size, d=1./sampling_rate))