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
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
|
"""
devices/analog_input.py
Analog Input device module — reads voltage/current channels.
Includes a simulation mode (no hardware required) for development/demo.
"""
import math
import random
import time
from typing import Any, Dict
from PyQt6.QtWidgets import (
QWidget, QVBoxLayout, QHBoxLayout, QLabel, QComboBox,
QDoubleSpinBox, QGroupBox, QCheckBox, QSpinBox, QFormLayout
)
from PyQt6.QtCore import Qt
from devices.base_device import BaseDevice, ChannelConfig, DeviceInfo, DeviceStatus
CHANNEL_COLORS = ["#00d4ff", "#ff6b35", "#7fff6e", "#ffcc00", "#c77dff", "#ff4d6d", "#4cc9f0", "#f72585"]
class AnalogInputDevice(BaseDevice):
"""
Analog voltage/current input module.
Supports up to 16 channels. In simulation mode, generates
realistic waveforms (sine, ramp, noise) for each channel.
Real hardware: override read_channels() with your SDK calls.
"""
DEVICE_TYPE = "analog_input"
ICON = "〜"
def __init__(self, device_id: str = "ai_0", num_channels: int = 4,
simulate: bool = True, sample_rate_hz: float = 10.0):
channels = [
ChannelConfig(
channel_id=f"ch{i}",
name=f"AI {i}",
unit="V",
min_value=-10.0,
max_value=10.0,
alarm_low=-8.0,
alarm_high=8.0,
color=CHANNEL_COLORS[i % len(CHANNEL_COLORS)],
)
for i in range(num_channels)
]
info = DeviceInfo(
device_id=device_id,
name="Analog Input",
device_type=self.DEVICE_TYPE,
description="Multi-channel analog voltage/current input",
manufacturer="Generic",
model="AI-16",
icon=self.ICON,
channels=channels,
)
super().__init__(info)
self.simulate = simulate
self.sample_rate_hz = sample_rate_hz
self._start_time = 0.0
# Sim parameters per channel
self._sim_params = [
{"freq": 0.5 + i * 0.3, "amp": 5.0, "offset": 0.0, "noise": 0.05, "mode": "sine"}
for i in range(num_channels)
]
# ------------------------------------------------------------------ #
# BaseDevice interface #
# ------------------------------------------------------------------ #
def connect(self) -> bool:
if self.simulate:
self._start_time = time.time()
self.status = DeviceStatus.SIMULATED
return True
# TODO: Replace with real hardware SDK init
# e.g. import nidaqmx; self._task = nidaqmx.Task(); ...
self.status = DeviceStatus.ERROR
return False
def disconnect(self) -> None:
self.status = DeviceStatus.DISCONNECTED
def read_channels(self) -> Dict[str, float]:
if self.simulate:
return self._simulate_read()
# TODO: Replace with real hardware read
return {}
def write_channel(self, channel_id: str, value: Any) -> bool:
# Analog inputs don't support write — subclass for AO
return False
def get_config_widget(self) -> QWidget:
return AnalogInputConfigWidget(self)
# ------------------------------------------------------------------ #
# Simulation #
# ------------------------------------------------------------------ #
def _simulate_read(self) -> Dict[str, float]:
t = time.time() - self._start_time
result = {}
for i, ch in enumerate(self.info.channels):
if not ch.enabled:
continue
p = self._sim_params[i]
if p["mode"] == "sine":
val = p["amp"] * math.sin(2 * math.pi * p["freq"] * t) + p["offset"]
elif p["mode"] == "ramp":
period = 1.0 / max(p["freq"], 0.01)
val = p["amp"] * ((t % period) / period) * 2 - p["amp"] + p["offset"]
elif p["mode"] == "square":
val = p["amp"] * math.copysign(1, math.sin(2 * math.pi * p["freq"] * t)) + p["offset"]
else:
val = p["offset"]
val += random.gauss(0, p["noise"] * p["amp"])
val = max(ch.min_value, min(ch.max_value, val))
result[ch.channel_id] = round(val, 4)
return result
# ------------------------------------------------------------------ #
# Config Widget #
# ------------------------------------------------------------------ #
class AnalogInputConfigWidget(QWidget):
def __init__(self, device: AnalogInputDevice):
super().__init__()
self.device = device
self._build_ui()
def _build_ui(self):
layout = QVBoxLayout(self)
layout.setContentsMargins(0, 0, 0, 0)
# Global settings
global_group = QGroupBox("Device Settings")
form = QFormLayout(global_group)
self.sim_check = QCheckBox("Simulation Mode")
self.sim_check.setChecked(self.device.simulate)
form.addRow(self.sim_check)
self.rate_spin = QDoubleSpinBox()
self.rate_spin.setRange(0.1, 1000.0)
self.rate_spin.setValue(self.device.sample_rate_hz)
self.rate_spin.setSuffix(" Hz")
form.addRow("Sample Rate:", self.rate_spin)
layout.addWidget(global_group)
# Per-channel
ch_group = QGroupBox("Channel Configuration")
ch_layout = QVBoxLayout(ch_group)
for i, ch in enumerate(self.device.info.channels):
row = QHBoxLayout()
en = QCheckBox(ch.name)
en.setChecked(ch.enabled)
row.addWidget(en)
mode_cb = QComboBox()
mode_cb.addItems(["sine", "ramp", "square", "dc"])
mode_cb.setCurrentText(self.device._sim_params[i]["mode"])
row.addWidget(QLabel("Mode:"))
row.addWidget(mode_cb)
freq_sp = QDoubleSpinBox()
freq_sp.setRange(0.01, 100.0)
freq_sp.setValue(self.device._sim_params[i]["freq"])
freq_sp.setSuffix(" Hz")
row.addWidget(QLabel("Freq:"))
row.addWidget(freq_sp)
ch_layout.addLayout(row)
layout.addWidget(ch_group)
layout.addStretch()
|
