jens
c6c1f26f32
- added filter by message type to nmea receiver - show case new features in MultiClient.main()
101 lines
2.5 KiB
Python
101 lines
2.5 KiB
Python
import socket
|
|
import selectors
|
|
import time
|
|
|
|
from msg_montainer import MsgContainer
|
|
from nmea import NmeaReceiver
|
|
from msg_sink import MsgSink
|
|
from msg_source import MsgSource
|
|
from packetizer import Packetizer
|
|
|
|
|
|
class MultiClient(MsgSource):
|
|
def __init__(self, servers: list[dict]):
|
|
MsgSource.__init__(self)
|
|
self.sel = None
|
|
self.sock_list = [{'name': s['name'], 'addr': (s['host'], s['port'])} for s in servers]
|
|
self.receivers = {}
|
|
for server in self.sock_list:
|
|
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
|
sock.setblocking(False)
|
|
server['sock'] = sock
|
|
|
|
def register_recv(self, name: str, rx: MsgSink):
|
|
self.receivers[name] = {'rx': rx}
|
|
|
|
def find_by_addr(self, addr):
|
|
for sock in self.sock_list:
|
|
if sock['addr'] == addr:
|
|
return sock
|
|
return None
|
|
|
|
def connect(self):
|
|
self.sel = selectors.DefaultSelector()
|
|
events = selectors.EVENT_READ
|
|
for sock in self.sock_list:
|
|
print(f"Starting connection {sock['name']} to {sock['addr']}")
|
|
self.sel.register(sock['sock'], events, data=None)
|
|
sock['sock'].connect_ex(sock['addr'])
|
|
|
|
def disconnect(self):
|
|
for sock in self.sock_list:
|
|
sock['sock'].close()
|
|
|
|
def event_loop(self):
|
|
try:
|
|
while True:
|
|
events = self.sel.select(timeout=1)
|
|
rx_data = []
|
|
for key, mask in events:
|
|
descr = self.find_by_addr(key.fileobj.getpeername())
|
|
name = descr['name']
|
|
|
|
if name in self.receivers:
|
|
rx = self.receivers[name]['rx']
|
|
if mask & selectors.EVENT_READ:
|
|
timestamp = time.time()
|
|
data = key.fileobj.recv(16)
|
|
rx.on_recv(MsgContainer(name, timestamp, data))
|
|
|
|
except KeyboardInterrupt:
|
|
print("Caught keyboard interrupt, exiting")
|
|
finally:
|
|
self.sel.close()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
server_list = [
|
|
{'name': "neo-f9p", 'host': "192.168.22.93", 'port': 8721},
|
|
{'name': "zed-x20p", 'host': "192.168.22.93", 'port': 8731}
|
|
]
|
|
|
|
# TCPIP multi source receiver
|
|
mc = MultiClient(server_list)
|
|
|
|
# NMEA multi source receiver
|
|
receiver = NmeaReceiver()
|
|
# receiver.filter_by_msg_type('.*')
|
|
receiver.filter_by_msg_type('^.?GPGSV.*')
|
|
receiver.filter_by_msg_type('^.?GNGLL.*')
|
|
|
|
# Packetizer for "zed-x20p"
|
|
packetizer_0 = Packetizer()
|
|
packetizer_0.register_recv("zed-x20p", receiver)
|
|
|
|
# Packetizer for "neo-f9p"
|
|
packetizer_1 = Packetizer()
|
|
packetizer_1.register_recv("neo-f9p", receiver)
|
|
|
|
# Register both packetizers at multi source receiver
|
|
mc.register_recv("zed-x20p", packetizer_0)
|
|
mc.register_recv("neo-f9p", packetizer_1)
|
|
|
|
# Connect all sources
|
|
mc.connect()
|
|
|
|
# receive and block
|
|
mc.event_loop()
|
|
|
|
# code never reached
|
|
mc.disconnect()
|