mirror of
https://github.com/DJ2LS/FreeDATA
synced 2024-05-14 08:04:33 +00:00
319 lines
11 KiB
Python
319 lines
11 KiB
Python
#!/usr/bin/env python3
|
|
|
|
|
|
import structlog
|
|
import threading
|
|
import websocket
|
|
import time
|
|
from queues import AUDIO_TRANSMIT_QUEUE, AUDIO_RECEIVED_QUEUE
|
|
|
|
class TCICtrl:
|
|
def __init__(self, hostname='127.0.0.1', port=50001):
|
|
# websocket.enableTrace(True)
|
|
self.log = structlog.get_logger("TCI")
|
|
|
|
self.audio_received_queue = AUDIO_RECEIVED_QUEUE
|
|
self.audio_transmit_queue = AUDIO_TRANSMIT_QUEUE
|
|
|
|
self.hostname = str(hostname)
|
|
self.port = str(port)
|
|
|
|
self.ws = ''
|
|
|
|
tci_thread = threading.Thread(
|
|
target=self.connect,
|
|
name="TCI THREAD",
|
|
daemon=True,
|
|
)
|
|
tci_thread.start()
|
|
|
|
# flag if we're receiving a tx_chrono
|
|
self.tx_chrono = False
|
|
|
|
# audio related parameters, will be updated by tx chrono
|
|
self.sample_rate = None
|
|
self.format = None
|
|
self.codec = None
|
|
self.audio_length = None
|
|
self.crc = None
|
|
self.channel = None
|
|
|
|
self.frequency = None
|
|
self.bandwidth = None
|
|
self.mode = None
|
|
self.alc = None
|
|
self.meter = None
|
|
self.level = None
|
|
self.ptt = None
|
|
|
|
def connect(self):
|
|
self.log.info(
|
|
"[TCI] Starting TCI thread!", ip=self.hostname, port=self.port
|
|
)
|
|
self.ws = websocket.WebSocketApp(
|
|
f"ws://{self.hostname}:{self.port}",
|
|
on_open=self.on_open,
|
|
on_message=self.on_message,
|
|
on_error=self.on_error,
|
|
on_close=self.on_close,
|
|
)
|
|
|
|
self.ws.run_forever(reconnect=5) # Set dispatcher to automatic reconnection, 5 second reconnect delay if con>
|
|
# rel.signal(2, rel.abort) # Keyboard Interrupt
|
|
# rel.dispatch()
|
|
|
|
def on_message(self, ws, message):
|
|
|
|
# ready message
|
|
# we need to wait until radio is ready before we can push commands
|
|
if message == "ready;":
|
|
self.ws.send('audio_samplerate:8000;')
|
|
self.ws.send('audio_stream_channels:1;')
|
|
self.ws.send('audio_stream_sample_type:int16;')
|
|
self.ws.send('audio_stream_samples:1200;')
|
|
self.ws.send('audio_start:0;')
|
|
|
|
# tx chrono frame
|
|
if len(message) in {64}:
|
|
receiver = message[:4]
|
|
sample_rate = int.from_bytes(message[4:8], "little")
|
|
format = int.from_bytes(message[8:12], "little")
|
|
codec = int.from_bytes(message[12:16], "little")
|
|
crc = int.from_bytes(message[16:20], "little")
|
|
audio_length = int.from_bytes(message[20:24], "little")
|
|
type = int.from_bytes(message[24:28], "little")
|
|
channel = int.from_bytes(message[28:32], "little")
|
|
reserved1 = int.from_bytes(message[32:36], "little")
|
|
reserved2 = int.from_bytes(message[36:40], "little")
|
|
reserved3 = int.from_bytes(message[40:44], "little")
|
|
reserved4 = int.from_bytes(message[44:48], "little")
|
|
reserved5 = int.from_bytes(message[48:52], "little")
|
|
reserved6 = int.from_bytes(message[52:56], "little")
|
|
reserved7 = int.from_bytes(message[56:60], "little")
|
|
reserved8 = int.from_bytes(message[60:64], "little")
|
|
if type == 3:
|
|
self.tx_chrono = True
|
|
|
|
self.sample_rate = sample_rate
|
|
self.format = format
|
|
self.codec = codec
|
|
self.audio_length = audio_length
|
|
self.channel = channel
|
|
self.crc = crc
|
|
|
|
# audio frame
|
|
if len(message) in {576, 2464, 4160}:
|
|
# audio received
|
|
receiver = message[:4]
|
|
sample_rate = int.from_bytes(message[4:8], "little")
|
|
format = int.from_bytes(message[8:12], "little")
|
|
codec = int.from_bytes(message[12:16], "little")
|
|
crc = int.from_bytes(message[16:20], "little")
|
|
audio_length = int.from_bytes(message[20:24], "little")
|
|
type = int.from_bytes(message[24:28], "little")
|
|
channel = int.from_bytes(message[28:32], "little")
|
|
reserved1 = int.from_bytes(message[32:36], "little")
|
|
reserved2 = int.from_bytes(message[36:40], "little")
|
|
reserved3 = int.from_bytes(message[40:44], "little")
|
|
reserved4 = int.from_bytes(message[44:48], "little")
|
|
reserved5 = int.from_bytes(message[48:52], "little")
|
|
reserved6 = int.from_bytes(message[52:56], "little")
|
|
reserved7 = int.from_bytes(message[56:60], "little")
|
|
reserved8 = int.from_bytes(message[60:64], "little")
|
|
audio_data = message[64:]
|
|
self.audio_received_queue.put(audio_data)
|
|
|
|
|
|
if len(message)< 64:
|
|
# find frequency
|
|
if bytes(message, "utf-8").startswith(b"vfo:0,0,"):
|
|
splitted_message = message.split("vfo:0,0,")
|
|
self.frequency = splitted_message[1][:-1]
|
|
|
|
# find mode
|
|
if bytes(message, "utf-8").startswith(b"modulation:0,"):
|
|
splitted_message = message.split("modulation:0,")
|
|
self.mode = splitted_message[1][:-1]
|
|
|
|
# find ptt
|
|
#if bytes(message, "utf-8").startswith(b"trx:0,"):
|
|
# splitted_message = message.split("trx:0,")
|
|
# self.ptt = splitted_message[1][:-1]
|
|
|
|
# find bandwidth
|
|
#if message.startswith("rx_filter_band:0,"):
|
|
# splitted_message = message.split("rx_filter_band:0,")
|
|
# bandwidths = splitted_message[1]
|
|
# splitted_bandwidths = bandwidths.split(",")
|
|
# lower_bandwidth = int(splitted_bandwidths[0])
|
|
# upper_bandwidth = int(splitted_bandwidths[1][:-1])
|
|
# self.bandwidth = upper_bandwidth - lower_bandwidth
|
|
|
|
|
|
|
|
|
|
|
|
def on_error(self, ws, error):
|
|
self.log.error(
|
|
"[TCI] Error FreeDATA to TCI rig!", ip=self.hostname, port=self.port, e=error
|
|
)
|
|
|
|
def on_close(self, ws, close_status_code, close_msg):
|
|
self.log.warning(
|
|
"[TCI] Closed FreeDATA to TCI connection!", ip=self.hostname, port=self.port, statu=close_status_code,
|
|
msg=close_msg
|
|
)
|
|
|
|
def on_open(self, ws):
|
|
self.ws = ws
|
|
self.log.info(
|
|
"[TCI] Connected FreeDATA to TCI rig!", ip=self.hostname, port=self.port
|
|
)
|
|
|
|
self.log.info(
|
|
"[TCI] Init...", ip=self.hostname, port=self.port
|
|
)
|
|
|
|
def push_audio(self, data_out):
|
|
#print(data_out)
|
|
|
|
"""
|
|
# audio[:4] = receiver.to_bytes(4,byteorder='little', signed=False)
|
|
audio[4:8] = sample_rate.to_bytes(4, byteorder='little', signed=False)
|
|
audio[8:12] = format.to_bytes(4, byteorder='little', signed=False)
|
|
audio[12:16] = codec.to_bytes(4, byteorder='little', signed=False)
|
|
audio[16:20] = crc.to_bytes(4, byteorder='little', signed=False)
|
|
audio[20:24] = audio_length.to_bytes(4, byteorder='little', signed=False)
|
|
audio[24:28] = int(2).to_bytes(4, byteorder='little', signed=True)
|
|
audio[28:32] = channel.to_bytes(4, byteorder='little', signed=False)
|
|
audio[32:36] = reserved1.to_bytes(4, byteorder='little', signed=False)
|
|
audio[36:40] = reserved2.to_bytes(4, byteorder='little', signed=False)
|
|
audio[40:44] = reserved3.to_bytes(4, byteorder='little', signed=False)
|
|
audio[44:48] = reserved4.to_bytes(4, byteorder='little', signed=False)
|
|
audio[48:52] = reserved5.to_bytes(4, byteorder='little', signed=False)
|
|
audio[52:56] = reserved6.to_bytes(4, byteorder='little', signed=False)
|
|
audio[56:60] = reserved7.to_bytes(4, byteorder='little', signed=False)
|
|
audio[60:64] = reserved8.to_bytes(4, byteorder='little', signed=False)
|
|
"""
|
|
|
|
while not self.tx_chrono:
|
|
time.sleep(0.01)
|
|
|
|
#print(len(data_out))
|
|
#print(self.sample_rate)
|
|
#print(self.audio_length)
|
|
#print(self.channel)
|
|
#print(self.crc)
|
|
#print(self.codec)
|
|
#print(self.tx_chrono)
|
|
|
|
if self.tx_chrono:
|
|
#print("#############")
|
|
#print(len(data_out))
|
|
#print(len(bytes(data_out)))
|
|
#print("-------------")
|
|
audio = bytearray(4096 + 64)
|
|
|
|
audio[64:64 + len(bytes(data_out))] = bytes(data_out)
|
|
audio[4:8] = self.sample_rate.to_bytes(4, byteorder='little', signed=False)
|
|
# audio[8:12] = format.to_bytes(4,byteorder='little', signed=False)
|
|
audio[12:16] = self.codec.to_bytes(4, byteorder='little', signed=False)
|
|
audio[16:20] = self.crc.to_bytes(4, byteorder='little', signed=False)
|
|
audio[20:24] = self.audio_length.to_bytes(4, byteorder='little', signed=False)
|
|
audio[24:28] = int(2).to_bytes(4, byteorder='little', signed=False)
|
|
audio[28:32] = self.channel.to_bytes(4, byteorder='little', signed=False)
|
|
# audio[32:36] = reserved1.to_bytes(4,byteorder='little', signed=False)
|
|
# audio[36:40] = reserved2.to_bytes(4,byteorder='little', signed=False)
|
|
# audio[40:44] = reserved3.to_bytes(4,byteorder='little', signed=False)
|
|
# audio[44:48] = reserved4.to_bytes(4,byteorder='little', signed=False)
|
|
# audio[48:52] = reserved5.to_bytes(4,byteorder='little', signed=False)
|
|
# audio[52:56] = reserved6.to_bytes(4,byteorder='little', signed=False)
|
|
# audio[56:60] = reserved7.to_bytes(4,byteorder='little', signed=False)
|
|
|
|
self.ws.send(audio, websocket.ABNF.OPCODE_BINARY)
|
|
|
|
def set_ptt(self, state):
|
|
if state:
|
|
self.ws.send('trx:0,true,tci;')
|
|
else:
|
|
|
|
self.ws.send('trx:0,false;')
|
|
self.tx_chrono = False
|
|
|
|
def get_frequency(self):
|
|
""" """
|
|
self.ws.send('VFO:0,0;')
|
|
return self.frequency
|
|
|
|
def get_mode(self):
|
|
""" """
|
|
self.ws.send('MODULATION:0;')
|
|
return self.mode
|
|
|
|
def get_level(self):
|
|
""" """
|
|
return self.level
|
|
|
|
def get_alc(self):
|
|
""" """
|
|
return self.alc
|
|
|
|
def get_meter(self):
|
|
""" """
|
|
return self.meter
|
|
|
|
def get_bandwidth(self):
|
|
""" """
|
|
return self.bandwidth
|
|
|
|
def get_strength(self):
|
|
""" """
|
|
return None
|
|
|
|
def set_bandwidth(self):
|
|
""" """
|
|
return None
|
|
|
|
def set_mode(self, mode):
|
|
"""
|
|
|
|
Args:
|
|
mode:
|
|
|
|
Returns:
|
|
|
|
"""
|
|
self.ws.send(f'MODULATION:0,{str(mode)};')
|
|
return None
|
|
|
|
def set_frequency(self, frequency):
|
|
"""
|
|
|
|
Args:
|
|
frequency:
|
|
|
|
Returns:
|
|
|
|
"""
|
|
self.ws.send(f'VFO:0,0,{str(frequency)};')
|
|
return None
|
|
|
|
def get_status(self):
|
|
"""
|
|
|
|
Args:
|
|
mode:
|
|
|
|
Returns:
|
|
|
|
"""
|
|
return "connected"
|
|
|
|
def get_ptt(self):
|
|
""" """
|
|
self.ws.send('trx:0;')
|
|
return self.ptt
|
|
|
|
def close_rig(self):
|
|
""" """
|
|
return
|