[feat](trx-rs): show live wfm stereo detect state

Co-authored-by: Codex <codex@openai.com>
Signed-off-by: Stan Grams <sjg@haxx.space>
This commit is contained in:
2026-02-28 21:44:32 +01:00
parent 862f0200bb
commit b722787ada
9 changed files with 81 additions and 0 deletions
+1
View File
@@ -663,6 +663,7 @@ async fn refresh_state_with_retry(
/// Read current state from the rig via CAT.
async fn refresh_state_from_cat(rig: &mut Box<dyn RigCat>, state: &mut RigState) -> DynResult<()> {
let (freq, mode, vfo) = rig.get_status().await?;
state.filter = rig.filter_state();
state.control.enabled = Some(true);
state.apply_freq(freq);
state.apply_mode(mode);
@@ -405,6 +405,10 @@ pub struct WfmStereoDecoder {
diff_denoise: MultibandStereoBlend,
/// Whether multiband stereo denoising is active.
denoise_enabled: bool,
/// Smoothed pilot-derived stereo detection strength in [0, 1].
stereo_detect_level: f32,
/// Hysteretic pilot-lock result used by the UI.
stereo_detected: bool,
/// FM discriminator gain normalization.
///
/// `demod_fm` outputs `atan2(…)/π ≈ 2·Δf/fs` for small deviations.
@@ -471,6 +475,8 @@ impl WfmStereoDecoder {
deemph_r: Deemphasis::new(audio_rate.max(1) as f32, deemphasis_us),
diff_denoise: MultibandStereoBlend::new(audio_rate.max(1) as f32),
denoise_enabled,
stereo_detect_level: 0.0,
stereo_detected: false,
fm_gain: composite_rate_f / (2.0 * 75_000.0),
sum_hist: [0.0; 4],
diff_hist: [0.0; 4],
@@ -509,6 +515,19 @@ impl WfmStereoDecoder {
let pilot_mag = (i * i + q * q).sqrt().max(pilot_tone.abs());
let stereo_blend = (pilot_mag * 40.0).clamp(0.0, 1.0);
let detect_coeff = if stereo_blend > self.stereo_detect_level {
0.0008
} else {
0.0002
};
self.stereo_detect_level += detect_coeff * (stereo_blend - self.stereo_detect_level);
if self.stereo_detected {
if self.stereo_detect_level < 0.35 {
self.stereo_detected = false;
}
} else if self.stereo_detect_level > 0.6 {
self.stereo_detected = true;
}
// --- RDS ---
let rds_quality = (0.35 + pilot_mag * 20.0).clamp(0.35, 1.0);
@@ -619,6 +638,10 @@ impl WfmStereoDecoder {
pub fn reset_rds(&mut self) {
self.rds_decoder.reset();
}
pub fn stereo_detected(&self) -> bool {
self.stereo_detected
}
}
/// Selects the demodulation algorithm for a channel.
@@ -568,6 +568,13 @@ impl ChannelDsp {
self.wfm_decoder.as_ref().and_then(WfmStereoDecoder::rds_data)
}
pub fn wfm_stereo_detected(&self) -> bool {
self.wfm_decoder
.as_ref()
.map(WfmStereoDecoder::stereo_detected)
.unwrap_or(false)
}
pub fn reset_rds(&mut self) {
if let Some(decoder) = &mut self.wfm_decoder {
decoder.reset_rds();
@@ -495,6 +495,12 @@ impl RigCat for SoapySdrRig {
}
fn filter_state(&self) -> Option<RigFilterState> {
let wfm_stereo_detected = self
.pipeline
.channel_dsps
.get(self.primary_channel_idx)
.and_then(|dsp| dsp.lock().ok().map(|d| d.wfm_stereo_detected()))
.unwrap_or(false);
Some(RigFilterState {
bandwidth_hz: self.bandwidth_hz,
fir_taps: self.fir_taps,
@@ -502,6 +508,7 @@ impl RigCat for SoapySdrRig {
wfm_deemphasis_us: self.wfm_deemphasis_us,
wfm_stereo: self.wfm_stereo,
wfm_denoise: self.wfm_denoise,
wfm_stereo_detected,
})
}