Bläddra i källkod

feat: add analysis level to refinement result

master
Jan Svabenik 9 timmar sedan
förälder
incheckning
f046bfb9c5
4 ändrade filer med 4 tillägg och 2 borttagningar
  1. +1
    -1
      cmd/sdrd/phase_state_test.go
  2. +1
    -1
      cmd/sdrd/pipeline_runtime.go
  3. +1
    -0
      internal/pipeline/phases.go
  4. +1
    -0
      internal/pipeline/phases_test.go

+ 1
- 1
cmd/sdrd/phase_state_test.go Visa fil

@@ -10,7 +10,7 @@ func TestPhaseStateCarriesPhaseResults(t *testing.T) {
ps := &phaseState{
surveillance: pipeline.SurveillanceResult{NoiseFloor: -90, Scheduled: []pipeline.ScheduledCandidate{{Candidate: pipeline.Candidate{ID: 1}, Priority: 5}}},
refinementInput: pipeline.RefinementInput{Scheduled: []pipeline.ScheduledCandidate{{Candidate: pipeline.Candidate{ID: 1}, Priority: 5}}, SampleRate: 2048000, FFTSize: 2048, CenterHz: 7.1e6},
refinement: pipeline.RefinementResult{Decisions: []pipeline.SignalDecision{{ShouldRecord: true}}, Candidates: []pipeline.Candidate{{ID: 1}}},
refinement: pipeline.RefinementResult{Level: pipeline.AnalysisLevel{Name: "refinement"}, Decisions: []pipeline.SignalDecision{{ShouldRecord: true}}, Candidates: []pipeline.Candidate{{ID: 1}}},
}
if ps.surveillance.NoiseFloor != -90 || len(ps.surveillance.Scheduled) != 1 {
t.Fatalf("unexpected surveillance state: %+v", ps.surveillance)


+ 1
- 1
cmd/sdrd/pipeline_runtime.go Visa fil

@@ -364,7 +364,7 @@ func (rt *dspRuntime) refineSignals(art *spectrumArtifacts, input pipeline.Refin
maxDecode := rt.cfg.Resources.MaxDecodeJobs
enforceDecisionBudgets(decisions, maxRecord, maxDecode)
rt.det.UpdateClasses(signals)
return pipeline.RefinementResult{Signals: signals, Decisions: decisions, Candidates: selectedCandidates}
return pipeline.RefinementResult{Level: input.Level, Signals: signals, Decisions: decisions, Candidates: selectedCandidates}
}

func (rt *dspRuntime) updateRDS(now time.Time, rec *recorder.Manager, sig *detector.Signal, cls *classifier.Classification) {


+ 1
- 0
internal/pipeline/phases.go Visa fil

@@ -43,6 +43,7 @@ type RefinementInput struct {
}

type RefinementResult struct {
Level AnalysisLevel `json:"level"`
Signals []detector.Signal `json:"signals"`
Decisions []SignalDecision `json:"decisions,omitempty"`
Candidates []Candidate `json:"candidates,omitempty"`


+ 1
- 0
internal/pipeline/phases_test.go Visa fil

@@ -8,6 +8,7 @@ import (

func TestRefinementResultCarriesDecisions(t *testing.T) {
res := RefinementResult{
Level: AnalysisLevel{Name: "refinement"},
Signals: []detector.Signal{{ID: 1}},
Decisions: []SignalDecision{{ShouldRecord: true}},
Candidates: []Candidate{{ID: 1}},


Laddar…
Avbryt
Spara