diff options
| -rw-r--r-- | dpd/src/Model.py | 24 |
1 files changed, 24 insertions, 0 deletions
diff --git a/dpd/src/Model.py b/dpd/src/Model.py index a7bc63b..78445cf 100644 --- a/dpd/src/Model.py +++ b/dpd/src/Model.py @@ -1,5 +1,9 @@ # -*- coding: utf-8 -*- +import numpy as np +import logging +import matplotlib.pyplot as plt + class Model: """Calculates new coefficients using the measurement and the old coefficients""" @@ -8,6 +12,26 @@ class Model: self.coefs = coefs def get_next_coefs(self, txframe_aligned, rxframe_aligned): + txframe_aligned = txframe_aligned / np.median(np.abs(txframe_aligned)) + rxframe_aligned = rxframe_aligned / np.median(np.abs(rxframe_aligned)) + + logging.debug("txframe: min %f, max %f, median %f" % + (np.min(np.abs(txframe_aligned)), + np.max(np.abs(txframe_aligned)), + np.median(np.abs(txframe_aligned)) + )) + + logging.debug("rxframe: min %f, max %f, median %f" % + (np.min(np.abs(rxframe_aligned)), + np.max(np.abs(rxframe_aligned)), + np.median(np.abs(rxframe_aligned)) + )) + + plt.plot(rxframe_aligned[:1024], label="rxframe") + plt.plot(txframe_aligned[:1024], label="txframe") + plt.legend() + plt.savefig("/tmp/") + return self.coefs # The MIT License (MIT) |