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
|
from time import time_ns, sleep
from typing import Optional, Tuple, Sequence, Union
import numpy as np
from picoscope.ps4000 import PS4000
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, PS6000]):
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()
|
