pico_streaming/
events.rs

1use parking_lot::Mutex;
2use pico_common::PicoChannel;
3use std::{
4    collections::HashMap,
5    sync::{Arc, Weak},
6};
7
8pub trait NewDataHandler: Send + Sync {
9    fn handle_event(&self, value: &StreamingEvent);
10}
11
12#[derive(Clone)]
13pub struct EventsInner {
14    pub listeners: Vec<Weak<dyn NewDataHandler>>,
15}
16
17impl EventsInner {
18    pub fn new() -> Self {
19        EventsInner {
20            listeners: Default::default(),
21        }
22    }
23}
24
25#[derive(Clone)]
26pub struct StreamingEvents {
27    inner: Arc<Mutex<EventsInner>>,
28}
29
30impl StreamingEvents {
31    pub fn new() -> Self {
32        StreamingEvents {
33            inner: Arc::new(Mutex::new(EventsInner::new())),
34        }
35    }
36
37    #[tracing::instrument(level = "trace", skip(self, observer))]
38    pub fn subscribe(&self, observer: Arc<dyn NewDataHandler>) {
39        self.inner.lock().listeners.push(Arc::downgrade(&observer));
40    }
41
42    #[tracing::instrument(level = "trace", skip(self, value))]
43    pub fn emit(&self, value: StreamingEvent) {
44        for listener in self.inner.lock().listeners.iter() {
45            if let Some(listener) = listener.upgrade() {
46                listener.handle_event(&value);
47            }
48        }
49    }
50}
51
52impl Default for StreamingEvents {
53    fn default() -> Self {
54        StreamingEvents::new()
55    }
56}
57
58#[derive(Clone)]
59/// Events returned by the `PicoStreamingDevice`
60pub struct StreamingEvent {
61    pub length: usize,
62    pub samples_per_second: u32,
63    pub channels: HashMap<PicoChannel, RawChannelDataBlock>,
64}
65
66#[derive(Clone)]
67/// A struct containing raw channel data and scaling factors to get scaled samples
68pub struct RawChannelDataBlock {
69    pub multiplier: f64,
70    pub samples: Vec<i16>,
71}
72
73impl RawChannelDataBlock {
74    pub fn scale_samples(&self) -> Vec<f64> {
75        self.samples
76            .iter()
77            .map(|v| *v as f64 * self.multiplier)
78            .collect()
79    }
80
81    #[inline(always)]
82    pub fn scale_sample(&self, index: usize) -> f64 {
83        self.samples[index] as f64 * self.multiplier
84    }
85}
pico_streaming/events.rs

pico_streaming/
events.rs
