summaryrefslogtreecommitdiffstats
path: root/lab_control/sds1000xe.py
blob: d86c1c7df9fcd7c82566ecf3b5574ab846379318 (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
import socket
import re
from lab_control.oscilloscope import Oscilloscope

class SDS1000XE(Oscilloscope):
    PORT = 5025
    TIMEOUT = 0.2

    AVAILABLE_CHANNELS = range(1, 5)

    def __init__(self, ip):
        self._socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        self._socket.connect((ip, SDS1000XE.PORT))
        self._socket.settimeout(SDS1000XE.TIMEOUT)

    def measureAmplitude(self, channel: int) -> float:
        assert channel in SDS1000XE.AVAILABLE_CHANNELS

        query = f"C{channel}:PAVA? AMPL"
        self._socket.sendall(query.encode())

        try:
            response = self._socket.recv(4096).decode()
            m = re.search(r"C(?P<responseChannel>\d):PAVA AMPL,(?P<rawMeasurement>.+)V", response)
            measurement = float(m.group("rawMeasurement"))
        except (TimeoutError, AttributeError) as e:
            measurement = None
        
        return measurement

    def measurePkToPk(self, channel: int) -> float:
        pass

    def measureRMS(self, channel: int) -> float:
        pass

    def measureFrequency(self, channel: int) -> float:
        pass