No puede seleccionar más de 25 temas Los temas deben comenzar con una letra o número, pueden incluir guiones ('-') y pueden tener hasta 35 caracteres de largo.

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