129 lines
3.2 KiB
Python
129 lines
3.2 KiB
Python
import socket
|
|
import selectors
|
|
import time
|
|
|
|
from msg_container import MsgContainer
|
|
from packetizer import Packetizer
|
|
from transceiver import Transceiver
|
|
from a_backend import ABackend
|
|
from nmea.GLL import Gll
|
|
from nmea.GSA import Gsa
|
|
from nmea.GSV import Gsv
|
|
from nmea.GGA import Gga
|
|
|
|
class NetworkBackend(ABackend):
|
|
def __init__(self, servers: list[dict]):
|
|
self.sel = None
|
|
self.sock_list = [{'name': s['name'], 'addr': (s['host'], s['port']), 'xcvr': None} for s in servers]
|
|
|
|
def create_xcvr(self, name: str) -> Transceiver:
|
|
obj = Transceiver(self, name)
|
|
if not self.register_xcvr(obj):
|
|
raise Exception(f"No socket found for \"{name}\"")
|
|
return obj
|
|
|
|
self.sel.register(name, selectors.EVENT_READ, data=None)
|
|
def register_xcvr(self, xcvr: Transceiver):
|
|
found = False
|
|
for sock in self.sock_list:
|
|
if xcvr.name in sock['name']:
|
|
found = True
|
|
sock['xcvr'] = xcvr
|
|
break
|
|
|
|
return found
|
|
def find_by_addr(self, addr):
|
|
for sock in self.sock_list:
|
|
if sock['addr'] == addr:
|
|
return sock
|
|
return None
|
|
|
|
def find_by_name(self, name: str):
|
|
for sock in self.sock_list:
|
|
if sock['name'] == name:
|
|
return sock
|
|
return None
|
|
|
|
def _prepare(self):
|
|
for server in self.sock_list:
|
|
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
|
sock.setblocking(False)
|
|
server['sock'] = sock
|
|
|
|
def connect(self):
|
|
self._prepare()
|
|
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=100)
|
|
for key, mask in events:
|
|
descr = self.find_by_addr(key.fileobj.getpeername())
|
|
name = descr['name']
|
|
sock = self.find_by_name(name)
|
|
rx = sock['xcvr']
|
|
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 = NetworkBackend(server_list)
|
|
for grm in [" neo-f9p", "zed-x20p"]:
|
|
rx_list = [
|
|
Gll(grm, 'GP'),
|
|
Gsa(grm, 'GP'),
|
|
Gsv(grm, 'GP'),
|
|
Gga(grm, 'GP'),
|
|
Gga(grm, 'GN')
|
|
]
|
|
# Topology
|
|
# Receiver-0.0 \
|
|
# Receiver-0.1 + Packetizer-0 - Transceiver-0 \
|
|
# + Backend
|
|
# Receiver-1.0 + Packetizer-1 - Transceiver-1 /
|
|
# Receiver-1.1 /
|
|
|
|
# Create transceiver
|
|
transceiver = mc.create_xcvr(grm)
|
|
|
|
# Create Packetizer
|
|
packetizer = Packetizer()
|
|
|
|
# Register packetizer to transceiver
|
|
transceiver.register_recv(packetizer)
|
|
|
|
# Register Receiver to packetizer
|
|
for rx in rx_list:
|
|
packetizer.register_recv(rx)
|
|
|
|
# Connect all sources
|
|
mc.connect()
|
|
|
|
# receive and block
|
|
mc.event_loop()
|
|
|
|
# code never reached
|
|
mc.disconnect()
|