2015-06-05 14:52:41 +08:00
|
|
|
import asyncio
|
|
|
|
import logging
|
|
|
|
import socket
|
|
|
|
import struct
|
2015-06-05 19:11:41 +08:00
|
|
|
from operator import itemgetter
|
2015-06-05 14:52:41 +08:00
|
|
|
|
2015-06-05 19:11:41 +08:00
|
|
|
from quamash import QtGui, QtCore
|
2015-06-05 14:52:41 +08:00
|
|
|
from pyqtgraph import dockarea
|
|
|
|
|
|
|
|
from artiq.tools import TaskObject
|
|
|
|
|
|
|
|
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
|
|
|
2015-06-05 19:11:41 +08:00
|
|
|
class _TTLWidget(QtGui.QFrame):
|
|
|
|
def __init__(self, force_out, name):
|
|
|
|
self.force_out = force_out
|
|
|
|
|
|
|
|
QtGui.QFrame.__init__(self)
|
|
|
|
|
|
|
|
self.setFrameShape(QtGui.QFrame.Panel)
|
|
|
|
self.setFrameShadow(QtGui.QFrame.Raised)
|
|
|
|
|
|
|
|
grid = QtGui.QGridLayout()
|
|
|
|
self.setLayout(grid)
|
|
|
|
label = QtGui.QLabel(name)
|
|
|
|
label.setAlignment(QtCore.Qt.AlignCenter)
|
|
|
|
grid.addWidget(label, 1, 1)
|
|
|
|
|
|
|
|
self._direction = QtGui.QLabel()
|
|
|
|
self._value = QtGui.QLabel()
|
|
|
|
self._direction.setAlignment(QtCore.Qt.AlignCenter)
|
|
|
|
self._value.setAlignment(QtCore.Qt.AlignCenter)
|
|
|
|
self.set_value(0, False, False)
|
|
|
|
grid.addWidget(self._direction, 2, 1)
|
|
|
|
grid.addWidget(self._value, 3, 1, 6, 1)
|
|
|
|
|
|
|
|
def set_value(self, value, oe, override):
|
|
|
|
value = "1" if value else "0"
|
|
|
|
if override:
|
|
|
|
value = "<b>" + value + "</b>"
|
|
|
|
color = " color=\"red\""
|
|
|
|
else:
|
|
|
|
color = ""
|
|
|
|
self._value.setText("<font size=\"9\"{}>{}</font>".format(
|
|
|
|
color, value))
|
|
|
|
oe = oe or self.force_out
|
|
|
|
direction = "OUT" if oe else "IN"
|
|
|
|
self._direction.setText("<font size=\"1\">" + direction + "</font>")
|
|
|
|
|
|
|
|
|
2015-06-05 14:52:41 +08:00
|
|
|
class _DeviceManager:
|
|
|
|
def __init__(self, init):
|
2015-06-05 19:11:41 +08:00
|
|
|
self.ddb = dict()
|
|
|
|
self.ttl_cb = lambda: None
|
|
|
|
self.ttl_widgets = dict()
|
|
|
|
for k, v in init.items():
|
|
|
|
self[k] = v
|
2015-06-05 14:52:41 +08:00
|
|
|
|
|
|
|
def __setitem__(self, k, v):
|
2015-06-05 19:11:41 +08:00
|
|
|
self.ddb[k] = v
|
|
|
|
if k in self.ttl_widgets:
|
|
|
|
del self[k]
|
|
|
|
if not isinstance(v, dict):
|
|
|
|
return
|
|
|
|
try:
|
|
|
|
if v["type"] == "local" and v["module"] == "artiq.coredevice.ttl":
|
|
|
|
channel = v["arguments"]["channel"]
|
|
|
|
force_out = v["class"] == "TTLOut"
|
|
|
|
self.ttl_widgets[channel] = _TTLWidget(force_out, k)
|
|
|
|
self.ttl_cb()
|
|
|
|
except KeyError:
|
|
|
|
pass
|
|
|
|
|
|
|
|
def __delitem__(self, k):
|
|
|
|
del self.ddb[k]
|
|
|
|
if k in self.ttl_widgets:
|
|
|
|
del self.ttl_widgets[k]
|
|
|
|
self.ttl_cb()
|
2015-06-05 14:52:41 +08:00
|
|
|
|
|
|
|
def get_core_addr(self):
|
|
|
|
try:
|
2015-06-05 19:11:41 +08:00
|
|
|
comm = self.ddb["comm"]
|
|
|
|
while isinstance(comm, str):
|
|
|
|
comm = self.ddb[comm]
|
|
|
|
return comm["arguments"]["host"]
|
2015-06-05 14:52:41 +08:00
|
|
|
except KeyError:
|
|
|
|
return None
|
|
|
|
|
|
|
|
|
|
|
|
class MonInjTTLDock(dockarea.Dock, TaskObject):
|
|
|
|
def __init__(self):
|
|
|
|
dockarea.Dock.__init__(self, "TTL", size=(1500, 500))
|
|
|
|
self.dm = _DeviceManager(dict())
|
|
|
|
self.transport = None
|
|
|
|
|
2015-06-05 19:11:41 +08:00
|
|
|
self.grid = QtGui.QGridLayout()
|
|
|
|
gridw = QtGui.QWidget()
|
|
|
|
gridw.setLayout(self.grid)
|
|
|
|
self.addWidget(gridw)
|
|
|
|
|
2015-06-05 14:52:41 +08:00
|
|
|
@asyncio.coroutine
|
|
|
|
def start(self):
|
|
|
|
loop = asyncio.get_event_loop()
|
2015-06-05 19:11:41 +08:00
|
|
|
yield from loop.create_datagram_endpoint(lambda: self,
|
|
|
|
family=socket.AF_INET)
|
2015-06-05 14:52:41 +08:00
|
|
|
TaskObject.start(self)
|
|
|
|
|
|
|
|
@asyncio.coroutine
|
|
|
|
def stop(self):
|
|
|
|
yield from TaskObject.stop(self)
|
|
|
|
if self.transport is not None:
|
|
|
|
self.transport.close()
|
|
|
|
self.transport = None
|
|
|
|
|
|
|
|
def connection_made(self, transport):
|
|
|
|
self.transport = transport
|
|
|
|
|
|
|
|
def datagram_received(self, data, addr):
|
2015-06-05 19:11:41 +08:00
|
|
|
ttl_levels, ttl_oes = struct.unpack(">QQ", data)
|
|
|
|
for channel, w in self.dm.ttl_widgets.items():
|
|
|
|
w.set_value(ttl_levels & (1 << channel),
|
|
|
|
ttl_oes & (1 << channel),
|
|
|
|
False)
|
2015-06-05 14:52:41 +08:00
|
|
|
|
|
|
|
def error_received(self, exc):
|
|
|
|
logger.warning("datagram endpoint error")
|
|
|
|
|
|
|
|
def connection_lost(self, exc):
|
|
|
|
self.transport = None
|
|
|
|
|
|
|
|
@asyncio.coroutine
|
|
|
|
def _do(self):
|
|
|
|
while True:
|
|
|
|
yield from asyncio.sleep(0.2)
|
|
|
|
ca = self.dm.get_core_addr()
|
|
|
|
if ca is None:
|
|
|
|
logger.warning("could not find core device address")
|
|
|
|
elif self.transport is None:
|
|
|
|
logger.warning("datagram endpoint not available")
|
|
|
|
else:
|
|
|
|
# MONINJ_REQ_MONITOR
|
|
|
|
self.transport.sendto(b"\x01", (ca, 3250))
|
|
|
|
|
2015-06-05 19:11:41 +08:00
|
|
|
def layout_ttl_widgets(self):
|
|
|
|
w = self.grid.itemAt(0)
|
|
|
|
while w is not None:
|
|
|
|
self.grid.removeItem(w)
|
|
|
|
w = self.grid.itemAt(0)
|
|
|
|
for i, (_, w) in enumerate(sorted(self.dm.ttl_widgets.items(),
|
|
|
|
key=itemgetter(0))):
|
|
|
|
self.grid.addWidget(w, i // 4, i % 4)
|
|
|
|
|
2015-06-05 14:52:41 +08:00
|
|
|
def init_devices(self, d):
|
|
|
|
self.dm = _DeviceManager(d)
|
2015-06-05 19:11:41 +08:00
|
|
|
self.dm.ttl_cb = self.layout_ttl_widgets
|
|
|
|
self.layout_ttl_widgets()
|
2015-06-05 14:52:41 +08:00
|
|
|
return self.dm
|
|
|
|
|
|
|
|
|
|
|
|
class MonInjDDSDock(dockarea.Dock):
|
|
|
|
def __init__(self):
|
|
|
|
dockarea.Dock.__init__(self, "DDS", size=(1500, 500))
|
|
|
|
|
|
|
|
def init_devices(self, d):
|
|
|
|
return d
|