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()