83 lines
2.2 KiB
Rust
83 lines
2.2 KiB
Rust
use crate::theme::meter;
|
|
use ratatui::buffer::Buffer;
|
|
use ratatui::layout::Rect;
|
|
use ratatui::style::Color;
|
|
use ratatui::widgets::Widget;
|
|
|
|
const DB_MIN: f32 = -48.0;
|
|
const DB_MAX: f32 = 3.0;
|
|
const DB_RANGE: f32 = DB_MAX - DB_MIN;
|
|
|
|
pub struct VuMeter {
|
|
left: f32,
|
|
right: f32,
|
|
}
|
|
|
|
impl VuMeter {
|
|
pub fn new(left: f32, right: f32) -> Self {
|
|
Self { left, right }
|
|
}
|
|
|
|
fn amplitude_to_db(amp: f32) -> f32 {
|
|
if amp <= 0.0 {
|
|
DB_MIN
|
|
} else {
|
|
(20.0 * amp.log10()).clamp(DB_MIN, DB_MAX)
|
|
}
|
|
}
|
|
|
|
fn db_to_normalized(db: f32) -> f32 {
|
|
(db - DB_MIN) / DB_RANGE
|
|
}
|
|
|
|
fn row_to_color(row_position: f32) -> Color {
|
|
if row_position > 0.9 {
|
|
meter::HIGH
|
|
} else if row_position > 0.75 {
|
|
meter::MID
|
|
} else {
|
|
meter::LOW
|
|
}
|
|
}
|
|
}
|
|
|
|
impl Widget for VuMeter {
|
|
fn render(self, area: Rect, buf: &mut Buffer) {
|
|
if area.width < 3 || area.height == 0 {
|
|
return;
|
|
}
|
|
|
|
let height = area.height as usize;
|
|
let half_width = area.width / 2;
|
|
let gap = 1u16;
|
|
|
|
let left_db = Self::amplitude_to_db(self.left);
|
|
let right_db = Self::amplitude_to_db(self.right);
|
|
let left_norm = Self::db_to_normalized(left_db);
|
|
let right_norm = Self::db_to_normalized(right_db);
|
|
|
|
let left_rows = (left_norm * height as f32).round() as usize;
|
|
let right_rows = (right_norm * height as f32).round() as usize;
|
|
|
|
for row in 0..height {
|
|
let y = area.y + area.height - 1 - row as u16;
|
|
let row_position = (row as f32 + 0.5) / height as f32;
|
|
let color = Self::row_to_color(row_position);
|
|
|
|
for col in 0..half_width.saturating_sub(gap) {
|
|
let x = area.x + col;
|
|
if row < left_rows {
|
|
buf[(x, y)].set_char(' ').set_bg(color);
|
|
}
|
|
}
|
|
|
|
for col in 0..half_width.saturating_sub(gap) {
|
|
let x = area.x + half_width + gap + col;
|
|
if x < area.x + area.width && row < right_rows {
|
|
buf[(x, y)].set_char(' ').set_bg(color);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|