Update several functions to match changes in firmware

Author: bessman

pslab/instrument/oscilloscope.py

@@ -38,8 +38,8 @@ def __init__(self, device: ConnectionHandler | None = None):
         self._trigger_voltage = None
         self._trigger_enabled = False
         self._trigger_channel = "CH1"
-        self._set_gain("CH1", 1)
-        self._set_gain("CH2", 1)
+        # self._set_gain("CH1", 1)
+        # self._set_gain("CH2", 1)
 
     def capture(
         self,
@@ -375,12 +375,6 @@ def select_range(self, channel: str, voltage_range: Union[int, float]):
         self._set_gain(channel, gain)
 
     def _set_gain(self, channel: str, gain: int):
-        spi_config_supported = self._check_spi_config()
-
-        if not spi_config_supported:
-            spi_parameters = SPIMaster.get_parameters()
-            spi = SPIMaster(self._device)  # Initializing SPIMaster will reset config.
-
         self._channels[channel].gain = gain
         pga = self._channels[channel].programmable_gain_amplifier
         gain_idx = GAIN_VALUES.index(gain)
@@ -390,9 +384,6 @@ def _set_gain(self, channel: str, gain: int):
         self._device.send_byte(gain_idx)
         self._device.get_ack()
 
-        if not spi_config_supported:
-            spi.set_parameters(*spi_parameters)
-
     @staticmethod
     def _check_spi_config() -> bool:
         """Check whether current SPI config is supported by PGA.

pslab/protocol.py

@@ -193,36 +193,9 @@
 BAUD230400_LEGACY = Byte.pack(8)
 BAUD1000000_LEGACY = Byte.pack(9)
 
-# /*-----------NRFL01 radio module----------*/
-NRFL01 = Byte.pack(13)
-NRF_SETUP = Byte.pack(1)
-NRF_RXMODE = Byte.pack(2)
-NRF_TXMODE = Byte.pack(3)
-NRF_POWER_DOWN = Byte.pack(4)
-NRF_RXCHAR = Byte.pack(5)
-NRF_TXCHAR = Byte.pack(6)
-NRF_HASDATA = Byte.pack(7)
-NRF_FLUSH = Byte.pack(8)
-NRF_WRITEREG = Byte.pack(9)
-NRF_READREG = Byte.pack(10)
-NRF_GETSTATUS = Byte.pack(11)
-NRF_WRITECOMMAND = Byte.pack(12)
-NRF_WRITEPAYLOAD = Byte.pack(13)
-NRF_READPAYLOAD = Byte.pack(14)
-NRF_WRITEADDRESS = Byte.pack(15)
-NRF_TRANSACTION = Byte.pack(16)
-NRF_START_TOKEN_MANAGER = Byte.pack(17)
-NRF_STOP_TOKEN_MANAGER = Byte.pack(18)
-NRF_TOTAL_TOKENS = Byte.pack(19)
-NRF_REPORTS = Byte.pack(20)
-NRF_WRITE_REPORT = Byte.pack(21)
-NRF_DELETE_REPORT_ROW = Byte.pack(22)
-
-NRF_WRITEADDRESSES = Byte.pack(23)
-
 # ---------Non standard IO protocols--------
 
-NONSTANDARD_IO = Byte.pack(14)
+NONSTANDARD_IO = Byte.pack(13)
 # HX711_HEADER = Byte.pack(1)
 HCSR04_HEADER = Byte.pack(2)
 # AM2302_HEADER = Byte.pack(3)

pslab/sciencelab.py

@@ -36,6 +36,7 @@ class ScienceLab:
 
     def __init__(self, device: ConnectionHandler | None = None):
         self.device = device if device is not None else autoconnect()
+        self.version = self.device.get_version()
         self.logic_analyzer = LogicAnalyzer(device=self.device)
         self.oscilloscope = Oscilloscope(device=self.device)
         self.waveform_generator = WaveformGenerator(device=self.device)
@@ -163,8 +164,10 @@ def rgb_led(self, colors: List, output: str = "RGB", order: str = "GRB"):
 
         >>> psl.rgb_led([[10,0,0],[0,10,10],[10,0,10]], output="SQ1", order="RGB")
         """
-        if "6" in self.device.version:
+        if "6" in self.version:
             pins = {"ONBOARD": 0, "SQ1": 1, "SQ2": 2, "SQ3": 3, "SQ4": 4}
+            if output == "RGB":
+                output = "ONBOARD"
         else:
             pins = {"RGB": CP.SET_RGB1, "PGC": CP.SET_RGB2, "SQ1": CP.SET_RGB3}
 
@@ -189,24 +192,13 @@ def rgb_led(self, colors: List, output: str = "RGB", order: str = "GRB"):
                 f"Invalid order: {order}. order must contain 'R', 'G', and 'B'."
             )
 
-        self.device.send_byte(CP.COMMON)
-
-        if "6" in self.device.version:
-            self.device.send_byte(CP.SET_RGB_COMMON)
-        else:
-            self.device.send_byte(pin)
-
-        self.device.send_byte(len(colors) * 3)
-
-        for color in colors:
-            self.device.send_byte(color[order.index("R")])
-            self.device.send_byte(color[order.index("G")])
-            self.device.send_byte(color[order.index("B")])
-
-        if "6" in self.device.version:
-            self.device.send_byte(pin)
-
-        self.device.get_ack()
+        cmd = CP.COMMON + CP.SET_RGB_COMMON
+        args = CP.Byte.pack(pin)
+        args += CP.Byte.pack(len(colors) * 3)
+        args += bytes(
+            color[order.index(channel)] for channel in "RGB" for color in colors
+        )
+        self.device.exchange(cmd, args)
 
     def _read_program_address(self, address: int):
         """Return the value stored at the specified address in program memory.