resolved TODO I had forgotton to handle channel configuration

Signed-off-by: {Johann Heyszl} <[email protected]>
lowRISC · Sep 26, 2023 · e1d10ee · e1d10ee
1 parent 97bac6a
commit e1d10ee
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/cw/waverunner.py b/cw/waverunner.py
@@ -372,10 +372,11 @@ def capture_and_transfer_waves(self):
         res = self._ask("WAIT 10;*OPC?")
         assert res == "*OPC 1"
         # Transfer and parse waveform data.
-        # TODO test this!
-        # data = self._ask_raw(b"C3:WF? DAT1")
-        data = self._ask_raw(b"C1:WF? DAT1")
-        # data = self._ask_raw(f"{self.acqu_channel}:WF? DAT1")
+        if self.acqu_channel == "C1": data = self._ask_raw(b"C1:WF? DAT1")
+        elif self.acqu_channel == "C2": data = self._ask_raw(b"C2:WF? DAT1")
+        elif self.acqu_channel == "C3": data = self._ask_raw(b"C3:WF? DAT1")
+        elif self.acqu_channel == "C4": data = self._ask_raw(b"C4:WF? DAT1")
+        else: print("WAVERUNNER: Error: Channel selection invalid")
         waves = self._parse_waveform(data)
         return waves