From 7909267b8839bdae970060f69c8ba084c7243a2e Mon Sep 17 00:00:00 2001
From: Harry Ho <hh@m-labs.hk>
Date: Mon, 22 Feb 2021 17:43:48 +0800
Subject: [PATCH] Add RP data collection/plotting scripts

* rp_get_sayma_data.py: to be installed on Creotech remote.
* plot_sayma_data.py: to be installed on local.
---
 plot_sayma_data.py   | 37 ++++++++++++++++++++++++++
 rp_get_sayma_data.py | 63 ++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 100 insertions(+)
 create mode 100644 plot_sayma_data.py
 create mode 100644 rp_get_sayma_data.py

diff --git a/plot_sayma_data.py b/plot_sayma_data.py
new file mode 100644
index 0000000..e0b5c47
--- /dev/null
+++ b/plot_sayma_data.py
@@ -0,0 +1,37 @@
+import numpy as np
+import matplotlib.pyplot as plot
+from scipy import signal, constants
+
+def rp_raw_to_numpy(rp_raw):
+    # Convert raw buffer strings to numpy arrays
+    buff_string = rp_raw.split(',')
+    return np.array(list(map(float, buff_string)))
+
+def main():
+    y1_raw = None
+    y2_raw = None
+
+    with open('rp_y1_raw.bin', 'rb') as f:
+        y1_raw = f.read().decode('utf-8')
+    with open('rp_y2_raw.bin', 'rb') as f:
+        y2_raw = f.read().decode('utf-8')
+    if None in [y1_raw, y2_raw]:
+        raise IOError("Raw RP string files cannot be opened.")
+
+    y1 = rp_raw_to_numpy(y1_raw)
+    y2 = rp_raw_to_numpy(y2_raw)
+
+    t = np.arange(y1.shape[0])/125e6
+    y = np.c_[y1, y2].T
+    z = signal.decimate(y*np.exp(1j*2*np.pi*9e6*t), q=10, ftype="fir", zero_phase=True)[:, 10:]
+    z = signal.decimate(z, q=10, ftype="fir", zero_phase=True)[:, 10:]
+    angle = np.angle(np.mean(z[0]*z[1].conj()))
+
+    print(angle)
+
+    plot.plot(y1)
+    plot.plot(y2)
+    plot.show()
+
+if __name__ == "__main__":
+    main()
diff --git a/rp_get_sayma_data.py b/rp_get_sayma_data.py
new file mode 100644
index 0000000..7684eca
--- /dev/null
+++ b/rp_get_sayma_data.py
@@ -0,0 +1,63 @@
+import socket
+
+
+class RPSCPI:
+    def connect(self, host):
+        self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+        self.sock.connect((host, 5000))
+        self.sock_f = self.sock.makefile()
+
+    def close(self):
+        self.sock_f.close()
+        self.sock.close()
+
+    def sendmsg(self, msg):
+        self.sock.send(msg.encode() + b"\r\n")
+
+    def recvmsg(self):
+        return self.sock_f.readline().strip()
+
+    def trigger(self):
+        self.sendmsg("ACQ:START")
+        self.sendmsg("ACQ:TRIG NOW")
+        while True:
+            self.sendmsg("ACQ:TRIG:STAT?")
+            if self.recvmsg() == "TD":
+                break
+
+    def get_data(self, channel):
+        self.sendmsg("ACQ:SOUR{}:DATA?".format(channel))
+        return self.recvmsg()[1:-1]
+
+def main():
+    rp = RPSCPI()
+    rp.connect("192.168.1.104")
+    try:
+        rp.trigger()
+        y1_raw = rp.get_data(1)
+        y2_raw = rp.get_data(2)
+
+        with open('rp_y1_raw.bin', 'wb') as f:
+            f.write(y1_raw.encode('utf-8'))
+            print("Succesfully written y1 raw string from RP.")
+        with open('rp_y2_raw.bin', 'wb') as f:
+            f.write(y2_raw.encode('utf-8'))
+            print("Succesfully written y2 raw string from RP.")
+
+        # DEBUGGING
+        y1 = [float(i) for i in y1_raw.split(',')]
+        y2 = [float(i) for i in y2_raw.split(',')]
+        with open('rp_y1_raw.txt', 'w') as f:
+            for i in y1:
+                f.write(str(i) + '\n')
+            print("[DEBUG] Succesfully written y1 human-readable data.")
+        with open('rp_y2_raw.txt', 'w') as f:
+            for i in y2:
+                f.write(str(i) + '\n')
+            print("[DEBUG] Succesfully written y2 human-readable data.")
+
+    finally:
+        rp.close()
+
+if __name__ == "__main__":
+    main()