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
|
from typing import Optional, Tuple, Sequence, Set
import numpy as np
from chipwhisperer.capture.scopes.OpenADC import OpenADC
from public import public
from .base import Scope
@public
class ChipWhispererScope(Scope): # pragma: no cover
"""A ChipWhisperer based scope."""
def __init__(self, scope: OpenADC):
self.scope = scope
self.triggers: Set[str] = set()
def open(self) -> None:
self.scope.con()
@property
def channels(self) -> Sequence[str]:
return []
def setup_frequency(self, frequency: int, samples: int) -> Tuple[int, int]:
self.scope.clock.clkgen_freq = frequency
self.scope.samples = samples
return self.scope.clock.freq_ctr, self.scope.samples
def setup_channel(self, channel: str, coupling: str, range: float, enable: bool) -> None:
pass
def setup_trigger(self, channel: str, threshold: float, direction: str, delay: int,
timeout: int, enable: bool) -> None:
if enable:
self.triggers.add(channel)
elif channel in self.triggers:
self.triggers.remove(channel)
self.scope.adc.basic_mode = direction
self.scope.trigger.triggers = " OR ".join(self.triggers)
def setup_capture(self, channel: str, enable: bool) -> None:
pass
def arm(self) -> None:
self.scope.arm()
def capture(self, channel: str, timeout: Optional[int] = None) -> Optional[np.ndarray]:
self.scope.capture()
return self.scope.get_last_trace()
def stop(self) -> None:
pass
def close(self) -> None:
self.scope.dis()
|
