78 lines
2.3 KiB
Python
78 lines
2.3 KiB
Python
"""Neutral preprocessing-set helpers for Kamil ADC acquisition."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import time
|
|
|
|
import numpy as np
|
|
|
|
from python_app.models.dataset_model import ComboKey, SweepCollection, TraceData
|
|
from python_app.models.run_config_model import ComboModel, RunConfigModel
|
|
|
|
|
|
def build_kamil_adc_neutral_s21_sets(
|
|
config: RunConfigModel,
|
|
point_count: int,
|
|
) -> tuple[SweepCollection, SweepCollection]:
|
|
"""Build S21 calibration/reference collections that leave input S21 unchanged."""
|
|
if not config.is_kamil_adc:
|
|
raise ValueError("Neutral Kamil ADC sets require radar.model='kamil_adc'")
|
|
|
|
points = int(point_count)
|
|
if points <= 0:
|
|
raise ValueError("Kamil ADC neutral set point count must be > 0")
|
|
|
|
combos = list(config.combos)
|
|
if not combos:
|
|
combos = RunConfigModel.build_full_combos(config.input_switch.positions, config.output_switch.positions)
|
|
if not combos:
|
|
raise ValueError("Kamil ADC neutral sets require at least one switch combo")
|
|
|
|
frequency_hz = np.linspace(
|
|
float(config.radar.sweep.start_hz),
|
|
float(config.radar.sweep.stop_hz),
|
|
points,
|
|
dtype=np.float32,
|
|
)
|
|
now_ns = time.monotonic_ns()
|
|
calibration = _neutral_collection(
|
|
combos=combos,
|
|
frequency_hz=frequency_hz,
|
|
s21_value=np.complex64(1.0 + 0.0j),
|
|
monotonic_ns=now_ns,
|
|
)
|
|
reference = _neutral_collection(
|
|
combos=combos,
|
|
frequency_hz=frequency_hz,
|
|
s21_value=np.complex64(0.0 + 0.0j),
|
|
monotonic_ns=now_ns,
|
|
)
|
|
return calibration, reference
|
|
|
|
|
|
def _neutral_collection(
|
|
*,
|
|
combos: list[ComboModel],
|
|
frequency_hz: np.ndarray,
|
|
s21_value: np.complex64,
|
|
monotonic_ns: int,
|
|
) -> SweepCollection:
|
|
traces = []
|
|
for combo in combos:
|
|
point_count = int(frequency_hz.size)
|
|
traces.append(
|
|
TraceData(
|
|
combo=ComboKey(input_pos=int(combo.input), output_pos=int(combo.output)),
|
|
frequency_hz=frequency_hz.copy(),
|
|
s11=np.zeros(point_count, dtype=np.complex64),
|
|
s21=np.full(point_count, s21_value, dtype=np.complex64),
|
|
)
|
|
)
|
|
return SweepCollection(
|
|
collection_id=1,
|
|
monotonic_ns=monotonic_ns,
|
|
traces=traces,
|
|
capture_start_ns=monotonic_ns,
|
|
capture_end_ns=monotonic_ns,
|
|
)
|