tunein-cli / src / visualization / oscilloscope.rs

use crossterm::event::{Event, KeyCode, KeyModifiers};
use ratatui::{
    style::Style,
    text::Span,
    widgets::{Axis, GraphType},
};

use crate::{
    app::{update_value_f, update_value_i},
    input::Matrix,
};

use super::{DataSet, Dimension, DisplayMode, GraphConfig};

#[derive(Default)]
pub struct Oscilloscope {
    pub triggering: bool,
    pub falling_edge: bool,
    pub threshold: f64,
    pub depth: u32,
    pub peaks: bool,
}

impl DisplayMode for Oscilloscope {
    fn from_args(_opts: &crate::cfg::SourceOptions) -> Self {
        Oscilloscope::default()
    }

    fn mode_str(&self) -> &'static str {
        "oscillo"
    }

    fn channel_name(&self, index: usize) -> String {
        match index {
            0 => "L".into(),
            1 => "R".into(),
            _ => format!("{}", index),
        }
    }

    fn header(&self, _: &GraphConfig) -> String {
        if self.triggering {
            format!(
                "{} {:.0}{} trigger",
                if self.falling_edge { "v" } else { "^" },
                self.threshold,
                if self.depth > 1 {
                    format!(":{}", self.depth)
                } else {
                    "".into()
                },
            )
        } else {
            "live".into()
        }
    }

    fn axis(&self, cfg: &GraphConfig, dimension: Dimension) -> Axis<'_> {
        let (name, bounds) = match dimension {
            Dimension::X => ("time -", [0.0, cfg.samples as f64]),
            Dimension::Y => ("| amplitude", [-cfg.scale, cfg.scale]),
        };
        let mut a = Axis::default();
        if cfg.show_ui {
            // TODO don't make it necessary to check show_ui inside here
            a = a.title(Span::styled(name, Style::default().fg(cfg.labels_color)));
        }
        a.style(Style::default().fg(cfg.axis_color)).bounds(bounds)
    }

    fn references(&self, cfg: &GraphConfig) -> Vec<DataSet> {
        vec![DataSet::new(
            None,
            vec![(0.0, 0.0), (cfg.samples as f64, 0.0)],
            cfg.marker_type,
            GraphType::Line,
            cfg.axis_color,
        )]
    }

    fn process(&mut self, cfg: &GraphConfig, data: &Matrix<f64>) -> Vec<DataSet> {
        let mut out = Vec::new();

        let mut trigger_offset = 0;
        if self.depth == 0 {
            self.depth = 1
        }
        if self.triggering {
            for i in 0..data[0].len() {
                if triggered(&data[0], i, self.threshold, self.depth, self.falling_edge) {
                    // triggered
                    break;
                }
                trigger_offset += 1;
            }
        }

        if self.triggering {
            out.push(DataSet::new(
                Some("T".into()),
                vec![(0.0, self.threshold)],
                cfg.marker_type,
                GraphType::Scatter,
                cfg.labels_color,
            ));
        }

        for (n, channel) in data.iter().enumerate().rev() {
            let (mut min, mut max) = (0.0, 0.0);
            let mut tmp = Vec::new();
            for (i, sample) in channel.iter().enumerate() {
                if *sample < min {
                    min = *sample
                };
                if *sample > max {
                    max = *sample
                };
                if i >= trigger_offset {
                    tmp.push(((i - trigger_offset) as f64, *sample));
                }
            }

            if self.peaks {
                out.push(DataSet::new(
                    None,
                    vec![(0.0, min), (0.0, max)],
                    cfg.marker_type,
                    GraphType::Scatter,
                    cfg.palette(n),
                ))
            }

            out.push(DataSet::new(
                Some(self.channel_name(n)),
                tmp,
                cfg.marker_type,
                if cfg.scatter {
                    GraphType::Scatter
                } else {
                    GraphType::Line
                },
                cfg.palette(n),
            ));
        }

        out
    }

    fn handle(&mut self, event: Event) {
        if let Event::Key(key) = event {
            let magnitude = match key.modifiers {
                KeyModifiers::SHIFT => 10.0,
                KeyModifiers::CONTROL => 5.0,
                KeyModifiers::ALT => 0.2,
                _ => 1.0,
            };
            match key.code {
                KeyCode::PageUp => {
                    update_value_f(&mut self.threshold, 250.0, magnitude, 0.0..32768.0)
                }
                KeyCode::PageDown => {
                    update_value_f(&mut self.threshold, -250.0, magnitude, 0.0..32768.0)
                }
                KeyCode::Char('t') => self.triggering = !self.triggering,
                KeyCode::Char('e') => self.falling_edge = !self.falling_edge,
                KeyCode::Char('p') => self.peaks = !self.peaks,
                KeyCode::Char('=') => update_value_i(&mut self.depth, true, 1, 1.0, 1..65535),
                KeyCode::Char('-') => update_value_i(&mut self.depth, false, 1, 1.0, 1..65535),
                KeyCode::Char('+') => update_value_i(&mut self.depth, true, 10, 1.0, 1..65535),
                KeyCode::Char('_') => update_value_i(&mut self.depth, false, 10, 1.0, 1..65535),
                KeyCode::Esc => {
                    self.triggering = false;
                }
                _ => {}
            }
        }
    }
}

#[allow(clippy::collapsible_else_if)] // TODO can this be made nicer?
fn triggered(data: &[f64], index: usize, threshold: f64, depth: u32, falling_edge: bool) -> bool {
    if data.len() < index + (1 + depth as usize) {
        return false;
    }
    if falling_edge {
        if data[index] >= threshold {
            for i in 1..=depth as usize {
                if data[index + i] >= threshold {
                    return false;
                }
            }
            true
        } else {
            false
        }
    } else {
        if data[index] <= threshold {
            for i in 1..=depth as usize {
                if data[index + i] <= threshold {
                    return false;
                }
            }
            true
        } else {
            false
        }
    }
}