From 02a0b7419fc5b7aaf18fb9be4ed0888defa77e2d Mon Sep 17 00:00:00 2001 From: andreas128 Date: Wed, 27 Sep 2017 12:49:04 +0200 Subject: Move consts to Const.py; Cleanup --- dpd/apply_adapt_dumps.py | 12 ++++++------ dpd/src/ExtractStatistic.py | 6 ++---- dpd/src/MER.py | 2 +- dpd/src/Measure_Shoulders.py | 4 ++-- dpd/src/Model_Poly.py | 5 ++--- dpd/src/Symbol_align.py | 2 +- dpd/src/Test_data.py | 4 ++-- 7 files changed, 16 insertions(+), 19 deletions(-) diff --git a/dpd/apply_adapt_dumps.py b/dpd/apply_adapt_dumps.py index 7b661c9..790a64a 100755 --- a/dpd/apply_adapt_dumps.py +++ b/dpd/apply_adapt_dumps.py @@ -46,7 +46,7 @@ import src.Agc as Agc import src.TX_Agc as TX_Agc import argparse -import src.const +import src.Const import src.Symbol_align import src.Measure_Shoulders import src.MER @@ -105,20 +105,20 @@ samplerate = cli_args.samplerate searchpath = cli_args.searchpath target_median = cli_args.target_median -c = src.const.const(samplerate, target_median) +c = src.Const.Const(samplerate, target_median, -1, -1, -1) SA = src.Symbol_align.Symbol_align(c) MER = src.MER.MER(c) -MS = src.Measure_Shoulders.Measure_Shoulder(c, plot=False) +MS = src.Measure_Shoulders.Measure_Shoulders(c) meas = Measure.Measure(samplerate, port, num_req) -extStat = ExtractStatistic.ExtractStatistic(c, plot=True) +extStat = ExtractStatistic.ExtractStatistic(c) adapt = Adapt.Adapt(port_rc, coef_path) dpddata = adapt.get_predistorter() if cli_args.lut: - model = Model.Lut(c, plot=True) + model = Model.Lut(c) else: - model = Model.Poly(c, plot=True) + model = Model.Poly(c) adapt.set_predistorter(model.get_dpd_data()) adapt.set_digital_gain(digital_gain) adapt.set_txgain(txgain) 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 = { -- cgit v1.2.3