aboutsummaryrefslogtreecommitdiffstats
path: root/correlate_with_ref.py
diff options
context:
space:
mode:
Diffstat (limited to 'correlate_with_ref.py')
-rwxr-xr-xcorrelate_with_ref.py29
1 files changed, 24 insertions, 5 deletions
diff --git a/correlate_with_ref.py b/correlate_with_ref.py
index ab55682..23c9c5c 100755
--- a/correlate_with_ref.py
+++ b/correlate_with_ref.py
@@ -33,13 +33,27 @@ T_NULL = 2656
T_TF = 196608
class CIR_Correlate:
- def __init__(self, iq_filename, iq_format):
- """Read phase reference from fixed file and load IQ data from
- iq_filename. iq_format must be fc64 or u8"""
+ def __init__(self, iq_filename="", iq_format=None, iq_data=None):
+ """Either call with iq_filename, or with iq_data containing
+ a np.array with the data.
+
+ This class will then read phase reference from fixed file and
+ load IQ data from iq_filename, or use iq_data directly.
+
+ iq_format must be fc64 or u8"""
+
+ if iq_format is None:
+ raise ValueError("Incorrect initialisation")
+
self.phase_ref = np.fromfile("phasereference.2048000.fc64.iq", np.complex64)
if iq_format == "u8":
- channel_u8_interleaved = np.fromfile(iq_filename, np.uint8)
+ if iq_filename:
+ channel_u8_interleaved = np.fromfile(iq_filename, np.uint8)
+ elif iq_data is not None:
+ channel_u8_interleaved = iq_data
+ else:
+ raise ValueError("Must give iq_filename or iq_data")
channel_u8_iq = channel_u8_interleaved.reshape(int(len(channel_u8_interleaved)/2), 2)
# This directly converts to fc64
channel_fc64_unscaled = channel_u8_iq[...,0] + np.complex64(1j) * channel_u8_iq[...,1]
@@ -47,7 +61,12 @@ class CIR_Correlate:
channel_fc64_dc_comp = channel_fc64_scaled - np.average(channel_fc64_scaled)
self.channel_out = channel_fc64_dc_comp
elif iq_format == "fc64":
- self.channel_out = np.fromfile(iq_filename, np.complex64)
+ if iq_filename:
+ self.channel_out = np.fromfile(iq_filename, np.complex64)
+ elif iq_data is not None:
+ self.channel_out = iq_data
+ else:
+ raise ValueError("Must give iq_filename or iq_data")
else:
raise ValueError("Unsupported format {}".format(iq_format))