def queue_persistence(self, queue, tdir, qdir):
self.assertEqual(queue.qsize(), 0)
for request in self.fake_requests:
queue.put(request)
self.assertEqual(queue.qsize(), len(self.fake_requests))
del queue
queue = Queue(qdir, tempdir=tdir)
self.assertEqual(queue.qsize(), len(self.fake_requests))
for _ in range(len(self.fake_requests)):
queue.get()
self.assertEqual(queue.qsize(), 0)
def test_ClearOldFile(self):
"""put until reaching chunksize, then get without calling task_done"""
q = Queue(self.path, chunksize=10)
for i in range(15):
q.put('var1')
for i in range(11):
q.get()
q = Queue(self.path, chunksize=10)
self.assertEqual(q.qsize(), 15)
for i in range(11):
q.get()
q.task_done()
self.assertEqual(q.qsize(), 4)
def test_OpenCloseSingle(self):
"""Write 1 item, close, reopen checking if same item is there"""
q = Queue(self.path)
q.put('var1')
del q
q = Queue(self.path)
self.assertEqual(1, q.qsize())
self.assertEqual('var1', q.get())
q.task_done()
def test_PartialWrite(self):
"""Test recovery from previous crash w/ partial write"""
q = Queue(self.path)
for i in range(100):
q.put('var%d' % i)
del q
with open(os.path.join(self.path, 'q00000'), 'ab') as f:
pickle.dump('文字化け', f)
q = Queue(self.path)
self.assertEqual(100, q.qsize())
for i in range(100):
self.assertEqual('var%d' % i, q.get())
q.task_done()
with self.assertRaises(Empty):
q.get_nowait()
def test_GarbageOnHead(self):
"""Adds garbage to the queue head and let the internal integrity
checks fix it"""
q = Queue(self.path)
q.put('var1')
del q
with open(os.path.join(self.path, 'q00001'), 'a') as fd:
fd.write('garbage')
q = Queue(self.path)
q.put('var2')
self.assertEqual(2, q.qsize())
self.assertEqual('var1', q.get())
q.task_done()
def test_OpenCloseOneHundred(self):
"""Write 1000 items, close, reopen checking if all items are there"""
q = Queue(self.path)
for i in range(1000):
q.put('var%d' % i)
del q
q = Queue(self.path)
self.assertEqual(1000, q.qsize())
for i in range(1000):
data = q.get()
self.assertEqual('var%d' % i, data)
q.task_done()
with self.assertRaises(Empty):
q.get_nowait()
# assert adding another one still works
q.put('foobar')
data = q.get()
class MyApp(QMainWindow):
def __init__(self):
super().__init__()
self.ui=Ui_MainWindow()
self.ui.setupUi(self)
#self.Ctl_timer = QTimer()
##self.Ctl_timer.setSingleShot(True)
#self.Ctl_timer.setInterval(70)
#self.Ctl_timer.timeout.connect(self.Ctl_loop)
#self.Ctl_timer.start()
self.N=1024*8
self.fmt_r='Q'*self.N
self.ip="192.168.1.2"
self.port_s=8889
self.port_r=8888
self.Int=500 #unit us
self.Scan=5000000 #unit ms, setting trigger
self.Sim=50 #unit ns, setting simulation
self.counter=0
#UNIT and DATA initiation
self._socket_s = None
self._socket_t = None
self._unit_connected_to = None
self._unit_connected= False
self.q=Queue('tmp',maxsize=self.N*2)
#self.q=Queue('tmp')
self.para_changed()
self.threadpool = QThreadPool()
self.line,=self.ui.widget.canvas.ax.plot([1],[1],'b.',markersize=0.3)
self.ui.widget.canvas.ax.set_ylim(-1,1)
self.ui.widget.canvas.ax.set_xlim(0,100)
self.ui.widget.canvas.draw()
self.ui.statusbar.showMessage(f"Software started")
def para_changed(self):
self.ip=self.ui.IP_line.text()
self.port_s=int(self.ui.Port_s_line.text())
self.port_r=int(self.ui.Port_r_line.text())
self.Int=int(self.ui.Int_line.text())*50
self.Scan=int(self.ui.Scan_line.text())*50000
self.Sim=int(int(self.ui.Sim_line.text())/20)
if self._unit_connected:
self.Pls_control()
def Pls_control(self):
i=0
if i!=1:
i=self.Pls_control_handler()
def Pls_control_handler(self):
fmt_tran='I'*3
mark=0
with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as self._socket_s:
tran_raw=struct.pack(fmt_tran,self.Int,self.Scan,self.Sim)
try:
self._socket_s.connect((self.ip, self.port_s))
self._socket_s.sendall(tran_raw)
self._socket_s.settimeout(1.0)
mark=self._socket_s.recv(4)
if mark==b'':
mark=self._socket_s.recv(4)
return 0
except Exception as e:
self.ui.statusbar.showMessage(f"control data send failed")
self._socket_s.close()
return 0
self.ui.statusbar.showMessage(f"control data sended")
self._socket_s.close()
return 1
def server_connect(self):
if self._unit_connected==False:
try:
self._socket_r= socket.socket(socket.AF_INET, socket.SOCK_STREAM) # instantiate
self._socket_r.connect((self.ip, self.port_r)) # connect to the server
#self._socket_s= socket.socket() # instantiate
#self._socket_s.connect((self.ip, self.port_s)) # connect to the server
self.ui.IP_line.setEnabled(False)
self.ui.Port_s_line.setEnabled(False)
self.ui.Port_r_line.setEnabled(False)
self.ui.Connect_button.setText("Disconnect")
self._unit_connected_to = True
self._unit_connected = True
self.ui.statusbar.showMessage(f"server connected")
self.Pls_control()
except Exception as ex:
self._unit_connected_to =False
self._unit_connected =False
self.ui.statusbar.showMessage(str(ex))
#self.ui.statusbar.showMessage(f"server connect error!")
self.ui.IP_line.setEnabled(True)
self.ui.Port_s_line.setEnabled(True)
self.ui.Port_r_line.setEnabled(True)
self.ui.Connect_button.setText("Connect")
worker_recv = Worker(self.Ctl_loop) # Any other args, kwargs are passed to the run function
worker_plot = Worker(self.plot_loop) # Any other args, kwargs are passed to the run function
#worker.signals.result.connect(self.print_output)
#worker.signals.finished.connect(self.thread_complete)
#worker.signals.progress.connect(self.progress_fn)
self.threadpool.start(worker_recv)
self.threadpool.start(worker_plot)
else:
self.counter=0
self.threadpool.releaseThread()
self.threadpool.releaseThread()
#self._socket_r.close() # instantiate
#self._socket_s.close() # instantiate
self.ui.IP_line.setEnabled(True)
self.ui.Port_s_line.setEnabled(True)
self.ui.Port_r_line.setEnabled(True)
self.ui.Connect_button.setText("Connect")
self._unit_connected_to = False
self._unit_connected = False
self.ui.statusbar.showMessage(f"server disconnectted")
def plot_loop(self):
si=2147483648;
while self._unit_connected:
while self._unit_connected & (self.q.qsize()>1):
x=[]
y=[]
tmp=1
while tmp<=si:
try:
raw=self.q.get_nowait()
except Exception as e:
self.ui.statusbar.showMessage(str(e))
ar=struct.unpack('II',struct.pack('Q',raw))
x_tmp=ar[0]
tmp=ar[1]
x.append(x_tmp)
y.append(tmp)
y[-1]=y[-1]-si
x=np.array(x)*20e-6
self.ui.widget.canvas.ax.set_ylim(0,np.mean(y)*1.2)
self.ui.widget.canvas.ax.set_xlim(0,self.Scan*20e-6)
self.line.set_ydata(y)
self.line.set_xdata(x)
self.ui.widget.canvas.draw()
time.sleep((self.Scan*20e-9)*0.5)
self.ui.widget.canvas.flush_events()
self.ui.statusbar.showMessage(f"data plotting, "+str(self.counter))
self.counter=self.counter+1
#self.Plot_timer.start()
def Ctl_loop(self):
while self._unit_connected:
if self._unit_connected_to:
self.ui.statusbar.showMessage(f"data receiving")
data_raw = self._socket_r.recv(8*self.N,socket.MSG_WAITALL) # receive response
else:
self._socket_r= socket.socket(socket.AF_INET, socket.SOCK_STREAM) # instantiate
self._socket_r.connect((self.ip, self.port_r)) # connect to the server
self.ui.statusbar.showMessage(f"data receiving")
data_raw = self._socket_r.recv(8*self.N,socket.MSG_WAITALL) # receive response
self.ui.statusbar.showMessage(f"data received")
data=struct.unpack(self.fmt_r,data_raw)
#self.plot_data(data)
try:
list(map(self.q.put_nowait,data))
except Exception as e:
self.ui.statusbar.showMessage(str(e))
self._socket_r.close()
self._unit_connected_to=False
#self.Ctl_timer.start()
def plot_data(self,data):
si=2147483648;
x=[]
y=[]
for i in range(self.N):
x.append(data[i*2])
y.append(data[i*2+1])
y=np.array(y)
I=y>si
y[I]=y[I]-si
x=np.array(x)*20e-6
##self.ui.statusbar.showMessage(f"data plotting, "+str(self.counter))
##self.counter=self.counter+1
self.ui.widget.canvas.ax.set_ylim(0,np.max(y)*1.2)
self.ui.widget.canvas.ax.set_xlim(0,self.Scan*20e-6)
##self.line.set_ydata(y)
##self.line.set_xdata(x)
##self.ui.widget.canvas.draw()
##self.ui.widget.canvas.flush_events()
II=np.where(np.diff(x)<0)[0]
III=[0]
for i in II:
III.append(i)
III.append(len(y)-1)
sl=len(II)+1;
print(II)
for i in range(sl):
xx=x[III[i]:III[i+1]]
yy=y[III[i]:III[i+1]]
self.line.set_ydata(y)
self.line.set_xdata(x)
self.ui.widget.canvas.draw()
self.ui.widget.canvas.flush_events()
self.ui.statusbar.showMessage(f"data plotting, "+str(self.counter))
self.counter=self.counter+1