6 files changed, 10 insertions, 13 deletions
diff --git a/dpd/src/ExtractStatistic.py b/dpd/src/ExtractStatistic.py
index 306c401..bf9eba5 100644
--- a/dpd/src/ExtractStatistic.py
+++ b/dpd/src/ExtractStatistic.py
@@ -34,9 +34,7 @@ class ExtractStatistic:
     """Calculate a low variance RX value for equally spaced tx values
     of a predefined range"""
 
-    def __init__(self,
-                 c,
-                 plot=False):
+    def __init__(self, c):
         self.c = c
 
         self.n_meas = 0
@@ -58,7 +56,7 @@ class ExtractStatistic:
         for i in range(c.ES_n_bins):
             self.rx_values.append(None)
 
-        self.plot = plot
+        self.plot = c.ES_plot
 
     def _plot_and_log(self):
         if logging.getLogger().getEffectiveLevel() == logging.DEBUG and self.plot:
diff --git a/dpd/src/MER.py b/dpd/src/MER.py
index 393e55c..69c94f9 100644
--- a/dpd/src/MER.py
+++ b/dpd/src/MER.py
@@ -13,7 +13,7 @@ try:
 except:
     logging_path = "/tmp/"
 
-import src.const
+import src.Const
 import numpy as np
 import matplotlib
 matplotlib.use('agg')
diff --git a/dpd/src/Measure_Shoulders.py b/dpd/src/Measure_Shoulders.py
index acb05b4..710e800 100644
--- a/dpd/src/Measure_Shoulders.py
+++ b/dpd/src/Measure_Shoulders.py
@@ -61,7 +61,7 @@ def shoulder_from_sig_offset(arg):
     return peak-shoulder, peak, shoulder
 
 
-class Measure_Shoulder:
+class Measure_Shoulders:
     """Calculate difference between the DAB signal and the shoulder hight in the
     power spectrum"""
 
@@ -75,7 +75,7 @@ class Measure_Shoulder:
         dt = datetime.datetime.now().isoformat()
         fig_path = logging_path + "/" + dt + "_sync_subsample_aligned.svg"
 
-        fft = calc_fft_db(signal, 100)
+        fft = calc_fft_db(signal, 100, 10)
         peak, idxs_peak = self._calc_peak(fft)
         shoulder, idxs_sh = self._calc_shoulder_hight(fft, self.c)
 
diff --git a/dpd/src/Model_Poly.py b/dpd/src/Model_Poly.py
index e799f1e..6a74bea 100644
--- a/dpd/src/Model_Poly.py
+++ b/dpd/src/Model_Poly.py
@@ -5,15 +5,12 @@
 # http://www.opendigitalradio.org
 # Licence: The MIT License, see notice at the end of this file
 
-import datetime
 import os
 import logging
 
 logging_path = os.path.dirname(logging.getLoggerClass().root.handlers[0].baseFilename)
 
 import numpy as np
-import matplotlib.pyplot as plt
-from sklearn import linear_model
 
 import src.Model_AM as Model_AM
 import src.Model_PM as Model_PM
@@ -24,6 +21,7 @@ def assert_np_float32(x):
     assert x.dtype == np.float32
     assert x.flags.contiguous
 
+
 def _check_input_get_next_coefs(tx_abs, rx_abs, phase_diff):
     assert_np_float32(tx_abs)
     assert_np_float32(rx_abs)
@@ -64,6 +62,7 @@ class Poly:
         return self.coefs_am, self.coefs_pm
 
     def train(self, tx_abs, rx_abs, phase_diff):
+        # type: (np.ndarray, np.ndarray, np.ndarray) -> (str, np.ndarray, np.ndarray)
         _check_input_get_next_coefs(tx_abs, rx_abs, phase_diff)
 
         coefs_am_new = self.model_am.get_next_coefs(tx_abs, rx_abs, self.coefs_am)
diff --git a/dpd/src/Symbol_align.py b/dpd/src/Symbol_align.py
index c9ae239..e21e793 100644
--- a/dpd/src/Symbol_align.py
+++ b/dpd/src/Symbol_align.py
@@ -15,7 +15,7 @@ except:
     logging_path = "/tmp/"
 
 import numpy as np
-import src.const
+import src.Const
 import scipy
 import matplotlib
 matplotlib.use('agg')
diff --git a/dpd/src/Test_data.py b/dpd/src/Test_data.py
index 9dd0913..67f4dff 100644
--- a/dpd/src/Test_data.py
+++ b/dpd/src/Test_data.py
@@ -14,7 +14,7 @@ try:
 except:
     logging_path = "/tmp/"
 
-import src.const
+import src.Const
 import src.Dab_Util
 import numpy as np
 import matplotlib
@@ -40,7 +40,7 @@ class Test_data:
         plt.plot(np.angle(np.fft.fftshift(np.fft.fft(tx_orig))), 'p')
         """
 
-        self.c = src.const.const(sample_rate)
+        self.c = src.Const.Const(sample_rate)
         self.du = src.Dab_Util.Dab_Util(sample_rate)
 
         self.file_paths = {