occheung
/
artiq
forked from M-Labs/artiq
1
0
Fork 0
Code Pull Requests Activity

lda: style

This commit is contained in:
Sebastien Bourdeauducq 2015-02-18 11:31:44 -07:00
parent dc132bd248
commit 05824fd3be
1 changed files with 10 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
artiq/devices/lda/driver.py
View File

@ -41,13 +41,12 @@ class Ldasim:
"""Stores the new attenuation value and prints it to console. """Stores the new attenuation value and prints it to console.
:param attenuation: The attenuation value in dB. :param attenuation: The attenuation value in dB.
:type attenuation: int, float or Fraction
""" """
step = self.get_att_step_size() step = self.get_att_step_size()
if isinstance(attenuation, Quantity): if isinstance(attenuation, Quantity):
check_unit(attenuation, 'dB') check_unit(attenuation, "dB")
att = attenuation att = attenuation
else: else:
att = attenuation*dB att = attenuation*dB
@ -55,13 +54,13 @@ class Ldasim:
att = round(att/step)*step att = round(att/step)*step
if att > self.get_att_max(): if att > self.get_att_max():
raise ValueError('Cannot set attenuation {} > {}' raise ValueError("Cannot set attenuation {} > {}"
.format(att, self.get_att_max())) .format(att, self.get_att_max()))
elif att < 0*dB: elif att < 0*dB:
raise ValueError('Cannot set attenuation {} < 0'.format(att)) raise ValueError("Cannot set attenuation {} < 0".format(att))
else: else:
att = round(att.amount*4)/4. * dB att = round(att.amount*4)/4. * dB
print("[LDA-sim] setting attenuation to {}".format(att)) print("setting attenuation to {}".format(att))
self._attenuation = att self._attenuation = att
@ -118,7 +117,6 @@ class Lda:
def close(self): def close(self):
"""Close the device. """Close the device.
""" """
self.hidapi.hid_close(self._dev) self.hidapi.hid_close(self._dev)
def get_att_step_size(self): def get_att_step_size(self):
@ -174,9 +172,9 @@ class Lda:
def get(self, command, length, timeout=1000): def get(self, command, length, timeout=1000):
"""Sends a GET command to read back some value of the Lab Brick device. """Sends a GET command to read back some value of the Lab Brick device.
:param int command: Command ID, most significant bit must be cleared. :param command: Command ID, most significant bit must be cleared.
:param int length: Length of the command, "count" in the datasheet. :param length: Length of the command, "count" in the datasheet.
:param int timeout: Timeout of the HID read in ms. :param timeout: Timeout of the HID read in ms.
:return: Returns the value read from the device. :return: Returns the value read from the device.
:rtype: bytes :rtype: bytes
""" """
@ -207,13 +205,12 @@ class Lda:
"""Sets attenuation value of the Lab Brick device. """Sets attenuation value of the Lab Brick device.
:param attenuation: Attenuation value in dB. :param attenuation: Attenuation value in dB.
:type attenuation: int, float or Fraction
""" """
step = self.get_att_step_size() step = self.get_att_step_size()
if isinstance(attenuation, Quantity): if isinstance(attenuation, Quantity):
check_unit(attenuation, 'dB') check_unit(attenuation, "dB")
att = attenuation att = attenuation
else: else:
att = attenuation*dB att = attenuation*dB
@ -221,9 +218,9 @@ class Lda:
att = round(att/step)*step att = round(att/step)*step
if att > self.get_att_max(): if att > self.get_att_max():
raise ValueError('Cannot set attenuation {} > {}' raise ValueError("Cannot set attenuation {} > {}"
.format(att, self.get_att_max())) .format(att, self.get_att_max()))
elif att < 0*dB: elif att < 0*dB:
raise ValueError('Cannot set attenuation {} < 0'.format(att)) raise ValueError("Cannot set attenuation {} < 0".format(att))
else: else:
self.set(0x8d, bytes([int(round(att.amount*4))])) self.set(0x8d, bytes([int(round(att.amount*4))]))