package main

import (
	"math"

	"sdr-wideband-suite/internal/demod/gpudemod"
	"sdr-wideband-suite/internal/detector"
	"sdr-wideband-suite/internal/telemetry"
)

const useStreamingOraclePath = true  // keep true during C2-C so the real native path is continuously compared against the corrected oracle
const useStreamingProductionPath = false // keep false until the new production path is explicitly activated in runtime bring-up

var streamingOracleRunner *gpudemod.CPUOracleRunner

func buildStreamingJobs(sampleRate int, centerHz float64, signals []detector.Signal, aqCfg extractionConfig) ([]gpudemod.StreamingExtractJob, error) {
	jobs := make([]gpudemod.StreamingExtractJob, len(signals))
	decimTarget := 200000
	bwMult := aqCfg.bwMult
	if bwMult <= 0 {
		bwMult = 1.0
	}
	firTaps := aqCfg.firTaps
	if firTaps <= 0 {
		firTaps = 101
	}
	for i, sig := range signals {
		bw := sig.BWHz * bwMult
		sigMHz := sig.CenterHz / 1e6
		isWFM := (sigMHz >= 87.5 && sigMHz <= 108.0) ||
			(sig.Class != nil && (sig.Class.ModType == "WFM" || sig.Class.ModType == "WFM_STEREO"))
		outRate := decimTarget
		if isWFM {
			outRate = wfmStreamOutRate
			if bw < wfmStreamMinBW {
				bw = wfmStreamMinBW
			}
		} else if bw < 20000 {
			bw = 20000
		}
		if _, err := gpudemod.ExactIntegerDecimation(sampleRate, outRate); err != nil {
			return nil, err
		}
		offset := sig.CenterHz - centerHz
		jobs[i] = gpudemod.StreamingExtractJob{
			SignalID:   sig.ID,
			OffsetHz:   offset,
			Bandwidth:  bw,
			OutRate:    outRate,
			NumTaps:    firTaps,
			ConfigHash: gpudemod.StreamingConfigHash(sig.ID, offset, bw, outRate, firTaps, sampleRate),
		}
	}
	return jobs, nil
}

func resetStreamingOracleRunner() {
	if streamingOracleRunner != nil {
		streamingOracleRunner.ResetAllStates()
	}
}

func extractForStreamingOracle(
	allIQ []complex64,
	sampleRate int,
	centerHz float64,
	signals []detector.Signal,
	aqCfg extractionConfig,
	coll *telemetry.Collector,
) ([][]complex64, []int, error) {
	out := make([][]complex64, len(signals))
	rates := make([]int, len(signals))
	jobs, err := buildStreamingJobs(sampleRate, centerHz, signals, aqCfg)
	if err != nil {
		return nil, nil, err
	}
	if streamingOracleRunner == nil || streamingOracleRunner.SampleRate != sampleRate {
		streamingOracleRunner = gpudemod.NewCPUOracleRunner(sampleRate)
	}
	results, err := streamingOracleRunner.StreamingExtract(allIQ, jobs)
	if err != nil {
		return nil, nil, err
	}
	for i, res := range results {
		out[i] = res.IQ
		rates[i] = res.Rate
		observeStreamingResult(coll, "streaming.oracle", res)
	}
	return out, rates, nil
}

func phaseIncForOffset(sampleRate int, offsetHz float64) float64 {
	return -2.0 * math.Pi * offsetHz / float64(sampleRate)
}