adam/LoraSA
1
0
Fork 0
mirror of https://github.com/Genaker/LoraSA.git synced 2026-08-12 03:42:52 +02:00
Code Issues Packages Projects Releases Wiki Activity

Cleaned up, linted, etc...

Browse Source
This commit is contained in:
KonradIT
2024-10-14 23:45:35 +02:00
parent 1c65cd745a
commit e6c92e4eb0
1 changed files with 49 additions and 68 deletions
Download Patch File Download Diff File Expand all files Collapse all files
SpectrumScan.py
+49 -68
View File
@@ -7,32 +7,21 @@ import sys
import numpy as np import numpy as np
import matplotlib as mpl import matplotlib as mpl
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
import json
from datetime import datetime from datetime import datetime
from argparse import RawTextHelpFormatter from argparse import RawTextHelpFormatter
import json # Constants
SCAN_WIDTH = 33 # number of samples in each scanline
OUT_PATH = "out" # output path for saved files
# number of samples in each scanline # Default settings
SCAN_WIDTH = 33
# scanline Serial start/end markers
SCAN_MARK_START = 'SCAN '
SCAN_MARK_FREQ = 'FREQ '
SCAN_MARK_END = ' END'
# output path
OUT_PATH = 'out'
# default settings
DEFAULT_BAUDRATE = 115200 DEFAULT_BAUDRATE = 115200
DEFAULT_COLOR_MAP = 'viridis' DEFAULT_COLOR_MAP = 'viridis'
DEFAULT_SCAN_LEN = 200 DEFAULT_SCAN_LEN = 200
DEFAULT_RSSI_OFFSET = -11 DEFAULT_RSSI_OFFSET = -11
# Print iterations progress def print_progress_bar(iteration, total, prefix='', suffix='', decimals=1, length=50, fill='█', print_end="\r"):
# from https://stackoverflow.com/questions/3173320/text-progress-bar-in-terminal-with-block-characters
def printProgressBar (iteration, total, prefix = '', suffix = '', decimals = 1, length = 50, fill = '█', printEnd = "\r"):
""" """
Call in a loop to create terminal progress bar Call in a loop to create terminal progress bar
@params: @params:
@@ -43,80 +32,72 @@ def printProgressBar (iteration, total, prefix = '', suffix = '', decimals = 1,
decimals - Optional : positive number of decimals in percent complete (Int) decimals - Optional : positive number of decimals in percent complete (Int)
length - Optional : character length of bar (Int) length - Optional : character length of bar (Int)
fill - Optional : bar fill character (Str) fill - Optional : bar fill character (Str)
printEnd - Optional : end character (e.g. "\r", "\r\n") (Str) print_end - Optional : end character (e.g. "\r", "\r\n") (Str)
""" """
percent = ("{0:." + str(decimals) + "f}").format(100 * (iteration / float(total))) percent = ("{0:." + str(decimals) + "f}").format(100 * (iteration / float(total)))
filledLength = int(length * iteration // total) filled_length = int(length * iteration // total)
bar = fill * filledLength + '-' * (length - filledLength) bar = fill * filled_length + '-' * (length - filled_length)
print(f'\r{prefix} |{bar}| {percent}% {suffix}', end = printEnd) print(f'\r{prefix} |{bar}| {percent}% {suffix}', end=print_end)
if iteration == total: if iteration == total:
print() print()
def parse_line(line): def parse_line(line):
json_line = json.loads(line) """Parse a JSON line from the serial input."""
return json_line return json.loads(line)
def main(): def main():
parser = argparse.ArgumentParser(formatter_class=RawTextHelpFormatter, description=''' parser = argparse.ArgumentParser(formatter_class=RawTextHelpFormatter, description='''\
Parse serial data from LOG_DATA_JSON functionality. Parse serial data from LOG_DATA_JSON functionality.
1. #define LOG_DATA_JSON true - add this line in main.cpp, upload to device 1. #define LOG_DATA_JSON true - add this line in main.cpp, upload to device
2. Run the script with appropriate arguments. 2. Run the script with appropriate arguments.
3. Once the scan is complete, output files will be saved to out/ 3. Once the scan is complete, output files will be saved to out/
''') ''')
parser.add_argument('port', parser.add_argument('port', type=str, help='COM port to connect to the device')
type=str, parser.add_argument('--speed', default=DEFAULT_BAUDRATE, type=int,
help='COM port to connect to the device') help=f'COM port baudrate (defaults to {DEFAULT_BAUDRATE})')
parser.add_argument('--speed', parser.add_argument('--map', default=DEFAULT_COLOR_MAP, type=str,
default=DEFAULT_BAUDRATE, help=f'Matplotlib color map to use for the output (defaults to "{DEFAULT_COLOR_MAP}")')
type=int, parser.add_argument('--len', default=DEFAULT_SCAN_LEN, type=int,
help=f'COM port baudrate (defaults to {DEFAULT_BAUDRATE})') help=f'Number of scanlines to record (defaults to {DEFAULT_SCAN_LEN})')
parser.add_argument('--map', parser.add_argument('--offset', default=DEFAULT_RSSI_OFFSET, type=int,
default=DEFAULT_COLOR_MAP, help=f'Default RSSI offset in dBm (defaults to {DEFAULT_RSSI_OFFSET})')
type=str, parser.add_argument('--freq', default=-1, type=float,
help=f'Matplotlib color map to use for the output (defaults to "{DEFAULT_COLOR_MAP}")') help='Default starting frequency in MHz')
parser.add_argument('--len',
default=DEFAULT_SCAN_LEN,
type=int,
help=f'Number of scanlines to record (defaults to {DEFAULT_SCAN_LEN})')
parser.add_argument('--offset',
default=DEFAULT_RSSI_OFFSET,
type=int,
help=f'Default RSSI offset in dBm (defaults to {DEFAULT_RSSI_OFFSET})')
parser.add_argument('--freq',
default=-1,
type=float,
help=f'Default starting frequency in MHz')
args = parser.parse_args() args = parser.parse_args()
# create the result array # Create the result array
scan_len = args.len
arr = np.zeros((scan_len, SCAN_WIDTH)) arr = np.zeros((scan_len, SCAN_WIDTH))
# scanline counter # Scanline counter
row = 0 row = 0
# list of frequencies # List of frequencies
freq_list = [] freq_list = []
# open the COM port # Open the COM port
with serial.Serial(args.port, args.speed, timeout=None) as com: with serial.Serial(args.port, args.speed, timeout=None) as com:
while(True): while True:
# update the progress bar # Update the progress bar
printProgressBar(row, scan_len) print_progress_bar(row, scan_len)
# read a single line # Read a single line
try: try:
line = com.readline().decode('utf-8') line = com.readline().decode('utf-8').strip()
except: except UnicodeDecodeError:
continue continue
print(line)
if "{" in line: if line.startswith("{"):
try: try:
data = parse_line(line) data = parse_line(line)
except: except json.JSONDecodeError:
continue continue
# get the lowest frequency for now, could be averaged too.
freq = data["low_range_freq"] freq = data["low_range_freq"]
# value in negative, eg: -70
rssi = int(data["value"]) rssi = int(data["value"])
if freq not in freq_list: if freq not in freq_list:
@@ -125,24 +106,24 @@ def main():
col = freq_list.index(freq) col = freq_list.index(freq)
arr[row][col] = rssi arr[row][col] = rssi
# increment the row counter # Increment the row counter
row = row + 1 row += 1
# check if we're done # Check if we're done
if (row >= scan_len): if row >= scan_len:
break break
# create the figure # Create the figure
fig, ax = plt.subplots(figsize=(12, 8)) fig, ax = plt.subplots(figsize=(12, 8))
# display the result as heatmap # Display the result as heatmap
extent = [0, scan_len, freq_list[0], freq_list[-1]] extent = [0, scan_len, freq_list[0], freq_list[-1]]
im = ax.imshow(arr.T, cmap=args.map, extent=extent, aspect='auto', origin='lower') im = ax.imshow(arr.T, cmap=args.map, extent=extent, aspect='auto', origin='lower')
fig.colorbar(im, label='RSSI (dBm)') fig.colorbar(im, label='RSSI (dBm)')
# set some properties and show # Set plot properties and show
timestamp = datetime.now().strftime('%y-%m-%d %H-%M-%S') timestamp = datetime.now().strftime('%y-%m-%d %H-%M-%S')
title = f'RadioLib SX126x Spectral Scan {timestamp}' title = f'LoraSA Spectral Scan {timestamp}'
plt.xlabel("Time (sample)") plt.xlabel("Time (sample)")
plt.ylabel("Frequency (MHz)") plt.ylabel("Frequency (MHz)")
fig.suptitle(title) fig.suptitle(title)