tek/bin/tek.rs

#[allow(unused_imports)] use std::sync::Arc;
#[allow(unused_imports)] use clap::{self, Parser, Subcommand, ValueEnum};
#[allow(unused_imports)] use tek::{
    *,
    jack::*,
    tek_input::*,
    tek_output::*,
    tek_tui::{*, ratatui::prelude::Color}
};

#[derive(Debug, Parser)]
#[command(version, about, long_about = None)]
pub struct TekCli {
    /// Which app to initialize
    #[command(subcommand)] mode: TekMode,
    /// Name of JACK client
    #[arg(short='t', long)] name: Option<String>,
    /// Whether to attempt to become transport master
    #[arg(short='S', long, default_value_t = false)] sync_lead: bool,
    /// Whether to sync to external transport master
    #[arg(short='s', long, default_value_t = true)] sync_follow: bool,
    /// Initial tempo in beats per minute
    #[arg(short='b', long, default_value = None)] bpm: Option<f64>,
}

#[derive(Debug, Clone, Subcommand)]
pub enum TekMode {
    /// A standalone transport view.
    Clock,
    /// A MIDI sequencer.
    Sequencer {
        /// Whether to include a transport toolbar (default: true)
        #[arg(short='t', long, default_value_t = true)] show_clock: bool,
        /// MIDI outs to connect to (multiple instances accepted)
        #[arg(short='i', long)] midi_from: Vec<String>,
        /// MIDI ins to connect to (multiple instances accepted)
        #[arg(short='o', long)] midi_to: Vec<String>,
    },
    /// A MIDI-controlled audio sampler.
    Sampler {
        /// MIDI outs to connect to (multiple instances accepted)
        #[arg(short='i', long)] midi_from: Vec<String>,
        /// Audio outs to connect to left input
        #[arg(short='l', long)] l_from: Vec<String>,
        /// Audio outs to connect to right input
        #[arg(short='r', long)] r_from: Vec<String>,
        /// Audio ins to connect from left output
        #[arg(short='L', long)] l_to: Vec<String>,
        /// Audio ins to connect from right output
        #[arg(short='R', long)] r_to: Vec<String>,
    },
    /// Sequencer and sampler together.
    Groovebox {
        /// Whether to include a transport toolbar (default: true)
        #[arg(short='t', long, default_value_t = true)] show_clock: bool,
        /// MIDI outs to connect to (multiple instances accepted)
        #[arg(short='i', long)] midi_from: Vec<String>,
        /// MIDI ins to connect to (multiple instances accepted)
        #[arg(short='o', long)] midi_to: Vec<String>,
        /// Audio outs to connect to left input
        #[arg(short='l', long)] l_from: Vec<String>,
        /// Audio outs to connect to right input
        #[arg(short='r', long)] r_from: Vec<String>,
        /// Audio ins to connect from left output
        #[arg(short='L', long)] l_to: Vec<String>,
        /// Audio ins to connect from right output
        #[arg(short='R', long)] r_to: Vec<String>,
    },
    /// Multi-track MIDI sequencer.
    Arranger {
        /// Whether to include a transport toolbar (default: true)
        #[arg(short='t', long, default_value_t = true)] show_clock: bool,
        /// MIDI outs to connect to (multiple instances accepted)
        #[arg(short='i', long)] midi_from: Vec<String>,
        /// MIDI ins to connect to (multiple instances accepted)
        #[arg(short='o', long)] midi_to: Vec<String>,
        /// Audio outs to connect to left input
        #[arg(short='l', long)] l_from: Vec<String>,
        /// Audio outs to connect to right input
        #[arg(short='r', long)] r_from: Vec<String>,
        /// Audio ins to connect from left output
        #[arg(short='L', long)] l_to: Vec<String>,
        /// Audio ins to connect from right output
        #[arg(short='R', long)] r_to: Vec<String>,
        /// Number of tracks
        #[arg(short = 'x', long, default_value_t = 16)] tracks: usize,
        /// Width of tracks
        #[arg(short = 'w', long, default_value_t = 6)] track_width: usize,
        /// Number of scenes
        #[arg(short = 'y', long, default_value_t = 8)] scenes: usize,
    },
    /// TODO: A MIDI-controlled audio mixer
    Mixer,
    /// TODO: A customizable channel strip
    Track,
    /// TODO: An audio plugin host
    Plugin,
}

/// Application entrypoint.
pub fn main () -> Usually<()> {
    use TekMode::*;
    let cli    = TekCli::parse();
    let name   = cli.name.as_ref().map_or("tek", |x|x.as_str());
    let jack   = JackConnection::new(name)?;
    let engine = Tui::new()?;
    Ok(match cli.mode {
        Clock =>
            engine.run(&jack.activate_with(|jack|Ok(
                crate::TransportTui::new(jack)?
            ))?)?,
        Sequencer { midi_from, midi_to, .. } =>
            engine.run(&jack.activate_with(|jack|Ok({
                let mut app  = crate::Sequencer::new(jack)?;
                let jack     = jack.read().unwrap();
                let midi_in  = jack.register_port("i", MidiIn::default())?;
                let midi_out = jack.register_port("o", MidiOut::default())?;
                connect_from(&jack, &midi_in, &midi_from)?;
                connect_to(&jack, &midi_out, &midi_to)?;
                app.player.midi_ins.push(midi_in);
                app.player.midi_outs.push(midi_out);
                app
            }))?)?,
        Sampler { midi_from, l_from, r_from, l_to, r_to, .. } =>
            engine.run(&jack.activate_with(|jack|Ok(
                tek::SamplerTui {
                    cursor:  (0, 0),
                    editing: None,
                    mode:    None,
                    size:    Measure::new(),
                    note_lo: 36.into(),
                    note_pt: 36.into(),
                    color:   ItemPalette::from(Color::Rgb(64, 128, 32)),
                    state:   tek::Sampler::new(
                        jack, &"sampler",
                        &midi_from,
                        &[&l_from, &r_from],
                        &[&l_to, &r_to],
                    )?,
                }
            ))?)?,
        Groovebox { midi_from, midi_to, l_from, r_from, l_to, r_to, .. } =>
            engine.run(&jack.activate_with(|jack|Ok({
                let app = tek::Groovebox::new(
                    jack,
                    &midi_from, &midi_to,
                    &[&l_from, &r_from],
                    &[&l_to,   &r_to],
                )?;
                if let Some(bpm) = cli.bpm {
                    app.clock().timebase.bpm.set(bpm);
                }
                if cli.sync_lead {
                    jack.read().unwrap().client().register_timebase_callback(false, |mut state|{
                        app.clock().playhead.update_from_sample(state.position.frame() as f64);
                        state.position.bbt = Some(app.clock().bbt());
                        state.position
                    })?
                } else if cli.sync_follow {
                    jack.read().unwrap().client().register_timebase_callback(false, |state|{
                        app.clock().playhead.update_from_sample(state.position.frame() as f64);
                        state.position
                    })?
                }
                app
            }))?)?,
        Arranger { scenes, tracks, track_width, midi_from, midi_to, .. } =>
            engine.run(&jack.activate_with(|jack|Ok({
                let mut app = crate::Arranger::try_from(jack)?;
                app.color = ItemPalette::random();
                app.tracks_add(tracks, track_width, midi_from.as_slice(), midi_to.as_slice())?;
                app.scenes_add(scenes)?;
                app
            }))?)?,
        _ => todo!()
    })
}

#[allow(unused)]
fn connect_from (jack: &JackConnection, input: &Port<MidiIn>, ports: &[String]) -> Usually<()> {
    for port in ports.iter() {
        if let Some(port) = jack.port_by_name(port).as_ref() {
            jack.client().connect_ports(port, input)?;
        } else {
            panic!("Missing MIDI output: {port}. Use jack_lsp to list all port names.");
        }
    }
    Ok(())
}

#[allow(unused)]
fn connect_to (jack: &JackConnection, output: &Port<MidiOut>, ports: &[String]) -> Usually<()> {
    for port in ports.iter() {
        if let Some(port) = jack.port_by_name(port).as_ref() {
            jack.client().connect_ports(output, port)?;
        } else {
            panic!("Missing MIDI input: {port}. Use jack_lsp to list all port names.");
        }
    }
    Ok(())
}

#[allow(unused)]
fn connect_audio_from (jack: &JackConnection, input: &Port<AudioIn>, ports: &[String]) -> Usually<()> {
    for port in ports.iter() {
        if let Some(port) = jack.port_by_name(port).as_ref() {
            jack.client().connect_ports(port, input)?;
        } else {
            panic!("Missing MIDI output: {port}. Use jack_lsp to list all port names.");
        }
    }
    Ok(())
}

#[allow(unused)]
fn connect_audio_to (jack: &JackConnection, output: &Port<AudioOut>, ports: &[String]) -> Usually<()> {
    for port in ports.iter() {
        if let Some(port) = jack.port_by_name(port).as_ref() {
            jack.client().connect_ports(output, port)?;
        } else {
            panic!("Missing MIDI input: {port}. Use jack_lsp to list all port names.");
        }
    }
    Ok(())
}


#[test] fn verify_cli () {
    use clap::CommandFactory;
    TekCli::command().debug_assert();
}