summaryrefslogtreecommitdiffstats
path: root/lab_control/jds6600.py
blob: 7e7640d767c46ac9e1e3edff0648e466590aad97 (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
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[int, int]:
        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