101 lines
3.4 KiB
Rust
101 lines
3.4 KiB
Rust
|
use eframe::egui;
|
||
|
|
use eframe::egui::{Color32, Context, Id, Margin, PointerButton, Pos2, Rangef, Response, Stroke, TopBottomPanel, Ui, Vec2, Widget};
|
||
|
|
use crate::nome::LOOP_LENGTH;
|
||
|
|
|
||
|
|
pub enum NoteType {
|
||
|
|
Sine,
|
||
|
|
Triangle,
|
||
|
|
Sawtooth,
|
||
|
|
Square,
|
||
|
|
Noise
|
||
|
|
}
|
||
|
|
|
||
|
|
impl NoteType {
|
||
|
|
fn color(&self) -> Color32 {
|
||
|
|
match self {
|
||
|
|
NoteType::Sine => Color32::from_rgb(3, 211, 252),
|
||
|
|
NoteType::Triangle => Color32::from_rgb(252, 202, 3),
|
||
|
|
NoteType::Sawtooth => Color32::from_rgb(252, 18, 61),
|
||
|
|
NoteType::Square => Color32::from_rgb(4, 219, 51),
|
||
|
|
NoteType::Noise => Color32::from_rgb(120, 120, 120),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
pub struct Grid {
|
||
|
|
note_type: NoteType,
|
||
|
|
volume: f32,
|
||
|
|
notes: Vec<bool>,
|
||
|
|
pub id: String
|
||
|
|
}
|
||
|
|
|
||
|
|
impl Grid {
|
||
|
|
pub fn for_note_type(note_type: NoteType, counter: usize) -> Self {
|
||
|
|
Self {
|
||
|
|
note_type,
|
||
|
|
volume: 1.0,
|
||
|
|
notes: vec![false; (LOOP_LENGTH * LOOP_LENGTH) as usize],
|
||
|
|
id: format!("Track {}", counter)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn show(&mut self, ctx: &Context, cursor: f32) {
|
||
|
|
let id = Id::from(self.id.clone());
|
||
|
|
let win = egui::Window::new(&self.id).resizable(false).id(id).scroll([false, false]);
|
||
|
|
win.show(ctx, |ui| {
|
||
|
|
self.frame(ui, cursor)
|
||
|
|
});
|
||
|
|
}
|
||
|
|
|
||
|
|
fn frame(&mut self, ui: &mut Ui, cursor: f32) {
|
||
|
|
egui::Frame::new().inner_margin(3).show(ui, |ui| {
|
||
|
|
let dim = 20.0 * LOOP_LENGTH as f32;
|
||
|
|
ui.set_width(dim);
|
||
|
|
ui.set_height(dim);
|
||
|
|
let rect = ui.clip_rect().translate(Vec2::new(6.0, 6.0));
|
||
|
|
//ui.painter().debug_rect(rect, Color32::from_rgb(0, 255, 0), "blah");
|
||
|
|
for (i, lit) in self.notes.iter().enumerate() {
|
||
|
|
let (x, y) = (i as u32 % LOOP_LENGTH, i as u32 / LOOP_LENGTH);
|
||
|
|
let center = Pos2::new(
|
||
|
|
(x * 20 + 10) as f32 + rect.left(),
|
||
|
|
(y * 20 + 10) as f32 + rect.top());
|
||
|
|
if *lit {
|
||
|
|
ui.painter().circle_filled(center, 10.0, self.note_type.color());
|
||
|
|
} else {
|
||
|
|
ui.painter().circle_stroke(center, 10.0, (1.0, Color32::from_gray(0x88)));
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
ui.painter().vline(
|
||
|
|
cursor * dim + rect.left(),
|
||
|
|
Rangef::new(rect.top(), rect.top() + dim),
|
||
|
|
(1.0, self.note_type.color())
|
||
|
|
);
|
||
|
|
|
||
|
|
if ui.response().contains_pointer() {
|
||
|
|
ui.input(|input| {
|
||
|
|
if input.pointer.button_clicked(PointerButton::Primary) {
|
||
|
|
if let Some(pos) = input.pointer.latest_pos() {
|
||
|
|
println!("window: {} pos: {} {}", self.id, pos.x - rect.left(), pos.y - rect.top());
|
||
|
|
let x = ((pos.x - rect.left()) / 20.0).floor() as usize;
|
||
|
|
let y = ((pos.y - rect.top()) / 20.0).floor() as usize;
|
||
|
|
let n = x + y * LOOP_LENGTH as usize;
|
||
|
|
self.notes[n] = !self.notes[n]
|
||
|
|
}
|
||
|
|
}
|
||
|
|
})
|
||
|
|
}
|
||
|
|
});
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn notes(&self, beat: u32) -> Vec<u32> {
|
||
|
|
let mut notes = vec![];
|
||
|
|
for y in 0..LOOP_LENGTH {
|
||
|
|
if self.notes[(y * LOOP_LENGTH + beat) as usize] {
|
||
|
|
notes.push(y)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
notes
|
||
|
|
}
|
||
|
|
}
|