summaryrefslogtreecommitdiff
path: root/daq_system/devices/serial_device.py
blob: 842a4010d428d5b8f9702e1a4f029e063a0bc244 (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
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
"""
devices/serial_device.py

Serial / UART device module — reads data from a serial port.
Parses CSV-format lines: "ch0,ch1,ch2,...\\n"
"""

import random
import time
from typing import Any, Dict

from PyQt6.QtWidgets import (
    QWidget, QVBoxLayout, QFormLayout, QGroupBox,
    QComboBox, QSpinBox, QLineEdit, QPushButton, QLabel
)
from devices.base_device import BaseDevice, ChannelConfig, DeviceInfo, DeviceStatus

SER_COLORS = ["#7fff6e", "#4cc9f0", "#f72585", "#00d4ff"]


class SerialDevice(BaseDevice):
    DEVICE_TYPE = "serial"
    ICON = "⇌"

    def __init__(self, device_id: str = "ser_0", port: str = "COM3",
                 baud_rate: int = 115200, num_channels: int = 4,
                 simulate: bool = True):
        channels = [
            ChannelConfig(
                channel_id=f"s{i}", name=f"Serial {i}",
                unit="", min_value=0.0, max_value=1023.0,
                color=SER_COLORS[i % len(SER_COLORS)]
            )
            for i in range(num_channels)
        ]
        info = DeviceInfo(
            device_id=device_id, name="Serial / UART",
            device_type=self.DEVICE_TYPE,
            description=f"Serial port {port} @ {baud_rate} baud",
            icon=self.ICON, channels=channels
        )
        super().__init__(info)
        self.port = port
        self.baud_rate = baud_rate
        self.simulate = simulate
        self._serial = None  # Replace with serial.Serial() for real hardware
        self._t0 = 0.0

    def connect(self) -> bool:
        if self.simulate:
            self._t0 = time.time()
            self.status = DeviceStatus.SIMULATED
            return True
        try:
            import serial
            self._serial = serial.Serial(self.port, self.baud_rate, timeout=0.1)
            self.status = DeviceStatus.CONNECTED
            return True
        except Exception as e:
            print(f"[SerialDevice] Connect failed: {e}")
            self.status = DeviceStatus.ERROR
            return False

    def disconnect(self) -> None:
        if self._serial:
            try:
                self._serial.close()
            except Exception:
                pass
        self.status = DeviceStatus.DISCONNECTED

    def read_channels(self) -> Dict[str, float]:
        if self.simulate:
            return self._simulate_read()
        if not self._serial or not self._serial.is_open:
            return {}
        try:
            line = self._serial.readline().decode("utf-8").strip()
            if not line:
                return {}
            parts = line.split(",")
            return {
                ch.channel_id: float(parts[i])
                for i, ch in enumerate(self.info.channels)
                if i < len(parts)
            }
        except Exception:
            return {}

    def _simulate_read(self) -> Dict[str, float]:
        t = time.time() - self._t0
        import math
        return {
            ch.channel_id: round(512 + 400 * math.sin(2 * 3.14159 * (0.2 + i * 0.15) * t)
                                 + random.gauss(0, 5), 1)
            for i, ch in enumerate(self.info.channels)
        }

    def write_channel(self, channel_id: str, value: Any) -> bool:
        if self._serial and self._serial.is_open:
            try:
                cmd = f"{channel_id}:{value}\n"
                self._serial.write(cmd.encode())
                return True
            except Exception:
                return False
        return False

    def get_config_widget(self) -> QWidget:
        w = QWidget()
        layout = QVBoxLayout(w)
        grp = QGroupBox("Serial Port Settings")
        form = QFormLayout(grp)

        self._port_edit = QLineEdit(self.port)
        form.addRow("Port:", self._port_edit)

        self._baud_cb = QComboBox()
        self._baud_cb.addItems(["9600", "19200", "38400", "57600", "115200", "230400", "460800"])
        self._baud_cb.setCurrentText(str(self.baud_rate))
        form.addRow("Baud Rate:", self._baud_cb)

        parity_cb = QComboBox()
        parity_cb.addItems(["None", "Even", "Odd"])
        form.addRow("Parity:", parity_cb)

        bits_cb = QComboBox()
        bits_cb.addItems(["8", "7"])
        form.addRow("Data Bits:", bits_cb)

        apply_btn = QPushButton("Apply & Reconnect")
        form.addRow(apply_btn)

        layout.addWidget(grp)
        layout.addStretch()
        return w