Wideband autonomous SDR analysis engine forked from sdr-visual-suite
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

936 lines
25KB

  1. package main
  2. import (
  3. "context"
  4. "encoding/json"
  5. "flag"
  6. "log"
  7. "math"
  8. "net/http"
  9. "os"
  10. "os/signal"
  11. "path/filepath"
  12. "runtime/debug"
  13. "sort"
  14. "strconv"
  15. "strings"
  16. "sync"
  17. "syscall"
  18. "time"
  19. "github.com/gorilla/websocket"
  20. "sdr-visual-suite/internal/classifier"
  21. "sdr-visual-suite/internal/config"
  22. "sdr-visual-suite/internal/detector"
  23. "sdr-visual-suite/internal/dsp"
  24. "sdr-visual-suite/internal/events"
  25. fftutil "sdr-visual-suite/internal/fft"
  26. "sdr-visual-suite/internal/fft/gpufft"
  27. "sdr-visual-suite/internal/mock"
  28. "sdr-visual-suite/internal/recorder"
  29. "sdr-visual-suite/internal/runtime"
  30. "sdr-visual-suite/internal/sdr"
  31. "sdr-visual-suite/internal/sdrplay"
  32. )
  33. type SpectrumFrame struct {
  34. Timestamp int64 `json:"ts"`
  35. CenterHz float64 `json:"center_hz"`
  36. SampleHz int `json:"sample_rate"`
  37. FFTSize int `json:"fft_size"`
  38. Spectrum []float64 `json:"spectrum_db"`
  39. Thresholds []float64 `json:"thresholds,omitempty"`
  40. NoiseFloor float64 `json:"noise_floor,omitempty"`
  41. Signals []detector.Signal `json:"signals"`
  42. }
  43. type client struct {
  44. conn *websocket.Conn
  45. send chan []byte
  46. done chan struct{}
  47. closeOnce sync.Once
  48. }
  49. type hub struct {
  50. mu sync.Mutex
  51. clients map[*client]struct{}
  52. frameCnt int64
  53. lastLogTs time.Time
  54. }
  55. type gpuStatus struct {
  56. mu sync.RWMutex
  57. Available bool `json:"available"`
  58. Active bool `json:"active"`
  59. Error string `json:"error"`
  60. }
  61. type signalSnapshot struct {
  62. mu sync.RWMutex
  63. signals []detector.Signal
  64. }
  65. func (s *signalSnapshot) set(sig []detector.Signal) {
  66. s.mu.Lock()
  67. defer s.mu.Unlock()
  68. s.signals = append([]detector.Signal(nil), sig...)
  69. }
  70. func (s *signalSnapshot) get() []detector.Signal {
  71. s.mu.RLock()
  72. defer s.mu.RUnlock()
  73. return append([]detector.Signal(nil), s.signals...)
  74. }
  75. func (g *gpuStatus) set(active bool, err error) {
  76. g.mu.Lock()
  77. defer g.mu.Unlock()
  78. g.Active = active
  79. if err != nil {
  80. g.Error = err.Error()
  81. } else {
  82. g.Error = ""
  83. }
  84. }
  85. func (g *gpuStatus) snapshot() gpuStatus {
  86. g.mu.RLock()
  87. defer g.mu.RUnlock()
  88. return gpuStatus{Available: g.Available, Active: g.Active, Error: g.Error}
  89. }
  90. func newHub() *hub {
  91. return &hub{clients: map[*client]struct{}{}, lastLogTs: time.Now()}
  92. }
  93. func (h *hub) add(c *client) {
  94. h.mu.Lock()
  95. defer h.mu.Unlock()
  96. h.clients[c] = struct{}{}
  97. log.Printf("ws connected (%d clients)", len(h.clients))
  98. }
  99. func (h *hub) remove(c *client) {
  100. c.closeOnce.Do(func() { close(c.done) })
  101. h.mu.Lock()
  102. defer h.mu.Unlock()
  103. delete(h.clients, c)
  104. log.Printf("ws disconnected (%d clients)", len(h.clients))
  105. }
  106. func (h *hub) broadcast(frame SpectrumFrame) {
  107. b, err := json.Marshal(frame)
  108. if err != nil {
  109. log.Printf("marshal frame: %v", err)
  110. return
  111. }
  112. h.mu.Lock()
  113. clients := make([]*client, 0, len(h.clients))
  114. for c := range h.clients {
  115. clients = append(clients, c)
  116. }
  117. h.mu.Unlock()
  118. for _, c := range clients {
  119. select {
  120. case c.send <- b:
  121. default:
  122. h.remove(c)
  123. }
  124. }
  125. h.frameCnt++
  126. if time.Since(h.lastLogTs) > 2*time.Second {
  127. h.lastLogTs = time.Now()
  128. log.Printf("broadcast frames=%d clients=%d", h.frameCnt, len(clients))
  129. }
  130. }
  131. type sourceManager struct {
  132. mu sync.RWMutex
  133. src sdr.Source
  134. newSource func(cfg config.Config) (sdr.Source, error)
  135. }
  136. func (m *sourceManager) Restart(cfg config.Config) error {
  137. m.mu.Lock()
  138. defer m.mu.Unlock()
  139. old := m.src
  140. _ = old.Stop()
  141. next, err := m.newSource(cfg)
  142. if err != nil {
  143. _ = old.Start()
  144. m.src = old
  145. return err
  146. }
  147. if err := next.Start(); err != nil {
  148. _ = next.Stop()
  149. _ = old.Start()
  150. m.src = old
  151. return err
  152. }
  153. m.src = next
  154. return nil
  155. }
  156. func (m *sourceManager) Stats() sdr.SourceStats {
  157. m.mu.RLock()
  158. defer m.mu.RUnlock()
  159. if sp, ok := m.src.(sdr.StatsProvider); ok {
  160. return sp.Stats()
  161. }
  162. return sdr.SourceStats{}
  163. }
  164. func (m *sourceManager) Flush() {
  165. m.mu.RLock()
  166. defer m.mu.RUnlock()
  167. if fl, ok := m.src.(sdr.Flushable); ok {
  168. fl.Flush()
  169. }
  170. }
  171. func newSourceManager(src sdr.Source, newSource func(cfg config.Config) (sdr.Source, error)) *sourceManager {
  172. return &sourceManager{src: src, newSource: newSource}
  173. }
  174. func (m *sourceManager) Start() error {
  175. m.mu.RLock()
  176. defer m.mu.RUnlock()
  177. return m.src.Start()
  178. }
  179. func (m *sourceManager) Stop() error {
  180. m.mu.RLock()
  181. defer m.mu.RUnlock()
  182. return m.src.Stop()
  183. }
  184. func (m *sourceManager) ReadIQ(n int) ([]complex64, error) {
  185. m.mu.RLock()
  186. defer m.mu.RUnlock()
  187. return m.src.ReadIQ(n)
  188. }
  189. func (m *sourceManager) ApplyConfig(cfg config.Config) error {
  190. m.mu.Lock()
  191. defer m.mu.Unlock()
  192. if updatable, ok := m.src.(sdr.ConfigurableSource); ok {
  193. if err := updatable.UpdateConfig(cfg.SampleRate, cfg.CenterHz, cfg.GainDb, cfg.AGC, cfg.TunerBwKHz); err == nil {
  194. return nil
  195. }
  196. }
  197. old := m.src
  198. _ = old.Stop()
  199. next, err := m.newSource(cfg)
  200. if err != nil {
  201. _ = old.Start()
  202. return err
  203. }
  204. if err := next.Start(); err != nil {
  205. _ = next.Stop()
  206. _ = old.Start()
  207. return err
  208. }
  209. m.src = next
  210. return nil
  211. }
  212. type dspUpdate struct {
  213. cfg config.Config
  214. det *detector.Detector
  215. window []float64
  216. dcBlock bool
  217. iqBalance bool
  218. useGPUFFT bool
  219. }
  220. func pushDSPUpdate(ch chan dspUpdate, update dspUpdate) {
  221. select {
  222. case ch <- update:
  223. default:
  224. select {
  225. case <-ch:
  226. default:
  227. }
  228. ch <- update
  229. }
  230. }
  231. func main() {
  232. var cfgPath string
  233. var mockFlag bool
  234. flag.StringVar(&cfgPath, "config", "config.yaml", "path to config YAML")
  235. flag.BoolVar(&mockFlag, "mock", false, "use synthetic IQ source")
  236. flag.Parse()
  237. cfg, err := config.Load(cfgPath)
  238. if err != nil {
  239. log.Fatalf("load config: %v", err)
  240. }
  241. cfgManager := runtime.New(cfg)
  242. gpuState := &gpuStatus{Available: gpufft.Available()}
  243. newSource := func(cfg config.Config) (sdr.Source, error) {
  244. if mockFlag {
  245. src := mock.New(cfg.SampleRate)
  246. if updatable, ok := interface{}(src).(sdr.ConfigurableSource); ok {
  247. _ = updatable.UpdateConfig(cfg.SampleRate, cfg.CenterHz, cfg.GainDb, cfg.AGC, cfg.TunerBwKHz)
  248. }
  249. return src, nil
  250. }
  251. src, err := sdrplay.New(cfg.SampleRate, cfg.CenterHz, cfg.GainDb, cfg.TunerBwKHz)
  252. if err != nil {
  253. return nil, err
  254. }
  255. if updatable, ok := src.(sdr.ConfigurableSource); ok {
  256. _ = updatable.UpdateConfig(cfg.SampleRate, cfg.CenterHz, cfg.GainDb, cfg.AGC, cfg.TunerBwKHz)
  257. }
  258. return src, nil
  259. }
  260. src, err := newSource(cfg)
  261. if err != nil {
  262. log.Fatalf("sdrplay init failed: %v (try --mock or build with -tags sdrplay)", err)
  263. }
  264. srcMgr := newSourceManager(src, newSource)
  265. if err := srcMgr.Start(); err != nil {
  266. log.Fatalf("source start: %v", err)
  267. }
  268. defer srcMgr.Stop()
  269. if err := os.MkdirAll(filepath.Dir(cfg.EventPath), 0o755); err != nil {
  270. log.Fatalf("event path: %v", err)
  271. }
  272. eventFile, err := os.OpenFile(cfg.EventPath, os.O_CREATE|os.O_APPEND|os.O_WRONLY, 0o644)
  273. if err != nil {
  274. log.Fatalf("open events: %v", err)
  275. }
  276. defer eventFile.Close()
  277. eventMu := &sync.RWMutex{}
  278. det := detector.New(cfg.Detector, cfg.SampleRate, cfg.FFTSize)
  279. window := fftutil.Hann(cfg.FFTSize)
  280. h := newHub()
  281. dspUpdates := make(chan dspUpdate, 1)
  282. ctx, cancel := context.WithCancel(context.Background())
  283. defer cancel()
  284. decodeMap := buildDecoderMap(cfg)
  285. recMgr := recorder.New(cfg.SampleRate, cfg.FFTSize, recorder.Policy{
  286. Enabled: cfg.Recorder.Enabled,
  287. MinSNRDb: cfg.Recorder.MinSNRDb,
  288. MinDuration: mustParseDuration(cfg.Recorder.MinDuration, 1*time.Second),
  289. MaxDuration: mustParseDuration(cfg.Recorder.MaxDuration, 300*time.Second),
  290. PrerollMs: cfg.Recorder.PrerollMs,
  291. RecordIQ: cfg.Recorder.RecordIQ,
  292. RecordAudio: cfg.Recorder.RecordAudio,
  293. AutoDemod: cfg.Recorder.AutoDemod,
  294. AutoDecode: cfg.Recorder.AutoDecode,
  295. MaxDiskMB: cfg.Recorder.MaxDiskMB,
  296. OutputDir: cfg.Recorder.OutputDir,
  297. ClassFilter: cfg.Recorder.ClassFilter,
  298. RingSeconds: cfg.Recorder.RingSeconds,
  299. }, cfg.CenterHz, decodeMap)
  300. sigSnap := &signalSnapshot{}
  301. go runDSP(ctx, srcMgr, cfg, det, window, h, eventFile, eventMu, dspUpdates, gpuState, recMgr, sigSnap)
  302. upgrader := websocket.Upgrader{CheckOrigin: func(r *http.Request) bool {
  303. origin := r.Header.Get("Origin")
  304. if origin == "" || origin == "null" {
  305. return true
  306. }
  307. // allow same-host or any local IP
  308. return true
  309. }}
  310. http.HandleFunc("/ws", func(w http.ResponseWriter, r *http.Request) {
  311. conn, err := upgrader.Upgrade(w, r, nil)
  312. if err != nil {
  313. log.Printf("ws upgrade failed: %v (origin: %s)", err, r.Header.Get("Origin"))
  314. return
  315. }
  316. c := &client{conn: conn, send: make(chan []byte, 32), done: make(chan struct{})}
  317. h.add(c)
  318. defer func() {
  319. h.remove(c)
  320. _ = conn.Close()
  321. }()
  322. conn.SetReadDeadline(time.Now().Add(60 * time.Second))
  323. conn.SetPongHandler(func(string) error {
  324. conn.SetReadDeadline(time.Now().Add(60 * time.Second))
  325. return nil
  326. })
  327. go func() {
  328. ping := time.NewTicker(30 * time.Second)
  329. defer ping.Stop()
  330. for {
  331. select {
  332. case msg, ok := <-c.send:
  333. if !ok {
  334. return
  335. }
  336. _ = conn.SetWriteDeadline(time.Now().Add(200 * time.Millisecond))
  337. if err := conn.WriteMessage(websocket.TextMessage, msg); err != nil {
  338. return
  339. }
  340. case <-ping.C:
  341. _ = conn.SetWriteDeadline(time.Now().Add(5 * time.Second))
  342. if err := conn.WriteMessage(websocket.PingMessage, nil); err != nil {
  343. log.Printf("ws ping error: %v", err)
  344. return
  345. }
  346. }
  347. }
  348. }()
  349. for {
  350. _, _, err := conn.ReadMessage()
  351. if err != nil {
  352. return
  353. }
  354. }
  355. })
  356. http.HandleFunc("/api/config", func(w http.ResponseWriter, r *http.Request) {
  357. w.Header().Set("Content-Type", "application/json")
  358. switch r.Method {
  359. case http.MethodGet:
  360. _ = json.NewEncoder(w).Encode(cfgManager.Snapshot())
  361. case http.MethodPost:
  362. var update runtime.ConfigUpdate
  363. if err := json.NewDecoder(r.Body).Decode(&update); err != nil {
  364. http.Error(w, "invalid json", http.StatusBadRequest)
  365. return
  366. }
  367. prev := cfgManager.Snapshot()
  368. next, err := cfgManager.ApplyConfig(update)
  369. if err != nil {
  370. http.Error(w, err.Error(), http.StatusBadRequest)
  371. return
  372. }
  373. sourceChanged := prev.CenterHz != next.CenterHz || prev.SampleRate != next.SampleRate || prev.GainDb != next.GainDb || prev.AGC != next.AGC || prev.TunerBwKHz != next.TunerBwKHz
  374. if sourceChanged {
  375. if err := srcMgr.ApplyConfig(next); err != nil {
  376. cfgManager.Replace(prev)
  377. http.Error(w, "failed to apply source config", http.StatusInternalServerError)
  378. return
  379. }
  380. }
  381. if err := config.Save(cfgPath, next); err != nil {
  382. log.Printf("config save failed: %v", err)
  383. }
  384. detChanged := prev.Detector.ThresholdDb != next.Detector.ThresholdDb ||
  385. prev.Detector.MinDurationMs != next.Detector.MinDurationMs ||
  386. prev.Detector.HoldMs != next.Detector.HoldMs ||
  387. prev.Detector.EmaAlpha != next.Detector.EmaAlpha ||
  388. prev.Detector.HysteresisDb != next.Detector.HysteresisDb ||
  389. prev.Detector.MinStableFrames != next.Detector.MinStableFrames ||
  390. prev.Detector.GapToleranceMs != next.Detector.GapToleranceMs ||
  391. prev.Detector.CFARMode != next.Detector.CFARMode ||
  392. prev.Detector.CFARGuardCells != next.Detector.CFARGuardCells ||
  393. prev.Detector.CFARTrainCells != next.Detector.CFARTrainCells ||
  394. prev.Detector.CFARRank != next.Detector.CFARRank ||
  395. prev.Detector.CFARScaleDb != next.Detector.CFARScaleDb ||
  396. prev.Detector.CFARWrapAround != next.Detector.CFARWrapAround ||
  397. prev.SampleRate != next.SampleRate ||
  398. prev.FFTSize != next.FFTSize
  399. windowChanged := prev.FFTSize != next.FFTSize
  400. var newDet *detector.Detector
  401. var newWindow []float64
  402. if detChanged {
  403. newDet = detector.New(next.Detector, next.SampleRate, next.FFTSize)
  404. }
  405. if windowChanged {
  406. newWindow = fftutil.Hann(next.FFTSize)
  407. }
  408. pushDSPUpdate(dspUpdates, dspUpdate{
  409. cfg: next,
  410. det: newDet,
  411. window: newWindow,
  412. dcBlock: next.DCBlock,
  413. iqBalance: next.IQBalance,
  414. useGPUFFT: next.UseGPUFFT,
  415. })
  416. _ = json.NewEncoder(w).Encode(next)
  417. default:
  418. http.Error(w, "method not allowed", http.StatusMethodNotAllowed)
  419. }
  420. })
  421. http.HandleFunc("/api/sdr/settings", func(w http.ResponseWriter, r *http.Request) {
  422. w.Header().Set("Content-Type", "application/json")
  423. if r.Method != http.MethodPost {
  424. http.Error(w, "method not allowed", http.StatusMethodNotAllowed)
  425. return
  426. }
  427. var update runtime.SettingsUpdate
  428. if err := json.NewDecoder(r.Body).Decode(&update); err != nil {
  429. http.Error(w, "invalid json", http.StatusBadRequest)
  430. return
  431. }
  432. prev := cfgManager.Snapshot()
  433. next, err := cfgManager.ApplySettings(update)
  434. if err != nil {
  435. http.Error(w, err.Error(), http.StatusBadRequest)
  436. return
  437. }
  438. if prev.AGC != next.AGC || prev.TunerBwKHz != next.TunerBwKHz {
  439. if err := srcMgr.ApplyConfig(next); err != nil {
  440. cfgManager.Replace(prev)
  441. http.Error(w, "failed to apply sdr settings", http.StatusInternalServerError)
  442. return
  443. }
  444. }
  445. if prev.DCBlock != next.DCBlock || prev.IQBalance != next.IQBalance {
  446. pushDSPUpdate(dspUpdates, dspUpdate{
  447. cfg: next,
  448. dcBlock: next.DCBlock,
  449. iqBalance: next.IQBalance,
  450. })
  451. }
  452. if err := config.Save(cfgPath, next); err != nil {
  453. log.Printf("config save failed: %v", err)
  454. }
  455. _ = json.NewEncoder(w).Encode(next)
  456. })
  457. http.HandleFunc("/api/stats", func(w http.ResponseWriter, r *http.Request) {
  458. w.Header().Set("Content-Type", "application/json")
  459. _ = json.NewEncoder(w).Encode(srcMgr.Stats())
  460. })
  461. http.HandleFunc("/api/gpu", func(w http.ResponseWriter, r *http.Request) {
  462. w.Header().Set("Content-Type", "application/json")
  463. _ = json.NewEncoder(w).Encode(gpuState.snapshot())
  464. })
  465. http.HandleFunc("/api/events", func(w http.ResponseWriter, r *http.Request) {
  466. w.Header().Set("Content-Type", "application/json")
  467. limit := 200
  468. if v := r.URL.Query().Get("limit"); v != "" {
  469. if parsed, err := strconv.Atoi(v); err == nil {
  470. limit = parsed
  471. }
  472. }
  473. var since time.Time
  474. if v := r.URL.Query().Get("since"); v != "" {
  475. if parsed, err := parseSince(v); err == nil {
  476. since = parsed
  477. } else {
  478. http.Error(w, "invalid since", http.StatusBadRequest)
  479. return
  480. }
  481. }
  482. snap := cfgManager.Snapshot()
  483. eventMu.RLock()
  484. evs, err := events.ReadRecent(snap.EventPath, limit, since)
  485. eventMu.RUnlock()
  486. if err != nil {
  487. http.Error(w, "failed to read events", http.StatusInternalServerError)
  488. return
  489. }
  490. _ = json.NewEncoder(w).Encode(evs)
  491. })
  492. http.HandleFunc("/api/signals", func(w http.ResponseWriter, r *http.Request) {
  493. w.Header().Set("Content-Type", "application/json")
  494. if sigSnap == nil {
  495. _ = json.NewEncoder(w).Encode([]detector.Signal{})
  496. return
  497. }
  498. _ = json.NewEncoder(w).Encode(sigSnap.get())
  499. })
  500. http.HandleFunc("/api/decoders", func(w http.ResponseWriter, r *http.Request) {
  501. w.Header().Set("Content-Type", "application/json")
  502. _ = json.NewEncoder(w).Encode(decoderKeys(cfgManager.Snapshot()))
  503. })
  504. http.HandleFunc("/api/recordings", func(w http.ResponseWriter, r *http.Request) {
  505. if r.Method != http.MethodGet {
  506. http.Error(w, "method not allowed", http.StatusMethodNotAllowed)
  507. return
  508. }
  509. w.Header().Set("Content-Type", "application/json")
  510. snap := cfgManager.Snapshot()
  511. list, err := recorder.ListRecordings(snap.Recorder.OutputDir)
  512. if err != nil {
  513. http.Error(w, "failed to list recordings", http.StatusInternalServerError)
  514. return
  515. }
  516. _ = json.NewEncoder(w).Encode(list)
  517. })
  518. http.HandleFunc("/api/recordings/", func(w http.ResponseWriter, r *http.Request) {
  519. w.Header().Set("Content-Type", "application/json")
  520. id := strings.TrimPrefix(r.URL.Path, "/api/recordings/")
  521. if id == "" {
  522. http.Error(w, "missing id", http.StatusBadRequest)
  523. return
  524. }
  525. snap := cfgManager.Snapshot()
  526. base := filepath.Clean(filepath.Join(snap.Recorder.OutputDir, id))
  527. if !strings.HasPrefix(base, filepath.Clean(snap.Recorder.OutputDir)) {
  528. http.Error(w, "invalid path", http.StatusBadRequest)
  529. return
  530. }
  531. if r.URL.Path == "/api/recordings/"+id+"/audio" {
  532. http.ServeFile(w, r, filepath.Join(base, "audio.wav"))
  533. return
  534. }
  535. if r.URL.Path == "/api/recordings/"+id+"/iq" {
  536. http.ServeFile(w, r, filepath.Join(base, "signal.cf32"))
  537. return
  538. }
  539. if r.URL.Path == "/api/recordings/"+id+"/decode" {
  540. mode := r.URL.Query().Get("mode")
  541. cmd := buildDecoderMap(cfgManager.Snapshot())[mode]
  542. if cmd == "" {
  543. http.Error(w, "decoder not configured", http.StatusBadRequest)
  544. return
  545. }
  546. meta, err := recorder.ReadMeta(filepath.Join(base, "meta.json"))
  547. if err != nil {
  548. http.Error(w, "meta read failed", http.StatusInternalServerError)
  549. return
  550. }
  551. audioPath := filepath.Join(base, "audio.wav")
  552. if _, errStat := os.Stat(audioPath); errStat != nil {
  553. audioPath = ""
  554. }
  555. res, err := recorder.DecodeOnDemand(cmd, filepath.Join(base, "signal.cf32"), meta.SampleRate, audioPath)
  556. if err != nil {
  557. http.Error(w, res.Stderr, http.StatusInternalServerError)
  558. return
  559. }
  560. _ = json.NewEncoder(w).Encode(res)
  561. return
  562. }
  563. // default: meta.json
  564. http.ServeFile(w, r, filepath.Join(base, "meta.json"))
  565. })
  566. http.HandleFunc("/api/demod", func(w http.ResponseWriter, r *http.Request) {
  567. if r.Method != http.MethodGet {
  568. http.Error(w, "method not allowed", http.StatusMethodNotAllowed)
  569. return
  570. }
  571. q := r.URL.Query()
  572. freq, _ := strconv.ParseFloat(q.Get("freq"), 64)
  573. bw, _ := strconv.ParseFloat(q.Get("bw"), 64)
  574. sec, _ := strconv.Atoi(q.Get("sec"))
  575. if sec < 1 {
  576. sec = 1
  577. }
  578. if sec > 10 {
  579. sec = 10
  580. }
  581. mode := q.Get("mode")
  582. data, _, err := recMgr.DemodLive(freq, bw, mode, sec)
  583. if err != nil {
  584. http.Error(w, err.Error(), http.StatusBadRequest)
  585. return
  586. }
  587. w.Header().Set("Content-Type", "audio/wav")
  588. _, _ = w.Write(data)
  589. })
  590. http.Handle("/", http.FileServer(http.Dir(cfg.WebRoot)))
  591. server := &http.Server{Addr: cfg.WebAddr}
  592. go func() {
  593. log.Printf("web listening on %s", cfg.WebAddr)
  594. if err := server.ListenAndServe(); err != nil && err != http.ErrServerClosed {
  595. log.Fatalf("server: %v", err)
  596. }
  597. }()
  598. stop := make(chan os.Signal, 1)
  599. signal.Notify(stop, os.Interrupt, syscall.SIGTERM)
  600. <-stop
  601. ctxTimeout, cancelTimeout := context.WithTimeout(context.Background(), 5*time.Second)
  602. defer cancelTimeout()
  603. _ = server.Shutdown(ctxTimeout)
  604. }
  605. func runDSP(ctx context.Context, srcMgr *sourceManager, cfg config.Config, det *detector.Detector, window []float64, h *hub, eventFile *os.File, eventMu *sync.RWMutex, updates <-chan dspUpdate, gpuState *gpuStatus, rec *recorder.Manager, sigSnap *signalSnapshot) {
  606. defer func() {
  607. if r := recover(); r != nil {
  608. log.Printf("FATAL: runDSP goroutine panic: %v\n%s", r, debug.Stack())
  609. }
  610. }()
  611. ticker := time.NewTicker(cfg.FrameInterval())
  612. defer ticker.Stop()
  613. logTicker := time.NewTicker(5 * time.Second)
  614. defer logTicker.Stop()
  615. enc := json.NewEncoder(eventFile)
  616. dcBlocker := dsp.NewDCBlocker(0.995)
  617. dcEnabled := cfg.DCBlock
  618. iqEnabled := cfg.IQBalance
  619. plan := fftutil.NewCmplxPlan(cfg.FFTSize)
  620. useGPU := cfg.UseGPUFFT
  621. var gpuEngine *gpufft.Engine
  622. if useGPU && gpuState != nil {
  623. snap := gpuState.snapshot()
  624. if snap.Available {
  625. if eng, err := gpufft.New(cfg.FFTSize); err == nil {
  626. gpuEngine = eng
  627. gpuState.set(true, nil)
  628. } else {
  629. gpuState.set(false, err)
  630. useGPU = false
  631. }
  632. } else {
  633. gpuState.set(false, nil)
  634. useGPU = false
  635. }
  636. } else if gpuState != nil {
  637. gpuState.set(false, nil)
  638. }
  639. gotSamples := false
  640. for {
  641. select {
  642. case <-ctx.Done():
  643. return
  644. case <-logTicker.C:
  645. st := srcMgr.Stats()
  646. log.Printf("stats: buf=%d drop=%d reset=%d last=%dms", st.BufferSamples, st.Dropped, st.Resets, st.LastSampleAgoMs)
  647. case upd := <-updates:
  648. prevFFT := cfg.FFTSize
  649. prevUseGPU := useGPU
  650. cfg = upd.cfg
  651. if rec != nil {
  652. rec.Update(cfg.SampleRate, cfg.FFTSize, recorder.Policy{
  653. Enabled: cfg.Recorder.Enabled,
  654. MinSNRDb: cfg.Recorder.MinSNRDb,
  655. MinDuration: mustParseDuration(cfg.Recorder.MinDuration, 1*time.Second),
  656. MaxDuration: mustParseDuration(cfg.Recorder.MaxDuration, 300*time.Second),
  657. PrerollMs: cfg.Recorder.PrerollMs,
  658. RecordIQ: cfg.Recorder.RecordIQ,
  659. RecordAudio: cfg.Recorder.RecordAudio,
  660. AutoDemod: cfg.Recorder.AutoDemod,
  661. AutoDecode: cfg.Recorder.AutoDecode,
  662. MaxDiskMB: cfg.Recorder.MaxDiskMB,
  663. OutputDir: cfg.Recorder.OutputDir,
  664. ClassFilter: cfg.Recorder.ClassFilter,
  665. RingSeconds: cfg.Recorder.RingSeconds,
  666. }, cfg.CenterHz, buildDecoderMap(cfg))
  667. }
  668. if upd.det != nil {
  669. det = upd.det
  670. }
  671. if upd.window != nil {
  672. window = upd.window
  673. plan = fftutil.NewCmplxPlan(cfg.FFTSize)
  674. }
  675. dcEnabled = upd.dcBlock
  676. iqEnabled = upd.iqBalance
  677. if cfg.FFTSize != prevFFT || cfg.UseGPUFFT != prevUseGPU {
  678. srcMgr.Flush()
  679. gotSamples = false
  680. if gpuEngine != nil {
  681. gpuEngine.Close()
  682. gpuEngine = nil
  683. }
  684. useGPU = cfg.UseGPUFFT
  685. if useGPU && gpuState != nil {
  686. snap := gpuState.snapshot()
  687. if snap.Available {
  688. if eng, err := gpufft.New(cfg.FFTSize); err == nil {
  689. gpuEngine = eng
  690. gpuState.set(true, nil)
  691. } else {
  692. gpuState.set(false, err)
  693. useGPU = false
  694. }
  695. } else {
  696. gpuState.set(false, nil)
  697. useGPU = false
  698. }
  699. } else if gpuState != nil {
  700. gpuState.set(false, nil)
  701. }
  702. }
  703. dcBlocker.Reset()
  704. ticker.Reset(cfg.FrameInterval())
  705. case <-ticker.C:
  706. iq, err := srcMgr.ReadIQ(cfg.FFTSize)
  707. if err != nil {
  708. log.Printf("read IQ: %v", err)
  709. if strings.Contains(err.Error(), "timeout") {
  710. if err := srcMgr.Restart(cfg); err != nil {
  711. log.Printf("restart failed: %v", err)
  712. }
  713. }
  714. continue
  715. }
  716. if rec != nil {
  717. rec.Ingest(time.Now(), iq)
  718. }
  719. if !gotSamples {
  720. log.Printf("received IQ samples")
  721. gotSamples = true
  722. }
  723. if dcEnabled {
  724. dcBlocker.Apply(iq)
  725. }
  726. if iqEnabled {
  727. dsp.IQBalance(iq)
  728. }
  729. var spectrum []float64
  730. if useGPU && gpuEngine != nil {
  731. if len(window) == len(iq) {
  732. for i := 0; i < len(iq); i++ {
  733. v := iq[i]
  734. w := float32(window[i])
  735. iq[i] = complex(real(v)*w, imag(v)*w)
  736. }
  737. }
  738. out, err := gpuEngine.Exec(iq)
  739. if err != nil {
  740. if gpuState != nil {
  741. gpuState.set(false, err)
  742. }
  743. useGPU = false
  744. spectrum = fftutil.SpectrumWithPlan(iq, nil, plan)
  745. } else {
  746. spectrum = fftutil.SpectrumFromFFT(out)
  747. }
  748. } else {
  749. spectrum = fftutil.SpectrumWithPlan(iq, window, plan)
  750. }
  751. for i := range spectrum {
  752. if math.IsNaN(spectrum[i]) || math.IsInf(spectrum[i], 0) {
  753. spectrum[i] = -200
  754. }
  755. }
  756. now := time.Now()
  757. finished, signals := det.Process(now, spectrum, cfg.CenterHz)
  758. thresholds := det.LastThresholds()
  759. noiseFloor := det.LastNoiseFloor()
  760. // enrich classification with temporal IQ features on per-signal snippet
  761. if len(iq) > 0 {
  762. for i := range signals {
  763. snip := extractSignalIQ(iq, cfg.SampleRate, cfg.CenterHz, signals[i].CenterHz, signals[i].BWHz)
  764. cls := classifier.Classify(classifier.SignalInput{FirstBin: signals[i].FirstBin, LastBin: signals[i].LastBin, SNRDb: signals[i].SNRDb}, spectrum, cfg.SampleRate, cfg.FFTSize, snip)
  765. signals[i].Class = cls
  766. }
  767. det.UpdateClasses(signals)
  768. }
  769. if sigSnap != nil {
  770. sigSnap.set(signals)
  771. }
  772. eventMu.Lock()
  773. for _, ev := range finished {
  774. _ = enc.Encode(ev)
  775. }
  776. eventMu.Unlock()
  777. if rec != nil && len(finished) > 0 {
  778. evCopy := make([]detector.Event, len(finished))
  779. copy(evCopy, finished)
  780. go rec.OnEvents(evCopy)
  781. }
  782. h.broadcast(SpectrumFrame{
  783. Timestamp: now.UnixMilli(),
  784. CenterHz: cfg.CenterHz,
  785. SampleHz: cfg.SampleRate,
  786. FFTSize: cfg.FFTSize,
  787. Spectrum: spectrum,
  788. Thresholds: thresholds,
  789. NoiseFloor: noiseFloor,
  790. Signals: signals,
  791. })
  792. }
  793. }
  794. }
  795. func mustParseDuration(raw string, fallback time.Duration) time.Duration {
  796. if raw == "" {
  797. return fallback
  798. }
  799. if d, err := time.ParseDuration(raw); err == nil {
  800. return d
  801. }
  802. return fallback
  803. }
  804. func buildDecoderMap(cfg config.Config) map[string]string {
  805. out := map[string]string{}
  806. if cfg.Decoder.FT8Cmd != "" {
  807. out["FT8"] = cfg.Decoder.FT8Cmd
  808. }
  809. if cfg.Decoder.WSPRCmd != "" {
  810. out["WSPR"] = cfg.Decoder.WSPRCmd
  811. }
  812. if cfg.Decoder.DMRCmd != "" {
  813. out["DMR"] = cfg.Decoder.DMRCmd
  814. }
  815. if cfg.Decoder.DStarCmd != "" {
  816. out["D-STAR"] = cfg.Decoder.DStarCmd
  817. }
  818. if cfg.Decoder.FSKCmd != "" {
  819. out["FSK"] = cfg.Decoder.FSKCmd
  820. }
  821. if cfg.Decoder.PSKCmd != "" {
  822. out["PSK"] = cfg.Decoder.PSKCmd
  823. }
  824. return out
  825. }
  826. func decoderKeys(cfg config.Config) []string {
  827. m := buildDecoderMap(cfg)
  828. keys := make([]string, 0, len(m))
  829. for k := range m {
  830. keys = append(keys, k)
  831. }
  832. sort.Strings(keys)
  833. return keys
  834. }
  835. func extractSignalIQ(iq []complex64, sampleRate int, centerHz float64, sigHz float64, bwHz float64) []complex64 {
  836. if len(iq) == 0 || sampleRate <= 0 {
  837. return nil
  838. }
  839. offset := sigHz - centerHz
  840. shifted := dsp.FreqShift(iq, sampleRate, offset)
  841. cutoff := bwHz / 2
  842. if cutoff < 200 {
  843. cutoff = 200
  844. }
  845. if cutoff > float64(sampleRate)/2-1 {
  846. cutoff = float64(sampleRate)/2 - 1
  847. }
  848. taps := dsp.LowpassFIR(cutoff, sampleRate, 101)
  849. filtered := dsp.ApplyFIR(shifted, taps)
  850. decim := sampleRate / 200000
  851. if decim < 1 {
  852. decim = 1
  853. }
  854. return dsp.Decimate(filtered, decim)
  855. }
  856. func parseSince(raw string) (time.Time, error) {
  857. if raw == "" {
  858. return time.Time{}, nil
  859. }
  860. if ms, err := strconv.ParseInt(raw, 10, 64); err == nil {
  861. if ms > 1e12 {
  862. return time.UnixMilli(ms), nil
  863. }
  864. return time.Unix(ms, 0), nil
  865. }
  866. if t, err := time.Parse(time.RFC3339Nano, raw); err == nil {
  867. return t, nil
  868. }
  869. return time.Parse(time.RFC3339, raw)
  870. }