59 lines
2.0 KiB
Python
59 lines
2.0 KiB
Python
"""Path and naming helpers for NPZ snapshot storage."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import re
|
|
from collections.abc import Sequence
|
|
|
|
|
|
def radar_key_from_config(
|
|
model_name: str,
|
|
serial: str,
|
|
sweep_start_hz: float,
|
|
sweep_stop_hz: float,
|
|
sweep_points: int,
|
|
ifbw_hz: float,
|
|
power_dbm: float,
|
|
extra_serials: Sequence[str] | None = None,
|
|
) -> str:
|
|
"""Build deterministic key for calibration/reference set lookup."""
|
|
serial_parts = [serial or "no_serial"]
|
|
if extra_serials:
|
|
serial_parts.extend(str(value).strip() or "no_serial" for value in extra_serials)
|
|
serial_part = "_".join(sanitize_path_component(value) for value in serial_parts)
|
|
start_token = _format_float_for_key(sweep_start_hz)
|
|
stop_token = _format_float_for_key(sweep_stop_hz)
|
|
points_token = "adc" if model_name.strip().lower() == "kamil_adc" else str(int(sweep_points))
|
|
ifbw_token = _format_float_for_key(ifbw_hz)
|
|
power_token = _format_float_for_key(power_dbm)
|
|
return (
|
|
f"{model_name}_{serial_part}"
|
|
f"_st{start_token}_sp{stop_token}"
|
|
f"_p{points_token}_if{ifbw_token}_pw{power_token}"
|
|
)
|
|
|
|
|
|
def collection_dir_name(index: int, collection_id: int, monotonic_ns: int) -> str:
|
|
"""Build canonical collection directory name for snapshot stages."""
|
|
return f"{index:04d}_id{int(collection_id)}_ns{int(monotonic_ns)}"
|
|
|
|
|
|
def sanitize_path_component(name: str) -> str:
|
|
"""Convert arbitrary string into filesystem-safe component."""
|
|
cleaned = re.sub(r"[^A-Za-z0-9._-]+", "_", name.strip())
|
|
cleaned = cleaned.strip("._")
|
|
return cleaned or "snapshot"
|
|
|
|
|
|
def format_float_for_key(value: float) -> str:
|
|
"""Format float compactly for deterministic key serialization."""
|
|
integer = int(round(value))
|
|
if abs(value - float(integer)) < 1e-6:
|
|
return str(integer)
|
|
return f"{value:.6f}".rstrip("0").rstrip(".")
|
|
|
|
|
|
def _format_float_for_key(value: float) -> str:
|
|
"""Private alias preserved for internal compatibility."""
|
|
return format_float_for_key(value)
|