mirror of https://github.com/DJ2LS/FreeDATA.git
smaller PEP8 cleanup
DJ2LS
GitHub
parent
0dccb0a3d4
commit
36b7ac91f8
1106
data_handler.py
1106
data_handler.py
File diff suppressed because it is too large
Load Diff
103
helpers.py
103
helpers.py
|
|
@ -7,31 +7,27 @@ Created on Fri Dec 25 21:25:14 2020
|
|||
"""
|
||||
|
||||
import time
|
||||
import threading
|
||||
import logging
|
||||
import crcengine
|
||||
import pyaudio
|
||||
import asyncio
|
||||
import crcengine
|
||||
|
||||
import data_handler
|
||||
|
||||
import static
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
def get_crc_8(data):
|
||||
crc_algorithm = crcengine.new('crc8-ccitt') #load crc8 library
|
||||
crc_algorithm = crcengine.new('crc8-ccitt') # load crc8 library
|
||||
crc_data = crc_algorithm(data)
|
||||
crc_data = crc_data.to_bytes(1, byteorder='big')
|
||||
crc_data = crc_data.to_bytes(1, byteorder='big')
|
||||
return crc_data
|
||||
|
||||
|
||||
def get_crc_16(data):
|
||||
crc_algorithm = crcengine.new('crc16-ccitt-false') #load crc16 library
|
||||
crc_algorithm = crcengine.new('crc16-ccitt-false') # load crc16 library
|
||||
crc_data = crc_algorithm(data)
|
||||
crc_data = crc_data.to_bytes(2, byteorder='big')
|
||||
return crc_data
|
||||
crc_data = crc_data.to_bytes(2, byteorder='big')
|
||||
return crc_data
|
||||
|
||||
|
||||
async def set_after_timeout():
|
||||
while True:
|
||||
|
|
@ -40,41 +36,45 @@ async def set_after_timeout():
|
|||
print("HALLOIOIOIOIOIOI")
|
||||
static.ARQ_RX_ACK_TIMEOUT = True
|
||||
await asyncio.sleep(1.1)
|
||||
#await asyncio.sleep(timeout)
|
||||
# await asyncio.sleep(timeout)
|
||||
#vars()[variable] = value
|
||||
|
||||
|
||||
|
||||
|
||||
def arq_disconnect_timeout():
|
||||
static.ARQ_WAIT_FOR_DISCONNECT = True
|
||||
logging.debug("ARQ_WAIT_FOR_DISCONNECT")
|
||||
|
||||
|
||||
logging.debug("ARQ_WAIT_FOR_DISCONNECT")
|
||||
|
||||
|
||||
def arq_ack_timeout():
|
||||
if static.CHANNEL_STATE == 'RECEIVING_SIGNALLING':
|
||||
static.ARQ_RX_ACK_TIMEOUT = True
|
||||
logging.debug("ARQ_RX_ACK_TIMEOUT")
|
||||
|
||||
|
||||
|
||||
def arq_rpt_timeout():
|
||||
if static.CHANNEL_STATE == 'RECEIVING_SIGNALLING':
|
||||
static.ARQ_RX_RPT_TIMEOUT = True
|
||||
logging.debug("ARQ_RX_RPT_TIMEOUT")
|
||||
static.ARQ_RX_RPT_TIMEOUT = True
|
||||
logging.debug("ARQ_RX_RPT_TIMEOUT")
|
||||
|
||||
|
||||
def arq_frame_timeout():
|
||||
if static.CHANNEL_STATE == 'RECEIVING_SIGNALLING':
|
||||
static.ARQ_RX_FRAME_TIMEOUT = True
|
||||
logging.debug("ARQ_RX_FRAME_TIMEOUT")
|
||||
|
||||
static.ARQ_RX_FRAME_TIMEOUT = True
|
||||
logging.debug("ARQ_RX_FRAME_TIMEOUT")
|
||||
|
||||
|
||||
def arq_reset_timeout(state):
|
||||
static.ARQ_RX_ACK_TIMEOUT = state
|
||||
static.ARQ_RX_FRAME_TIMEOUT = state
|
||||
static.ARQ_RX_RPT_TIMEOUT = state
|
||||
|
||||
def arq_reset_ack(state):
|
||||
|
||||
|
||||
def arq_reset_ack(state):
|
||||
static.ARQ_ACK_RECEIVED = state
|
||||
static.ARQ_RPT_RECEIVED = state
|
||||
static.ARQ_FRAME_ACK_RECEIVED = state
|
||||
|
||||
|
||||
|
||||
def arq_reset_frame_machine():
|
||||
arq_reset_timeout(False)
|
||||
arq_reset_ack(False)
|
||||
|
|
@ -84,42 +84,41 @@ def arq_reset_frame_machine():
|
|||
static.ARQ_TX_N_CURRENT_ARQ_FRAME = 0
|
||||
static.ARQ_TX_N_TOTAL_ARQ_FRAMES = 0
|
||||
static.ARQ_TX_N_CURRENT_ARQ_FRAME = 0
|
||||
|
||||
|
||||
static.ARQ_RX_N_CURRENT_ARQ_FRAME = 0
|
||||
static.ARQ_N_ARQ_FRAMES_PER_DATA_FRAME = 0
|
||||
static.ARQ_FRAME_BOF_RECEIVED = False
|
||||
static.ARQ_FRAME_EOF_RECEIVED = False
|
||||
|
||||
static.ARQ_FRAME_EOF_RECEIVED = False
|
||||
|
||||
static.TNC_STATE = 'IDLE'
|
||||
static.TNC_STATE = 'IDLE'
|
||||
static.ARQ_SEND_KEEP_ALIVE = True
|
||||
static.CHANNEL_STATE = 'RECEIVING_SIGNALLING'
|
||||
static.ARQ_READY_FOR_DATA = False
|
||||
|
||||
#start sending keep alive after some seconds
|
||||
|
||||
# start sending keep alive after some seconds
|
||||
#acktimer = threading.Timer(3.0, data_handler.arq_connect)
|
||||
#acktimer.start()
|
||||
#await asyncio.sleep(2)
|
||||
#modem.transmit_arq_connect()
|
||||
# acktimer.start()
|
||||
# await asyncio.sleep(2)
|
||||
# modem.transmit_arq_connect()
|
||||
|
||||
|
||||
def setup_logging():
|
||||
|
||||
|
||||
logging.basicConfig(format='%(asctime)s.%(msecs)03d %(levelname)s:\t%(message)s', datefmt='%H:%M:%S', level=logging.INFO)
|
||||
|
||||
logging.addLevelName( logging.DEBUG, "\033[1;36m%s\033[1;0m" % logging.getLevelName(logging.DEBUG))
|
||||
logging.addLevelName( logging.INFO, "\033[1;37m%s\033[1;0m" % logging.getLevelName(logging.INFO))
|
||||
logging.addLevelName( logging.WARNING, "\033[1;33m%s\033[1;0m" % logging.getLevelName(logging.WARNING))
|
||||
logging.addLevelName( logging.ERROR, "\033[1;31m%s\033[1;0m" % "FAILED")
|
||||
logging.addLevelName(logging.DEBUG, "\033[1;36m%s\033[1;0m" % logging.getLevelName(logging.DEBUG))
|
||||
logging.addLevelName(logging.INFO, "\033[1;37m%s\033[1;0m" % logging.getLevelName(logging.INFO))
|
||||
logging.addLevelName(logging.WARNING, "\033[1;33m%s\033[1;0m" % logging.getLevelName(logging.WARNING))
|
||||
logging.addLevelName(logging.ERROR, "\033[1;31m%s\033[1;0m" % "FAILED")
|
||||
#logging.addLevelName( logging.ERROR, "\033[1;31m%s\033[1;0m" % logging.getLevelName(logging.ERROR))
|
||||
logging.addLevelName( logging.CRITICAL, "\033[1;41m%s\033[1;0m" % logging.getLevelName(logging.CRITICAL))
|
||||
|
||||
logging.addLevelName( 25, "\033[1;32m%s\033[1;0m" % "SUCCESS")
|
||||
logging.addLevelName( 24, "\033[1;34m%s\033[1;0m" % "DATA")
|
||||
logging.addLevelName(logging.CRITICAL, "\033[1;41m%s\033[1;0m" % logging.getLevelName(logging.CRITICAL))
|
||||
|
||||
logging.addLevelName(25, "\033[1;32m%s\033[1;0m" % "SUCCESS")
|
||||
logging.addLevelName(24, "\033[1;34m%s\033[1;0m" % "DATA")
|
||||
|
||||
|
||||
# https://stackoverflow.com/questions/384076/how-can-i-color-python-logging-output
|
||||
#'DEBUG' : 37, # white
|
||||
#'INFO' : 36, # cyan
|
||||
#'WARNING' : 33, # yellow
|
||||
#'ERROR' : 31, # red
|
||||
#'CRITICAL': 41, # white on red bg
|
||||
|
||||
# 'DEBUG' : 37, # white
|
||||
# 'INFO' : 36, # cyan
|
||||
# 'WARNING' : 33, # yellow
|
||||
# 'ERROR' : 31, # red
|
||||
# 'CRITICAL': 41, # white on red bg
|
||||
|
|
|
|||
20
main.py
20
main.py
|
|
@ -17,23 +17,23 @@ if __name__ == '__main__':
|
|||
|
||||
# config logging
|
||||
helpers.setup_logging()
|
||||
|
||||
#--------------------------------------------GET PARAMETER INPUTS
|
||||
|
||||
# --------------------------------------------GET PARAMETER INPUTS
|
||||
PARSER = argparse.ArgumentParser(description='Simons TEST TNC')
|
||||
PARSER.add_argument('--rx', dest="audio_input_device", default=0, help="listening sound card", type=int)
|
||||
PARSER.add_argument('--tx', dest="audio_output_device", default=0, help="transmitting sound card", type=int)
|
||||
PARSER.add_argument('--port', dest="socket_port", default=3000, help="Socket port", type=int)
|
||||
#parser.add_argument('--mode', dest="freedv_data_mode", default=12, help="Set the mode.", type=int)
|
||||
|
||||
PARSER.add_argument('--port', dest="socket_port", default=3000, help="Socket port", type=int)
|
||||
# parser.add_argument('--mode', dest="freedv_data_mode", default=12, help="Set the mode.", type=int)
|
||||
|
||||
ARGS = PARSER.parse_args()
|
||||
|
||||
#static.FREEDV_DATA_MODE = args.freedv_data_mode
|
||||
|
||||
# static.FREEDV_DATA_MODE = args.freedv_data_mode
|
||||
static.AUDIO_INPUT_DEVICE = ARGS.audio_input_device
|
||||
static.AUDIO_OUTPUT_DEVICE = ARGS.audio_output_device
|
||||
static.PORT = ARGS.socket_port
|
||||
|
||||
#--------------------------------------------START CMD SERVER
|
||||
import sock # we need to wait until we got all parameters from argparse first
|
||||
|
||||
# --------------------------------------------START CMD SERVER
|
||||
import sock # we need to wait until we got all parameters from argparse first
|
||||
|
||||
CMD_SERVER_THREAD = threading.Thread(target=sock.start_cmd_socket, name="cmd server")
|
||||
CMD_SERVER_THREAD.start()
|
||||
|
|
|
|||
447
modem.py
447
modem.py
|
|
@ -24,68 +24,63 @@ import data_handler
|
|||
import Hamlib
|
||||
|
||||
|
||||
|
||||
class RF():
|
||||
|
||||
def __init__(self):
|
||||
#-------------------------------------------- LOAD FREEDV
|
||||
|
||||
def __init__(self):
|
||||
# -------------------------------------------- LOAD FREEDV
|
||||
libname = pathlib.Path().absolute() / "codec2/build_linux/src/libcodec2.so"
|
||||
self.c_lib = ctypes.CDLL(libname)
|
||||
#--------------------------------------------CREATE PYAUDIO INSTANCE
|
||||
# --------------------------------------------CREATE PYAUDIO INSTANCE
|
||||
self.p = pyaudio.PyAudio()
|
||||
#--------------------------------------------GET SUPPORTED SAMPLE RATES FROM SOUND DEVICE
|
||||
# --------------------------------------------GET SUPPORTED SAMPLE RATES FROM SOUND DEVICE
|
||||
#static.AUDIO_SAMPLE_RATE_RX = int(self.p.get_device_info_by_index(static.AUDIO_INPUT_DEVICE)['defaultSampleRate'])
|
||||
#static.AUDIO_SAMPLE_RATE_TX = int(self.p.get_device_info_by_index(static.AUDIO_OUTPUT_DEVICE)['defaultSampleRate'])
|
||||
static.AUDIO_SAMPLE_RATE_TX = 8000
|
||||
static.AUDIO_SAMPLE_RATE_RX = 8000
|
||||
#--------------------------------------------OPEN AUDIO CHANNEL RX
|
||||
self.stream_rx = self.p.open(format=pyaudio.paInt16,
|
||||
channels=static.AUDIO_CHANNELS,
|
||||
rate=static.AUDIO_SAMPLE_RATE_RX,
|
||||
frames_per_buffer=static.AUDIO_FRAMES_PER_BUFFER,
|
||||
input=True,
|
||||
input_device_index=static.AUDIO_INPUT_DEVICE,
|
||||
)
|
||||
#--------------------------------------------OPEN AUDIO CHANNEL TX
|
||||
# --------------------------------------------OPEN AUDIO CHANNEL RX
|
||||
self.stream_rx = self.p.open(format=pyaudio.paInt16,
|
||||
channels=static.AUDIO_CHANNELS,
|
||||
rate=static.AUDIO_SAMPLE_RATE_RX,
|
||||
frames_per_buffer=static.AUDIO_FRAMES_PER_BUFFER,
|
||||
input=True,
|
||||
input_device_index=static.AUDIO_INPUT_DEVICE,
|
||||
)
|
||||
# --------------------------------------------OPEN AUDIO CHANNEL TX
|
||||
self.stream_tx = self.p.open(format=pyaudio.paInt16,
|
||||
channels=1,
|
||||
rate=static.AUDIO_SAMPLE_RATE_TX,
|
||||
frames_per_buffer=static.AUDIO_FRAMES_PER_BUFFER, #n_nom_modem_samples
|
||||
output=True,
|
||||
output_device_index=static.AUDIO_OUTPUT_DEVICE, #static.AUDIO_OUTPUT_DEVICE
|
||||
)
|
||||
|
||||
|
||||
channels=1,
|
||||
rate=static.AUDIO_SAMPLE_RATE_TX,
|
||||
frames_per_buffer=static.AUDIO_FRAMES_PER_BUFFER, # n_nom_modem_samples
|
||||
output=True,
|
||||
output_device_index=static.AUDIO_OUTPUT_DEVICE, # static.AUDIO_OUTPUT_DEVICE
|
||||
)
|
||||
|
||||
self.streambuffer = bytes(0)
|
||||
self.audio_writing_to_stream = False
|
||||
#--------------------------------------------START DECODER THREAD
|
||||
self.audio_writing_to_stream = False
|
||||
# --------------------------------------------START DECODER THREAD
|
||||
FREEDV_DECODER_THREAD_10 = threading.Thread(target=self.receive, args=[10], name="FREEDV_DECODER_THREAD_10")
|
||||
FREEDV_DECODER_THREAD_10.start()
|
||||
|
||||
|
||||
FREEDV_DECODER_THREAD_11 = threading.Thread(target=self.receive, args=[11], name="FREEDV_DECODER_THREAD_11")
|
||||
FREEDV_DECODER_THREAD_11.start()
|
||||
|
||||
|
||||
FREEDV_DECODER_THREAD_12 = threading.Thread(target=self.receive, args=[12], name="FREEDV_DECODER_THREAD_12")
|
||||
FREEDV_DECODER_THREAD_12.start()
|
||||
|
||||
|
||||
FREEDV_DECODER_THREAD_14 = threading.Thread(target=self.receive, args=[static.FREEDV_SIGNALLING_MODE], name="FREEDV_DECODER_THREAD_14")
|
||||
FREEDV_DECODER_THREAD_14.start()
|
||||
|
||||
|
||||
FREEDV_PLAYBACK_THREAD = threading.Thread(target=self.play_audio, name="FREEDV_DECODER_THREAD_14")
|
||||
FREEDV_PLAYBACK_THREAD.start()
|
||||
|
||||
|
||||
#--------------------------------------------CONFIGURE HAMLIB
|
||||
|
||||
# --------------------------------------------CONFIGURE HAMLIB
|
||||
Hamlib.rig_set_debug(Hamlib.RIG_DEBUG_NONE)
|
||||
|
||||
# Init RIG_MODEL_DUMMY
|
||||
self.my_rig = Hamlib.Rig(Hamlib.RIG_MODEL_DUMMY)
|
||||
self.my_rig.set_conf("rig_pathname", "/dev/Rig")
|
||||
self.my_rig.set_conf("retry", "5")
|
||||
self.my_rig.open ()
|
||||
|
||||
|
||||
|
||||
self.my_rig.open()
|
||||
|
||||
if static.HAMLIB_PTT_TYPE == 'RIG_PTT_RIG':
|
||||
self.hamlib_ptt_type = Hamlib.RIG_PTT_RIG
|
||||
elif static.HAMLIB_PTT_TYPE == 'RIG_PTT_SERIAL_DTR':
|
||||
|
|
@ -98,284 +93,282 @@ class RF():
|
|||
self.hamlib_ptt_type = Hamlib.RIG_PTT_RIG_MICDATA
|
||||
elif static.HAMLIB_PTT_TYPE == 'RIG_PTT_CM108':
|
||||
self.hamlib_ptt_type = Hamlib.RIG_PTT_CM108
|
||||
else:# static.HAMLIB_PTT_TYPE == 'RIG_PTT_NONE':
|
||||
else: # static.HAMLIB_PTT_TYPE == 'RIG_PTT_NONE':
|
||||
self.hamlib_ptt_type = Hamlib.RIG_PTT_NONE
|
||||
|
||||
self.my_rig.set_ptt(self.hamlib_ptt_type,0)
|
||||
|
||||
#--------------------------------------------------------------------------------------------------------
|
||||
|
||||
self.my_rig.set_ptt(self.hamlib_ptt_type, 0)
|
||||
|
||||
# --------------------------------------------------------------------------------------------------------
|
||||
def play_audio(self):
|
||||
|
||||
while True:
|
||||
|
||||
while True:
|
||||
time.sleep(0.01)
|
||||
#state_before_transmit = static.CHANNEL_STATE
|
||||
|
||||
while len(self.streambuffer) > 0:
|
||||
time.sleep(0.01)
|
||||
time.sleep(0.01)
|
||||
if len(self.streambuffer) > 0:
|
||||
#print(self.streambuffer)
|
||||
# print(self.streambuffer)
|
||||
self.audio_writing_to_stream = True
|
||||
self.stream_tx.write(self.streambuffer)
|
||||
self.streambuffer = bytes()
|
||||
#static.CHANNEL_STATE = state_before_transmit
|
||||
self.audio_writing_to_stream = False
|
||||
#--------------------------------------------------------------------------------------------------------
|
||||
def transmit_signalling(self,data_out):
|
||||
# --------------------------------------------------------------------------------------------------------
|
||||
|
||||
def transmit_signalling(self, data_out):
|
||||
|
||||
state_before_transmit = static.CHANNEL_STATE
|
||||
static.CHANNEL_STATE = 'SENDING_SIGNALLING'
|
||||
static.CHANNEL_STATE = 'SENDING_SIGNALLING'
|
||||
static.PTT_STATE = True
|
||||
self.my_rig.set_ptt(self.hamlib_ptt_type,1)
|
||||
|
||||
self.my_rig.set_ptt(self.hamlib_ptt_type, 1)
|
||||
|
||||
self.c_lib.freedv_open.restype = ctypes.POINTER(ctypes.c_ubyte)
|
||||
freedv = self.c_lib.freedv_open(static.FREEDV_SIGNALLING_MODE)
|
||||
bytes_per_frame = int(self.c_lib.freedv_get_bits_per_modem_frame(freedv)/8)
|
||||
payload_per_frame = bytes_per_frame -2
|
||||
bytes_per_frame = int(self.c_lib.freedv_get_bits_per_modem_frame(freedv) / 8)
|
||||
payload_per_frame = bytes_per_frame - 2
|
||||
n_nom_modem_samples = self.c_lib.freedv_get_n_nom_modem_samples(freedv)
|
||||
n_tx_modem_samples = self.c_lib.freedv_get_n_tx_modem_samples(freedv) #get n_tx_modem_samples which defines the size of the modulation object
|
||||
n_tx_preamble_modem_samples = self.c_lib.freedv_get_n_tx_preamble_modem_samples(freedv)
|
||||
|
||||
n_tx_modem_samples = self.c_lib.freedv_get_n_tx_modem_samples(freedv) # get n_tx_modem_samples which defines the size of the modulation object
|
||||
n_tx_preamble_modem_samples = self.c_lib.freedv_get_n_tx_preamble_modem_samples(freedv)
|
||||
|
||||
mod_out = ctypes.c_short * n_tx_modem_samples
|
||||
mod_out = mod_out()
|
||||
mod_out_preamble = ctypes.c_short * n_tx_preamble_modem_samples #*2 #1760 for mode 10,11,12 #4000 for mode 9
|
||||
mod_out_preamble = ctypes.c_short * n_tx_preamble_modem_samples # *2 #1760 for mode 10,11,12 #4000 for mode 9
|
||||
mod_out_preamble = mod_out_preamble()
|
||||
|
||||
buffer = bytearray(payload_per_frame) # use this if CRC16 checksum is required ( DATA1-3)
|
||||
buffer[:len(data_out)] = data_out # set buffersize to length of data which will be send
|
||||
buffer = bytearray(payload_per_frame) # use this if CRC16 checksum is required ( DATA1-3)
|
||||
buffer[:len(data_out)] = data_out # set buffersize to length of data which will be send
|
||||
|
||||
crc = ctypes.c_ushort(self.c_lib.freedv_gen_crc16(bytes(buffer), payload_per_frame)) # generate CRC16
|
||||
crc = crc.value.to_bytes(2, byteorder='big') # convert crc to 2 byte hex string
|
||||
crc = crc.value.to_bytes(2, byteorder='big') # convert crc to 2 byte hex string
|
||||
buffer += crc # append crc16 to buffer
|
||||
data = (ctypes.c_ubyte * bytes_per_frame).from_buffer_copy(buffer)
|
||||
|
||||
self.c_lib.freedv_rawdatapreambletx(freedv, mod_out_preamble)
|
||||
self.c_lib.freedv_rawdatatx(freedv,mod_out,data) # modulate DATA and safe it into mod_out pointer
|
||||
|
||||
txbuffer = bytearray()
|
||||
|
||||
self.c_lib.freedv_rawdatapreambletx(freedv, mod_out_preamble)
|
||||
self.c_lib.freedv_rawdatatx(freedv, mod_out, data) # modulate DATA and safe it into mod_out pointer
|
||||
|
||||
txbuffer = bytearray()
|
||||
txbuffer += bytes(mod_out_preamble)
|
||||
txbuffer += bytes(mod_out)
|
||||
|
||||
|
||||
# -------------- transmit audio
|
||||
logging.debug("SENDING SIGNALLING FRAME " + str(data_out))
|
||||
|
||||
self.streambuffer = bytes()
|
||||
|
||||
self.streambuffer = bytes()
|
||||
self.streambuffer = bytes(txbuffer)
|
||||
self.audio_writing_to_stream = True
|
||||
|
||||
#wait until audio has been processed
|
||||
# wait until audio has been processed
|
||||
while self.audio_writing_to_stream == True:
|
||||
time.sleep(0.01)
|
||||
static.CHANNEL_STATE = 'SENDING_SIGNALLING'
|
||||
#print("sending signalling...")
|
||||
|
||||
self.my_rig.set_ptt(self.hamlib_ptt_type,0)
|
||||
static.PTT_STATE = False
|
||||
|
||||
self.my_rig.set_ptt(self.hamlib_ptt_type, 0)
|
||||
static.PTT_STATE = False
|
||||
static.CHANNEL_STATE = state_before_transmit
|
||||
|
||||
self.c_lib.freedv_close(freedv)
|
||||
#time.sleep(0.5)
|
||||
#--------------------------------------------------------------------------------------------------------
|
||||
# GET ARQ BURST FRAME VOM BUFFER AND MODULATE IT
|
||||
|
||||
self.c_lib.freedv_close(freedv)
|
||||
# time.sleep(0.5)
|
||||
# --------------------------------------------------------------------------------------------------------
|
||||
# GET ARQ BURST FRAME VOM BUFFER AND MODULATE IT
|
||||
|
||||
def transmit_arq_burst(self):
|
||||
|
||||
self.my_rig.set_ptt(self.hamlib_ptt_type,1)
|
||||
|
||||
self.my_rig.set_ptt(self.hamlib_ptt_type, 1)
|
||||
static.PTT_STATE = True
|
||||
state_before_transmit = static.CHANNEL_STATE
|
||||
static.CHANNEL_STATE = 'SENDING_DATA'
|
||||
|
||||
self.c_lib.freedv_open.restype = ctypes.POINTER(ctypes.c_ubyte)
|
||||
freedv = self.c_lib.freedv_open(static.ARQ_DATA_CHANNEL_MODE)
|
||||
|
||||
static.FREEDV_DATA_BYTES_PER_FRAME = int(self.c_lib.freedv_get_bits_per_modem_frame(freedv)/8)
|
||||
static.FREEDV_DATA_PAYLOAD_PER_FRAME = static.FREEDV_DATA_BYTES_PER_FRAME -2
|
||||
|
||||
|
||||
static.FREEDV_DATA_BYTES_PER_FRAME = int(self.c_lib.freedv_get_bits_per_modem_frame(freedv) / 8)
|
||||
static.FREEDV_DATA_PAYLOAD_PER_FRAME = static.FREEDV_DATA_BYTES_PER_FRAME - 2
|
||||
|
||||
n_nom_modem_samples = self.c_lib.freedv_get_n_nom_modem_samples(freedv)
|
||||
n_tx_modem_samples = self.c_lib.freedv_get_n_tx_modem_samples(freedv)#*2 #get n_tx_modem_samples which defines the size of the modulation object
|
||||
n_tx_modem_samples = self.c_lib.freedv_get_n_tx_modem_samples(freedv) # *2 #get n_tx_modem_samples which defines the size of the modulation object
|
||||
n_tx_preamble_modem_samples = self.c_lib.freedv_get_n_tx_preamble_modem_samples(freedv)
|
||||
|
||||
|
||||
mod_out = ctypes.c_short * n_tx_modem_samples
|
||||
mod_out = mod_out()
|
||||
mod_out_preamble = ctypes.c_short * n_tx_preamble_modem_samples #*2 #1760 for mode 10,11,12 #4000 for mode 9
|
||||
mod_out_preamble = ctypes.c_short * n_tx_preamble_modem_samples # *2 #1760 for mode 10,11,12 #4000 for mode 9
|
||||
mod_out_preamble = mod_out_preamble()
|
||||
|
||||
self.c_lib.freedv_rawdatapreambletx(freedv, mod_out_preamble);
|
||||
self.c_lib.freedv_rawdatapreambletx(freedv, mod_out_preamble)
|
||||
|
||||
txbuffer = bytearray()
|
||||
txbuffer += bytes(mod_out_preamble)
|
||||
|
||||
if static.ARQ_RPT_RECEIVED == False:
|
||||
for n in range(0,static.ARQ_TX_N_FRAMES_PER_BURST):
|
||||
for n in range(0, static.ARQ_TX_N_FRAMES_PER_BURST):
|
||||
|
||||
#---------------------------BUILD ARQ BURST ---------------------------------------------------------------------
|
||||
frame_type = 10 + n + 1 #static.ARQ_TX_N_FRAMES_PER_BURST
|
||||
# ---------------------------BUILD ARQ BURST ---------------------------------------------------------------------
|
||||
frame_type = 10 + n + 1 # static.ARQ_TX_N_FRAMES_PER_BURST
|
||||
frame_type = bytes([frame_type])
|
||||
|
||||
payload_data = bytes(static.TX_BUFFER[static.ARQ_N_SENT_FRAMES + n])
|
||||
|
||||
|
||||
n_current_arq_frame = static.ARQ_N_SENT_FRAMES + n + 1
|
||||
static.ARQ_TX_N_CURRENT_ARQ_FRAME = n_current_arq_frame.to_bytes(2, byteorder='big')
|
||||
|
||||
|
||||
n_total_arq_frame = len(static.TX_BUFFER)
|
||||
static.ARQ_TX_N_TOTAL_ARQ_FRAMES = n_total_arq_frame.to_bytes(2, byteorder='big')
|
||||
|
||||
|
||||
arqframe = frame_type + \
|
||||
bytes([static.ARQ_TX_N_FRAMES_PER_BURST]) + \
|
||||
static.ARQ_TX_N_CURRENT_ARQ_FRAME + \
|
||||
static.ARQ_TX_N_TOTAL_ARQ_FRAMES + \
|
||||
static.DXCALLSIGN_CRC8 + \
|
||||
static.MYCALLSIGN_CRC8 + \
|
||||
payload_data
|
||||
|
||||
#print(arqframe)
|
||||
|
||||
buffer = bytearray(static.FREEDV_DATA_PAYLOAD_PER_FRAME) # create TX buffer
|
||||
buffer[:len(arqframe)] = arqframe # set buffersize to length of data which will be send
|
||||
|
||||
bytes([static.ARQ_TX_N_FRAMES_PER_BURST]) + \
|
||||
static.ARQ_TX_N_CURRENT_ARQ_FRAME + \
|
||||
static.ARQ_TX_N_TOTAL_ARQ_FRAMES + \
|
||||
static.DXCALLSIGN_CRC8 + \
|
||||
static.MYCALLSIGN_CRC8 + \
|
||||
payload_data
|
||||
|
||||
# print(arqframe)
|
||||
|
||||
buffer = bytearray(static.FREEDV_DATA_PAYLOAD_PER_FRAME) # create TX buffer
|
||||
buffer[:len(arqframe)] = arqframe # set buffersize to length of data which will be send
|
||||
|
||||
crc = ctypes.c_ushort(self.c_lib.freedv_gen_crc16(bytes(buffer), static.FREEDV_DATA_PAYLOAD_PER_FRAME)) # generate CRC16
|
||||
crc = crc.value.to_bytes(2, byteorder='big') # convert crc to 2 byte hex string
|
||||
crc = crc.value.to_bytes(2, byteorder='big') # convert crc to 2 byte hex string
|
||||
buffer += crc # append crc16 to buffer
|
||||
|
||||
data = (ctypes.c_ubyte * static.FREEDV_DATA_BYTES_PER_FRAME).from_buffer_copy(buffer)
|
||||
self.c_lib.freedv_rawdatatx(freedv,mod_out,data) # modulate DATA and safe it into mod_out pointer
|
||||
self.c_lib.freedv_rawdatatx(freedv, mod_out, data) # modulate DATA and safe it into mod_out pointer
|
||||
txbuffer += bytes(mod_out)
|
||||
|
||||
|
||||
elif static.ARQ_RPT_RECEIVED == True:
|
||||
|
||||
for n in range(0,len(static.ARQ_RPT_FRAMES)):
|
||||
|
||||
for n in range(0, len(static.ARQ_RPT_FRAMES)):
|
||||
|
||||
missing_frame = int.from_bytes(static.ARQ_RPT_FRAMES[n], "big")
|
||||
|
||||
#---------------------------BUILD ARQ BURST ---------------------------------------------------------------------
|
||||
frame_type = 10 + missing_frame #static.ARQ_TX_N_FRAMES_PER_BURST
|
||||
|
||||
# ---------------------------BUILD ARQ BURST ---------------------------------------------------------------------
|
||||
frame_type = 10 + missing_frame # static.ARQ_TX_N_FRAMES_PER_BURST
|
||||
frame_type = bytes([frame_type])
|
||||
|
||||
payload_data = bytes(static.TX_BUFFER[static.ARQ_N_SENT_FRAMES + missing_frame - 1])
|
||||
|
||||
|
||||
n_current_arq_frame = static.ARQ_N_SENT_FRAMES + missing_frame
|
||||
static.ARQ_TX_N_CURRENT_ARQ_FRAME = n_current_arq_frame.to_bytes(2, byteorder='big')
|
||||
|
||||
|
||||
n_total_arq_frame = len(static.TX_BUFFER)
|
||||
static.ARQ_TX_N_TOTAL_ARQ_FRAMES = n_total_arq_frame.to_bytes(2, byteorder='big')
|
||||
|
||||
arqframe = frame_type + \
|
||||
bytes([static.ARQ_TX_N_FRAMES_PER_BURST]) + \
|
||||
static.ARQ_TX_N_CURRENT_ARQ_FRAME + \
|
||||
static.ARQ_TX_N_TOTAL_ARQ_FRAMES + \
|
||||
static.DXCALLSIGN_CRC8 + \
|
||||
static.MYCALLSIGN_CRC8 + \
|
||||
payload_data
|
||||
|
||||
buffer = bytearray(static.FREEDV_DATA_PAYLOAD_PER_FRAME) # create TX buffer
|
||||
buffer[:len(arqframe)] = arqframe # set buffersize to length of data which will be send
|
||||
|
||||
arqframe = frame_type + \
|
||||
bytes([static.ARQ_TX_N_FRAMES_PER_BURST]) + \
|
||||
static.ARQ_TX_N_CURRENT_ARQ_FRAME + \
|
||||
static.ARQ_TX_N_TOTAL_ARQ_FRAMES + \
|
||||
static.DXCALLSIGN_CRC8 + \
|
||||
static.MYCALLSIGN_CRC8 + \
|
||||
payload_data
|
||||
|
||||
buffer = bytearray(static.FREEDV_DATA_PAYLOAD_PER_FRAME) # create TX buffer
|
||||
buffer[:len(arqframe)] = arqframe # set buffersize to length of data which will be send
|
||||
|
||||
crc = ctypes.c_ushort(self.c_lib.freedv_gen_crc16(bytes(buffer), static.FREEDV_DATA_PAYLOAD_PER_FRAME)) # generate CRC16
|
||||
crc = crc.value.to_bytes(2, byteorder='big') # convert crc to 2 byte hex string
|
||||
crc = crc.value.to_bytes(2, byteorder='big') # convert crc to 2 byte hex string
|
||||
buffer += crc # append crc16 to buffer
|
||||
|
||||
|
||||
data = (ctypes.c_ubyte * static.FREEDV_DATA_BYTES_PER_FRAME).from_buffer_copy(buffer)
|
||||
self.c_lib.freedv_rawdatatx(freedv,mod_out,data) # modulate DATA and safe it into mod_out pointer
|
||||
self.c_lib.freedv_rawdatatx(freedv, mod_out, data) # modulate DATA and safe it into mod_out pointer
|
||||
txbuffer += bytes(mod_out)
|
||||
|
||||
|
||||
|
||||
# -------------- transmit audio
|
||||
|
||||
|
||||
#self.stream_tx.write(bytes(txbuffer))
|
||||
self.streambuffer = bytes()
|
||||
|
||||
# self.stream_tx.write(bytes(txbuffer))
|
||||
self.streambuffer = bytes()
|
||||
self.streambuffer = bytes(txbuffer)
|
||||
self.audio_writing_to_stream = True
|
||||
|
||||
#wait until audio has been processed
|
||||
# wait until audio has been processed
|
||||
while self.audio_writing_to_stream == True:
|
||||
time.sleep(0.01)
|
||||
static.CHANNEL_STATE = 'SENDING_DATA'
|
||||
#print("sending data...")
|
||||
|
||||
|
||||
static.CHANNEL_STATE = 'RECEIVING_SIGNALLING'
|
||||
static.PTT_STATE = False
|
||||
self.my_rig.set_ptt(self.hamlib_ptt_type,0)
|
||||
|
||||
self.c_lib.freedv_close(freedv)
|
||||
#--------------------------------------------------------------------------------------------------------
|
||||
self.my_rig.set_ptt(self.hamlib_ptt_type, 0)
|
||||
|
||||
def receive(self,mode):
|
||||
self.c_lib.freedv_close(freedv)
|
||||
# --------------------------------------------------------------------------------------------------------
|
||||
|
||||
def receive(self, mode):
|
||||
force = True
|
||||
|
||||
self.c_lib.freedv_open.restype = ctypes.POINTER(ctypes.c_ubyte)
|
||||
|
||||
self.c_lib.freedv_open.restype = ctypes.POINTER(ctypes.c_ubyte)
|
||||
freedv = self.c_lib.freedv_open(mode)
|
||||
bytes_per_frame = int(self.c_lib.freedv_get_bits_per_modem_frame(freedv)/8)
|
||||
|
||||
bytes_per_frame = int(self.c_lib.freedv_get_bits_per_modem_frame(freedv) / 8)
|
||||
|
||||
if mode == static.FREEDV_SIGNALLING_MODE:
|
||||
static.FREEDV_SIGNALLING_BYTES_PER_FRAME = bytes_per_frame
|
||||
static.FREEDV_SIGNALLING_PAYLOAD_PER_FRAME = bytes_per_frame - 2
|
||||
|
||||
self.c_lib.freedv_set_frames_per_burst(freedv, 1);
|
||||
|
||||
|
||||
self.c_lib.freedv_set_frames_per_burst(freedv, 1)
|
||||
|
||||
elif mode == static.ARQ_DATA_CHANNEL_MODE:
|
||||
static.FREEDV_DATA_BYTES_PER_FRAME = bytes_per_frame
|
||||
static.FREEDV_DATA_PAYLOAD_PER_FRAME = bytes_per_frame - 2
|
||||
|
||||
self.c_lib.freedv_set_frames_per_burst(freedv, 0);
|
||||
|
||||
self.c_lib.freedv_set_frames_per_burst(freedv, 0)
|
||||
else:
|
||||
pass
|
||||
|
||||
|
||||
|
||||
|
||||
bytes_out = (ctypes.c_ubyte * bytes_per_frame)
|
||||
bytes_out = bytes_out() #get pointer to bytes_out
|
||||
bytes_out = bytes_out() # get pointer to bytes_out
|
||||
|
||||
while static.FREEDV_RECEIVE == True:
|
||||
time.sleep(0.05)
|
||||
|
||||
|
||||
# stuck in sync counter
|
||||
stuck_in_sync_counter = 0
|
||||
stuck_in_sync_10_counter = 0
|
||||
#
|
||||
|
||||
|
||||
# here we do an unsync to be sure, the modem is in idle state and ready for new data
|
||||
self.c_lib.freedv_set_sync(freedv, 0)
|
||||
|
||||
|
||||
# here we do a buffer cleanup before returning to demod loop
|
||||
dummy_mod = bytes(self.c_lib.freedv_nin(freedv))
|
||||
self.c_lib.freedv_rawdatarx(freedv, bytes_out, dummy_mod)
|
||||
|
||||
#demod loop
|
||||
|
||||
# demod loop
|
||||
while (static.CHANNEL_STATE == 'RECEIVING_DATA' and static.ARQ_DATA_CHANNEL_MODE == mode) or (static.CHANNEL_STATE == 'RECEIVING_SIGNALLING' and static.FREEDV_SIGNALLING_MODE == mode):
|
||||
time.sleep(0.01)
|
||||
|
||||
# refresh vars, so the correct parameters of the used mode are set
|
||||
static.FREEDV_DATA_BYTES_PER_FRAME = bytes_per_frame
|
||||
static.FREEDV_DATA_PAYLOAD_PER_FRAME = bytes_per_frame - 2
|
||||
|
||||
|
||||
nin = self.c_lib.freedv_nin(freedv)
|
||||
#nin = int(nin*(static.AUDIO_SAMPLE_RATE_RX/static.MODEM_SAMPLE_RATE))
|
||||
data_in = self.stream_rx.read(nin, exception_on_overflow = False)
|
||||
data_in = self.stream_rx.read(nin, exception_on_overflow=False)
|
||||
static.AUDIO_RMS = audioop.rms(data_in, 2)
|
||||
#self.c_lib.freedv_rawdatarx.argtype = [ctypes.POINTER(ctypes.c_ubyte), data_bytes_out, data_in] # check if really neccessary
|
||||
nbytes = self.c_lib.freedv_rawdatarx(freedv, bytes_out, data_in) # demodulate audio
|
||||
#logging.debug(self.c_lib.freedv_get_rx_status(freedv))
|
||||
# self.c_lib.freedv_rawdatarx.argtype = [ctypes.POINTER(ctypes.c_ubyte), data_bytes_out, data_in] # check if really neccessary
|
||||
nbytes = self.c_lib.freedv_rawdatarx(freedv, bytes_out, data_in) # demodulate audio
|
||||
# logging.debug(self.c_lib.freedv_get_rx_status(freedv))
|
||||
#print("listening-" + str(mode) + "-" + str(self.c_lib.freedv_get_rx_status(freedv)))
|
||||
|
||||
#-------------STUCK IN SYNC DETECTOR
|
||||
|
||||
# -------------STUCK IN SYNC DETECTOR
|
||||
stuck_in_sync_counter += 1
|
||||
if self.c_lib.freedv_get_rx_status(freedv) == 10:
|
||||
stuck_in_sync_10_counter += 1
|
||||
#self.c_lib.freedv_set_sync(freedv, 0)
|
||||
logging.warning("MODEM | SYNC 10 TRIGGER | M:" + str(mode) + " | " + str(static.CHANNEL_STATE))
|
||||
|
||||
|
||||
if stuck_in_sync_counter == 33 and self.c_lib.freedv_get_rx_status(freedv) == 10:
|
||||
logging.critical("MODEM | stuck in sync #1")
|
||||
self.c_lib.freedv_set_sync(freedv, 0) #FORCE UNSYNC
|
||||
self.c_lib.freedv_set_sync(freedv, 0) # FORCE UNSYNC
|
||||
stuck_in_sync_counter = 0
|
||||
stuck_in_sync_10_counter = 0
|
||||
|
||||
|
||||
if stuck_in_sync_counter >= 66 and stuck_in_sync_10_counter >= 2:
|
||||
logging.critical("MODEM | stuck in sync #2")
|
||||
self.c_lib.freedv_set_sync(freedv, 0) #FORCE UNSYNC
|
||||
stuck_in_sync_counter = 0
|
||||
self.c_lib.freedv_set_sync(freedv, 0) # FORCE UNSYNC
|
||||
stuck_in_sync_counter = 0
|
||||
stuck_in_sync_10_counter = 0
|
||||
#-----------------------------------
|
||||
|
||||
# -----------------------------------
|
||||
|
||||
# forward data only if broadcast or we are the receiver
|
||||
if nbytes == bytes_per_frame and bytes(bytes_out[1:2]) == static.MYCALLSIGN_CRC8 or bytes(bytes_out[1:2]) == b'\x01':
|
||||
self.calculate_ber(freedv)
|
||||
|
|
@ -384,120 +377,118 @@ class RF():
|
|||
stuck_in_sync_counter = 0
|
||||
stuck_in_sync_10_counter = 0
|
||||
#
|
||||
|
||||
|
||||
# CHECK IF FRAMETYPE IS BETWEEN 10 and 50 ------------------------
|
||||
frametype = int.from_bytes(bytes(bytes_out[:1]), "big")
|
||||
frame = frametype - 10
|
||||
n_frames_per_burst = int.from_bytes(bytes(bytes_out[1:2]), "big")
|
||||
|
||||
|
||||
#self.c_lib.freedv_set_frames_per_burst(freedv_data, n_frames_per_burst);
|
||||
|
||||
if 50 >= frametype >= 10:
|
||||
|
||||
if 50 >= frametype >= 10:
|
||||
if frame != 3 or force == True:
|
||||
|
||||
data_handler.arq_data_received(bytes(bytes_out[:-2])) #send payload data to arq checker without CRC16
|
||||
|
||||
data_handler.arq_data_received(bytes(bytes_out[:-2])) # send payload data to arq checker without CRC16
|
||||
|
||||
#print("static.ARQ_RX_BURST_BUFFER.count(None) " + str(static.ARQ_RX_BURST_BUFFER.count(None)))
|
||||
if static.ARQ_RX_BURST_BUFFER.count(None) <= 1:
|
||||
logging.debug("FULL BURST BUFFER ---> UNSYNC")
|
||||
self.c_lib.freedv_set_sync(freedv, 0)
|
||||
|
||||
|
||||
else:
|
||||
logging.critical("---------------------------SIMULATED MISSING FRAME")
|
||||
force = True
|
||||
|
||||
# BURST ACK
|
||||
elif frametype == 60:
|
||||
logging.debug("ACK RECEIVED....")
|
||||
data_handler.burst_ack_received()
|
||||
|
||||
logging.debug("ACK RECEIVED....")
|
||||
data_handler.burst_ack_received()
|
||||
|
||||
# FRAME ACK
|
||||
elif frametype == 61:
|
||||
logging.debug("FRAME ACK RECEIVED....")
|
||||
data_handler.frame_ack_received()
|
||||
|
||||
logging.debug("FRAME ACK RECEIVED....")
|
||||
data_handler.frame_ack_received()
|
||||
|
||||
# FRAME RPT
|
||||
elif frametype == 62:
|
||||
logging.debug("REPEAT REQUEST RECEIVED....")
|
||||
data_handler.burst_rpt_received(bytes_out[:-2])
|
||||
logging.debug("REPEAT REQUEST RECEIVED....")
|
||||
data_handler.burst_rpt_received(bytes_out[:-2])
|
||||
|
||||
# CQ FRAME
|
||||
elif frametype == 200:
|
||||
logging.debug("CQ RECEIVED....")
|
||||
data_handler.received_cq(bytes_out[:-2])
|
||||
|
||||
logging.debug("CQ RECEIVED....")
|
||||
data_handler.received_cq(bytes_out[:-2])
|
||||
|
||||
# PING FRAME
|
||||
elif frametype == 210:
|
||||
logging.debug("PING RECEIVED....")
|
||||
data_handler.received_ping(bytes_out[:-2])
|
||||
logging.debug("PING RECEIVED....")
|
||||
data_handler.received_ping(bytes_out[:-2])
|
||||
|
||||
# PING ACK
|
||||
elif frametype == 211:
|
||||
logging.debug("PING ACK RECEIVED....")
|
||||
data_handler.received_ping_ack(bytes_out[:-2])
|
||||
logging.debug("PING ACK RECEIVED....")
|
||||
data_handler.received_ping_ack(bytes_out[:-2])
|
||||
|
||||
# ARQ CONNECT
|
||||
elif frametype == 220:
|
||||
logging.debug("ARQ CONNECT RECEIVED....")
|
||||
data_handler.arq_received_connect(bytes_out[:-2])
|
||||
logging.debug("ARQ CONNECT RECEIVED....")
|
||||
data_handler.arq_received_connect(bytes_out[:-2])
|
||||
|
||||
# ARQ CONNECT ACK / KEEP ALIVE
|
||||
elif frametype == 221:
|
||||
logging.debug("ARQ CONNECT ACK RECEIVED / KEEP ALIVE....")
|
||||
data_handler.arq_received_connect_keep_alive(bytes_out[:-2])
|
||||
logging.debug("ARQ CONNECT ACK RECEIVED / KEEP ALIVE....")
|
||||
data_handler.arq_received_connect_keep_alive(bytes_out[:-2])
|
||||
|
||||
# ARQ CONNECT ACK / KEEP ALIVE
|
||||
elif frametype == 222:
|
||||
logging.debug("ARQ DISCONNECT RECEIVED")
|
||||
data_handler.arq_disconnect_received(bytes_out[:-2])
|
||||
logging.debug("ARQ DISCONNECT RECEIVED")
|
||||
data_handler.arq_disconnect_received(bytes_out[:-2])
|
||||
|
||||
# ARQ FILE TRANSFER RECEIVED!
|
||||
# ARQ FILE TRANSFER RECEIVED!
|
||||
elif frametype == 225:
|
||||
logging.debug("ARQ arq_received_data_channel_opener RECEIVED")
|
||||
data_handler.arq_received_data_channel_opener(bytes_out[:-2])
|
||||
|
||||
logging.debug("ARQ arq_received_data_channel_opener RECEIVED")
|
||||
data_handler.arq_received_data_channel_opener(bytes_out[:-2])
|
||||
|
||||
# ARQ CHANNEL IS OPENED
|
||||
elif frametype == 226:
|
||||
logging.debug("ARQ arq_received_channel_is_open RECEIVED")
|
||||
data_handler.arq_received_channel_is_open(bytes_out[:-2])
|
||||
|
||||
logging.debug("ARQ arq_received_channel_is_open RECEIVED")
|
||||
data_handler.arq_received_channel_is_open(bytes_out[:-2])
|
||||
|
||||
# ARQ CONNECT ACK / KEEP ALIVE
|
||||
elif frametype == 230:
|
||||
logging.debug("BEACON RECEIVED")
|
||||
|
||||
logging.debug("BEACON RECEIVED")
|
||||
data_handler.received_beacon(bytes_out[:-2])
|
||||
|
||||
else:
|
||||
logging.info("OTHER FRAME: " + str(bytes_out[:-2]))
|
||||
print(frametype)
|
||||
|
||||
|
||||
# DO UNSYNC AFTER LAST BURST by checking the frame nums agains the total frames per burst
|
||||
if frame == n_frames_per_burst:
|
||||
if frame == n_frames_per_burst:
|
||||
logging.debug("LAST FRAME ---> UNSYNC")
|
||||
self.c_lib.freedv_set_sync(freedv, 0) #FORCE UNSYNC
|
||||
|
||||
|
||||
self.c_lib.freedv_set_sync(freedv, 0) # FORCE UNSYNC
|
||||
|
||||
# clear bytes_out buffer to be ready for next frames after successfull decoding
|
||||
|
||||
|
||||
bytes_out = (ctypes.c_ubyte * bytes_per_frame)
|
||||
bytes_out = bytes_out() #get pointer to bytes_out
|
||||
|
||||
bytes_out = bytes_out() # get pointer to bytes_out
|
||||
|
||||
if mode == 14:
|
||||
self.c_lib.freedv_set_sync(freedv, 0)
|
||||
for i in range(0,10):
|
||||
for i in range(0, 10):
|
||||
dummy_mod = bytes(self.c_lib.freedv_nin(freedv))
|
||||
self.c_lib.freedv_rawdatarx(freedv, bytes_out, dummy_mod)
|
||||
else:
|
||||
# for debugging purposes to receive all data
|
||||
pass
|
||||
#print(bytes_out[:-2])
|
||||
|
||||
|
||||
def calculate_ber(self,freedv):
|
||||
# print(bytes_out[:-2])
|
||||
|
||||
def calculate_ber(self, freedv):
|
||||
Tbits = self.c_lib.freedv_get_total_bits(freedv)
|
||||
Terrs = self.c_lib.freedv_get_total_bit_errors(freedv)
|
||||
if Tbits != 0:
|
||||
ber = (Terrs/Tbits)*100
|
||||
ber = (Terrs / Tbits) * 100
|
||||
static.BER = int(ber)
|
||||
|
||||
self.c_lib.freedv_set_total_bit_errors(freedv,0)
|
||||
self.c_lib.freedv_set_total_bits(freedv,0)
|
||||
|
||||
|
||||
self.c_lib.freedv_set_total_bit_errors(freedv, 0)
|
||||
self.c_lib.freedv_set_total_bits(freedv, 0)
|
||||
|
|
|
|||
134
sock.py
134
sock.py
|
|
@ -9,7 +9,6 @@ Created on Fri Dec 25 21:25:14 2020
|
|||
import socketserver
|
||||
import threading
|
||||
import logging
|
||||
import time
|
||||
import json
|
||||
import asyncio
|
||||
|
||||
|
|
@ -20,124 +19,113 @@ import helpers
|
|||
|
||||
|
||||
|
||||
import asyncbg
|
||||
|
||||
|
||||
|
||||
class CMDTCPRequestHandler(socketserver.BaseRequestHandler):
|
||||
class CMDTCPRequestHandler(socketserver.BaseRequestHandler):
|
||||
|
||||
def handle(self):
|
||||
|
||||
encoding = 'utf-8'
|
||||
|
||||
encoding = 'utf-8'
|
||||
#data = str(self.request.recv(1024), 'utf-8')
|
||||
|
||||
|
||||
data = bytes()
|
||||
while True:
|
||||
chunk = self.request.recv(8192)#.strip()
|
||||
while True:
|
||||
chunk = self.request.recv(8192) # .strip()
|
||||
data += chunk
|
||||
if chunk.endswith(b'\n'):
|
||||
break
|
||||
data = data[:-1] # remove b'\n'
|
||||
data = str(data, 'utf-8')
|
||||
data = data[:-1] # remove b'\n'
|
||||
data = str(data, 'utf-8')
|
||||
|
||||
# SOCKETTEST ---------------------------------------------------
|
||||
if data == 'SOCKETTEST':
|
||||
#cur_thread = threading.current_thread()
|
||||
response = bytes("WELL DONE! YOU ARE ABLE TO COMMUNICATE WITH THE TNC", encoding)
|
||||
self.request.sendall(response)
|
||||
|
||||
|
||||
# CQ CQ CQ -----------------------------------------------------
|
||||
if data == 'CQCQCQ':
|
||||
asyncio.run(data_handler.transmit_cq())
|
||||
#asyncio.run(asyncbg.call(data_handler.transmit_cq))
|
||||
# asyncio.run(asyncbg.call(data_handler.transmit_cq))
|
||||
#######self.request.sendall(b'CALLING CQ')
|
||||
|
||||
|
||||
# PING ----------------------------------------------------------
|
||||
if data.startswith('PING:'):
|
||||
#send ping frame and wait for ACK
|
||||
# send ping frame and wait for ACK
|
||||
pingcommand = data.split('PING:')
|
||||
dxcallsign = pingcommand[1]
|
||||
#data_handler.transmit_ping(dxcallsign)
|
||||
# data_handler.transmit_ping(dxcallsign)
|
||||
##loop = asyncio.get_event_loop()
|
||||
##loop.create_task(data_handler.transmit_ping(dxcallsign))
|
||||
##loop.run()
|
||||
|
||||
#asyncio.new_event_loop()
|
||||
#asyncio.ensure_future(data_handler.transmit_ping(dxcallsign))
|
||||
|
||||
# loop.create_task(data_handler.transmit_ping(dxcallsign))
|
||||
# loop.run()
|
||||
|
||||
# asyncio.new_event_loop()
|
||||
# asyncio.ensure_future(data_handler.transmit_ping(dxcallsign))
|
||||
|
||||
asyncio.run(data_handler.transmit_ping(dxcallsign))
|
||||
|
||||
#asyncio.create_task(data_handler.transmit_ping(dxcallsign))
|
||||
#asyncio.run(data_handler.transmit_ping(dxcallsign))
|
||||
|
||||
|
||||
# asyncio.create_task(data_handler.transmit_ping(dxcallsign))
|
||||
# asyncio.run(data_handler.transmit_ping(dxcallsign))
|
||||
|
||||
# ARQ CONNECT TO CALLSIGN ----------------------------------------
|
||||
if data.startswith('ARQ:CONNECT:'):
|
||||
arqconnectcommand = data.split('ARQ:CONNECT:')
|
||||
dxcallsign = arqconnectcommand[1]
|
||||
static.DXCALLSIGN = bytes(dxcallsign, 'utf-8')
|
||||
static.DXCALLSIGN_CRC8 = helpers.get_crc_8(static.DXCALLSIGN)
|
||||
|
||||
|
||||
if static.ARQ_STATE == 'CONNECTED':
|
||||
# here we could disconnect
|
||||
pass
|
||||
|
||||
|
||||
if static.TNC_STATE == 'IDLE':
|
||||
# here we send an "CONNECT FRAME
|
||||
|
||||
|
||||
#ARQ_CONNECT_THREAD = threading.Thread(target=data_handler.arq_connect, name="ARQ_CONNECT")
|
||||
#ARQ_CONNECT_THREAD.start()
|
||||
# ARQ_CONNECT_THREAD.start()
|
||||
asyncio.run(data_handler.arq_connect())
|
||||
########self.request.sendall(bytes("CONNECTING", encoding))
|
||||
#data_handler.arq_connect()
|
||||
# data_handler.arq_connect()
|
||||
# ARQ DISCONNECT FROM CALLSIGN ----------------------------------------
|
||||
if data == 'ARQ:DISCONNECT':
|
||||
|
||||
#ARQ_DISCONNECT_THREAD = threading.Thread(target=data_handler.arq_disconnect, name="ARQ_DISCONNECT")
|
||||
#ARQ_DISCONNECT_THREAD.start()
|
||||
# ARQ_DISCONNECT_THREAD.start()
|
||||
asyncio.run(data_handler.arq_disconnect())
|
||||
|
||||
|
||||
########self.request.sendall(bytes("DISCONNECTING", encoding))
|
||||
#data_handler.arq_disconnect()
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
# data_handler.arq_disconnect()
|
||||
|
||||
if data.startswith('ARQ:OPEN_DATA_CHANNEL') and static.ARQ_STATE == 'CONNECTED':
|
||||
static.ARQ_READY_FOR_DATA = False
|
||||
static.TNC_STATE = 'BUSY'
|
||||
asyncio.run(data_handler.arq_open_data_channel())
|
||||
|
||||
if data.startswith('ARQ:DATA:') and static.ARQ_STATE == 'CONNECTED' and static.ARQ_READY_FOR_DATA == True:
|
||||
|
||||
if data.startswith('ARQ:DATA:') and static.ARQ_STATE == 'CONNECTED' and static.ARQ_READY_FOR_DATA == True:
|
||||
|
||||
static.TNC_STATE = 'BUSY'
|
||||
arqdata = data.split('ARQ:')
|
||||
data_out = bytes(arqdata[1], 'utf-8')
|
||||
|
||||
|
||||
ARQ_DATA_THREAD = threading.Thread(target=data_handler.arq_transmit, args=[data_out], name="ARQ_DATA")
|
||||
ARQ_DATA_THREAD.start()
|
||||
#asyncio.run(data_handler.arq_transmit(data_out))
|
||||
|
||||
# SETTINGS AND STATUS ---------------------------------------------
|
||||
ARQ_DATA_THREAD.start()
|
||||
# asyncio.run(data_handler.arq_transmit(data_out))
|
||||
|
||||
# SETTINGS AND STATUS ---------------------------------------------
|
||||
if data.startswith('SET:MYCALLSIGN:'):
|
||||
callsign = data.split('SET:MYCALLSIGN:')
|
||||
if bytes(callsign[1], encoding) == b'':
|
||||
self.request.sendall(b'INVALID CALLSIGN')
|
||||
else:
|
||||
else:
|
||||
static.MYCALLSIGN = bytes(callsign[1], encoding)
|
||||
static.MYCALLSIGN_CRC8 = helpers.get_crc_8(static.MYCALLSIGN)
|
||||
########self.request.sendall(static.MYCALLSIGN)
|
||||
# self.request.sendall(static.MYCALLSIGN)
|
||||
logging.info("CMD | MYCALLSIGN: " + str(static.MYCALLSIGN))
|
||||
|
||||
|
||||
if data == 'GET:MYCALLSIGN':
|
||||
self.request.sendall(bytes(static.MYCALLSIGN, encoding))
|
||||
|
||||
|
||||
|
||||
if data == 'GET:DXCALLSIGN':
|
||||
self.request.sendall(bytes(static.DXCALLSIGN, encoding))
|
||||
|
||||
self.request.sendall(bytes(static.DXCALLSIGN, encoding))
|
||||
|
||||
if data == 'GET:TNC_STATE':
|
||||
output = {
|
||||
"PTT_STATE": str(static.PTT_STATE),
|
||||
|
|
@ -149,7 +137,7 @@ class CMDTCPRequestHandler(socketserver.BaseRequestHandler):
|
|||
}
|
||||
jsondata = json.dumps(output)
|
||||
self.request.sendall(bytes(jsondata, encoding))
|
||||
|
||||
|
||||
if data == 'GET:DATA_STATE':
|
||||
output = {
|
||||
"RX_BUFFER_LENGTH": str(len(static.RX_BUFFER)),
|
||||
|
|
@ -159,51 +147,43 @@ class CMDTCPRequestHandler(socketserver.BaseRequestHandler):
|
|||
"ARQ_TX_N_CURRENT_ARQ_FRAME": str(int.from_bytes(bytes(static.ARQ_TX_N_CURRENT_ARQ_FRAME), "big")),
|
||||
"ARQ_TX_N_TOTAL_ARQ_FRAMES": str(int.from_bytes(bytes(static.ARQ_TX_N_TOTAL_ARQ_FRAMES), "big")),
|
||||
"ARQ_RX_FRAME_N_BURSTS": str(static.ARQ_RX_FRAME_N_BURSTS),
|
||||
"ARQ_RX_N_CURRENT_ARQ_FRAME": str(static.ARQ_RX_N_CURRENT_ARQ_FRAME),
|
||||
"ARQ_N_ARQ_FRAMES_PER_DATA_FRAME": str(static.ARQ_N_ARQ_FRAMES_PER_DATA_FRAME )
|
||||
"ARQ_RX_N_CURRENT_ARQ_FRAME": str(static.ARQ_RX_N_CURRENT_ARQ_FRAME),
|
||||
"ARQ_N_ARQ_FRAMES_PER_DATA_FRAME": str(static.ARQ_N_ARQ_FRAMES_PER_DATA_FRAME)
|
||||
}
|
||||
jsondata = json.dumps(output)
|
||||
self.request.sendall(bytes(jsondata, encoding))
|
||||
self.request.sendall(bytes(jsondata, encoding))
|
||||
|
||||
if data == 'GET:HEARD_STATIONS':
|
||||
output = []
|
||||
for i in range(0,len(static.HEARD_STATIONS)):
|
||||
output.append({"CALLSIGN" : str(static.HEARD_STATIONS[i][0], 'utf-8'), "TIMESTAMP" : static.HEARD_STATIONS[i][1]})
|
||||
|
||||
for i in range(0, len(static.HEARD_STATIONS)):
|
||||
output.append({"CALLSIGN": str(static.HEARD_STATIONS[i][0], 'utf-8'), "TIMESTAMP": static.HEARD_STATIONS[i][1]})
|
||||
|
||||
jsondata = json.dumps(output)
|
||||
self.request.sendall(bytes(jsondata, encoding))
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
if data.startswith('GET:RX_BUFFER:'):
|
||||
data = data.split('GET:RX_BUFFER:')
|
||||
bufferposition = int(data[1])-1
|
||||
bufferposition = int(data[1]) - 1
|
||||
if bufferposition == -1:
|
||||
if len(static.RX_BUFFER) > 0:
|
||||
self.request.sendall(static.RX_BUFFER[-1])
|
||||
|
||||
|
||||
if bufferposition <= len(static.RX_BUFFER) > 0:
|
||||
self.request.sendall(bytes(static.RX_BUFFER[bufferposition]))
|
||||
|
||||
self.request.sendall(bytes(static.RX_BUFFER[bufferposition]))
|
||||
|
||||
if data == 'DEL:RX_BUFFER':
|
||||
static.RX_BUFFER = []
|
||||
|
||||
#self.request.close()
|
||||
|
||||
|
||||
# self.request.close()
|
||||
|
||||
|
||||
def start_cmd_socket():
|
||||
|
||||
try:
|
||||
logging.info("SRV | STARTING TCP/IP SOCKET FOR CMD ON PORT: " + str(static.PORT))
|
||||
socketserver.TCPServer.allow_reuse_address = True #https://stackoverflow.com/a/16641793
|
||||
socketserver.TCPServer.allow_reuse_address = True # https://stackoverflow.com/a/16641793
|
||||
cmdserver = socketserver.TCPServer((static.HOST, static.PORT), CMDTCPRequestHandler)
|
||||
cmdserver.serve_forever()
|
||||
|
||||
|
||||
finally:
|
||||
cmdserver.server_close()
|
||||
|
|
|
|||
88
static.py
88
static.py
|
|
@ -15,50 +15,41 @@ DXCALLSIGN_CRC8 = b'A'
|
|||
|
||||
MYGRID = b''
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
#---------------------------------
|
||||
# ---------------------------------
|
||||
|
||||
# Server Defaults
|
||||
HOST = "localhost"
|
||||
PORT = 3000
|
||||
#---------------------------------
|
||||
# ---------------------------------
|
||||
|
||||
# HAMLIB DEFAULTS
|
||||
|
||||
#RIG_PTT_NONE
|
||||
#No PTT available
|
||||
# RIG_PTT_NONE
|
||||
# No PTT available
|
||||
|
||||
#RIG_PTT_RIG
|
||||
#Legacy PTT
|
||||
# RIG_PTT_RIG
|
||||
# Legacy PTT
|
||||
|
||||
#RIG_PTT_SERIAL_DTR
|
||||
#PTT control through serial DTR signal
|
||||
# RIG_PTT_SERIAL_DTR
|
||||
# PTT control through serial DTR signal
|
||||
|
||||
#RIG_PTT_SERIAL_RTS
|
||||
#PTT control through serial RTS signal
|
||||
# RIG_PTT_SERIAL_RTS
|
||||
# PTT control through serial RTS signal
|
||||
|
||||
#RIG_PTT_PARALLEL
|
||||
#PTT control through parallel port
|
||||
# RIG_PTT_PARALLEL
|
||||
# PTT control through parallel port
|
||||
|
||||
#RIG_PTT_RIG_MICDATA
|
||||
#Legacy PTT, supports RIG_PTT_ON_MIC/RIG_PTT_ON_DATA
|
||||
# RIG_PTT_RIG_MICDATA
|
||||
# Legacy PTT, supports RIG_PTT_ON_MIC/RIG_PTT_ON_DATA
|
||||
|
||||
#RIG_PTT_CM108
|
||||
#PTT control through CM108 GPIO pin
|
||||
# RIG_PTT_CM108
|
||||
# PTT control through CM108 GPIO pin
|
||||
|
||||
HAMLIB_PTT_TYPE = 'RIG_PTT_NONE'
|
||||
PTT_STATE = False
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
#-------------------------
|
||||
# -------------------------
|
||||
# FreeDV Defaults
|
||||
FREEDV_RECEIVE = True
|
||||
|
||||
|
|
@ -70,9 +61,9 @@ FREEDV_SIGNALLING_BYTES_PER_FRAME = 0
|
|||
FREEDV_SIGNALLING_PAYLOAD_PER_FRAME = 0
|
||||
|
||||
BER = 0
|
||||
#---------------------------------
|
||||
# ---------------------------------
|
||||
|
||||
#Audio Defaults
|
||||
# Audio Defaults
|
||||
AUDIO_INPUT_DEVICE = 1
|
||||
AUDIO_OUTPUT_DEVICE = 1
|
||||
#TX_SAMPLE_STATE = None
|
||||
|
|
@ -80,13 +71,13 @@ AUDIO_OUTPUT_DEVICE = 1
|
|||
|
||||
#AUDIO_SAMPLE_RATE_RX = 44100
|
||||
#AUDIO_SAMPLE_RATE_TX = 44100
|
||||
MODEM_SAMPLE_RATE = 8000 #8000
|
||||
MODEM_SAMPLE_RATE = 8000 # 8000
|
||||
AUDIO_FRAMES_PER_BUFFER = 2048
|
||||
AUDIO_CHANNELS = 1
|
||||
AUDIO_RMS = 0
|
||||
#---------------------------------
|
||||
# ---------------------------------
|
||||
|
||||
#ARQ DEFAULTS
|
||||
# ARQ DEFAULTS
|
||||
TX_N_MAX_RETRIES = 5
|
||||
TX_N_RETRIES = 0
|
||||
|
||||
|
|
@ -99,41 +90,41 @@ ARQ_RX_BURST_BUFFER = []
|
|||
ARQ_RX_FRAME_BUFFER = []
|
||||
ARQ_RX_FRAME_N_BURSTS = 0
|
||||
|
||||
## TX
|
||||
# TX
|
||||
ARQ_TX_N_CURRENT_ARQ_FRAME = 0
|
||||
ARQ_TX_N_TOTAL_ARQ_FRAMES = 0
|
||||
##
|
||||
|
||||
## RX
|
||||
ARQ_N_ARQ_FRAMES_PER_DATA_FRAME = 0 #total number of arq frames per data frame
|
||||
# RX
|
||||
ARQ_N_ARQ_FRAMES_PER_DATA_FRAME = 0 # total number of arq frames per data frame
|
||||
ARQ_RX_N_CURRENT_ARQ_FRAME = 0
|
||||
##
|
||||
|
||||
ARQ_N_RX_ARQ_FRAMES = 0 # total number of received frames
|
||||
ARQ_N_RX_FRAMES_PER_BURSTS = 0 # NUMBER OF FRAMES WE ARE WAITING FOR --> GOT DATA FROM RECEIVED FRAME
|
||||
ARQ_ACK_PAYLOAD_PER_FRAME = 0 # PAYLOAD per ACK frame
|
||||
ARQ_N_RX_ARQ_FRAMES = 0 # total number of received frames
|
||||
ARQ_N_RX_FRAMES_PER_BURSTS = 0 # NUMBER OF FRAMES WE ARE WAITING FOR --> GOT DATA FROM RECEIVED FRAME
|
||||
ARQ_ACK_PAYLOAD_PER_FRAME = 0 # PAYLOAD per ACK frame
|
||||
|
||||
ARQ_ACK_RECEIVED = False # set to 1 if ACK received
|
||||
ARQ_RX_ACK_TIMEOUT = False # set to 1 if timeut reached
|
||||
ARQ_RX_ACK_TIMEOUT_SECONDS = 10.0 #timeout for waiting for ACK frames
|
||||
ARQ_ACK_RECEIVED = False # set to 1 if ACK received
|
||||
ARQ_RX_ACK_TIMEOUT = False # set to 1 if timeut reached
|
||||
ARQ_RX_ACK_TIMEOUT_SECONDS = 10.0 # timeout for waiting for ACK frames
|
||||
|
||||
ARQ_FRAME_ACK_RECEIVED = False # set to 1 if FRAME ACK received
|
||||
ARQ_FRAME_ACK_RECEIVED = False # set to 1 if FRAME ACK received
|
||||
ARQ_RX_FRAME_TIMEOUT = False
|
||||
ARQ_RX_FRAME_TIMEOUT_SECONDS = 10.0
|
||||
|
||||
|
||||
ARQ_RX_RPT_TIMEOUT = False
|
||||
ARQ_RX_RPT_TIMEOUT_SECONDS = 10.0
|
||||
ARQ_RPT_RECEIVED = False #indicate if RPT frame has been received
|
||||
ARQ_RPT_FRAMES = [] #buffer for frames which are requested to repeat
|
||||
ARQ_RPT_RECEIVED = False # indicate if RPT frame has been received
|
||||
ARQ_RPT_FRAMES = [] # buffer for frames which are requested to repeat
|
||||
|
||||
FRAME_CRC = b''
|
||||
FRAME_BOF = b'\xAA\xAA' #here we define 2 bytes for the BOF
|
||||
FRAME_EOF = b'\xFF\xFF' #here we define 2 bytes for the EOF
|
||||
ARQ_FRAME_BOF_RECEIVED = False # status, if we received a BOF of a data frame
|
||||
ARQ_FRAME_EOF_RECEIVED = False # status, if we received a EOF of a data frame
|
||||
FRAME_BOF = b'\xAA\xAA' # here we define 2 bytes for the BOF
|
||||
FRAME_EOF = b'\xFF\xFF' # here we define 2 bytes for the EOF
|
||||
ARQ_FRAME_BOF_RECEIVED = False # status, if we received a BOF of a data frame
|
||||
ARQ_FRAME_EOF_RECEIVED = False # status, if we received a EOF of a data frame
|
||||
|
||||
ARQ_N_SENT_FRAMES = 0 #counter for already sent frames
|
||||
ARQ_N_SENT_FRAMES = 0 # counter for already sent frames
|
||||
|
||||
|
||||
# ARQ STATES:
|
||||
|
|
@ -177,4 +168,3 @@ RX_BUFFER_SIZE = 0
|
|||
|
||||
# ------- HEARD STATIOS BUFFER
|
||||
HEARD_STATIONS = []
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue