noptica2-sdr/gui.py

73 lines
2.0 KiB
Python
Raw Normal View History

2020-08-10 01:57:45 +08:00
import os
2020-08-10 18:56:08 +08:00
import sys
import subprocess
import time
2020-08-10 01:57:45 +08:00
import numpy as np
from sipyco import pyon
2020-08-10 18:56:08 +08:00
class ParentComm:
def __init__(self):
self.c_rfd, self.p_wfd = os.pipe()
self.p_rfd, self.c_wfd = os.pipe()
self.rf = open(int(self.p_rfd), "rb", 0)
self.wf = open(int(self.p_wfd), "wb", 0)
self.process = None
def get_address(self):
return "{},{}".format(self.c_rfd, self.c_wfd)
def read(self, n):
return self.rf.read(n)
def readline(self):
return self.rf.readline()
def write(self, data):
return self.wf.write(data)
2020-08-10 01:57:45 +08:00
2020-08-10 18:56:08 +08:00
def write_pyon(self, obj):
self.write(pyon.encode(obj).encode() + b"\n")
def close(self):
self.rf.close()
self.wf.close()
if self.process is not None:
self.process.wait()
def create_subprocess(self, *args):
env = os.environ.copy()
env["PYTHONUNBUFFERED"] = "1"
env["NOPTICA2_IPC"] = self.get_address()
self.process = subprocess.Popen(
*args, pass_fds={self.c_rfd, self.c_wfd},
env=env)
os.close(self.c_rfd)
os.close(self.c_wfd)
2020-08-10 19:22:58 +08:00
class GUI:
2020-08-10 19:55:34 +08:00
def __init__(self, freq_sample, freq_base, block_size):
2020-08-10 19:22:58 +08:00
self.impl = ParentComm()
self.impl.create_subprocess([sys.executable,
2020-08-10 19:28:45 +08:00
os.path.join(os.path.dirname(os.path.abspath(__file__)), "gui_impl.py"),
2020-08-10 19:55:34 +08:00
str(freq_sample), str(freq_base), str(block_size)])
2020-08-10 19:22:58 +08:00
2020-08-11 16:09:24 +08:00
def update_ref(self, block, peak_freq, locked):
obj = {"action": "update_ref", "block": block, "peak_freq": peak_freq, "locked": locked}
self.impl.write_pyon(obj)
def update_meas(self, block):
obj = {"action": "update_meas", "block": block}
2020-08-10 19:22:58 +08:00
self.impl.write_pyon(obj)
2020-08-11 17:48:11 +08:00
def update_position(self, position):
obj = {"action": "update_position", "position": position}
self.impl.write_pyon(obj)
2020-08-10 19:22:58 +08:00
def close(self):
obj = {"action": "terminate"}
self.impl.write_pyon(obj)
self.impl.close()