From 62f118dd40c12471b98b6eec555c8a1ad54dc696 Mon Sep 17 00:00:00 2001
From: andreas128 <Andreas>
Date: Fri, 22 Sep 2017 16:31:10 +0200
Subject: Set small phases offsets to zero

---
 dpd/src/Model_PM.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/dpd/src/Model_PM.py b/dpd/src/Model_PM.py
index 8d35051..eab5ec5 100644
--- a/dpd/src/Model_PM.py
+++ b/dpd/src/Model_PM.py
@@ -77,8 +77,10 @@ class Model_PM:
             plt.close(fig)
 
     def _discard_small_values(self, tx_dpd, phase_diff):
-        mask = tx_dpd > self.c.MPM_tx_min
-        return tx_dpd[mask], phase_diff[mask]
+        """ Assumes that the phase for small tx amplitudes is zero"""
+        mask = tx_dpd < self.c.MPM_tx_min
+        phase_diff[mask] = 0
+        return tx_dpd, phase_diff
 
     def poly(self, sig):
         return np.array([sig ** i for i in range(0, 5)]).T
-- 
cgit v1.2.3