summaryrefslogtreecommitdiffstats
path: root/lab_control/jds6600.py
blob: eda8306f2f18ef068aade81684db3fd54e8c6ff9 (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
import serial

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"
        print(f"Script writing request to port: {request}")
        self._port.write(request.encode())
        print(f"Script waiting for response")
        responseRaw = self._port.readline()
        return responseRaw.decode().strip()

    def setOn(self, channel: int) -> None:
        args = ",".join(["1" if i == channel else "0" for i in [1, 2]])
        response = self._sendRequest("w20", args)
        print(f"Script got response {response}")
        # TODO figure out error handling

    def setOff(self, channel: int) -> None:
        pass