added sn9000 support
This commit is contained in:
@@ -1,95 +0,0 @@
|
|||||||
"""Raw acquisition producer for LibreVNA multi-device mode."""
|
|
||||||
|
|
||||||
from __future__ import annotations
|
|
||||||
|
|
||||||
import argparse
|
|
||||||
import logging
|
|
||||||
from pathlib import Path
|
|
||||||
import signal
|
|
||||||
import threading
|
|
||||||
import time
|
|
||||||
|
|
||||||
from python_app.hardware_full.multi_device_service import MultiDeviceLibreVnaService
|
|
||||||
from python_app.models.run_config_model import RunConfigModel
|
|
||||||
from python_app.orchestration.shm import ShmRingWriter
|
|
||||||
from python_app.storage.npz.serialize import RAW_MAGIC, serialize_trace_collection
|
|
||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
|
||||||
|
|
||||||
|
|
||||||
def main() -> int:
|
|
||||||
"""Run producer process until config or signal requests exit."""
|
|
||||||
parser = argparse.ArgumentParser(description="Publish LibreVNA multi-device raw sweeps to SHM rings")
|
|
||||||
parser.add_argument("--config", required=True, type=Path, help="Path to run_config.json")
|
|
||||||
args = parser.parse_args()
|
|
||||||
|
|
||||||
logging.basicConfig(level=logging.INFO, format="%(levelname)s %(name)s: %(message)s")
|
|
||||||
stop_requested = threading.Event()
|
|
||||||
|
|
||||||
def request_stop(_signum: int, _frame: object) -> None:
|
|
||||||
stop_requested.set()
|
|
||||||
|
|
||||||
signal.signal(signal.SIGINT, request_stop)
|
|
||||||
signal.signal(signal.SIGTERM, request_stop)
|
|
||||||
|
|
||||||
config = RunConfigModel.load_from_path(args.config)
|
|
||||||
config.apply_device_model_constraints()
|
|
||||||
if not config.is_multi_device:
|
|
||||||
raise RuntimeError("multi_device_raw_producer requires radar.model='librevna_multi'")
|
|
||||||
|
|
||||||
raw_writer = ShmRingWriter(
|
|
||||||
config.rings.raw.name,
|
|
||||||
config.rings.raw.capacity,
|
|
||||||
config.rings.raw.slot_size_bytes,
|
|
||||||
)
|
|
||||||
raw_tap_writer = ShmRingWriter(
|
|
||||||
config.rings.raw_tap.name,
|
|
||||||
config.rings.raw_tap.capacity,
|
|
||||||
config.rings.raw_tap.slot_size_bytes,
|
|
||||||
)
|
|
||||||
radar = MultiDeviceLibreVnaService(
|
|
||||||
master_serial=config.radar.serial,
|
|
||||||
slave_serials=list(config.radar.multi_device.slave_serials),
|
|
||||||
force_external_reference=config.radar.multi_device.force_external_reference,
|
|
||||||
recovery_attempts=config.radar.multi_device.recovery_attempts,
|
|
||||||
backend_mode=config.radar.driver_mode,
|
|
||||||
)
|
|
||||||
|
|
||||||
try:
|
|
||||||
radar.open()
|
|
||||||
radar.configure(config.radar.sweep)
|
|
||||||
collection_id = 1
|
|
||||||
while not stop_requested.is_set():
|
|
||||||
collection_start = time.monotonic()
|
|
||||||
collection = radar.acquire_collection(collection_id=collection_id)
|
|
||||||
|
|
||||||
payload = serialize_trace_collection(collection, RAW_MAGIC)
|
|
||||||
if not raw_writer.push(payload):
|
|
||||||
raise RuntimeError(
|
|
||||||
f"Raw payload size {len(payload)} exceeds ring slot size {raw_writer.slot_size_bytes}"
|
|
||||||
)
|
|
||||||
if not raw_tap_writer.push(payload):
|
|
||||||
raise RuntimeError(
|
|
||||||
f"Raw tap payload size {len(payload)} exceeds ring slot size {raw_tap_writer.slot_size_bytes}"
|
|
||||||
)
|
|
||||||
if not config.runtime.continuous:
|
|
||||||
break
|
|
||||||
collection_duration_s = time.monotonic() - collection_start
|
|
||||||
if collection_id == 1 or collection_id % 20 == 0 or collection_duration_s > 2.0:
|
|
||||||
logger.info(
|
|
||||||
"multi-device collection %d acquired in %.3f s",
|
|
||||||
collection_id,
|
|
||||||
collection_duration_s,
|
|
||||||
)
|
|
||||||
collection_id += 1
|
|
||||||
finally:
|
|
||||||
radar.close()
|
|
||||||
raw_tap_writer.close()
|
|
||||||
raw_writer.close()
|
|
||||||
|
|
||||||
logger.info("multi-device raw producer stopped")
|
|
||||||
return 0
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
|
||||||
raise SystemExit(main())
|
|
||||||
Reference in New Issue
Block a user