awe/radar_system
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Rework radar system

Browse Source
This commit is contained in:
Ayzen
2026-10-06 12:32:26 +03:00
parent 552cdc688a
commit 5bb86b7165
299 changed files with 13537 additions and 27763 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python_app/tests/test_sar3d_plot.py
+96 -82
View File
@@ -3,22 +3,18 @@
from __future__ import annotations
import os
from types import SimpleNamespace
import unittest
from unittest import mock
import numpy as np
os.environ.setdefault("QT_QPA_PLATFORM", "offscreen")
from PyQt6.QtCore import Qt # noqa: E402
from PyQt6.QtWidgets import QApplication, QStackedWidget # noqa: E402
from PyQt6.QtCore import Qt, QSignalBlocker # noqa: E402
from PyQt6.QtWidgets import QApplication, QCheckBox, QComboBox, QDoubleSpinBox # noqa: E402
from python_app.gui.controllers.app_window_pipeline_mixin import AppWindowPipelineMixin # noqa: E402
from python_app.gui.controllers.app_window_config.live_processing_mixin import AppWindowLiveProcessingMixin # noqa: E402
from python_app.gui.controllers.app_window_plot_mixin import AppWindowPlotMixin # noqa: E402
from python_app.gui.controllers.app_window_plot.sar3d_plot_mixin import SAR3D_VIEWS, sar3d_status # noqa: E402
from python_app.gui.controllers.app_window_ui_mixin import AppWindowUiMixin # noqa: E402
from python_app.models.dataset_model import ResultCollection, ResultPayload # noqa: E402
from python_app.gui.controllers.renderers.sar3d import SAR3D_STATES, SAR3D_VIEWS, Sar3dRenderer # noqa: E402
from python_app.models.dataset_model import ComboKey, ResultCollection, ResultPayload, TraceData # noqa: E402
from python_app.tests.gui_test_support import widget
def _status(code: int, message: str) -> ResultPayload:
@@ -27,7 +23,8 @@ def _status(code: int, message: str) -> ResultPayload:
def _collection(*payloads: ResultPayload) -> ResultCollection:
return ResultCollection(collection_id=5, monotonic_ns=1500, collection_payloads=list(payloads))
return ResultCollection(collection_id=5, monotonic_ns=1500, collection_payloads=list(payloads),
capture_start_ns=1200, capture_end_ns=1400)
def _projections() -> ResultCollection:
@@ -43,114 +40,131 @@ def _projections() -> ResultCollection:
table=np.array([[2, 0.5, -0.2, 2]], dtype=np.float32)))
class _View(AppWindowPlotMixin, AppWindowUiMixin, AppWindowPipelineMixin, AppWindowLiveProcessingMixin):
def __init__(self) -> None:
self._processing_mode = SimpleNamespace(currentText=lambda: "sar3d")
self._plot_stack = QStackedWidget()
self._build_sar3d_plot_page()
class Sar3dPlotTest(unittest.TestCase):
@classmethod
def setUpClass(cls) -> None:
cls.app = QApplication.instance() or QApplication([])
def setUp(self) -> None:
self.view = _View()
self.addCleanup(self.view._plot_stack.close)
self.renderer = Sar3dRenderer()
self.addCleanup(self.renderer.widget.close)
def test_all_four_projections_have_visible_images_and_correct_peak_axes(self) -> None:
self.assertTrue(self.view._draw_results(_projections()))
self.assertEqual(self.view._sar3d_status_label.text(), "SAR3D: ready — CUDA")
self.assertTrue(self.renderer.draw(_projections()))
self.assertEqual(self.renderer.status_label.text(), "SAR3D: ready — CUDA")
positions = {"sar3d_front": (0.5, -0.2), "sar3d_side": (2, -0.2),
"sar3d_top": (0.5, 2), "sar3d_oblique": (0.5, -0.2)}
for name, position in positions.items():
image = self.view._sar3d_images[name]
image = self.renderer.images[name]
self.assertTrue(image.isVisible(), name)
self.assertGreater(image.boundingRect().width(), 0)
self.assertGreater(image.boundingRect().height(), 0)
xx, yy = self.view._sar3d_peaks[name].getData()
xx, yy = self.renderer.peaks[name].getData()
np.testing.assert_allclose([xx[0], yy[0]], position)
self.assertTrue(all(label.isVisible() for label in self.view._sar3d_axis_labels.values()))
self.assertTrue(all(label.isVisible() for label in self.renderer.axis_labels.values()))
self.renderer.clear()
self.assertEqual(self.renderer.status_label.text(), "SAR3D: waiting for measurements")
def test_unavailable_warning_clears_previous_images_and_labels_and_shows_reason(self) -> None:
self.view._draw_results(_projections())
self.renderer.draw(_projections())
message = "CUDA unavailable: драйвер <libcuda>"
self.assertFalse(self.view._draw_results(_collection(_status(3, message))))
self.assertEqual(self.view._sar3d_status_label.text(), f"SAR3D: warning — {message}")
self.assertEqual(self.view._sar3d_status_label.textFormat(), Qt.TextFormat.PlainText)
self.assertTrue(all(not image.isVisible() for image in self.view._sar3d_images.values()))
self.assertTrue(all(not label.isVisible() for label in self.view._sar3d_axis_labels.values()))
self.assertFalse(self.renderer.draw(_collection(_status(3, message))))
self.assertEqual(self.renderer.status_label.text(), f"SAR3D: warning — {message}")
self.assertEqual(self.renderer.status_label.textFormat(), Qt.TextFormat.PlainText)
self.assertTrue(all(not image.isVisible() for image in self.renderer.images.values()))
self.assertTrue(all(not label.isVisible() for label in self.renderer.axis_labels.values()))
def test_no_signal_and_missing_projections_explain_empty_panels(self) -> None:
self.assertFalse(self.view._draw_results(_collection(_status(1, "No signal in selected range"))))
self.assertIn("no signal", self.view._sar3d_status_label.text())
self.assertFalse(self.view._draw_results(_collection()))
self.assertEqual(self.view._sar3d_status_label.text(), "SAR3D: waiting for projection data")
self.assertFalse(self.renderer.draw(_collection(_status(1, "No signal in selected range"))))
self.assertIn("no signal", self.renderer.status_label.text())
self.assertFalse(self.renderer.draw(_collection()))
self.assertEqual(self.renderer.status_label.text(), "SAR3D: waiting for projection data")
def test_ready_cuda_status_and_clear_restores_waiting(self) -> None:
self.assertTrue(self.view._draw_results(_projections()))
self.assertEqual(self.view._sar3d_status_label.text(), "SAR3D: ready — CUDA")
self.view._clear_sar3d_plot()
self.assertEqual(self.view._sar3d_status_label.text(), "SAR3D: waiting for measurements")
def test_malformed_status_is_a_visible_error_instead_of_a_qt_exception(self) -> None:
for table in (np.empty((0, 0)), np.array([[2, np.nan]]), np.array([[2, 256]]),
np.array([[2, 0.5]]), np.array([[99]]), np.array([2])):
with self.subTest(table=table):
payload = _status(2, "failure")
payload.table = table
self.assertEqual(sar3d_status(_collection(payload)), (2, "Invalid SAR3D status payload"))
self.assertFalse(self.view._draw_results(_collection(payload)))
class Sar3dIntegrationTest(unittest.TestCase):
@classmethod
def setUpClass(cls) -> None:
cls.app = QApplication.instance() or QApplication([])
def setUp(self) -> None:
from python_app.tests.gui_test_support import isolated_window
self.view = isolated_window(self.addCleanup)
self.view._pipeline.timer.stop()
self.mode = self.view._view.processing.mode
with QSignalBlocker(self.mode):
self.mode.setCurrentText("sar3d")
self.view._plots.set_mode("sar3d")
self.view._plots.required_revision = None
self.renderer = self.view._plots.sar3d
def test_capture_preview_keeps_sar_page_and_uses_saved_trace_settings(self) -> None:
self.view._plots.set_mode("sar3d")
fields = self.view._view.processing
phase = widget(fields, "_show_phase_checkbox", QCheckBox)
unwrap = widget(fields, "_unwrap_phase_checkbox", QCheckBox)
fixed = widget(fields, "_pass_through_fixed_y_enabled", QCheckBox)
lower = widget(fields, "_pass_through_y_min_db", QDoubleSpinBox)
upper = widget(fields, "_pass_through_y_max_db", QDoubleSpinBox)
revision = self.view._live.revision
phase.setChecked(False)
unwrap.setChecked(True)
fixed.setChecked(True)
lower.setValue(-70)
upper.setValue(-5)
self.assertEqual(self.view._live.revision, revision)
trace = TraceData(ComboKey(0, 0), np.array([1., 2.]), np.array([0.1 + 0j, 0.2 + 0j]))
self.view._plots.draw_trace(trace, "Captured reference")
self.assertIs(self.view._plots.stack.currentWidget(), self.renderer.widget)
self.assertFalse(self.view._plots.trace.phase.curves)
self.assertTrue(self.view._config.processing_settings().pass_through.unwrap_phase)
np.testing.assert_allclose(self.view._plots.trace.magnitude.view.viewRange()[1], [-70, -5])
def test_single_capture_finishes_reported_failure_without_losing_its_result(self) -> None:
for code in (1, 2, 3):
with self.subTest(code=code):
result = _collection(_status(code, "processor explanation"))
self.view._result_history = [result]
self.view._single_capture_active = True
self.view._single_capture_start_ns = 1000
self.view._single_capture_seen_raw = True
self.view._single_capture_target_collection_id = 5
self.view._log = mock.Mock()
self.view._log_error = mock.Mock()
self.view._log_warning = mock.Mock()
self.view._status_label = mock.Mock()
self.view._stop_run = mock.Mock(side_effect=lambda: setattr(self.view, "_single_capture_active", False))
self.assertTrue(self.view._finish_single_capture_if_ready())
self.assertFalse(self.view._single_capture_active)
self.assertEqual(self.view._result_history, [result])
self.assertIn("processor explanation", self.view._sar3d_status_label.text())
state = {1: "no signal in selected range", 2: "error", 3: "warning"}[code]
self.view._status_label.setText.assert_called_once_with(f"Status: {state}")
log = self.view._log_error if code == 2 else self.view._log_warning
log.assert_called_once_with(
f"Single capture: SAR3D {state}: processor explanation")
self.view._history.results.clear()
self.view._history.record_result(result)
self.view._pipeline.single_capture_active = True
self.view._pipeline._single_capture_start_ns = 1000
self.view._pipeline._single_capture_target_key = (5, 1200)
with self.assertLogs("python_app.gui.controllers.pipeline_controller", level="WARNING") as logged:
self.assertTrue(self.view._pipeline._finish_single_capture_if_ready())
self.assertFalse(self.view._pipeline.single_capture_active)
self.assertEqual(list(self.view._history.results), [result])
self.assertIn("processor explanation", self.renderer.status_label.text())
state = SAR3D_STATES[code]
self.assertEqual(self.view._view.status.text(), f"Status: {state}")
self.assertEqual(logged.records[-1].levelname, "ERROR" if code == 2 else "WARNING")
self.assertEqual(logged.records[-1].getMessage(),
f"Single capture: SAR3D {state}: processor explanation")
def test_live_edit_retains_failure_reason_then_recovers_to_ready_images(self) -> None:
self.view._result_history = []
self.view._write_live_processing_config = mock.Mock()
self.view._show_exception = mock.Mock()
self.view._history.results.clear()
self.view._live.publish = mock.Mock()
self.view._log.show_exception = mock.Mock()
for result, expected in ((_collection(_status(2, "CUDA kernel failure")), "error"),
(_collection(_status(3, "Driver initialization unavailable")), "warning"),
(_collection(_status(2, "fewer than two frequency points")), "error"),
(_collection(_status(1, "No signal in selected depth range")), "no signal"),
(_projections(), "ready — CUDA")):
self.view._drain_results_until_quiet = mock.Mock(side_effect=[None, result])
self.view._on_processing_live_settings_changed()
self.assertIn(expected, self.view._sar3d_status_label.text())
self.view._show_exception.assert_not_called()
self.assertTrue(all(image.isVisible() for image in self.view._sar3d_images.values()))
self.view._pipeline.drain_results = mock.Mock(return_value=result)
self.view._live.changed()
self.assertIn(expected, self.renderer.status_label.text())
self.view._log.show_exception.assert_not_called()
self.assertTrue(all(image.isVisible() for image in self.renderer.images.values()))
def test_history_redraw_preserves_error_but_stale_revision_clears_old_images(self) -> None:
self.view._result_history = [_collection(_status(2, "processor error"))]
self.view._redraw_after_history_deletion()
self.assertEqual(self.view._sar3d_status_label.text(), "SAR3D: error — processor error")
self.view._draw_results(_projections())
self.view._required_processing_revision = 10
self.view._redraw_after_history_deletion()
self.assertEqual(self.view._sar3d_status_label.text(), "SAR3D: waiting for measurements")
self.assertTrue(all(not image.isVisible() for image in self.view._sar3d_images.values()))
self.view._history.record_result(_collection(_status(2, "processor error")))
self.view._plots.redraw()
self.assertEqual(self.renderer.status_label.text(), "SAR3D: error — processor error")
self.view._plots.draw(_projections())
self.view._plots.required_revision = 10
self.view._plots.redraw()
self.assertEqual(self.renderer.status_label.text(), "SAR3D: waiting for measurements")
self.assertTrue(all(not image.isVisible() for image in self.renderer.images.values()))
if __name__ == "__main__":