58 lines
1.5 KiB
Python
58 lines
1.5 KiB
Python
|
M
|
import pathlib
|
||
|
|
|
||
|
|
MAX_PLOT_POINTS = 100_000
|
||
|
|
COLORS = {
|
||
|
|
"primary": "#6366f1",
|
||
|
|
"secondary": "#8b5cf6",
|
||
|
|
"accent": "#06b6d4",
|
||
|
|
"dark": "#1e293b",
|
||
|
|
"light": "#f8fafc",
|
||
|
|
"text": "#334155",
|
||
|
|
"muted": "#64748b",
|
||
|
|
"success": "#10b981",
|
||
|
|
"warning": "#f59e0b",
|
||
|
|
"error": "#ef4444",
|
||
|
|
"purple": "#8b5cf6",
|
||
|
|
"magenta": "#d946ef",
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
def decimate(x, max_points=MAX_PLOT_POINTS):
|
||
|
|
if len(x) <= max_points:
|
||
|
|
return x
|
||
|
|
step = len(x) // max_points
|
||
|
|
return x[::step]
|
||
|
|
|
||
|
|
|
||
|
|
def extract_metadata_fields(metadata):
|
||
|
|
sample_rate = next((v for k, v in metadata.items() if "sample_rate" in k), 1)
|
||
|
|
center_freq = next((v for k, v in metadata.items() if "center_freq" in k), 0)
|
||
|
|
sdr = next((v for k, v in metadata.items() if "sdr" in k), "Unknown")
|
||
|
|
return sample_rate, center_freq, sdr
|
||
|
|
|
||
|
|
|
||
|
|
def set_path(output_path):
|
||
|
|
split_path = output_path.split("/")
|
||
|
|
|
||
|
|
if len(split_path) == 1:
|
||
|
|
folder = "images"
|
||
|
|
file = split_path[0]
|
||
|
|
elif len(split_path) > 2:
|
||
|
|
file = split_path[-1]
|
||
|
|
folder = "/".join(split_path[:-1])
|
||
|
|
else:
|
||
|
|
folder, file = split_path
|
||
|
|
|
||
|
|
split_file = file.split(".")
|
||
|
|
if len(split_file) == 2:
|
||
|
|
extension = split_file[1]
|
||
|
|
else:
|
||
|
|
extension = "no extension"
|
||
|
|
if extension != "png" and extension != "svg":
|
||
|
|
print(f"{extension} not supported, saving as .png.")
|
||
|
|
extension = "png"
|
||
|
|
file = file + ".png"
|
||
|
|
|
||
|
|
pathlib.Path(folder).mkdir(parents=True, exist_ok=True)
|
||
|
|
return "/".join([folder, file]), extension
|