tek/crates/tek_tui/src/todo_tui_sampler.rs

use crate::*;

/// The sampler plugin plays sounds.
pub struct SamplerView<E: Engine> {
    _engine:     PhantomData<E>,
    pub state:   Sampler,
    pub cursor:  (usize, usize),
    pub editing: Option<Arc<RwLock<Sample>>>,
    pub buffer:  Vec<Vec<f32>>,
    pub modal:   Arc<Mutex<Option<Box<dyn Exit + Send>>>>,
}

impl<E: Engine> SamplerView<E> {
    pub fn new (
        jack: &Arc<RwLock<JackClient>>,
        name: &str,
        mapped: Option<BTreeMap<u7, Arc<RwLock<Sample>>>>
    ) -> Usually<JackDevice<E>> {
        Jack::new(name)?
            .midi_in("midi")
            .audio_in("recL")
            .audio_in("recR")
            .audio_out("outL")
            .audio_out("outR")
            .run(|ports|Box::new(Self {
                _engine:  Default::default(),
                jack:     jack.clone(),
                name:     name.into(),
                cursor:   (0, 0),
                editing:  None,
                mapped:   mapped.unwrap_or_else(||BTreeMap::new()),
                unmapped: vec![],
                voices:   Arc::new(RwLock::new(vec![])),
                ports,
                buffer:   vec![vec![0.0;16384];2],
                output_gain: 0.5,
                modal:    Default::default()
            }))
    }
    /// Immutable reference to sample at cursor.
    pub fn sample (&self) -> Option<&Arc<RwLock<Sample>>> {
        for (i, sample) in self.mapped.values().enumerate() {
            if i == self.cursor.0 {
                return Some(sample)
            }
        }
        for (i, sample) in self.unmapped.iter().enumerate() {
            if i + self.mapped.len() == self.cursor.0 {
                return Some(sample)
            }
        }
        None
    }
}

/// A sound sample.
#[derive(Default, Debug)]
pub struct Sample {
    pub name:     String,
    pub start:    usize,
    pub end:      usize,
    pub channels: Vec<Vec<f32>>,
    pub rate:     Option<usize>,
}

/// Load sample from WAV and assign to MIDI note.
#[macro_export] macro_rules! sample {
    ($note:expr, $name:expr, $src:expr) => {{
        let (end, data) = read_sample_data($src)?;
        (
            u7::from_int_lossy($note).into(),
            Sample::new($name, 0, end, data).into()
        )
    }};
}

use std::fs::File;
use symphonia::core::codecs::CODEC_TYPE_NULL;
use symphonia::core::errors::Error;
use symphonia::core::io::MediaSourceStream;
use symphonia::core::probe::Hint;
use symphonia::core::audio::SampleBuffer;
use symphonia::default::get_codecs;

pub struct AddSampleModal {
    exited:  bool,
    dir:     PathBuf,
    subdirs: Vec<OsString>,
    files:   Vec<OsString>,
    cursor:  usize,
    offset:  usize,
    sample:  Arc<RwLock<Sample>>,
    voices:  Arc<RwLock<Vec<Voice>>>,
    _search: Option<String>,
}

impl Exit for AddSampleModal {
    fn exited (&self) -> bool {
        self.exited
    }
    fn exit (&mut self) {
        self.exited = true
    }
}

impl AddSampleModal {
    pub fn new (
        sample: &Arc<RwLock<Sample>>,
        voices: &Arc<RwLock<Vec<Voice>>>
    ) -> Usually<Self> {
        let dir = std::env::current_dir()?;
        let (subdirs, files) = scan(&dir)?;
        Ok(Self {
            exited: false,
            dir,
            subdirs,
            files,
            cursor: 0,
            offset: 0,
            sample: sample.clone(),
            voices: voices.clone(),
            _search: None
        })
    } 
    fn rescan (&mut self) -> Usually<()> {
        scan(&self.dir).map(|(subdirs, files)|{
            self.subdirs = subdirs;
            self.files = files;
        })
    }
    fn prev (&mut self) {
        self.cursor = self.cursor.saturating_sub(1);
    }
    fn next (&mut self) {
        self.cursor = self.cursor + 1;
    }
    fn try_preview (&mut self) -> Usually<()> {
        if let Some(path) = self.cursor_file() {
            if let Ok(sample) = Sample::from_file(&path) {
                *self.sample.write().unwrap() = sample;
                self.voices.write().unwrap().push(
                    Sample::play(&self.sample, 0, &u7::from(100u8))
                );
            }
            //load_sample(&path)?;
            //let src = std::fs::File::open(&path)?;
            //let mss = MediaSourceStream::new(Box::new(src), Default::default());
            //let mut hint = Hint::new();
            //if let Some(ext) = path.extension() {
                //hint.with_extension(&ext.to_string_lossy());
            //}
            //let meta_opts: MetadataOptions = Default::default();
            //let fmt_opts: FormatOptions = Default::default();
            //if let Ok(mut probed) =  symphonia::default::get_probe()
                //.format(&hint, mss, &fmt_opts, &meta_opts)
            //{
                //panic!("{:?}", probed.format.metadata());
            //};
        }
        Ok(())
    }
    fn cursor_dir (&self) -> Option<PathBuf> {
        if self.cursor < self.subdirs.len() {
            Some(self.dir.join(&self.subdirs[self.cursor]))
        } else {
            None
        }
    }
    fn cursor_file (&self) -> Option<PathBuf> {
        if self.cursor < self.subdirs.len() {
            return None
        }
        let index = self.cursor.saturating_sub(self.subdirs.len());
        if index < self.files.len() {
            Some(self.dir.join(&self.files[index]))
        } else {
            None
        }
    }
    fn pick (&mut self) -> Usually<bool> {
        if self.cursor == 0 {
            if let Some(parent) = self.dir.parent() {
                self.dir = parent.into();
                self.rescan()?;
                self.cursor = 0;
                return Ok(false)
            }
        }
        if let Some(dir) = self.cursor_dir() {
            self.dir = dir;
            self.rescan()?;
            self.cursor = 0;
            return Ok(false)
        }
        if let Some(path) = self.cursor_file() {
            let (end, channels) = read_sample_data(&path.to_string_lossy())?;
            let mut sample = self.sample.write().unwrap();
            sample.name = path.file_name().unwrap().to_string_lossy().into();
            sample.end = end;
            sample.channels = channels;
            return Ok(true)
        }
        return Ok(false)
    }
}

pub const KEYMAP_ADD_SAMPLE: &'static [KeyBinding<AddSampleModal>] = keymap!(AddSampleModal {
    [Esc,   NONE, "sampler/add/close", "close help dialog",       |modal: &mut AddSampleModal|{
        modal.exit();
        Ok(true)
    }],
    [Up,    NONE, "sampler/add/prev",  "select previous entry",   |modal: &mut AddSampleModal|{
        modal.prev();
        Ok(true)
    }],
    [Down,  NONE, "sampler/add/next",  "select next entry",       |modal: &mut AddSampleModal|{
        modal.next();
        Ok(true)
    }],
    [Enter, NONE, "sampler/add/enter", "activate selected entry", |modal: &mut AddSampleModal|{
        if modal.pick()? {
            modal.exit();
        }
        Ok(true)
    }],
    [Char('p'), NONE, "sampler/add/preview", "preview selected entry", |modal: &mut AddSampleModal|{
        modal.try_preview()?;
        Ok(true)
    }]
});

fn scan (dir: &PathBuf) -> Usually<(Vec<OsString>, Vec<OsString>)> {
    let (mut subdirs, mut files) = read_dir(dir)?
        .fold((vec!["..".into()], vec![]), |(mut subdirs, mut files), entry|{
            let entry = entry.expect("failed to read drectory entry");
            let meta = entry.metadata().expect("failed to read entry metadata");
            if meta.is_file() {
                files.push(entry.file_name());
            } else if meta.is_dir() {
                subdirs.push(entry.file_name());
            }
            (subdirs, files)
        });
    subdirs.sort();
    files.sort();
    Ok((subdirs, files))
}

impl Sample {
    fn from_file (path: &PathBuf) -> Usually<Self> {
        let mut sample = Self::default();
        sample.name = path.file_name().unwrap().to_string_lossy().into();
        // Use file extension if present
        let mut hint = Hint::new();
        if let Some(ext) = path.extension() {
            hint.with_extension(&ext.to_string_lossy());
        }
        let probed = symphonia::default::get_probe().format(
            &hint,
            MediaSourceStream::new(
                Box::new(File::open(path)?),
                Default::default(),
            ),
            &Default::default(),
            &Default::default()
        )?;
        let mut format = probed.format;
        let mut decoder = get_codecs().make(
            &format.tracks().iter()
                .find(|t| t.codec_params.codec != CODEC_TYPE_NULL)
                .expect("no tracks found")
                .codec_params,
            &Default::default()
        )?;
        loop {
            match format.next_packet() {
                Ok(packet) => {
                    // Decode a packet
                    let decoded = match decoder.decode(&packet) {
                        Ok(decoded) => decoded,
                        Err(err) => { return Err(err.into()); }
                    };
                    // Determine sample rate
                    let spec = *decoded.spec();
                    if let Some(rate) = sample.rate {
                        if rate != spec.rate as usize {
                            panic!("sample rate changed");
                        }
                    } else {
                        sample.rate = Some(spec.rate as usize);
                    }
                    // Determine channel count
                    while sample.channels.len() < spec.channels.count() {
                        sample.channels.push(vec![]);
                    }
                    // Load sample
                    let mut samples = SampleBuffer::new(
                        decoded.frames() as u64,
                        spec
                    );
                    if samples.capacity() > 0 {
                        samples.copy_interleaved_ref(decoded);
                        for frame in samples.samples().chunks(spec.channels.count()) {
                            for (chan, frame) in frame.iter().enumerate() {
                                sample.channels[chan].push(*frame)
                            }
                        }
                    }
                },
                Err(Error::IoError(_)) => break decoder.last_decoded(),
                Err(err) => return Err(err.into()),
            };
        };
        sample.end = sample.channels.iter().fold(0, |l, c|l + c.len());
        Ok(sample)
    }
}

impl Widget for SamplerView<Tui> {
    type Engine = Tui;
    fn layout (&self, to: [u16;2]) -> Perhaps<[u16;2]> {
        todo!()
    }
    fn render (&self, to: &mut TuiOutput) -> Usually<()> {
        tui_render_sampler(self, to)
    }
}

pub fn tui_render_sampler (sampler: &SamplerView<Tui>, to: &mut TuiOutput) -> Usually<()> {
    let [x, y, _, height] = to.area();
    let style = Style::default().gray();
    let title = format!(" {} ({})", sampler.name, sampler.voices.read().unwrap().len());
    to.blit(&title, x+1, y, Some(style.white().bold().not_dim()));
    let mut width = title.len() + 2;
    let mut y1 = 1;
    let mut j = 0;
    for (note, sample) in sampler.mapped.iter()
        .map(|(note, sample)|(Some(note), sample))
        .chain(sampler.unmapped.iter().map(|sample|(None, sample)))
    {
        if y1 >= height {
            break
        }
        let active = j == sampler.cursor.0;
        width = width.max(
            draw_sample(to, x, y + y1, note, &*sample.read().unwrap(), active)?
        );
        y1 = y1 + 1;
        j = j + 1;
    }
    let height = ((2 + y1) as u16).min(height);
    //Ok(Some([x, y, (width as u16).min(to.area().w()), height]))
    Ok(())
}

fn draw_sample (
    to: &mut TuiOutput, x: u16, y: u16, note: Option<&u7>, sample: &Sample, focus: bool
) -> Usually<usize> {
    let style = if focus { Style::default().green() } else { Style::default() };
    if focus {
        to.blit(&"🬴", x+1, y, Some(style.bold()));
    }
    let label1 = format!("{:3} {:12}",
        note.map(|n|n.to_string()).unwrap_or(String::default()),
        sample.name);
    let label2 = format!("{:>6} {:>6}  +0.0",
        sample.start,
        sample.end);
    to.blit(&label1, x+2, y, Some(style.bold()));
    to.blit(&label2, x+3+label1.len()as u16, y, Some(style));
    Ok(label1.len() + label2.len() + 4)
}

impl Widget for AddSampleModal {
    type Engine = Tui;
    fn layout (&self, to: [u16;2]) -> Perhaps<[u16;2]> {
        todo!()
        //Align::Center(()).layout(to)
    }
    fn render (&self, to: &mut TuiOutput) -> Usually<()> {
        todo!()
        //let area = to.area();
        //to.make_dim();
        //let area = center_box(
            //area,
            //64.max(area.w().saturating_sub(8)),
            //20.max(area.w().saturating_sub(8)),
        //);
        //to.fill_fg(area, Color::Reset);
        //to.fill_bg(area, Nord::bg_lo(true, true));
        //to.fill_char(area, ' ');
        //to.blit(&format!("{}", &self.dir.to_string_lossy()), area.x()+2, area.y()+1, Some(Style::default().bold()))?;
        //to.blit(&"Select sample:", area.x()+2, area.y()+2, Some(Style::default().bold()))?;
        //for (i, (is_dir, name)) in self.subdirs.iter()
            //.map(|path|(true, path))
            //.chain(self.files.iter().map(|path|(false, path)))
            //.enumerate()
            //.skip(self.offset)
        //{
            //if i >= area.h() as usize - 4 {
                //break
            //}
            //let t = if is_dir { "" } else { "" };
            //let line = format!("{t} {}", name.to_string_lossy());
            //let line = &line[..line.len().min(area.w() as usize - 4)];
            //to.blit(&line, area.x() + 2, area.y() + 3 + i as u16, Some(if i == self.cursor {
                //Style::default().green()
            //} else {
                //Style::default().white()
            //}))?;
        //}
        //Lozenge(Style::default()).draw(to)
    }
}