Fix complex float type

2 files changed, 4 insertions, 1 deletions

diff --git a/src/dab_util.py b/src/dab_util.py
index 3187036..ff26fa5 100644
--- a/src/dab_util.py
+++ b/src/dab_util.py
@@ -81,6 +81,7 @@ def subsample_align(sig1, sig2):
     """
     Returns an aligned version of sig1 and sig2 by cropping and subsample alignment
     """
+    assert(sig2.dtype == 'complex64')
     off_meas = lag_upsampling(sig2, sig1, n_up=1)
     off = int(abs(off_meas))
 
@@ -95,7 +96,9 @@ def subsample_align(sig1, sig2):
         sig1 = sig1[:-1]
         sig2 = sig2[:-1]
 
+    assert(sig2.dtype == 'complex64')
     sig2 = sa.subsample_align(sig2, sig1)
+    assert(sig2.dtype == 'complex64')
     return sig1, sig2
 
 def get_amp_ratio(ampl_1, ampl_2, a_out_abs, a_in_abs):
diff --git a/src/subsample_align.py b/src/subsample_align.py
index 1657131..76ec97f 100755
--- a/src/subsample_align.py
+++ b/src/subsample_align.py
@@ -70,7 +70,7 @@ def subsample_align(sig, ref_sig):
         # Prepare rotate_vec = fft_sig with rotated phase
         rotate_vec = np.exp(1j * best_tau * omega)
         rotate_vec[halflen] = np.cos(np.pi * best_tau)
-        return np.fft.ifft(rotate_vec * fft_sig)
+        return np.fft.ifft(rotate_vec * fft_sig).astype(np.complex64)
     else:
         #print("Could not optimize: " + optim_result.message)
         return np.zeros(0, dtype=np.complex64)