summaryrefslogtreecommitdiffstats
path: root/lab_control/jds6600.py
blob: 61371056f4a0e92e26048e68b3c25289c76d2f14 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
import serial
import re

from lab_control.function_generator import FunctionGenerator

class JDS6600(FunctionGenerator):
    def __init__(self, portName):
        self._port = serial.Serial(portName)
        self._port.baudrate = 115200
        self._port.bytesize = serial.EIGHTBITS
        self._port.stopbits = serial.STOPBITS_ONE
        self._port.parity = serial.PARITY_NONE

    def _sendRequest(self, opcode: str, args: str) -> str:
        request = f":{opcode}={args}.\r\n"
        self._port.write(request.encode())
        responseRaw = self._port.readline()
        return responseRaw.decode().strip()

    def closePort(self) -> None:
        self._port.close()

    def _queryOnOff(self) -> list[str, str]:
        # TODO double check query on the device
        response = self._sendRequest("r20", "")
        return [response[5], response[7]]

    def setOn(self, channel: int) -> None:
        state = self._queryOnOff()
        state[channel - 1] = "1"
        response = self._sendRequest("w20", f"{state[0]},{state[1]}")
        # TODO figure out error handling

    def setOff(self, channel: int) -> None:
        state = self._queryOnOff()
        state[channel - 1] = "0"
        response = self._sendRequest("w20", f"{state[0]},{state[1]}")
        # TODO figure out error handling

    def setFrequency(self, channel: int, frequency: float) -> None:
        opcode = f"w{channel + 23}"