80 lines
4.0 KiB
Python
80 lines
4.0 KiB
Python
"""@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()
|