119 lines
4.6 KiB
Python
119 lines
4.6 KiB
Python
"""Live processing settings model and atomic JSON writer."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from dataclasses import asdict, dataclass, field
|
|
import logging
|
|
from pathlib import Path
|
|
|
|
from python_app.file_io import atomic_write_json
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
@dataclass(slots=True)
|
|
class ProcessingLiveConfig:
|
|
"""Runtime-adjustable processing parameters shared with data processor."""
|
|
|
|
processor_mode: str = "pass_through"
|
|
bscan_axis: str = "abs"
|
|
bscan_cut_m: float = 0.824
|
|
bscan_max_depth_m: float = 1.0
|
|
bscan_gain: float = 1.0
|
|
bscan_start_freq_mhz: float = 100.0
|
|
bscan_stop_freq_mhz: float = 8800.0
|
|
tx_geometry: list[dict] | None = None
|
|
rx_geometry: list[dict] | None = None
|
|
gpr_input_positions: list[int] | None = None
|
|
gpr_output_positions: list[int] | None = None
|
|
gpr_min_depth_m: float = 2.0
|
|
gpr_max_depth_m: float = 14.0
|
|
gpr_range_comp_power: float = 0.1
|
|
gpr_object_min_frac: float = 0.7
|
|
gpr_score_mode: str = "combined"
|
|
# Backprojection intra-sweep speed-correction mode: "int_minus" (full
|
|
# correction) or "int_focus" (focusing residual only).
|
|
gpr_motion_mode: str = "int_minus"
|
|
gpr_max_detected_objects_to_draw: int = 5
|
|
gpr_draw_top_m_objects: int = 2
|
|
gpr_object_approach_min_frames: int = 3
|
|
gpr_speed_m_s: float = 0.0
|
|
gpr_look_angle_deg: float = 0.0
|
|
gpr_direction_sign: float = 1.0
|
|
gpr_start_freq_mhz: float = 3000.0
|
|
gpr_stop_freq_mhz: float = 6000.0
|
|
gpr_background_subtract_enabled: bool = True
|
|
gpr_background_mean_count: int = 10
|
|
gpr_remove_sidelobe_objects_enabled: bool = True
|
|
gpr_bp_ground_depth_m: float = 1.25
|
|
pair_depth_equalize: bool = True
|
|
# Visible X/Z window (metres). The locator and the desktop plot both clip
|
|
# detected objects to this window, so the socket broadcasts only what is shown.
|
|
gpr_visible_x_min_m: float = -2.0
|
|
gpr_visible_x_max_m: float = 2.0
|
|
gpr_visible_z_min_m: float = 0.0
|
|
gpr_visible_z_max_m: float = 14.0
|
|
# 3D SAR (sar3d processor) parameters, named after the reference notebook
|
|
# Multi_antenna_demonstrate_depthnorm.ipynb (SAR_FRAME_RADIUS, BP_GROUND_DEPTH_M, ...).
|
|
# The band, depth gate, antenna selection, speed and look angle are the
|
|
# shared gpr_* fields.
|
|
sar3d_frame_radius: int = 0
|
|
sar3d_voxel_m: float = 0.02 # SAR_VOXEL_M, one edge for X, Y and Z
|
|
sar3d_bp_ground_depth_m: float = 1.25
|
|
sar3d_comp_range_power: float = 0.3
|
|
sar3d_use_compensation: bool = True
|
|
sar3d_normalize_each_frame: bool = True
|
|
sar3d_x_size_m: float = 0.0
|
|
sar3d_y_min_m: float = -0.5
|
|
sar3d_y_max_m: float = 2.0
|
|
sar3d_z_size_m: float = 0.0
|
|
# When true, the C++ data_processor ignores socket-supplied vlc updates
|
|
# and keeps using `gpr_speed_m_s` from this file.
|
|
ignore_socket_speed: bool = False
|
|
reprocess_current_result: bool = True
|
|
history_clear_before_ns: int = 0
|
|
history_removed: list[dict[str, int]] = field(default_factory=list)
|
|
settings_revision: int = 0
|
|
|
|
def __post_init__(self) -> None:
|
|
"""Normalize optional list fields to concrete integer lists."""
|
|
if self.gpr_input_positions is None:
|
|
self.gpr_input_positions = []
|
|
else:
|
|
self.gpr_input_positions = [int(value) for value in self.gpr_input_positions]
|
|
if self.gpr_output_positions is None:
|
|
self.gpr_output_positions = []
|
|
else:
|
|
self.gpr_output_positions = [int(value) for value in self.gpr_output_positions]
|
|
|
|
def to_dict(self) -> dict[str, object]:
|
|
"""Convert live config to JSON-serializable dictionary."""
|
|
optional_geometry = {"tx_geometry", "rx_geometry"}
|
|
return {
|
|
name: value for name, value in asdict(self).items()
|
|
if name not in optional_geometry or value is not None
|
|
}
|
|
|
|
|
|
class ProcessingLiveConfigWriter:
|
|
"""Atomic writer for processing live-config file."""
|
|
|
|
def __init__(self, config_path: Path) -> None:
|
|
"""Create writer targeting `config_path`."""
|
|
self._config_path = config_path
|
|
self._config_path.parent.mkdir(parents=True, exist_ok=True)
|
|
|
|
@property
|
|
def path(self) -> Path:
|
|
"""Return destination config path."""
|
|
return self._config_path
|
|
|
|
def write(self, config: ProcessingLiveConfig) -> Path:
|
|
"""Atomically write config by temp-file replace."""
|
|
# Hot path: rewritten on every live knob change, so keep this at DEBUG.
|
|
logger.debug(
|
|
"Writing live processing config (mode=%s) to %s",
|
|
config.processor_mode,
|
|
self._config_path,
|
|
)
|
|
return atomic_write_json(self._config_path, config.to_dict())
|