Catch arctan2 errors

2013-04-01 16:38:50 -04:00 · 2013-04-01 16:38:50 -04:00 · 348c435d1e
commit 348c435d1e
parent 7f1c1a6c32
1 changed files with 5 additions and 1 deletions
--- a/src/sinefit.py
+++ b/src/sinefit.py
@ -172,7 +172,11 @@ def sfit4(data, fs):
    ## Extract results
    A = sqrt(s[0]*s[0] + s[1]*s[1]) # eqn B.21
    f0 = w / (2*pi)
-    phi = -arctan2(s[1], s[0]) # eqn B.22
+    try:
+        phi = -arctan2(s[1], s[0]) # eqn B.22
+    except TypeError:
+        # something broke down, just return zeros
+        return (0, 0, 0, 0)
    C = s[2]

    return (A, f0, phi, C)