a simple (oscillo|vector|spectro)-scope for your terminal
Something went wrong. Try again.
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113use ratatui::{ style::Style, text::Span, widgets::{Axis, GraphType},};
use crate::input::Matrix;
use super::{DataSet, Dimension, DisplayMode, GraphConfig};
#[derive(Default)]pub struct Vectorscope {}
impl DisplayMode for Vectorscope { fn mode_str(&self) -> &'static str { "vector" }
fn channel_name(&self, index: usize) -> String { format!("{}", index) }
fn header(&self, _: &GraphConfig) -> String { "live".into() }
fn axis(&'_ self, cfg: &GraphConfig, dimension: Dimension) -> Axis<'_> { let (name, bounds) = match dimension { Dimension::X => ("left -", [-cfg.scale, cfg.scale]), Dimension::Y => ("| right", [-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![(-cfg.scale, 0.0), (cfg.scale, 0.0)], cfg.marker_type, GraphType::Line, cfg.axis_color, ), DataSet::new( None, vec![(0.0, -cfg.scale), (0.0, cfg.scale)], cfg.marker_type, GraphType::Line, cfg.axis_color, ), ] }
fn process(&mut self, cfg: &GraphConfig, data: &Matrix<f64>) -> Vec<DataSet> { let mut out = Vec::new();
for (n, chunk) in data.chunks(2).enumerate() { let mut tmp = vec![]; match chunk.len() { 2 => { for i in 0..std::cmp::min(chunk[0].len(), chunk[1].len()) { if i > cfg.samples as usize { break; } tmp.push((chunk[0][i], chunk[1][i])); } } 1 => { for i in 0..chunk[0].len() { if i > cfg.samples as usize { break; } tmp.push((chunk[0][i], i as f64)); } } _ => continue, } // split it in two for easier coloring // TODO configure splitting in multiple parts? let pivot = tmp.len() / 2; out.push(DataSet::new( Some(self.channel_name((n * 2) + 1)), tmp[pivot..].to_vec(), cfg.marker_type, if cfg.scatter { GraphType::Scatter } else { GraphType::Line }, cfg.palette((n * 2) + 1), )); out.push(DataSet::new( Some(self.channel_name(n * 2)), tmp[..pivot].to_vec(), cfg.marker_type, if cfg.scatter { GraphType::Scatter } else { GraphType::Line }, cfg.palette(n * 2), )); }
out }}