refactored MsgSink and MsgSource
This commit is contained in:
@@ -1,7 +1,7 @@
|
|||||||
from abc import ABC, abstractmethod
|
from abc import ABC, abstractmethod
|
||||||
from msg_container import MsgContainer
|
from msg_container import MsgContainer
|
||||||
|
|
||||||
class MsgSink(ABC):
|
class MsgListener(ABC):
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
pass
|
pass
|
||||||
|
|
||||||
@@ -1,15 +1,15 @@
|
|||||||
from abc import ABC
|
from abc import ABC
|
||||||
from msg_container import MsgContainer
|
from msg_container import MsgContainer
|
||||||
from msg_sink import MsgSink
|
from msg_listener import MsgListener
|
||||||
|
|
||||||
class MsgSource(ABC):
|
class MsgTalker(ABC):
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
self.sinks = []
|
self.sinks = []
|
||||||
|
|
||||||
def register_recv(self, sink: MsgSink):
|
def register_listener(self, sink: MsgListener):
|
||||||
self.sinks.append(sink)
|
self.sinks.append(sink)
|
||||||
|
|
||||||
def call_sinks(self, msg: MsgContainer):
|
def call_listener(self, msg: MsgContainer):
|
||||||
for sink in self.sinks:
|
for sink in self.sinks:
|
||||||
if sink.is_msg(msg):
|
if sink.is_msg(msg):
|
||||||
sink.on_recv(msg)
|
sink.on_recv(msg)
|
||||||
+4
-4
@@ -175,16 +175,16 @@ if __name__ == "__main__":
|
|||||||
ubx_packetizer = UbxPacketizer()
|
ubx_packetizer = UbxPacketizer()
|
||||||
|
|
||||||
# Register NMEA-Packetizer to transceiver
|
# Register NMEA-Packetizer to transceiver
|
||||||
transceiver[grm].register_recv(nmea_packetizer)
|
transceiver[grm].register_listener(nmea_packetizer)
|
||||||
transceiver[grm].register_recv(ubx_packetizer)
|
transceiver[grm].register_listener(ubx_packetizer)
|
||||||
|
|
||||||
# Register Receiver to NMEA-Packetizer
|
# Register Receiver to NMEA-Packetizer
|
||||||
for rx in nmea_rx_list:
|
for rx in nmea_rx_list:
|
||||||
nmea_packetizer.register_recv(rx)
|
nmea_packetizer.register_listener(rx)
|
||||||
|
|
||||||
# Register Receiver to UBX-Packetizer
|
# Register Receiver to UBX-Packetizer
|
||||||
for rx in ubx_rx_list:
|
for rx in ubx_rx_list:
|
||||||
ubx_packetizer.register_recv(rx)
|
ubx_packetizer.register_listener(rx)
|
||||||
|
|
||||||
|
|
||||||
# Connect all sources
|
# Connect all sources
|
||||||
|
|||||||
@@ -1,12 +1,12 @@
|
|||||||
from msg_container import MsgContainer
|
from msg_container import MsgContainer
|
||||||
from msg_sink import MsgSink
|
from msg_listener import MsgListener
|
||||||
from msg_source import MsgSource
|
from msg_talker import MsgTalker
|
||||||
from struct import pack
|
from struct import pack
|
||||||
|
|
||||||
class NmeaPacketizer(MsgSink, MsgSource):
|
class NmeaPacketizer(MsgListener, MsgTalker):
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
MsgSink.__init__(self)
|
MsgListener.__init__(self)
|
||||||
MsgSource.__init__(self)
|
MsgTalker.__init__(self)
|
||||||
self.packet = b''
|
self.packet = b''
|
||||||
self.wait_sync = True
|
self.wait_sync = True
|
||||||
|
|
||||||
@@ -19,7 +19,7 @@ class NmeaPacketizer(MsgSink, MsgSource):
|
|||||||
timestamp = msg.timestamp
|
timestamp = msg.timestamp
|
||||||
self.wait_sync = False
|
self.wait_sync = False
|
||||||
if len(self.packet) > 0:
|
if len(self.packet) > 0:
|
||||||
self.call_sinks(MsgContainer(name, timestamp, self.packet))
|
self.call_listener(MsgContainer(name, timestamp, self.packet))
|
||||||
self.wait_sync = True
|
self.wait_sync = True
|
||||||
self.packet = b''
|
self.packet = b''
|
||||||
elif not self.wait_sync:
|
elif not self.wait_sync:
|
||||||
|
|||||||
@@ -1,8 +1,8 @@
|
|||||||
from msg_container import MsgContainer
|
from msg_container import MsgContainer
|
||||||
from msg_sink import MsgSink
|
from msg_listener import MsgListener
|
||||||
import re
|
import re
|
||||||
|
|
||||||
class AsciiReceiver(MsgSink):
|
class AsciiReceiver(MsgListener):
|
||||||
def __init__(self, name: str, filter: str):
|
def __init__(self, name: str, filter: str):
|
||||||
super().__init__()
|
super().__init__()
|
||||||
self.name = name
|
self.name = name
|
||||||
|
|||||||
@@ -1,7 +1,7 @@
|
|||||||
import struct
|
import struct
|
||||||
from msg_container import MsgContainer
|
from msg_container import MsgContainer
|
||||||
from msg_sink import MsgSink
|
from msg_listener import MsgListener
|
||||||
from msg_source import MsgSource
|
from msg_talker import MsgTalker
|
||||||
from ubx.ubx import UBX_SYNC_WORD, frame_parse, frame_create
|
from ubx.ubx import UBX_SYNC_WORD, frame_parse, frame_create
|
||||||
from receiver.ubx_receiver import UbxReceiver
|
from receiver.ubx_receiver import UbxReceiver
|
||||||
|
|
||||||
@@ -10,10 +10,10 @@ def ubx_pkt_debug(s: str):
|
|||||||
if UBX_PKT_DEBUG:
|
if UBX_PKT_DEBUG:
|
||||||
print(s)
|
print(s)
|
||||||
|
|
||||||
class UbxPacketizer(MsgSink, MsgSource):
|
class UbxPacketizer(MsgListener, MsgTalker):
|
||||||
def __init__(self, sync_bytes: bytes = UBX_SYNC_WORD):
|
def __init__(self, sync_bytes: bytes = UBX_SYNC_WORD):
|
||||||
MsgSink.__init__(self)
|
MsgListener.__init__(self)
|
||||||
MsgSource.__init__(self)
|
MsgTalker.__init__(self)
|
||||||
self.sync_bytes = sync_bytes
|
self.sync_bytes = sync_bytes
|
||||||
self.packet = b''
|
self.packet = b''
|
||||||
self.wait_sync = True
|
self.wait_sync = True
|
||||||
@@ -51,7 +51,7 @@ class UbxPacketizer(MsgSink, MsgSource):
|
|||||||
data, hdr = frame_parse(self.packet)
|
data, hdr = frame_parse(self.packet)
|
||||||
if data is not None:
|
if data is not None:
|
||||||
if hdr.length == len(data):
|
if hdr.length == len(data):
|
||||||
self.call_sinks(MsgContainer(msg.name, self.timestamp, self.packet))
|
self.call_listener(MsgContainer(msg.name, self.timestamp, self.packet))
|
||||||
self.packet = b''
|
self.packet = b''
|
||||||
self.wait_sync = True
|
self.wait_sync = True
|
||||||
|
|
||||||
@@ -61,9 +61,9 @@ class UbxPacketizer(MsgSink, MsgSource):
|
|||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
UBX_PKT_DEBUG = True
|
UBX_PKT_DEBUG = True
|
||||||
class MySink(MsgSink):
|
class MySink(MsgListener):
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
MsgSink.__init__(self)
|
MsgListener.__init__(self)
|
||||||
|
|
||||||
def on_recv(self, msg: MsgContainer):
|
def on_recv(self, msg: MsgContainer):
|
||||||
print(msg)
|
print(msg)
|
||||||
@@ -74,7 +74,7 @@ if __name__ == "__main__":
|
|||||||
ubx_rx = UbxReceiver('test', 1, 2)
|
ubx_rx = UbxReceiver('test', 1, 2)
|
||||||
|
|
||||||
pkt = UbxPacketizer()
|
pkt = UbxPacketizer()
|
||||||
pkt.register_recv(ubx_rx)
|
pkt.register_listener(ubx_rx)
|
||||||
|
|
||||||
pkt.reset()
|
pkt.reset()
|
||||||
frame = frame_create(_class=1, _id=2, _data=b'Hallo' + b'\xb5' + b',\x62Welt') + frame_create(_class=1, _id=2, _data=b',') + frame_create(_class=1, _id=2, _data=b'UBX') + frame_create(_class=1, _id=2, _data=b'World')
|
frame = frame_create(_class=1, _id=2, _data=b'Hallo' + b'\xb5' + b',\x62Welt') + frame_create(_class=1, _id=2, _data=b',') + frame_create(_class=1, _id=2, _data=b'UBX') + frame_create(_class=1, _id=2, _data=b'World')
|
||||||
|
|||||||
@@ -1,11 +1,11 @@
|
|||||||
from msg_container import MsgContainer
|
from msg_container import MsgContainer
|
||||||
from msg_sink import MsgSink
|
from msg_listener import MsgListener
|
||||||
from ubx.ubx import frame_parse, frame_parse_hdr
|
from ubx.ubx import frame_parse, frame_parse_hdr
|
||||||
from ubx.msg import UbxMsg
|
from ubx.msg import UbxMsg
|
||||||
|
|
||||||
class UbxReceiver(MsgSink):
|
class UbxReceiver(MsgListener):
|
||||||
def __init__(self, parser: UbxMsg, mclass: int|None = None, mid: int|None = None):
|
def __init__(self, parser: UbxMsg, mclass: int|None = None, mid: int|None = None):
|
||||||
MsgSink.__init__(self)
|
MsgListener.__init__(self)
|
||||||
self.parser = parser
|
self.parser = parser
|
||||||
self.mclass = mclass
|
self.mclass = mclass
|
||||||
self.mid = mid
|
self.mid = mid
|
||||||
|
|||||||
+4
-4
@@ -1,15 +1,15 @@
|
|||||||
from msg_source import MsgSource
|
from msg_talker import MsgTalker
|
||||||
from a_backend import ABackend
|
from a_backend import ABackend
|
||||||
from msg_container import *
|
from msg_container import *
|
||||||
|
|
||||||
class Transceiver(MsgSource):
|
class Transceiver(MsgTalker):
|
||||||
def __init__(self, backend: ABackend, name: str):
|
def __init__(self, backend: ABackend, name: str):
|
||||||
MsgSource.__init__(self)
|
MsgTalker.__init__(self)
|
||||||
self.backend = backend
|
self.backend = backend
|
||||||
self.name = name
|
self.name = name
|
||||||
|
|
||||||
def on_recv(self, msg: MsgContainer):
|
def on_recv(self, msg: MsgContainer):
|
||||||
self.call_sinks(msg)
|
self.call_listener(msg)
|
||||||
|
|
||||||
def send(self, data: bytes):
|
def send(self, data: bytes):
|
||||||
self.backend.send(self.name, data)
|
self.backend.send(self.name, data)
|
||||||
|
|||||||
Reference in New Issue
Block a user