summaryrefslogtreecommitdiffstats
path: root/lab_control/test/mock_lab.py
blob: 1016b9236f48758e04d3a843290b58c436ded4c4 (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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
from collections.abc import Callable

from lab_control.function_generator import FunctionGenerator
from lab_control.oscilloscope import Oscilloscope

class MockLab(FunctionGenerator, Oscilloscope):
    class FGChannelState:
        def __init__(self):
            self.on = False
            self.frequency = None
            self.amplitude = None
            self.function = None

    class OscChannelState:
        def __init__(self):
            self.testFunction = None
            self.connectedChannel = None
            self.voltsPerDiv = None

    def __init__(self):
        self.fgChannels = [MockLab.FGChannelState() for i in range(0, 2)]
        self.oscChannels = [MockLab.OscChannelState() for i in range(0, 4)]

    def setOn(self, channel: int) -> None:
        self.fgChannels[channel - 1].on = True

    def setOff(self, channel: int) -> None:
        self.fgChannels[channel - 1].on = False

    def setFrequency(self, channel: int, frequency: float) -> None:
        self.fgChannels[channel - 1].frequency = frequency

    def setAmplitude(self, channel: int, amplitude: float) -> None:
        self.fgChannels[channel - 1].amplitude = amplitude

    def setFunction(self, channel: int, function: int) -> None:
        self.fgChannels[channel - 1].function = function

    def measureAmplitude(self, channel: int) -> float:
        fgChannel = self.oscChannels[channel - 1].connectedChannel
        frequency = fgChannel.frequency if fgChannel.on else 0.0
        amplitude = fgChannel.amplitude if fgChannel.on else 0.0

        return self.oscChannels[channel - 1].testFunction(frequency) * amplitude

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

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

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

    def setVoltsPerDivision(self, channel: int, volts: float) -> None:
        self.oscChannels[channel - 1].voltsPerDiv = volts

    def getDivisionsDisplayed(self) -> int:
        return 12

    def setTestFunction(self, channel: int, f: Callable[[float], float]) -> None:
        self.oscChannels[channel - 1].testFunction = f

    def connectChannels(self, fg: int, osc: int) -> None:
        self.oscChannels[osc - 1].connectedChannel = self.fgChannels[fg - 1]

    def getFunctionGeneratorChannel(self, channel: int):
        return self.fgChannels[channel - 1]

    def getOscilloscopeChannel(self, channel: int):
        return self.oscChannels[channel - 1]