- added stream talker
- added stream listener
This commit is contained in:
+4
-5
@@ -5,11 +5,10 @@ import signal
|
||||
from threading import Thread
|
||||
from queue import Queue
|
||||
|
||||
from msg.container import MsgContainer
|
||||
from transceiver import Transceiver
|
||||
from a_backend import ABackend
|
||||
|
||||
from nmea import NmeaPacketizer
|
||||
from nmea.packetizer import NmeaPacketizer
|
||||
from nmea.messages import Gll, Gsa, Gsv, Gga, Rmc
|
||||
|
||||
from ubx.ubx import frame_create
|
||||
@@ -98,7 +97,7 @@ class NetworkBackend(ABackend):
|
||||
if mask & selectors.EVENT_READ:
|
||||
timestamp = time.time()
|
||||
data = key.fileobj.recv(64)
|
||||
sock['xcvr'].on_recv(MsgContainer(name, timestamp, data))
|
||||
sock['xcvr'].on_recv(data)
|
||||
if mask & selectors.EVENT_WRITE:
|
||||
data = sock['queue'].get()
|
||||
key.fileobj.send(data)
|
||||
@@ -165,8 +164,8 @@ if __name__ == "__main__":
|
||||
mc.register_xcvr(transceiver[grm])
|
||||
|
||||
# Create NMEA-Packetizer
|
||||
nmea_packetizer = NmeaPacketizer()
|
||||
ubx_packetizer = UbxPacketizer()
|
||||
nmea_packetizer = NmeaPacketizer(grm)
|
||||
ubx_packetizer = UbxPacketizer(grm)
|
||||
|
||||
# Register NMEA-Packetizer to transceiver
|
||||
transceiver[grm].register_listener(nmea_packetizer)
|
||||
|
||||
+9
-16
@@ -1,31 +1,24 @@
|
||||
from msg.container import MsgContainer
|
||||
from msg.listener import MsgListener
|
||||
from msg.talker import MsgTalker
|
||||
from stream.listener import StreamListener
|
||||
from struct import pack
|
||||
import time
|
||||
|
||||
class NmeaPacketizer(MsgListener, MsgTalker):
|
||||
def __init__(self):
|
||||
MsgListener.__init__(self)
|
||||
class NmeaPacketizer(StreamListener, MsgTalker):
|
||||
def __init__(self, name: str):
|
||||
StreamListener.__init__(self)
|
||||
MsgTalker.__init__(self)
|
||||
self.name = name
|
||||
self.packet = b''
|
||||
self.wait_sync = True
|
||||
|
||||
def on_recv(self, msg: MsgContainer):
|
||||
name = msg.name
|
||||
data = msg.data
|
||||
for n in range (len(data)):
|
||||
d = data[n]
|
||||
def on_recv(self, rx_data: bytes):
|
||||
for d in rx_data:
|
||||
if d == 13 or d == 10:
|
||||
timestamp = msg.timestamp
|
||||
self.wait_sync = False
|
||||
if len(self.packet) > 0:
|
||||
self.call_listener(MsgContainer(name, timestamp, self.packet))
|
||||
self.call_listener(MsgContainer(self.name, time.time(), self.packet))
|
||||
self.wait_sync = True
|
||||
self.packet = b''
|
||||
elif not self.wait_sync:
|
||||
self.packet += pack('B', d)
|
||||
|
||||
|
||||
|
||||
def is_msg(self, msg: MsgContainer):
|
||||
return True
|
||||
@@ -0,0 +1,9 @@
|
||||
from abc import ABC, abstractmethod
|
||||
|
||||
class StreamListener(ABC):
|
||||
def __init__(self):
|
||||
pass
|
||||
|
||||
@abstractmethod
|
||||
def on_recv(self, data: bytes):
|
||||
pass
|
||||
@@ -0,0 +1,13 @@
|
||||
from abc import ABC
|
||||
from stream.listener import StreamListener
|
||||
|
||||
class StreamTalker(ABC):
|
||||
def __init__(self):
|
||||
self.sinks = []
|
||||
|
||||
def register_listener(self, sink: StreamListener):
|
||||
self.sinks.append(sink)
|
||||
|
||||
def call_listener(self, data: bytes):
|
||||
for sink in self.sinks:
|
||||
sink.on_recv(data)
|
||||
+8
-6
@@ -1,15 +1,17 @@
|
||||
from msg.talker import MsgTalker
|
||||
from a_backend import ABackend
|
||||
from msg.container import *
|
||||
from stream.talker import StreamTalker
|
||||
from stream.listener import StreamListener
|
||||
|
||||
class Transceiver(MsgTalker):
|
||||
|
||||
class Transceiver(StreamListener, StreamTalker):
|
||||
def __init__(self, name: str):
|
||||
MsgTalker.__init__(self)
|
||||
StreamListener.__init__(self)
|
||||
StreamTalker.__init__(self)
|
||||
self.name = name
|
||||
self.backend = None
|
||||
|
||||
def on_recv(self, msg: MsgContainer):
|
||||
self.call_listener(msg)
|
||||
def on_recv(self, data: bytes):
|
||||
self.call_listener(data)
|
||||
|
||||
def on_register(self, backend: ABackend):
|
||||
self.backend = backend
|
||||
|
||||
+11
-15
@@ -1,7 +1,9 @@
|
||||
import struct
|
||||
import time
|
||||
from msg.container import MsgContainer
|
||||
from msg.listener import MsgListener
|
||||
from msg.talker import MsgTalker
|
||||
from stream.listener import StreamListener
|
||||
from ubx.ubx import UBX_SYNC_WORD, frame_parse, frame_create
|
||||
from ubx.listener import UbxListener
|
||||
|
||||
@@ -10,15 +12,15 @@ def ubx_pkt_debug(s: str):
|
||||
if UBX_PKT_DEBUG:
|
||||
print(s)
|
||||
|
||||
class UbxPacketizer(MsgListener, MsgTalker):
|
||||
def __init__(self, sync_bytes: bytes = UBX_SYNC_WORD):
|
||||
MsgListener.__init__(self)
|
||||
class UbxPacketizer(StreamListener, MsgTalker):
|
||||
def __init__(self, name: str, sync_bytes: bytes = UBX_SYNC_WORD):
|
||||
StreamListener.__init__(self)
|
||||
MsgTalker.__init__(self)
|
||||
self.name = name
|
||||
self.sync_bytes = sync_bytes
|
||||
self.packet = b''
|
||||
self.wait_sync = True
|
||||
self.sync_count = 0
|
||||
self.timestamp = 0
|
||||
self.d_save = b''
|
||||
|
||||
def reset(self):
|
||||
@@ -26,21 +28,19 @@ class UbxPacketizer(MsgListener, MsgTalker):
|
||||
self.wait_sync = True
|
||||
self.sync_count = 0
|
||||
|
||||
def on_recv(self, msg: MsgContainer):
|
||||
for n in range (0, len(msg.data)):
|
||||
d = msg.data[n]
|
||||
def on_recv(self, rx_data: bytes):
|
||||
for d in rx_data:
|
||||
if d == self.sync_bytes[self.sync_count]:
|
||||
self.d_save += struct.pack('B', d)
|
||||
self.sync_count += 1
|
||||
if self.sync_count == len(self.sync_bytes):
|
||||
ubx_pkt_debug(f"{msg.name}:UBX-sync")
|
||||
ubx_pkt_debug(f"{self.name}:UBX-sync")
|
||||
if len(self.packet) > 0:
|
||||
ubx_pkt_debug(f"{msg.name}:Remaining packet: {self.packet}")
|
||||
ubx_pkt_debug(f"{self.name}:Remaining packet: {self.packet}")
|
||||
self.sync_count = 0
|
||||
self.d_save = b''
|
||||
self.packet = b''
|
||||
self.wait_sync = False
|
||||
self.timestamp = msg.timestamp
|
||||
else:
|
||||
if self.sync_count > 0:
|
||||
self.packet += self.d_save
|
||||
@@ -51,14 +51,10 @@ class UbxPacketizer(MsgListener, MsgTalker):
|
||||
data, hdr = frame_parse(self.packet)
|
||||
if data is not None:
|
||||
if hdr.length == len(data):
|
||||
self.call_listener(MsgContainer(msg.name, self.timestamp, self.packet))
|
||||
self.call_listener(MsgContainer(self.name, time.time(), self.packet))
|
||||
self.packet = b''
|
||||
self.wait_sync = True
|
||||
|
||||
def is_msg(self, msg: MsgContainer):
|
||||
return True
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
UBX_PKT_DEBUG = True
|
||||
class MySink(MsgListener):
|
||||
|
||||
Reference in New Issue
Block a user