sdr-wideband-suite/internal/detector/detector_test.go

package detector

import (
	"testing"
	"time"

	"sdr-visual-suite/internal/classifier"
	"sdr-visual-suite/internal/config"
)

func TestDetectorCreatesEvent(t *testing.T) {
	d := New(config.DetectorConfig{
		ThresholdDb:     -10,
		MinDurationMs:   1,
		HoldMs:          10,
		EmaAlpha:        0.2,
		HysteresisDb:    3,
		MinStableFrames: 1,
		GapToleranceMs:  10,
		CFARMode:        "OFF",
		CFARGuardCells:  2,
		CFARTrainCells:  16,
		CFARRank:        24,
		CFARScaleDb:     6,
		CFARWrapAround:  true,
	}, 1000, 10)
	center := 0.0
	spectrum := []float64{-30, -30, -30, -5, -5, -30, -30, -30, -30, -30}
	now := time.Now()
	finished, signals := d.Process(now, spectrum, center)
	if len(finished) != 0 {
		t.Fatalf("expected no finished events yet")
	}
	if len(signals) != 1 {
		t.Fatalf("expected 1 signal, got %d", len(signals))
	}
	if signals[0].BWHz <= 0 {
		t.Fatalf("expected bandwidth > 0")
	}

	_, _ = d.Process(now.Add(5*time.Millisecond), spectrum, center)

	now2 := now.Add(50 * time.Millisecond)
	noSignal := make([]float64, len(spectrum))
	for i := range noSignal {
		noSignal[i] = -100
	}
	finished, _ = d.Process(now2, noSignal, center)
	if len(finished) != 1 {
		t.Fatalf("expected 1 finished event, got %d", len(finished))
	}
	if finished[0].Bandwidth <= 0 {
		t.Fatalf("event bandwidth not set")
	}
}

func TestSignalBandwidthExpansion(t *testing.T) {
	sampleRate := 2048000
	fftSize := 2048
	cfg := config.DetectorConfig{
		ThresholdDb:     -20,
		MinDurationMs:   1,
		HoldMs:          10,
		EmaAlpha:        1.0,
		HysteresisDb:    3,
		MinStableFrames: 1,
		GapToleranceMs:  10,
		CFARMode:        "OFF",
		CFARGuardCells:  2,
		CFARTrainCells:  8,
		CFARRank:        12,
		CFARScaleDb:     6,
		CFARWrapAround:  true,
		EdgeMarginDb:    3.0,
		MaxSignalBwHz:   150000,
		MergeGapHz:      5000,
	}
	d := New(cfg, sampleRate, fftSize)
	spectrum := make([]float64, fftSize)
	for i := range spectrum {
		spectrum[i] = -100
	}
	for i := 1000; i <= 1012; i++ {
		spectrum[i] = -20
	}
	for j := 1; j <= 5; j++ {
		level := -20 - float64(j)*3
		if 1000-j >= 0 {
			spectrum[1000-j] = level
		}
		if 1012+j < fftSize {
			spectrum[1012+j] = level
		}
	}
	now := time.Now()
	_, signals := d.Process(now, spectrum, 434e6)
	if len(signals) == 0 {
		t.Fatal("no signals detected")
	}
	sig := signals[0]
	expectedMinBW := 18.0 * 1000
	if sig.BWHz < expectedMinBW {
		t.Errorf("BW too narrow: got %.0f Hz, want >= %.0f Hz (FirstBin=%d LastBin=%d)", sig.BWHz, expectedMinBW, sig.FirstBin, sig.LastBin)
	}
}

func TestMergeGap(t *testing.T) {
	sampleRate := 2048000
	fftSize := 2048
	cfg := config.DetectorConfig{
		ThresholdDb:     -20,
		MinDurationMs:   1,
		HoldMs:          10,
		EmaAlpha:        1.0,
		HysteresisDb:    3,
		MinStableFrames: 1,
		GapToleranceMs:  10,
		CFARMode:        "OFF",
		MergeGapHz:      5000,
	}
	d := New(cfg, sampleRate, fftSize)
	spectrum := make([]float64, fftSize)
	for i := range spectrum {
		spectrum[i] = -100
	}
	for i := 500; i <= 505; i++ {
		spectrum[i] = -20
	}
	for i := 509; i <= 514; i++ {
		spectrum[i] = -20
	}
	now := time.Now()
	_, signals := d.Process(now, spectrum, 434e6)
	if len(signals) != 1 {
		t.Errorf("expected 1 merged signal, got %d", len(signals))
	}
	if len(signals) == 1 && signals[0].BWHz < 14000 {
		t.Errorf("merged BW too narrow: %.0f Hz", signals[0].BWHz)
	}
}

func TestGuardTrainHzScaling(t *testing.T) {
	sampleRate := 2048000
	guardHz := 500.0
	trainHz := 5000.0
	cfg := config.DetectorConfig{
		ThresholdDb:     -20,
		MinDurationMs:   1,
		HoldMs:          10,
		EmaAlpha:        1.0,
		HysteresisDb:    3,
		MinStableFrames: 1,
		GapToleranceMs:  10,
		CFARMode:        "OFF",
		CFARGuardHz:     guardHz,
		CFARTrainHz:     trainHz,
		CFARScaleDb:     6,
		CFARWrapAround:  true,
	}
	d1 := New(cfg, sampleRate, 2048)
	d2 := New(cfg, sampleRate, 65536)
	spec2048 := makeSignalSpectrum(2048, sampleRate, 500e3, 12e3, -20, -100)
	spec65536 := makeSignalSpectrum(65536, sampleRate, 500e3, 12e3, -20, -100)
	now := time.Now()
	_, sig1 := d1.Process(now, spec2048, 434e6)
	_, sig2 := d2.Process(now, spec65536, 434e6)
	if len(sig1) == 0 || len(sig2) == 0 {
		t.Fatalf("detection failed: fft2048=%d signals, fft65536=%d signals", len(sig1), len(sig2))
	}
	bw1 := sig1[0].BWHz
	bw2 := sig2[0].BWHz
	ratio := bw1 / bw2
	if ratio < 0.8 || ratio > 1.2 {
		t.Errorf("BW mismatch across FFT sizes: fft2048=%.0f Hz, fft65536=%.0f Hz", bw1, bw2)
	}
}

func TestClassStabilization(t *testing.T) {
	ev := &activeEvent{}
	histSize := 5
	switchRatio := 0.6

	wfm := &classifier.Classification{ModType: classifier.ClassWFM, Confidence: 0.8}
	nfm := &classifier.Classification{ModType: classifier.ClassNFM, Confidence: 0.7}

	ev.updateClass(wfm, histSize, switchRatio)
	if ev.class.ModType != classifier.ClassWFM {
		t.Fatalf("first class should be WFM, got %s", ev.class.ModType)
	}

	ev.updateClass(nfm, histSize, switchRatio)
	if ev.class.ModType != classifier.ClassWFM {
		t.Fatalf("should stay WFM after 1 NFM, got %s", ev.class.ModType)
	}

	ev.updateClass(nfm, histSize, switchRatio)
	if ev.class.ModType != classifier.ClassNFM {
		t.Fatalf("should switch to NFM after 2/3 majority, got %s", ev.class.ModType)
	}

	for i := 0; i < 5; i++ {
		ev.updateClass(wfm, histSize, switchRatio)
	}
	if ev.class.ModType != classifier.ClassWFM {
		t.Fatalf("should be WFM after 5 consecutive, got %s", ev.class.ModType)
	}

	ev.updateClass(nfm, histSize, switchRatio)
	if ev.class.ModType != classifier.ClassWFM {
		t.Fatalf("single outlier should not flip class, got %s", ev.class.ModType)
	}
}

func makeSignalSpectrum(fftSize int, sampleRate int, offsetHz float64, bwHz float64, signalDb float64, noiseDb float64) []float64 {
	spectrum := make([]float64, fftSize)
	for i := range spectrum {
		spectrum[i] = noiseDb
	}
	binWidth := float64(sampleRate) / float64(fftSize)
	centerBin := int(float64(fftSize)/2 + offsetHz/binWidth)
	halfBins := int((bwHz / 2) / binWidth)
	if halfBins < 1 {
		halfBins = 1
	}
	for i := centerBin - halfBins; i <= centerBin+halfBins; i++ {
		if i >= 0 && i < fftSize {
			spectrum[i] = signalDb
		}
	}
	return spectrum
}