107 lines
2.8 KiB
Python
107 lines
2.8 KiB
Python
#!/usr/bin/env python
|
|
|
|
from __future__ import print_function, division
|
|
import os
|
|
import sys
|
|
import argparse
|
|
import curses
|
|
from threading import Lock
|
|
import pyModeS as pms
|
|
from pyModeS.extra.tcpclient import BaseClient
|
|
from pyModeS.streamer.stream import Stream
|
|
from pyModeS.streamer.screen import Screen
|
|
|
|
LOCK = Lock()
|
|
ADSB_MSG = []
|
|
ADSB_TS = []
|
|
COMMB_MSG = []
|
|
COMMB_TS = []
|
|
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('--server', help='server address or IP', required=True)
|
|
parser.add_argument('--port', help='raw data port', required=True)
|
|
parser.add_argument('--rawtype', help='beast or avr', required=True)
|
|
parser.add_argument('--latlon', help='receiver position', nargs=2, metavar=('LAT', 'LON'), required=True)
|
|
parser.add_argument('--show-uncertainty', help='display uncertainty indicators', required=False)
|
|
args = parser.parse_args()
|
|
|
|
SERVER = args.server
|
|
PORT = int(args.port)
|
|
RAWTYPE = args.rawtype
|
|
LAT0 = float(args.latlon[0])
|
|
LON0 = float(args.latlon[1])
|
|
|
|
|
|
class ModesClient(BaseClient):
|
|
def __init__(self, host, port, rawtype):
|
|
super(ModesClient, self).__init__(host, port, rawtype)
|
|
|
|
def handle_messages(self, messages):
|
|
local_buffer_adsb_msg = []
|
|
local_buffer_adsb_ts = []
|
|
local_buffer_ehs_msg = []
|
|
local_buffer_ehs_ts = []
|
|
|
|
for msg, t in messages:
|
|
if len(msg) < 28: # only process long messages
|
|
continue
|
|
|
|
df = pms.df(msg)
|
|
|
|
if df == 17 or df == 18:
|
|
local_buffer_adsb_msg.append(msg)
|
|
local_buffer_adsb_ts.append(t)
|
|
elif df == 20 or df == 21:
|
|
local_buffer_ehs_msg.append(msg)
|
|
local_buffer_ehs_ts.append(t)
|
|
else:
|
|
continue
|
|
|
|
|
|
LOCK.acquire()
|
|
ADSB_MSG.extend(local_buffer_adsb_msg)
|
|
ADSB_TS.extend(local_buffer_adsb_ts)
|
|
COMMB_MSG.extend(local_buffer_ehs_msg)
|
|
COMMB_TS.extend(local_buffer_ehs_ts)
|
|
LOCK.release()
|
|
|
|
|
|
# redirect all stdout to null, avoiding messing up with the screen
|
|
sys.stdout = open(os.devnull, 'w')
|
|
|
|
client = ModesClient(host=SERVER, port=PORT, rawtype=RAWTYPE)
|
|
client.daemon = True
|
|
client.start()
|
|
|
|
stream = Stream(lat0=LAT0, lon0=LON0)
|
|
|
|
try:
|
|
screen = Screen()
|
|
screen.daemon = True
|
|
screen.start()
|
|
|
|
while True:
|
|
if len(ADSB_MSG) > 200:
|
|
LOCK.acquire()
|
|
stream.process_raw(ADSB_TS, ADSB_MSG, COMMB_TS, COMMB_MSG)
|
|
ADSB_MSG = []
|
|
ADSB_TS = []
|
|
COMMB_MSG = []
|
|
COMMB_TS = []
|
|
LOCK.release()
|
|
|
|
acs = stream.get_aircraft()
|
|
try:
|
|
screen.update_data(acs)
|
|
screen.update()
|
|
except KeyboardInterrupt:
|
|
raise
|
|
except:
|
|
continue
|
|
|
|
except KeyboardInterrupt:
|
|
sys.exit(0)
|
|
|
|
finally:
|
|
curses.endwin()
|