Extract reusable device protocols, ports and logic analyzers from SETGUI
This commit is contained in:
79
python/set_devices/bus_demo.py
Normal file
79
python/set_devices/bus_demo.py
Normal file
@@ -0,0 +1,79 @@
|
||||
"""@file bus_demo.py
|
||||
@brief Генератор трафика полевой шины: разбор проверяется без железа.
|
||||
|
||||
Повторяет то, что мост выдаёт в RS485: готовые транспортные кадры со
|
||||
всеми типами сообщений ProtoCAN. Модуль не импортирует Qt — планировщик
|
||||
живёт во вкладке, а сам генератор проверяется host-тестами.
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import random
|
||||
from datetime import datetime
|
||||
|
||||
from . import can_transport as tr
|
||||
from protocan import protocan as pc
|
||||
|
||||
#: Типы сообщений, которые генератор выдаёт по кругу случайно.
|
||||
KINDS = ("pulse", "analog", "modbus", "gas", "status", "error", "discrete")
|
||||
|
||||
|
||||
class BusDemoSource:
|
||||
"""@brief Источник демонстрационных кадров шины.
|
||||
|
||||
@param rng Источник случайности; передаётся в тестах для повторяемости.
|
||||
"""
|
||||
|
||||
def __init__(self, rng: random.Random | None = None) -> None:
|
||||
self._rng = rng or random.Random()
|
||||
self._seq = 0
|
||||
|
||||
def next_frame(self) -> bytes:
|
||||
"""@brief Собирает очередной кадр так же, как его отдал бы мост.
|
||||
|
||||
@return Готовые байты транспортного кадра, включая SOF и CRC.
|
||||
"""
|
||||
self._seq = (self._seq + 1) & 0xFF
|
||||
kind = self._rng.choice(KINDS)
|
||||
dev_type, dev_id = 1, 2
|
||||
if kind == "pulse":
|
||||
cid = pc.ProtoCanId.build(1, 1, dev_type, dev_id, pc.MsgType.PULSE, 0)
|
||||
data = bytes([self._seq])
|
||||
elif kind == "analog":
|
||||
sid = self._rng.randint(1, 40)
|
||||
atype = self._rng.choice([pc.AnalogType.U, pc.AnalogType.I, pc.AnalogType.T])
|
||||
cid = pc.ProtoCanId.build(0, 1, dev_type, dev_id, pc.MsgType.ANALOG,
|
||||
pc.merge_analog(atype, sid))
|
||||
tag = {pc.AnalogType.U: b"US", pc.AnalogType.I: b"IS",
|
||||
pc.AnalogType.T: b"TS"}[atype]
|
||||
data = tag + ("%04d" % sid).encode()
|
||||
elif kind == "modbus":
|
||||
adr, cnt = self._rng.randint(0, 0x200), self._rng.randint(1, 4)
|
||||
cid = pc.ProtoCanId.build(0, 1, dev_type, dev_id, pc.MsgType.MODBUS_HOLDING,
|
||||
pc.merge_modbus(adr, cnt))
|
||||
data = b"".join(self._rng.randint(0, 0xFFFF).to_bytes(2, "little")
|
||||
for _ in range(cnt))
|
||||
elif kind == "gas":
|
||||
adr = self._rng.randint(0, 0x400)
|
||||
cid = pc.ProtoCanId.build(0, 1, dev_type, dev_id,
|
||||
pc.MsgType.GENERAL_ADDRESS_SPACE, adr)
|
||||
data = b"".join(self._rng.randint(0, 0xFFFF).to_bytes(2, "little")
|
||||
for _ in range(self._rng.randint(1, 4)))
|
||||
elif kind == "status":
|
||||
now = datetime.now()
|
||||
cid = pc.ProtoCanId.build(1, 1, dev_type, dev_id, pc.MsgType.BROADCAST,
|
||||
pc.merge_broadcast(pc.BroadcastType.STATUS, 0))
|
||||
data = bytes([now.hour, now.minute, now.second, now.year % 100,
|
||||
now.month, now.day, now.isoweekday() % 7])
|
||||
elif kind == "error":
|
||||
cid = pc.ProtoCanId.build(0, 1, dev_type, dev_id, pc.MsgType.ERROR,
|
||||
pc.merge_error(self._rng.randint(0, 5), 0xFF))
|
||||
data = b""
|
||||
else:
|
||||
dtype = self._rng.choice(list(pc.DiscreteType)[:7])
|
||||
cid = pc.ProtoCanId.build(0, 1, dev_type, dev_id, pc.MsgType.DISCRETE,
|
||||
pc.merge_discrete(dtype, self._rng.randint(0, 20)))
|
||||
data = bytes(self._rng.randint(0, 255)
|
||||
for _ in range(self._rng.randint(1, 8)))
|
||||
return tr.Frame(seq=self._seq, flags=tr.FLAG_IDE, can_id=cid,
|
||||
data=data).encode()
|
||||
Reference in New Issue
Block a user