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
|
from time import time_ns, sleep
from typing import Optional, Tuple, Sequence, Union
import numpy as np
from picoscope.ps4000 import PS4000
from picoscope.ps5000 import PS5000
from picoscope.ps6000 import PS6000
from public import public
from .base import Scope
@public
class PicoScopeAlt(Scope): # pragma: no cover
def __init__(self, ps: Union[PS4000, PS5000, PS6000]):
super().__init__()
self.ps = ps
def open(self) -> None:
self.ps.open()
@property
def channels(self) -> Sequence[str]:
return list(self.ps.CHANNELS.keys())
def setup_frequency(self, frequency: int, samples: int) -> Tuple[int, int]:
actual_frequency, max_samples = self.ps.setSamplingFrequency(frequency, samples)
if max_samples < samples:
samples = max_samples
return actual_frequency, samples
def setup_channel(self, channel: str, coupling: str, range: float, enable: bool) -> None:
self.ps.setChannel(channel, coupling, range, 0.0, enable)
def setup_trigger(self, channel: str, threshold: float, direction: str, delay: int,
timeout: int, enable: bool) -> None:
self.ps.setSimpleTrigger(channel, threshold, direction, delay, timeout, enable)
def setup_capture(self, channel: str, enable: bool) -> None:
pass
def arm(self) -> None:
self.ps.runBlock()
def capture(self, channel: str, timeout: Optional[int] = None) -> Optional[np.ndarray]:
start = time_ns()
while not self.ps.isReady():
sleep(0.001)
if timeout is not None and (time_ns() - start) / 1e6 >= timeout:
return None
return self.ps.getDataV(channel)
def stop(self) -> None:
self.ps.stop()
def close(self) -> None:
self.ps.close()
|
