wip: histogram 🚧

This commit is contained in:
Dheepak Krishnamurthy 2023-09-22 01:41:56 -04:00 committed by Josh McKinney
parent b06557bbaf
commit 8be13ca74e
No known key found for this signature in database
GPG key ID: 722287396A903BC5
2 changed files with 157 additions and 201 deletions

View file

@ -1,111 +1,108 @@
use crossterm::{
event::{self, DisableMouseCapture, EnableMouseCapture, Event, KeyCode},
execute,
terminal::{disable_raw_mode, enable_raw_mode, EnterAlternateScreen, LeaveAlternateScreen},
};
use ratatui::{
backend::{Backend, CrosstermBackend},
layout::{Constraint, Direction, Layout},
style::{Color, Style},
widgets::{Block, Borders, Histogram},
Frame, Terminal,
};
// use crossterm::{
// event::{self, DisableMouseCapture, EnableMouseCapture, Event, KeyCode},
// execute,
// terminal::{disable_raw_mode, enable_raw_mode, EnterAlternateScreen, LeaveAlternateScreen},
// };
// use ratatui::{
// backend::{Backend, CrosstermBackend},
// layout::{Constraint, Direction, Layout},
// style::{Color, Style},
// widgets::{Block, Borders, Histogram},
// Frame, Terminal,
// };
use std::{
error::Error,
io,
time::{Duration, Instant},
};
use rand::{rngs::ThreadRng, thread_rng, Rng};
struct App {
data: Vec<u64>,
size: usize,
rng: ThreadRng,
}
impl App {
fn new(size: usize) -> App {
let data = vec![0; size];
App {
data,
rng: thread_rng(),
size,
}
}
fn on_tick(&mut self) {
for i in 0..self.size {
self.data[i] = self.rng.gen_range(0..100);
}
}
}
//
// use rand::{rngs::ThreadRng, thread_rng, Rng};
//
// struct App {
// data: Vec<u64>,
// size: usize,
// rng: ThreadRng,
// }
//
// impl App {
// fn new(size: usize) -> App {
// let data = vec![0; size];
// App {
// data,
// rng: thread_rng(),
// size,
// }
// }
//
// fn on_tick(&mut self) {
// for i in 0..self.size {
// self.data[i] = self.rng.gen_range(0..100);
// }
// }
// }
//
fn main() -> Result<(), Box<dyn Error>> {
// setup terminal
enable_raw_mode()?;
let mut stdout = io::stdout();
execute!(stdout, EnterAlternateScreen, EnableMouseCapture)?;
let backend = CrosstermBackend::new(stdout);
let mut terminal = Terminal::new(backend)?;
// create app and run it
let tick_rate = Duration::from_millis(250);
let app = App::new(100);
let res = run_app(&mut terminal, app, tick_rate);
// restore terminal
disable_raw_mode()?;
execute!(
terminal.backend_mut(),
LeaveAlternateScreen,
DisableMouseCapture
)?;
terminal.show_cursor()?;
if let Err(err) = res {
println!("{:?}", err)
}
// // setup terminal
// enable_raw_mode()?;
// let mut stdout = io::stdout();
// execute!(stdout, EnterAlternateScreen, EnableMouseCapture)?;
// let backend = CrosstermBackend::new(stdout);
// let mut terminal = Terminal::new(backend)?;
//
// // create app and run it
// let tick_rate = Duration::from_millis(250);
// let app = App::new(100);
// let res = run_app(&mut terminal, app, tick_rate);
//
// // restore terminal
// disable_raw_mode()?;
// execute!(
// terminal.backend_mut(),
// LeaveAlternateScreen,
// DisableMouseCapture
// )?;
// terminal.show_cursor()?;
//
// if let Err(err) = res {
// println!("{:?}", err)
// }
//
Ok(())
}
fn run_app<B: Backend>(
terminal: &mut Terminal<B>,
mut app: App,
tick_rate: Duration,
) -> io::Result<()> {
let mut last_tick = Instant::now();
loop {
terminal.draw(|f| ui(f, &app))?;
let timeout = tick_rate
.checked_sub(last_tick.elapsed())
.unwrap_or_else(|| Duration::from_secs(0));
if crossterm::event::poll(timeout)? {
if let Event::Key(key) = event::read()? {
if let KeyCode::Char('q') = key.code {
return Ok(());
}
}
}
if last_tick.elapsed() >= tick_rate {
app.on_tick();
last_tick = Instant::now();
}
}
}
fn ui<B: Backend>(f: &mut Frame<B>, app: &App) {
let chunks = Layout::default()
.direction(Direction::Vertical)
.margin(2)
.constraints([Constraint::Percentage(100)].as_ref())
.split(f.size());
let histogram = Histogram::default()
.block(Block::default().title("Data1").borders(Borders::ALL))
.data(&app.data, 10)
.bar_style(Style::default().fg(Color::Yellow))
.value_style(Style::default().fg(Color::Black).bg(Color::Yellow));
f.render_widget(histogram, chunks[0]);
}
//
// fn run_app<B: Backend>(
// terminal: &mut Terminal<B>,
// mut app: App,
// tick_rate: Duration,
// ) -> io::Result<()> { let mut last_tick = Instant::now(); loop { terminal.draw(|f| ui(f, &app))?;
//
// let timeout = tick_rate
// .checked_sub(last_tick.elapsed())
// .unwrap_or_else(|| Duration::from_secs(0));
// if crossterm::event::poll(timeout)? {
// if let Event::Key(key) = event::read()? {
// if let KeyCode::Char('q') = key.code {
// return Ok(());
// }
// }
// }
// if last_tick.elapsed() >= tick_rate {
// app.on_tick();
// last_tick = Instant::now();
// }
// }
// }
//
// fn ui<B: Backend>(f: &mut Frame<B>, app: &App) {
// let chunks = Layout::default()
// .direction(Direction::Vertical)
// .margin(2)
// .constraints([Constraint::Percentage(100)].as_ref())
// .split(f.size());
// let histogram = Histogram::default()
// .block(Block::default().title("Data1").borders(Borders::ALL))
// .data(&app.data, 10)
// .bar_style(Style::default().fg(Color::Yellow))
// .value_style(Style::default().fg(Color::Black).bg(Color::Yellow));
// f.render_widget(histogram, chunks[0]);
// }

View file

@ -1,3 +1,5 @@
use unicode_width::UnicodeWidthStr;
use crate::{
buffer::Buffer,
layout::Rect,
@ -5,7 +7,13 @@ use crate::{
symbols,
widgets::{Block, Widget},
};
use unicode_width::UnicodeWidthStr;
#[derive(Debug, Clone, Copy, PartialEq)]
struct Bucket {
low: f64,
high: f64,
count: usize,
}
/// A bar chart specialized for showing histograms
///
@ -41,12 +49,12 @@ pub struct Histogram<'a> {
/// Style for the widget
style: Style,
/// Slice of values to plot on the chart
data: &'a [u64],
data: &'a [f64],
/// each bucket keeps a count of the data points that fall into it
/// buckets[0] counts items where 0 <= x < bucket_size
/// buckets[1] counts items where bucket_size <= x < 2*bucket_size
/// buckets[0].count counts items where buckets[0].low <= x < buckets[0].high
/// buckets[1].count counts items where buckets[1].low <= x < buckets[1].high
/// etc.
buckets: Vec<u64>,
buckets: Vec<Bucket>,
/// Value necessary for a bar to reach the maximum height (if no value is specified,
/// the maximum value in the data is taken as reference)
max: Option<u64>,
@ -73,28 +81,39 @@ impl<'a> Default for Histogram<'a> {
}
impl<'a> Histogram<'a> {
pub fn data(mut self, data: &'a [u64], n_buckets: u64) -> Histogram<'a> {
pub fn data(mut self, data: &'a [f64], n_buckets: u64) -> Histogram<'a> {
self.data = data;
let min = *self.data.iter().min().unwrap();
let max = *self.data.iter().max().unwrap() + 1;
let bucket_size: u64 = ((max - min) as f64 / n_buckets as f64).ceil() as u64;
self.buckets = vec![0; n_buckets as usize];
let min = data.iter().cloned().fold(f64::NAN, f64::min);
let max = data.iter().cloned().fold(f64::NAN, f64::max);
let bucket_size = (max - min) / n_buckets as f64;
self.buckets = Vec::with_capacity(n_buckets as usize);
// initialize buckets
self.values = Vec::with_capacity(n_buckets as usize);
for v in 0..n_buckets {
self.values.push(format!("{}", v * bucket_size));
for i in 0..n_buckets {
let start = min + bucket_size * i as f64;
let bucket = Bucket {
low: start,
high: start + bucket_size,
count: 0,
};
self.buckets.push(bucket);
self.values
.push(format!("[{:.1}, {:.1})", bucket.low, bucket.high));
}
// bucketize data
for &x in self.data.iter() {
let idx: usize = ((x - min) / bucket_size) as usize;
self.buckets[idx] += 1;
let idx: usize = ((x - min) / bucket_size).floor() as usize;
if idx < self.buckets.len() {
self.buckets[idx].count += 1;
} else {
// TODO: decide what to do with excess
}
}
self.max = Some(*self.buckets.iter().max().unwrap());
self
}
@ -142,87 +161,27 @@ impl<'a> Histogram<'a> {
impl<'a> Widget for Histogram<'a> {
fn render(mut self, area: Rect, buf: &mut Buffer) {
buf.set_style(area, self.style);
let chart_area = match self.block.take() {
Some(b) => {
let inner_area = b.inner(area);
b.render(area, buf);
inner_area
}
None => area,
};
if chart_area.height < 2 {
return;
}
let n_bars = self.buckets.len() as u16;
let bar_width: u16 = (chart_area.width - (n_bars + 1) * self.bar_gap) / n_bars;
let max = self
.max
.unwrap_or_else(|| self.buckets.iter().copied().max().unwrap_or_default());
let mut data = self
.buckets
.iter()
.take(n_bars as usize)
.map(|&v| v * u64::from(chart_area.height - 1) * 8 / std::cmp::max(max, 1))
.collect::<Vec<u64>>();
for j in (0..chart_area.height - 1).rev() {
for (i, d) in data.iter_mut().enumerate() {
let symbol = match d {
0 => self.bar_set.empty,
1 => self.bar_set.one_eighth,
2 => self.bar_set.one_quarter,
3 => self.bar_set.three_eighths,
4 => self.bar_set.half,
5 => self.bar_set.five_eighths,
6 => self.bar_set.three_quarters,
7 => self.bar_set.seven_eighths,
_ => self.bar_set.full,
};
for x in 0..bar_width {
buf.get_mut(
chart_area.left() + i as u16 * (bar_width + self.bar_gap) + x,
chart_area.top() + j,
)
.set_symbol(symbol)
.set_style(self.bar_style);
}
if *d > 8 {
*d -= 8;
} else {
*d = 0;
}
}
}
for (i, &value) in self.buckets.iter().enumerate() {
let label = &self.values[i];
if value != 0 {
let value_label = format!("{}", &self.buckets[i]);
let width = value_label.width() as u16;
if width < bar_width {
buf.set_string(
chart_area.left()
+ i as u16 * (bar_width + self.bar_gap)
+ (bar_width - width) / 2,
chart_area.bottom() - 2,
value_label,
self.value_style,
);
}
}
buf.set_stringn(
chart_area.left() + i as u16 * (bar_width + self.bar_gap),
chart_area.bottom() - 1,
label,
bar_width as usize,
self.label_style,
);
}
}
}
#[cfg(test)]
mod tests {
use pretty_assertions::{assert_eq, assert_ne};
use super::*;
use crate::assert_buffer_eq;
#[test]
fn test_compute_bins() {
let data = [0.0, 0.5, 1.0, 1.5, 2.0, 2.5, 3.0];
let hist = Histogram::default().data(&data, 3);
assert_eq!(hist.buckets.len(), 3);
assert_eq!(hist.buckets[0].count, 2); // 0.0, 0.5
assert_eq!(hist.buckets[1].count, 2); // 1.0, 1.5
assert_eq!(hist.buckets[2].count, 2); // 2.0, 2.5
}
#[test]
fn test_render_histogram() {}
}