forked from M-Labs/artiq
devices/thorlabs_tcube: remove unnecessary {read,write}_exactly functions
This commit is contained in:
parent
f5187eb140
commit
76eadc0a61
|
@ -113,22 +113,6 @@ class MGMSG(Enum):
|
||||||
PZ_GET_TPZ_IOSETTINGS = 0x07D6
|
PZ_GET_TPZ_IOSETTINGS = 0x07D6
|
||||||
|
|
||||||
|
|
||||||
def _write_exactly(f, data):
|
|
||||||
remaining = len(data)
|
|
||||||
pos = 0
|
|
||||||
while remaining:
|
|
||||||
written = f.write(data[pos:])
|
|
||||||
remaining -= written
|
|
||||||
pos += written
|
|
||||||
|
|
||||||
|
|
||||||
def _read_exactly(f, n):
|
|
||||||
r = bytes()
|
|
||||||
while len(r) < n:
|
|
||||||
r += f.read(n - len(r))
|
|
||||||
return r
|
|
||||||
|
|
||||||
|
|
||||||
class Direction:
|
class Direction:
|
||||||
def __init__(self, direction):
|
def __init__(self, direction):
|
||||||
if direction not in (1, 2):
|
if direction not in (1, 2):
|
||||||
|
@ -179,15 +163,15 @@ class Message:
|
||||||
self.param1, self.param2, self.dest, self.src)
|
self.param1, self.param2, self.dest, self.src)
|
||||||
|
|
||||||
def send(self, port):
|
def send(self, port):
|
||||||
_write_exactly(port, self.pack())
|
port.write(self.pack())
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def recv(cls, port):
|
def recv(cls, port):
|
||||||
(header, ) = st.unpack("6s", _read_exactly(port, 6))
|
(header, ) = st.unpack("6s", port.read(6))
|
||||||
data = b""
|
data = b""
|
||||||
if header[4] & 0x80:
|
if header[4] & 0x80:
|
||||||
(length, ) = st.unpack("<H", header[2:4])
|
(length, ) = st.unpack("<H", header[2:4])
|
||||||
data = _read_exactly(port, length)
|
data = port.read(length)
|
||||||
return cls.unpack(header + data)
|
return cls.unpack(header + data)
|
||||||
|
|
||||||
@property
|
@property
|
||||||
|
@ -750,7 +734,7 @@ class Tdc(Tcube):
|
||||||
raise MsgError("Error: Please disconnect the TDC001")
|
raise MsgError("Error: Please disconnect the TDC001")
|
||||||
elif msg_id == MGMSG.HW_RESPONSE:
|
elif msg_id == MGMSG.HW_RESPONSE:
|
||||||
raise MsgError("Hardware error, please disconnect "
|
raise MsgError("Hardware error, please disconnect "
|
||||||
+ "and reconnect the TDC001")
|
"and reconnect the TDC001")
|
||||||
elif msg_id == MGMSG.HW_RICHRESPONSE:
|
elif msg_id == MGMSG.HW_RICHRESPONSE:
|
||||||
(code, ) = st.unpack("<H", data[2:4])
|
(code, ) = st.unpack("<H", data[2:4])
|
||||||
raise MsgError("Hardware error {}: {}"
|
raise MsgError("Hardware error {}: {}"
|
||||||
|
|
Loading…
Reference in New Issue