119 lines
3.4 KiB
Python
119 lines
3.4 KiB
Python
import serial
|
|
import time
|
|
import logging as log
|
|
|
|
|
|
SAMPLING_RATE_KHZ = 16
|
|
FRAME_DUR_MS = 10
|
|
PRESET = f'{SAMPLING_RATE_KHZ}_2_1'
|
|
|
|
if SAMPLING_RATE_KHZ == 8:
|
|
BITRATE_KBPS = 16
|
|
elif SAMPLING_RATE_KHZ == 16:
|
|
BITRATE_KBPS = 32
|
|
elif SAMPLING_RATE_KHZ == 24:
|
|
BITRATE_KBPS = 48
|
|
else:
|
|
raise NotImplemented()
|
|
|
|
def write_to_serial_read_respone(port, cmd, timeout = 2):
|
|
# Initialize serial connection
|
|
ser = serial.Serial(timeout = timeout)
|
|
ser.port = port
|
|
ser.baudrate = 115200
|
|
ser.bytesize = serial.EIGHTBITS
|
|
ser.parity = serial.PARITY_NONE
|
|
ser.stopbits = serial.STOPBITS_ONE
|
|
|
|
try:
|
|
# Try to open the serial connection
|
|
#if not ser.is_open:
|
|
ser.open()
|
|
|
|
# Send string to serial port and get response
|
|
command = f"{cmd.strip()}\r\n"
|
|
ser.write(command.encode())
|
|
time.sleep(1) # wait a bit for response
|
|
|
|
readlines = []
|
|
for _ in range(20):
|
|
line = ser.readline().decode('utf-8').strip()
|
|
if not line:
|
|
ser.close()
|
|
break
|
|
else:
|
|
readlines.append(line)
|
|
|
|
except serial.SerialException as e:
|
|
print(f"Error communicating with serial port: {e}")
|
|
finally:
|
|
# Close serial connection before returning
|
|
if ser.is_open:
|
|
ser.close()
|
|
return readlines
|
|
|
|
|
|
|
|
def gen_broadcast_config_cmd(preset, broadcast_config: dict):
|
|
"""
|
|
Writes broadcaster configuration to the given serial port.
|
|
|
|
Args:
|
|
serial_port (str): Device path of the serial port (e.g., '/dev/ttyACM0')
|
|
preset (str): Preset string used in nac preset line
|
|
broadcast_names (list): List of names for each broadcast group
|
|
"""
|
|
cmds = []
|
|
for ch, file_name in broadcast_config.items():
|
|
|
|
cmds.append(f"nac preset {preset} {ch}")
|
|
cmds.append(f"nac broadcast_name broadcast{ch} {ch}")
|
|
cmds.append(f"nac file select_play_once {file_name} {ch} 0 0")
|
|
cmds.append(f"nac num_bises 1 {ch} 0")
|
|
|
|
return cmds
|
|
|
|
# TODO: Advertising interval wird ungelmäßig bei mehr als 3 broadcasts 10ms -> 1s< bei 24kHz sampling rate
|
|
BROADCAST_CONFIG = {
|
|
0: f"announcement_{SAMPLING_RATE_KHZ}_{FRAME_DUR_MS}_{BITRATE_KBPS}_de.lc3",
|
|
1: f"announcement_{SAMPLING_RATE_KHZ}_{FRAME_DUR_MS}_{BITRATE_KBPS}_en.lc3",
|
|
2: f"announcement_{SAMPLING_RATE_KHZ}_{FRAME_DUR_MS}_{BITRATE_KBPS}_fr.lc3",
|
|
#3: f"announcement_{SAMPLING_RATE_KHZ}_{FRAME_DUR_MS}_{BITRATE_KBPS}_es.lc3",
|
|
#4: f"announcement_{SAMPLING_RATE_KHZ}_{FRAME_DUR_MS}_{BITRATE_KBPS}_it.lc3"
|
|
}
|
|
|
|
def broadcaster_config():
|
|
|
|
import subprocess
|
|
|
|
PORT = "/dev/ttyACM0"
|
|
|
|
total_ret= ""
|
|
|
|
cmds = gen_broadcast_config_cmd(PRESET, BROADCAST_CONFIG)
|
|
|
|
subprocess.run(["nrfjprog", "--reset", "-s", "1050109484"], check=True)
|
|
|
|
time.sleep(2)
|
|
ret = write_to_serial_read_respone(PORT, f"nac en_usb_mass", timeout=0.1)
|
|
total_ret += "\n".join(ret)
|
|
log.info("\n".join(ret))
|
|
time.sleep(1)
|
|
|
|
|
|
for cmd in cmds:
|
|
ret = write_to_serial_read_respone(PORT, cmd, timeout=0.1)
|
|
log.info("\n".join(ret))
|
|
total_ret += "\n".join(ret)
|
|
|
|
time.sleep(1)
|
|
|
|
for i in BROADCAST_CONFIG.keys():
|
|
ret = write_to_serial_read_respone(PORT, f"nac start_idx {i}", timeout=0.1)
|
|
total_ret += "\n".join(ret)
|
|
log.info("\n".join(ret))
|
|
time.sleep(0.2)
|
|
|
|
return total_ret
|
|
|