sdr-visual-suite/internal/cfar/caso.go

package cfar

type caso struct {
	guard      int
	train      int
	scaleDb    float64
	wrapAround bool
}

func newCASO(cfg Config) CFAR {
	return &caso{guard: cfg.GuardCells, train: cfg.TrainCells, scaleDb: cfg.ScaleDb, wrapAround: cfg.WrapAround}
}

func (c *caso) Thresholds(spectrum []float64) []float64 {
	n := len(spectrum)
	if n == 0 {
		return nil
	}
	out := make([]float64, n)
	train := c.train
	guard := c.guard
	if train == 0 {
		return out
	}
	inv := 1.0 / float64(train)

	at := func(i int) float64 {
		if c.wrapAround {
			return spectrum[((i%n)+n)%n]
		}
		if i < 0 || i >= n {
			return spectrum[clampInt(i, 0, n-1)]
		}
		return spectrum[i]
	}

	var leftSum, rightSum float64
	for k := 1; k <= train; k++ {
		leftSum += at(0 - guard - k)
		rightSum += at(0 + guard + k)
	}
	lm := leftSum * inv
	rm := rightSum * inv
	noise := lm
	if rm < noise {
		noise = rm
	}
	out[0] = noise + c.scaleDb

	for i := 1; i < n; i++ {
		leftSum -= at(i - 1 - guard - train)
		leftSum += at(i - guard - 1)
		rightSum -= at(i - 1 + guard + 1)
		rightSum += at(i + guard + train)
		lm = leftSum * inv
		rm = rightSum * inv
		noise = lm
		if rm < noise {
			noise = rm
		}
		out[i] = noise + c.scaleDb
	}
	return out
}